File size: 510 Bytes
01095ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import pickle
from query_data import get_chain
if __name__ == "__main__":
with open("vectorstore.pkl", "rb") as f:
vectorstore = pickle.load(f)
qa_chain = get_chain(vectorstore)
chat_history = []
print("Chat with your docs!")
while True:
print("Human:")
question = input()
result = qa_chain({"question": question, "chat_history": chat_history})
chat_history.append((question, result["answer"]))
print("AI:")
print(result["answer"])
|