From bbec77152de8ca4ac32a68ee6db6146b3adf0b66 Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Tue, 2 Dec 2025 13:41:50 +0800
Subject: [PATCH] update import path

---
 lang_agent/graphs/react.py      |  2 +-
 lang_agent/graphs/routing.py    |  2 +-
 lang_agent/graphs/tool_nodes.py | 11 +++++++----
 3 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/lang_agent/graphs/react.py b/lang_agent/graphs/react.py
index 24ec65a..0279ca1 100644
--- a/lang_agent/graphs/react.py
+++ b/lang_agent/graphs/react.py
@@ -4,7 +4,7 @@ import tyro
 import jax
 
 from lang_agent.config import KeyConfig
-from lang_agent.tool_manager import ToolManager, ToolManagerConfig
+from lang_agent.components.tool_manager import ToolManager, ToolManagerConfig
 from lang_agent.base import GraphBase
 
 from langchain.chat_models import init_chat_model
diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py
index a746b7a..598dd77 100644
--- a/lang_agent/graphs/routing.py
+++ b/lang_agent/graphs/routing.py
@@ -10,7 +10,7 @@ import glob
 import time
 
 from lang_agent.config import KeyConfig
-from lang_agent.tool_manager import ToolManager, ToolManagerConfig
+from lang_agent.components.tool_manager import ToolManager, ToolManagerConfig
 from lang_agent.base import GraphBase, ToolNodeBase
 from lang_agent.graphs.graph_states import State
 from lang_agent.graphs.tool_nodes import AnnotatedToolNode, ToolNodeConfig
diff --git a/lang_agent/graphs/tool_nodes.py b/lang_agent/graphs/tool_nodes.py
index 90cd47f..70d427a 100644
--- a/lang_agent/graphs/tool_nodes.py
+++ b/lang_agent/graphs/tool_nodes.py
@@ -3,9 +3,10 @@ from typing import Type, TypedDict, Literal, Dict, List, Tuple
 import tyro
 import os.path as osp
 import time
+from loguru import logger
 
 from lang_agent.config import InstantiateConfig, KeyConfig
-from lang_agent.tool_manager import ToolManager
+from lang_agent.components.tool_manager import ToolManager
 from lang_agent.base import ToolNodeBase
 from lang_agent.graphs.graph_states import State, ChattyToolState
 from lang_agent.utils import make_llm, words_only
@@ -152,7 +153,7 @@ class ChattyToolNode(ToolNodeBase):
 
             # NOTE: words generate faster than speech
             content = words_only(outs[-1].content)
-            time.sleep(len(content) * 0.22) # 0.22 = sec/words
+            time.sleep(len(content) * 0.20) # 0.22 = sec/words
 
         
         return {"chatty_messages": {"messages":outs}}
@@ -165,8 +166,10 @@ class ChattyToolNode(ToolNodeBase):
                 SystemMessage(
                     "REPEAT THE LAST MESSAGE AND DO NOTHING ELSE!"
                 ),
-                tool_msgs[-1].content
+                HumanMessage(tool_msgs[-1].content)
               ]
+        
+        logger.info("!!!!!!!!!!!!! tool result is out !!!!!!!!!!!!!!!!!!!!!")
         self.reit_llm.invoke(inp)
         return {}
 
@@ -223,4 +226,4 @@ def debug_chatty_node():
             print(chunk.content, end="", flush=True)
 
 if __name__ == "__main__":
-    pass
\ No newline at end of file
+    debug_chatty_node()
\ No newline at end of file