Update query_data.py

query_data.py

@@ -1,5 +1,5 @@
 from langchain.prompts.prompt import PromptTemplate
-from langchain.llms import OpenAI
+from langchain.llms import OpenAIChat
 from langchain.chains import ChatVectorDBChain
 
 _template = """Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.
@@ -22,9 +22,10 @@ Question: {question}
 Answer in Markdown:"""
 QA_PROMPT = PromptTemplate(template=template, input_variables=["question", "context"])
 
+prefix_messages = [{"role": "system", "content": "You are a helpful assistant that is very good at answering questions about conversational AI and the bot forge."}]
 
 def get_chain(vectorstore):
-    llm = OpenAI(temperature=0)
+    llm = OpenAIChat(temperature=0,prefix_messages)
     qa_chain = ChatVectorDBChain.from_llm(
         llm,
         vectorstore,