change

real_agent.py

@@ -7,7 +7,7 @@ from langgraph.prebuilt import tools_condition
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain_openai import ChatOpenAI
 import os
-from tools import hub_stats_tool, extract_image_text, add, divide, multiply, subtract, calculate_cargo_travel_time
+from tools import hub_stats_tool, extract_image_text, add, divide, multiply, subtract, calculate_cargo_travel_time, web_search_tool
 from retriever import guest_info_tool
 import gradio as gr
 from langfuse.langchain import CallbackHandler
@@ -28,14 +28,14 @@ def call_build():
     langfuse_handler = CallbackHandler()
     
     # Initialize the web search tool
-    search_tool = GoogleSearchTool(provider = "serpapi")
-    search_tool.name = "GoogleSearchTool"
-    visit_webpage_tool = VisitWebpageTool()
-    visit_webpage_tool.name = "VisitWebpageTool"
+   # search_tool = GoogleSearchTool(provider = "serpapi")
+   # search_tool.name = "GoogleSearchTool"
+   # visit_webpage_tool = VisitWebpageTool()
+   # visit_webpage_tool.name = "VisitWebpageTool"
     # Generate the chat interface, including the tools
     
     chat = ChatOpenAI(model="gpt-4o")
-    tools = [guest_info_tool,  search_tool, hub_stats_tool, visit_webpage_tool, extract_image_text, add, divide, multiply, subtract, calculate_cargo_travel_time]
+    tools = [guest_info_tool, hub_stats_tool, extract_image_text, add, divide, multiply, subtract, calculate_cargo_travel_time, web_search_tool]
     chat_with_tools = chat.bind_tools(tools)
     
     # Generate the AgentState and Agent graph