tols

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