agent_real2

real_agent.py

@@ -7,10 +7,11 @@ from langgraph.prebuilt import tools_condition
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain_openai import ChatOpenAI
 import os
-from tools import DuckDuckGoSearchRun, weather_info_tool, hub_stats_tool
+from tools import hub_stats_tool
 from retriever import guest_info_tool
 import gradio as gr
 from langfuse.langchain import CallbackHandler
+from smolagents import VisitWebpageTool, GoogleSearchTool
 
 
 os.environ["LANGFUSE_PUBLIC_KEY"] = "pk-lf-842782bf-0e8d-4f50-8fb0-157464d5fe28" 
@@ -19,21 +20,22 @@ os.environ["LANGFUSE_HOST"] = "https://us.cloud.langfuse.com"
 
 os.environ["OPENAI_API_KEY"] = "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"
 
+os.environ["SERPAPI_API_KEY"] = "82491710ce6c6aa2d767d7796023a5ca11aa182fcda2a703cd44aea8d6da4489"
+
 
 def call_build():
     langfuse_handler = CallbackHandler()
     
     # Initialize the web search tool
-    new_search_tool = DuckDuckGoSearchRun()
+
+    search_tool = GoogleSearchTool()
     
     
     # Generate the chat interface, including the tools
-    #llm = ChatOpenAI(model="gpt-4o")
     
     chat = ChatOpenAI(model="gpt-4o")
-    tools = [guest_info_tool, new_search_tool, weather_info_tool, hub_stats_tool]
+    tools = [guest_info_tool, search_tool, hub_stats_tool, VisitWebpageTool()]
     chat_with_tools = chat.bind_tools(tools)
-    #llm_with_tools = llm.bind_tools(tools, parallel_tool_calls=False)
     
     # Generate the AgentState and Agent graph
     class AgentState(TypedDict):