宝宝消化不良腹泻,应该采取什么措施呢?

您好,我家宝宝四个月了,母乳喂养,目前十七斤重,大便一直稀黄带有颗粒,咨询过医生,说是消化不良不用管。可是最近这两天便便很稀,偶尔呈水样,还是黄色,目前宝宝精神很好,睡眠也还好,食量也和平时一样,我想问一下,这种情况的腹泻需要采取什么措施吗?