From 96e247a4ca3357dafc6107d8a978d48da818bc66 Mon Sep 17 00:00:00 2001 From: goulustis Date: Mon, 5 Jan 2026 23:01:47 +0800 Subject: [PATCH] no tool for chat model --- lang_agent/graphs/routing.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py index 1e5e3d0..ee2315d 100644 --- a/lang_agent/graphs/routing.py +++ b/lang_agent/graphs/routing.py @@ -57,10 +57,6 @@ class Route(BaseModel): class RoutingGraph(GraphBase): def __init__(self, config: RoutingConfig): self.config = config - - # NOTE: tool that the chatbranch should have - self.chat_tool_names = ["retrieve", - "get_resources"] self._build_modules() @@ -68,10 +64,6 @@ class RoutingGraph(GraphBase): self.streamable_tags:List[List[str]] = self.tool_node.get_streamable_tags() + [["route_chat_llm"]] - - def _get_chat_tools(self, man:ToolManager): - return [lang_tool for lang_tool in man.get_list_langchain_tools() if lang_tool.name in self.chat_tool_names] - def _build_modules(self): self.chat_llm = init_chat_model(model=self.config.llm_name, model_provider=self.config.llm_provider, @@ -90,7 +82,7 @@ class RoutingGraph(GraphBase): self.router = self.fast_llm.with_structured_output(Route) tool_manager:ToolManager = self.config.tool_manager_config.setup() - self.chat_model = create_agent(self.chat_llm, self._get_chat_tools(tool_manager), checkpointer=self.memory) + self.chat_model = create_agent(self.chat_llm, [], checkpointer=self.memory) self.tool_node:ToolNodeBase = self.config.tool_node_config.setup(tool_manager=tool_manager, memory=self.memory)