Update azure_openai.py

azure_openai.py

@@ -45,7 +45,10 @@ def qt(systemMessgae, history, temp, tokens, file):
     temperature=temp, 
     max_tokens=tokens  # Name of the deployment for identification
 )
-    userMessage = """{conversationToDate}"""
+    userMessage = """## Converstaion to date: {conversationToDate}
+    
+    ## Create Optimised Query
+    """
 
     system_message_template = SystemMessagePromptTemplate.from_template(systemMessgae)
     human_message_template = HumanMessagePromptTemplate.from_template(userMessage)
@@ -61,7 +64,7 @@ def qt(systemMessgae, history, temp, tokens, file):
     })
 
 
-def get_response(chat_history, qte, knowledge, temp1, temp2, tokens1, tokens2, persona2SystemMessage, persona2UserMessage, file):
+def get_response(chat_history, qte, knowledge, temp2, tokens2, persona2SystemMessage, persona2UserMessage, file):
     asset = read_file(file)
     llm = AzureChatOpenAI(
         openai_api_version=OPENAI_API_VERSION,
@@ -82,12 +85,11 @@ def get_response(chat_history, qte, knowledge, temp1, temp2, tokens1, tokens2, p
     prompt = ChatPromptTemplate.from_messages([system_message_template, persona2UserMessage])
 
     chain = prompt | llm | StrOutputParser()
-    
-    print(asset)
 
     return chain.stream({
         "assetGlossary": asset,
-        "query": chat_history,
+        "query": qte,
+        "chatHistory": chat_history,
         "knowledge": knowledge
     })