feat: 查询直属下级员工工具

app/controller/add_langgraph_chat_route.py

@@ -12,10 +12,7 @@ from starlette.responses import StreamingResponse
 
 from app.controller.add_user_route import oauth2_scheme
 from app.model.ConversationModel import ConversationService
-from app.tools.tool_add import tool_add
+from app.tools.tool_list import tool_list
-from app.tools.tool_book_hotel import tool_book_hotel
-from app.tools.tool_get_datetime import tool_get_datetime
-from app.tools.tool_multiple import tool_multiply
 from app.utils.db_utils import AsyncSessionDep
 from app.utils.llm_utils import create_llm
 from app.utils.postgres_checkpointer import PostgresCheckpointerManager, AsyncPostgresSaverDep
@@ -41,12 +38,7 @@ class ChatAgent:
     if not ChatAgent.agent:
       ChatAgent.agent = create_react_agent(
         model=create_llm("huoshan-think-pro"),
-        tools=[
+        tools=tool_list,
-          tool_book_hotel,
-          tool_get_datetime,
-          tool_add,
-          tool_multiply,
-        ],
         checkpointer=await PostgresCheckpointerManager.get_instance(),
         prompt="""
         - 你是一名擅长使用工具的智能助手，你需要根据用户问题来进行回答，请使用中文进行回答。

app/tools/tool_list.py

@@ -0,0 +1,13 @@
+from app.tools.tool_add import tool_add
+from app.tools.tool_book_hotel import tool_book_hotel
+from app.tools.tool_get_datetime import tool_get_datetime
+from app.tools.tool_multiple import tool_multiply
+from app.tools.tool_query_direct_subordinates import tool_query_direct_subordinates
+
+tool_list = [
+  tool_book_hotel,
+  tool_get_datetime,
+  tool_add,
+  tool_multiply,
+  tool_query_direct_subordinates,
+]

app/tools/tool_query_direct_subordinates.py

@@ -0,0 +1,15 @@
+from langchain_core.tools import tool
+
+
+@tool(
+  name_or_callable="tool_query_direct_subordinates",
+  description="查询直接下级员工信息"
+)
+def tool_query_direct_subordinates() -> float:
+  return [
+    "已经查询完毕",
+    {
+      "component": "DirectSubordinates",
+      "props": {},
+    }
+  ]