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

add debugging tool

Browse Source
This commit is contained in:
goulustis
2025-10-15 20:47:22 +08:00
parent 183a5a605d
commit fe3c8913c9
1 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lang_agent/pipeline.py
View File

@@ -106,15 +106,26 @@ class Pipeline:
def get_ws_url(self): def get_ws_url(self):
return f"ws://{self.config.host}:{self.config.port}" return f"ws://{self.config.host}:{self.config.port}"
def chat(self, inp:str): 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}}
if as_stream:
for step in self.agent.stream(*inp, stream_mode="values"):
step["messages"][-1].pretty_print()
return
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")) # print(pipeline.chat("I'm steve"))
print(pipeline.chat("what is my name?")) # print(pipeline.chat("what is my name?"))
# pipeline.chat("use the calculator tool to calculate what is 900 * 321", as_stream=True)