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

better debugging

Browse Source
This commit is contained in:
goulustis
2025-10-15 20:51:23 +08:00
parent e26c3dd70d
commit bcb9ead2fb
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
lang_agent/pipeline.py
View File

@@ -115,17 +115,16 @@ class Pipeline:
if as_stream: if as_stream:
for step in self.agent.stream(*inp, stream_mode="values"): for step in self.agent.stream(*inp, stream_mode="values"):
step["messages"][-1].pretty_print() step["messages"][-1].pretty_print()
out = step
return else:
out = self.invoke(*inp) out = self.invoke(*inp)
return out['messages'][-1].content return out['messages'][-1].content
# if __name__ == "__main__": if __name__ == "__main__":
# pipeline:Pipeline = PipelineConfig().setup() pipeline:Pipeline = PipelineConfig().setup()
# print(pipeline.chat("I'm steve")) u = pipeline.chat("use the calculator tool to calculate what is 900 * 321", as_stream=True)
# print(pipeline.chat("what is my name?")) print("================out")
print(u)
# pipeline.chat("use the calculator tool to calculate what is 900 * 321", as_stream=True)