Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

no tool for chat model

Browse Source
This commit is contained in:
goulustis
2026-01-05 23:01:47 +08:00
parent ff69e4728c
commit 96e247a4ca
1 changed files with 1 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lang_agent/graphs/routing.py
View File

@@ -57,10 +57,6 @@ class Route(BaseModel):
class RoutingGraph(GraphBase): class RoutingGraph(GraphBase):
def __init__(self, config: RoutingConfig): def __init__(self, config: RoutingConfig):
self.config = config self.config = config
# NOTE: tool that the chatbranch should have
self.chat_tool_names = ["retrieve",
"get_resources"]
self._build_modules() 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"]] 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): def _build_modules(self):
self.chat_llm = init_chat_model(model=self.config.llm_name, self.chat_llm = init_chat_model(model=self.config.llm_name,
model_provider=self.config.llm_provider, model_provider=self.config.llm_provider,
@@ -90,7 +82,7 @@ class RoutingGraph(GraphBase):
self.router = self.fast_llm.with_structured_output(Route) self.router = self.fast_llm.with_structured_output(Route)
tool_manager:ToolManager = self.config.tool_manager_config.setup() 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, self.tool_node:ToolNodeBase = self.config.tool_node_config.setup(tool_manager=tool_manager,
memory=self.memory) memory=self.memory)