From 8b88669bbcbb282a9085a80a529d463d649b058a Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Thu, 6 Nov 2025 17:10:08 +0800
Subject: [PATCH] add some symple logging

---
 lang_agent/graphs/routing.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py
index e298695..5e1c0d0 100644
--- a/lang_agent/graphs/routing.py
+++ b/lang_agent/graphs/routing.py
@@ -28,7 +28,7 @@ from langgraph.checkpoint.memory import MemorySaver
 class RoutingConfig(KeyConfig):
     _target: Type = field(default_factory=lambda: RoutingGraph)
 
-    llm_name: str = "qwen-flash"
+    llm_name: str = "qwen-plus"
     """name of llm"""
 
     llm_provider:str = "openai"
@@ -97,12 +97,24 @@ class RoutingGraph(GraphBase):
             state = self.workflow.invoke({"inp": nargs})
         
         msg_list = jax.tree.leaves(state)
+
+        for e in msg_list:
+            if isinstance(e, BaseMessage):
+                e.pretty_print()
+                
         if as_raw:
             return msg_list
 
         return msg_list[-1].content 
     
     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(kwargs) == 0, "due to inp assumptions"