Update utils.py

utils.py

@@ -147,14 +147,15 @@ def translate_to_english(llm, user_query):
         You are an AI assistant that translates a text to English. \
         Do not generate any irrelavant text, only return the translation."""
 
-    message1 = SystemMessage(content=sys_prompt)
-
-    message2 = HumanMessage(
-        content=user_query
-    )
-    message_log = [message1, message2]
+    messages = [
+    (
+        "system",
+        sys_prompt,
+    ),
+    ("human", user_query),
+    ]
     
-    output = llm.invoke(message_log)
+    output = llm.invoke(messages)
     pred = output.content
     
     return pred
@@ -165,14 +166,15 @@ def translate(llm, user_query, to_translate):
         Do not generate any irrelavant text, only return the translation. \
         If the user query is in English, then don't do anything just return the original text, no translation is required there.""".format(user_query)
 
-    message1 = SystemMessage(content=sys_prompt)
-
-    message2 = HumanMessage(
-        content=to_translate
-    )
-    message_log = [message1, message2]
+    messages = [
+    (
+        "system",
+        sys_prompt,
+    ),
+    ("human", to_translate),
+    ]
     
-    output = llm.invoke(message_log)
+    output = llm.invoke(messages)
     pred = output.content
     
     return pred