15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

chore: update agent configuration to include recursion limit in astream call
Some checks are pending
sync2gitee / repo-sync (push) Waiting to run
Details
Typos Check / Spell Check with Typos (push) Waiting to run
Details

This commit is contained in:
CaptainB 2025-12-15 11:33:55 +08:00
parent 967eedeca3
commit 067cd73325
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/application/flow/tools.py
View File

@ -313,8 +313,9 @@ async def _yield_mcp_response(chat_model, message_list, mcp_servers, mcp_output_
try: try:
client = MultiServerMCPClient(json.loads(mcp_servers)) client = MultiServerMCPClient(json.loads(mcp_servers))
tools = await client.get_tools() tools = await client.get_tools()
agent = create_react_agent(chat_model, tools).configure(recursion_limit=(int(CONFIG.get("LANGCHAIN_GRAPH_RECURSION_LIMIT", '25')))) agent = create_react_agent(chat_model, tools)
response = agent.astream({"messages": message_list}, stream_mode='messages') recursion_limit = int(CONFIG.get("LANGCHAIN_GRAPH_RECURSION_LIMIT", '25'))
response = agent.astream({"messages": message_list}, config={"recursion_limit": recursion_limit}, stream_mode='messages')
# 用于存储工具调用信息（按 tool_id）以及按 index 聚合分片 # 用于存储工具调用信息（按 tool_id）以及按 index 聚合分片
tool_calls_info = {} tool_calls_info = {}