tool update

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 # import re
 # import json
 from typing_extensions import TypedDict
-# from typing import List
+from typing import Dict
 
 import operator
 from typing import  Annotated
@@ -28,7 +28,7 @@ tools = [wiki_ret, arxiv_ret, tavily_ret, handle_file_tool]
 llm = ChatGroq(api_key=groq_api_key, model="gemma2-9b-it")
 llm_with_tools = llm.bind_tools(tools)
 class GraphState(TypedDict):
-    messages: str #Annotated[list, operator.add]
+    messages: Annotated[Dict, operator.add]
     context: str
 
 # --- Basic Agent Definition ---
@@ -74,7 +74,7 @@ class BasicAgent:
         # Call the LLM
         messages = [HumanMessage(content=prompt)]
         print("messages: ", messages)
-        response = (llm_with_tools.invoke(messages)).content
+        response = llm_with_tools.invoke(messages)
         print("response: ", response)
         # cleaned_text = re.sub(r"<think>.*?</think>", "", response.content, flags=re.DOTALL)