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

make streamable_tags property

Browse Source
This commit is contained in:
goulustis
2026-01-05 21:09:15 +08:00
parent 99b100cce1
commit 18494b1aea
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lang_agent/graphs/routing.py
View File

@@ -137,7 +137,6 @@ class RoutingGraph(GraphBase):
async def _astream_result(self, *nargs, **kwargs) -> AsyncIterator[str]: async def _astream_result(self, *nargs, **kwargs) -> AsyncIterator[str]:
"""Async streaming using LangGraph's astream method.""" """Async streaming using LangGraph's astream method."""
streamable_tags = self.tool_node.get_streamable_tags() + [["route_chat_llm"]]
async def text_iterator(): async def text_iterator():
async for chunk, metadata in self.workflow.astream( async for chunk, metadata in self.workflow.astream(
@@ -150,7 +149,7 @@ class RoutingGraph(GraphBase):
chunk, metadata = metadata chunk, metadata = metadata
tags = metadata.get("tags") tags = metadata.get("tags")
if not (tags in streamable_tags): if not (tags in self.streamable_tags):
continue continue
if isinstance(chunk, (BaseMessageChunk, BaseMessage)) and getattr(chunk, "content", None): if isinstance(chunk, (BaseMessageChunk, BaseMessage)) and getattr(chunk, "content", None):