Hello You are having pulmonary tuberculosis & complete treatment of tuberculosis is for 6 - 9 months depending upon the lesion and different doctors . As you mentioned that you took treatment for just 2 months and now asking for treatment for cough. This is a blunder and you are doing a great mistake . In my opinion at once consult your doctor and get his full treatment as he advise. Mind it intermittent treatment of tuberculosis develop Multi drug resistant Tuberculosis ( a very dangerous disease ). So this is the cause of concern and immediately consult your doctor.