update tools

app.py

@@ -28,42 +28,36 @@ class State(TypedDict):
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        graph_builder = StateGraph(State)
+        gb = StateGraph(State)
 
-        # ツールを定義
+        # 1) define LLM + tool
         tool = TavilySearchResults(max_results=5)
-        tools = [tool]
         llm = ChatOpenAI(model="gpt-4o", temperature=0)
-        llm_with_tools = llm.bind_tools(tools)
+        llm_with_tools = llm.bind_tools([tool])
 
-        # ノード登録：まず「チャット」ノード
+        # 2) chatbot node
         def chatbot_node(state: State):
-            # LLM+ツールを呼び出し、メッセージ履歴を更新
-            response_msg = llm_with_tools.invoke(state["messages"])
-            return {"messages": state["messages"] + [response_msg]}
+            resp = llm_with_tools.invoke(state["messages"])
+            return {"messages": state["messages"] + [resp]}
 
-        graph_builder.add_node("chatbot", chatbot_node)
+        gb.add_node("chatbot", chatbot_node)
 
-        # ツールノード登録
-        tool_node = ToolNode(tools=tools)
-        graph_builder.add_node("tools", tool_node)
+        # 3) tool node must be named "tools"
+        tools_node = ToolNode(tools=[tool])
+        gb.add_node("tools", tools_node)
 
-        # 条件付きエッジ：チャットノード→（必要なら）ツール→チャットノード
-        graph_builder.add_conditional_edges("chatbot", tools_condition)
-        # スタートからチャットへ
-        graph_builder.add_edge(START, "chatbot")
-        # ツールノードからチャットノードへ戻る
-        graph_builder.add_edge("tools", "chatbot")
+        # 4) hook up the conditional edges
+        gb.add_conditional_edges("chatbot", tools_condition)
+        gb.add_edge(START, "chatbot")
+        gb.add_edge("tools", "chatbot")
 
-        # グラフをコンパイルして保持
-        self.graph = graph_builder.compile()
+        # compile once
+        self.graph = gb.compile()
 
     def __call__(self, question: str) -> str:
-        init_state = {"messages": [question]}
-        # グラフを実行
-        final_state = self.graph(init_state)
-        # 最後のメッセージを返す
-        return final_state["messages"][-1]
+        state = {"messages": [question]}
+        out = self.graph(state)
+        return out["messages"][-1]
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """