ch

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, web_search_tool
+from tools import hub_stats_tool, extract_image_text, add, divide, multiply, subtract, calculate_cargo_travel_time, web_search_tool, wiki_search_tool, arvix_search_tool
 from retriever import guest_info_tool
 import gradio as gr
 from langfuse.langchain import CallbackHandler
@@ -35,7 +35,7 @@ def call_build():
     # Generate the chat interface, including the tools
     
     chat = ChatOpenAI(model="gpt-4o")
-    tools = [guest_info_tool, hub_stats_tool, extract_image_text, add, divide, multiply, subtract, calculate_cargo_travel_time, web_search_tool]
+    tools = [guest_info_tool, hub_stats_tool, extract_image_text, add, divide, multiply, subtract, calculate_cargo_travel_time, web_search_tool, wiki_search_tool, arvix_search_tool]
     chat_with_tools = chat.bind_tools(tools)
     
     # Generate the AgentState and Agent graph