Upload app.py

app.py

@@ -69,23 +69,6 @@ def translate(target, text):
   res = response.json()
   return res[0]["translations"][0]["text"]
 
-'''
-def get_response(user_input):
-        Takes user_input in English and invokes the infer API for response.
-
-        Parameters:
-            user_input (string): User Query in English.
-        Returns:
-            res (string): Response from the LLM.
-    url = f"https://whole-icons-hammer.loca.lt/infer/{user_id}"
-    headers = {"Content-Type": "application/x-www-form-urlencoded"}
-    data = {"user_input": user_input}
-    response = requests.post(url, headers=headers, data=data)
-    print(response)
-    res = response.json()
-    return res["data"]
-'''
-
 def infer(user_input: str):
     '''
         Returns the translated response from the LLM in response to a user query.
@@ -110,14 +93,10 @@ def infer(user_input: str):
         ]
     )
 
-    runnable = prompt | agent_executor # define a chain
 
-    conversation = RunnableSequence( # wrap the chain along with chat history and user input
-        runnable,
-    )
 
-    response = conversation.invoke( # invoke the chain by giving the user input and the chat history
-        {"user_input": user_input},
+    response = agent_executor.invoke( # invoke the chain by giving the user input and the chat history
+        {"messages": [HumanMessage(content=user_input)]},
         config={"configurable": {"thread_id":user_id}}
     )