Currently `query_data.py` doesn't maintain context of the conversation. Would be good to have that.
Currently
query_data.pydoesn't maintain context of the conversation. Would be good to have that.