+ updated for langgraph

app.py

@@ -8,6 +8,7 @@ from langraph_agent import build_graph
 import asyncio
 import aiohttp
 from langfuse.langchain import CallbackHandler
+from langchain_core.messages import HumanMessage
 
 # Initialize Langfuse CallbackHandler for LangGraph/Langchain (tracing)
 langfuse_handler = CallbackHandler()
@@ -23,10 +24,14 @@ class BasicAgent:
         self.agent = build_graph()
         print("BasicAgent initialized.")
     async def aquery(self, question: str) -> str:
+        messages = [HumanMessage(content=question)]
+        # messages = self.agent.invoke({"messages": messages}, config={"callbacks": [langfuse_handler]})
+         
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        response = await self.agent.invoke(question, config={"callbacks": [langfuse_handler]})
-        print(f"Agent returning fixed answer: {response}")
-        return response
+        response = await self.agent.invoke({"messages": messages}, config={"callbacks": [langfuse_handler]})
+        answer = response['messages'][-1].content  
+        print(f"Agent returning fixed answer: {answer}")
+        return answer
 
 # Global cache for answers (in-memory)
 cached_answers = None