tool update

app.py

@@ -24,9 +24,9 @@ from agent_utils import wiki_ret, arxiv_ret, tavily_ret, handle_file_tool
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 groq_api_key = os.getenv("GROQ_API_KEY")
-
+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]
     context: str
@@ -70,7 +70,7 @@ class BasicAgent:
         # Just make up a task_id.
         # Call the LLM
         messages = [HumanMessage(content=prompt)]
-        response = (llm.invoke(messages)).content
+        response = (llm_with_tools.invoke(messages)).content
         # cleaned_text = re.sub(r"<think>.*?</think>", "", response.content, flags=re.DOTALL)
 
         # json_start = response.find('{')
@@ -82,7 +82,7 @@ class BasicAgent:
     def create_graph(self):
         builder = StateGraph(GraphState)
         builder.add_node("agent", self.agent)
-        builder.add_node("tools", ToolNode(tools = [wiki_ret, arxiv_ret, tavily_ret, handle_file_tool]))
+        builder.add_node("tools", ToolNode(tools = tools))
 
         builder.add_edge(START, "agent")
         builder.add_conditional_edges("agent", tools_condition)