async update

app.py

@@ -27,7 +27,7 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         try:
-            response = await self.agent.invoke({"messages": messages}, config={"callbacks": [langfuse_handler]})
+            response = await self.agent.ainvoke({"messages": messages}, config={"callbacks": [langfuse_handler]})
             print(f"Agent raw response: {response}")
             if not response or 'messages' not in response or not response['messages']:
                 print("Agent response missing or empty 'messages'. Returning AGENT ERROR.")

langraph_agent.py

@@ -85,13 +85,13 @@ def modulus(a: int, b: int) -> int:
     return a % b
 
 @tool
-def wiki_search(query: str) -> str:
+def wiki_search(input: str) -> str:
     """Search Wikipedia for a query and return maximum 2 results.
     
     Args:
-        query: The search query."""
+        input: The search query."""
     try:
-        search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
+        search_docs = WikipediaLoader(query=input, load_max_docs=2).load()
         if not search_docs:
             return {"wiki_results": "No Wikipedia results found for the query."}
         formatted_search_docs = "\n\n---\n\n".join(
@@ -105,13 +105,13 @@ def wiki_search(query: str) -> str:
         return {"wiki_results": f"Error searching Wikipedia: {e}"}
 
 @tool
-def web_search(query: str) -> str:
+def web_search(input: str) -> str:
     """Search Tavily for a query and return maximum 3 results.
     
     Args:
-        query: The search query."""
+        input: The search query."""
     try:
-        search_docs = TavilySearchResults(max_results=3).invoke(query=query)
+        search_docs = TavilySearchResults(max_results=3).invoke(query=input)
         if not search_docs:
             return {"web_results": "No web search results found for the query."}
         formatted_search_docs = "\n\n---\n\n".join(
@@ -125,13 +125,13 @@ def web_search(query: str) -> str:
         return {"web_results": f"Error searching web: {e}"}
 
 @tool
-def arvix_search(query: str) -> str:
+def arvix_search(input: str) -> str:
     """Search Arxiv for a query and return maximum 3 result.
     
     Args:
-        query: The search query."""
+        input: The search query."""
     try:
-        search_docs = ArxivLoader(query=query, load_max_docs=3).load()
+        search_docs = ArxivLoader(query=input, load_max_docs=3).load()
         if not search_docs:
             return {"arvix_results": "No Arxiv results found for the query."}
         formatted_search_docs = "\n\n---\n\n".join(