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

update invoke

Browse Source
This commit is contained in:
goulustis
2025-10-22 16:29:45 +08:00
parent 77252c3536
commit c86ce019b2
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lang_agent/graphs/routing.py
View File

@@ -59,7 +59,13 @@ class RoutingGraph(GraphBase):
def invoke(self, *nargs, as_stream:bool=False, **kwargs): def invoke(self, *nargs, as_stream:bool=False, **kwargs):
assert len(kwargs) == 0, "due to inp assumptions" assert len(kwargs) == 0, "due to inp assumptions"
state = self.workflow.invoke({"inp": nargs}) if as_stream:
for step in self.workflow.stream({"inp": nargs}, stream_mode="values", **kwargs):
step["messages"][-1].pretty_print()
state = step
else:
state = self.workflow.invoke({"inp": nargs})
return state["output"] return state["output"]
def _build_modules(self): def _build_modules(self):