s

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, wiki_search_tool, arvix_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, excel_search_tool
 from retriever import guest_info_tool
 import gradio as gr
 from langfuse.langchain import CallbackHandler
@@ -18,7 +18,7 @@ with open("system_prompt.txt", "r", encoding="utf-8") as f:
     system_prompt = f.read()
 
 # System message
-    sys_msg = SystemMessage(content=system_prompt)
+    
 
     
     
@@ -30,7 +30,6 @@ with open("system_prompt.txt", "r", encoding="utf-8") as f:
     
 def call_build():
     
-    
     # Initialize the web search tool
    # search_tool = GoogleSearchTool(provider = "serpapi")
    # search_tool.name = "GoogleSearchTool"
@@ -39,7 +38,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, wiki_search_tool, arvix_search_tool]
+    tools = [guest_info_tool, hub_stats_tool, excel_search_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
@@ -50,7 +49,7 @@ def call_build():
         return {
             "messages": [chat_with_tools.invoke(state["messages"])],
         }
-    
+    sys_msg = SystemMessage(content=system_prompt)
     ## The graph
     builder = StateGraph(AgentState)