Update app.py

app.py

@@ -19,23 +19,22 @@ class BasicAgent:
         self.graph = build_graph(provider="groq")  # Using Groq as default
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question: {question}")
-        # Wrap the question in a HumanMessage from langchain_core
-        messages = [HumanMessage(content=question)]
-        config = {"configurable": {"thread_id": f"eval_{hash(question)}"}}
-        
         try:
-            result = self.graph.invoke({"messages": messages}, config)
+            messages = [HumanMessage(content=question)]
+            state = {
+                "messages": messages,
+                "query": question,
+                "agent_type": "",
+                "final_answer": "",
+                "perf": {},
+                "agno_resp": ""
+            }
+            result = self.graph.invoke(state)
             answer = result['messages'][-1].content
-            
-            # Extract final answer if present
-            if "FINAL ANSWER:" in answer:
-                return answer.split("FINAL ANSWER:")[-1].strip()
-            else:
-                return answer.strip()
-                
+            return answer[14:]
         except Exception as e:
-            return f"Error: {str(e)}"
+            return(e)
+
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """