From cdc30c3c3254d564c09180fe73f7c452af1ded48 Mon Sep 17 00:00:00 2001 From: goulustis Date: Mon, 5 Jan 2026 20:50:29 +0800 Subject: [PATCH] make streamable tags a thing --- lang_agent/graphs/routing.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py index 10e2cfa..dffe26f 100644 --- a/lang_agent/graphs/routing.py +++ b/lang_agent/graphs/routing.py @@ -65,10 +65,11 @@ class RoutingGraph(GraphBase): self._build_modules() self.workflow = self._build_graph() + + self.streamable_tags:List[List[str]] = self.tool_node.get_streamable_tags() + [["route_chat_llm"]] def _stream_result(self, *nargs, **kwargs): - streamable_tags = self.tool_node.get_streamable_tags() + [["route_chat_llm"]] def text_iterator(): for chunk, metadata in self.workflow.stream({"inp": nargs}, @@ -79,7 +80,7 @@ class RoutingGraph(GraphBase): chunk, metadata = metadata tags = metadata.get("tags") - if not (tags in streamable_tags): + if not (tags in self.streamable_tags): continue if isinstance(chunk, (BaseMessageChunk, BaseMessage)) and getattr(chunk, "content", None):