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

bug fix + debug codes

Browse Source
This commit is contained in:
goulustis
2025-11-21 17:01:57 +08:00
parent 355b87102a
commit b7cf0bc983
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lang_agent/graphs/tool_nodes.py
View File

@@ -137,7 +137,7 @@ class ChattyToolNode(GraphBase):
]}, state["inp"][1] ]}, state["inp"][1]
outs.append(self.chatty_agent.invoke(*inp)) outs.append(self.chatty_agent.invoke(*inp))
return {"chatty_message": outs} return {"chatty_messages": outs}
def _handoff_node(self, state:ChattyToolState): def _handoff_node(self, state:ChattyToolState):
@@ -198,11 +198,12 @@ if __name__ == "__main__":
query = "use calculator to calculate 33*42" query = "use calculator to calculate 33*42"
input = {"inp" : ({"messages":[SystemMessage("you are a kind helper"), HumanMessage(query)]}, input = {"inp" : ({"messages":[SystemMessage("you are a kind helper"), HumanMessage(query)]},
{"configurable": {"thread_id": '3'}})} {"configurable": {"thread_id": '3'}})}
inp = input["inp"] inp = input
graph = chatty_node.workflow graph = chatty_node.workflow
for chunk, metadata in graph.stream(*inp, stream_mode="messages"): # for chunk, metadata in graph.stream(inp, stream_mode="messages"):
if isinstance(chunk, (BaseMessageChunk, BaseMessage)) and getattr(chunk, "content", None): # if isinstance(chunk, (BaseMessageChunk, BaseMessage)) and getattr(chunk, "content", None):
print(chunk.content, end="", flush=True) # print(chunk.content, end="", flush=True)
out = graph.invoke(inp)
assert 0