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

move comments

Browse Source
This commit is contained in:
goulustis
2025-10-15 21:21:45 +08:00
parent d5c7845d7a
commit 7f6a954342
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lang_agent/pipeline.py
View File

@@ -76,6 +76,9 @@ class Pipeline:
def invoke(self, *nargs, as_stream:bool=False, **kwargs): def invoke(self, *nargs, as_stream:bool=False, **kwargs):
"""
as_stream (bool): for debug only, gets the agent to print its thoughts
"""
if as_stream: if as_stream:
for step in self.agent.stream(*nargs, stream_mode="values", **kwargs): for step in self.agent.stream(*nargs, stream_mode="values", **kwargs):
@@ -116,9 +119,6 @@ class Pipeline:
def chat(self, inp:str, as_stream:bool=False): def chat(self, inp:str, as_stream:bool=False):
"""
as_stream (bool): for debug only, gets the agent to print its thoughts
"""
inp = {"messages":[HumanMessage(inp)]}, {"configurable": {"thread_id": 3}} inp = {"messages":[HumanMessage(inp)]}, {"configurable": {"thread_id": 3}}
out = self.invoke(*inp, as_stream=as_stream) out = self.invoke(*inp, as_stream=as_stream)