Update agent.py

agent.py

@@ -13,7 +13,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from smolagents import CodeAgent, HfApiModel
 
 # Import our custom tools from their modules
-from tools import search_tool
+from tools import basic_tools
 
 # HF Chat interface
 # llm = HuggingFaceEndpoint(
@@ -29,8 +29,7 @@ chat = ChatGoogleGenerativeAI(
     max_retries=2,
 )
 
-tools = [search_tool]
-chat_with_tools = chat.bind_tools(tools)
+chat_with_tools = chat.bind_tools(basic_tools)
 memory = MemorySaver()
 
 # Generate the AgentState and Agent graph
@@ -47,7 +46,7 @@ builder = StateGraph(AgentState)
 
 # Define nodes: these do the work
 builder.add_node("assistant", assistant)
-builder.add_node("tools", ToolNode(tools))
+builder.add_node("tools", ToolNode(basic_tools))
 
 # Define edges: these determine how the control flow moves
 builder.add_edge(START, "assistant")