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="")