Update app.py

app.py

@@ -50,14 +50,14 @@ try:
     )
 
     system_message = SystemMessage(content="You are a helpful assistant.")
-    human_template = """Based on the following context generate an answer for the question. If the answer is not available say I dont know.
+    human_template = """Based on the following context generate an response for the user query. If the response is not available say I dont know.
     Context: {context}
 
-    Question: {question}
+    Query: {query}
 
     Answer:"""
     human_message = HumanMessagePromptTemplate.from_template(template=human_template)
-    llm = ChatOpenAI(model="gpt-3.5-turbo-1106", streaming=True)
+    llm = ChatOpenAI(model="gpt-4-0125-preview", streaming=True)
     prompt = ChatPromptTemplate.from_messages(messages=[system_message, human_message])
     parser = StrOutputParser()
     prompt_chain = prompt | llm | parser
@@ -110,7 +110,7 @@ async def start_chat():
 async def main(message: cl.Message):
     retriever = cl.user_session.get("retriever")
     retriever_chain = RunnableParallel(
-        {"context": retriever, "question": RunnablePassthrough()}
+        {"context": retriever, "query": RunnablePassthrough()}
     )
 
     out = cl.Message(content="")