Update query_data.py

query_data.py

@@ -1,6 +1,7 @@
 from langchain.prompts.prompt import PromptTemplate
 from langchain.llms import OpenAI
 from langchain.chains import ChatVectorDBChain
+from langchain.chat_models import ChatOpenAI
 
 _template = """Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.
 You can assume the question about the syllabus of the H2 Economics, H2 History and H2 Geography A-Level Examinations in Singapore.
@@ -24,7 +25,7 @@ QA_PROMPT = PromptTemplate(template=template, input_variables=["question", "cont
 
 
 def get_chain(vectorstore):
-    llm = OpenAI(temperature=0, model_name = "text-davinci-003")
+    llm = ChatOpenAI(temperature=0)
     qa_chain = ChatVectorDBChain.from_llm(
         llm,
         vectorstore,