check

app.py

@@ -4,6 +4,7 @@ from langgraph.prebuilt import create_react_agent
 from search_agent import tools
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
 from search_agent import tools
+from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
@@ -51,14 +52,23 @@ def respond(
 
     #     response += token
     #     yield response
-
-    inputs = {"messages": [(role, content) for role, content in messages]}
+    def convert(msg):
+        if msg["role"] in ["user", "human"]:
+            return HumanMessage(content=msg["content"])
+        elif msg["role"] in ["assistant", "ai"]:
+            return AIMessage(content=msg["content"])
+        elif msg["role"] == "system":
+            return SystemMessage(content=msg["content"])
+        else:
+            raise ValueError(f"Unsupported role: {msg['role']}")
+        
+    inputs = {"messages": [convert(m) for m in messages]}
     
     # Get the response from the agent (this integrates your agent with the model)
     agent_response = graph.invoke(inputs)  # Process the inputs through your agent
     
     # Return the final message from the agent
-    return agent_response['messages'][-1][1]
+    return agent_response['messages'][-1][1].content
 
 
 """