fix typos for QueryAgent class

[shao-shuai]

Hi,

Was trying this notebook and couldn't run QueryAgent.

Seems like there are a few typos, please refer to the changes and let me know if you have any questions.

Really like this repo, particularly the part that teaches low level load distribution of LLM work.

Thanks, Shuai