I am so sorry your baby is sick. I also encourage the use of probiotics.
here is the link to the puppy lemon law in Arizona:
Puppy Lemon Law - Arizona
I will say again for future readers to this post; do not take a puppy under 12 wks old that doesn't weigh in at least 2lbs.