SkyLiCoder/WiseGrade
1
0
Fork 1
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

feat: raise configurable chat tool call limit

Browse Source
This commit is contained in:
李傲琦
2026-09-26 20:36:05 +08:00
parent 84bb5abb28
commit 47e60187c7
5 changed files with 49 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/wisegrade/chat_api.py
+11 -6
View File
@@ -10,6 +10,7 @@ from sqlalchemy.orm import Session, sessionmaker
from .actions import action_dict, confirm, prepare, require_scope
from .agent_tools import execute, tool_specs
from .config import get_settings
from .db import get_db
from .llm import ModelError
from .models import AgentRun, ChatMessage, Conversation, PendingAction, Project, ScoringBatch, now
@@ -113,7 +114,10 @@ def _run_agent(factory, conversation_id, run_id, user_text):
"project_id": conversation.project_id}, ensure_ascii=False)}]
messages.extend({"role": m.role, "content": m.content[:4000]} for m in history)
yield _event("run", {"id": run.id, "status": "running"})
for round_number in range(4):
max_tool_calls = get_settings().chat_max_tool_calls
attempted_calls = 0
# One extra model response lets the agent answer after its last allowed tool call.
for _ in range(max_tool_calls + 1):
reply = client.tool_call(messages, tool_specs())
calls = reply.get("tool_calls") or []
if not calls:
@@ -121,16 +125,17 @@ def _run_agent(factory, conversation_id, run_id, user_text):
if not isinstance(content, str) or not content.strip():
raise ModelError("对话模型未返回文字或工具调用")
db.add(ChatMessage(conversation_id=conversation_id, role="assistant", content=content,
citations=citations[:20], tool_calls=used[:8]))
citations=citations[:20], tool_calls=used))
run.status, run.finished_at = "completed", now()
conversation.updated_at = now()
db.commit()
completed = True
yield _event("message", {"content": content, "citations": citations[:20], "tool_calls": used[:8]})
yield _event("message", {"content": content, "citations": citations[:20], "tool_calls": used})
yield _event("done", {"run_id": run.id})
return
if len(calls) + len(used) > 8:
raise ModelError("单轮工具调用超过上限 8 次")
if attempted_calls + len(calls) > max_tool_calls:
raise ModelError(f"本次消息工具调用超过上限 {max_tool_calls} 次")
attempted_calls += len(calls)
messages.append({"role": "assistant", "content": reply.get("content"), "tool_calls": calls})
for call in calls:
name = call.get("function", {}).get("name", "")
@@ -155,7 +160,7 @@ def _run_agent(factory, conversation_id, run_id, user_text):
messages.append({"role": "tool", "tool_call_id": call_id,
"content": json.dumps({"error": error[:500]}, ensure_ascii=False)})
yield _event("tool", {"name": name, "status": "failed", "error": error[:500]})
raise ModelError("模型工具调用轮数超过上限 4 次")
raise ModelError(f"本次消息模型调用超过上限 {max_tool_calls + 1} 轮")
except Exception as exc:
run.status, run.error, run.finished_at = "failed", str(exc)[:500], now()
db.commit()