A Probiotic is an excellent idea. This one is the best on the market but expensive.
A gram stain is a test a vet does on the birds crop and bottom area. It test for various microorganisms that should not be there and also fungi like yeast. It is usually only done by an avian vet.