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

add some symple logging

Browse Source
This commit is contained in:
goulustis
2025-11-06 17:10:08 +08:00
parent d0026c936b
commit 8b88669bbc
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
lang_agent/graphs/routing.py
View File

@@ -28,7 +28,7 @@ from langgraph.checkpoint.memory import MemorySaver
class RoutingConfig(KeyConfig): class RoutingConfig(KeyConfig):
_target: Type = field(default_factory=lambda: RoutingGraph) _target: Type = field(default_factory=lambda: RoutingGraph)
llm_name: str = "qwen-flash" llm_name: str = "qwen-plus"
"""name of llm""" """name of llm"""
llm_provider:str = "openai" llm_provider:str = "openai"
@@ -97,12 +97,24 @@ class RoutingGraph(GraphBase):
state = self.workflow.invoke({"inp": nargs}) state = self.workflow.invoke({"inp": nargs})
msg_list = jax.tree.leaves(state) msg_list = jax.tree.leaves(state)
for e in msg_list:
if isinstance(e, BaseMessage):
e.pretty_print()
if as_raw: if as_raw:
return msg_list return msg_list
return msg_list[-1].content return msg_list[-1].content
def _validate_input(self, *nargs, **kwargs): def _validate_input(self, *nargs, **kwargs):
print("\033[93m====================INPUT MESSAGES=============================\033[0m")
for e in nargs[0]["messages"]:
if isinstance(e, BaseMessage):
e.pretty_print()
print("\033[93m====================END INPUT MESSAGES=============================\033[0m")
print(f"\033[93 model used: {self.config.llm_name}\033[0m")
assert len(nargs[0]["messages"]) >= 2, "need at least 1 system and 1 human message" assert len(nargs[0]["messages"]) >= 2, "need at least 1 system and 1 human message"
assert len(kwargs) == 0, "due to inp assumptions" assert len(kwargs) == 0, "due to inp assumptions"