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

This commit is contained in:
martsforever
2025-09-04 22:18:57 +08:00
parent b123fdb3ec
commit 1ada5aaffd
3 changed files with 30 additions and 10 deletions
+2 -10
View File
@@ -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_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_list import tool_list
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=[
tool_book_hotel,
tool_get_datetime,
tool_add,
tool_multiply,
],
tools=tool_list,
checkpointer=await PostgresCheckpointerManager.get_instance(),
prompt="""
- 你是一名擅长使用工具的智能助手,你需要根据用户问题来进行回答,请使用中文进行回答。