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

moved to debug

Browse Source
This commit is contained in:
goulustis
2025-11-25 13:19:40 +08:00
parent a674e313b9
commit 5db82c62d8
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lang_agent/graphs/tool_nodes.py
View File

@@ -195,7 +195,8 @@ tool_node_dict = {
tool_node_union = tyro.extras.subcommand_type_from_defaults(tool_node_dict, prefix_names=False) tool_node_union = tyro.extras.subcommand_type_from_defaults(tool_node_dict, prefix_names=False)
AnnotatedToolNode = tyro.conf.OmitSubcommandPrefixes[tyro.conf.SuppressFixed[tool_node_union]] AnnotatedToolNode = tyro.conf.OmitSubcommandPrefixes[tyro.conf.SuppressFixed[tool_node_union]]
if __name__ == "__main__":
def debug_chatty_node():
from langchain_core.messages.base import BaseMessageChunk from langchain_core.messages.base import BaseMessageChunk
from langchain_core.messages import BaseMessage from langchain_core.messages import BaseMessage
@@ -222,3 +223,6 @@ if __name__ == "__main__":
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)
if __name__ == "__main__":
pass