236 lines
7.6 KiB
Python
236 lines
7.6 KiB
Python
"""unify system tools as reusable tool resources
|
|
|
|
Revision ID: 20260804_0012
|
|
Revises: 20260804_0011
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Sequence
|
|
import json
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
revision: str = "20260804_0012"
|
|
down_revision: str | Sequence[str] | None = "20260804_0011"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
SYSTEM_TOOLS = {
|
|
"end_conversation": {
|
|
"id": "tool_end_call_default",
|
|
"name": "结束对话",
|
|
"function_name": "end_conversation",
|
|
"description": (
|
|
"礼貌地结束本次对话。当用户明确告别、表示任务已完成"
|
|
"或要求挂断时调用。"
|
|
),
|
|
},
|
|
"update_state": {
|
|
"id": "tool_update_state_default",
|
|
"name": "更新状态",
|
|
"function_name": "update_state",
|
|
"description": (
|
|
"静默更新本次对话中已经声明并明确授权的动态变量。"
|
|
"只提交本轮获得或确认的信息，更新后继续当前回答。"
|
|
),
|
|
},
|
|
"skip_turn": {
|
|
"id": "tool_skip_turn_default",
|
|
"name": "跳过本轮",
|
|
"function_name": "skip_turn",
|
|
"description": (
|
|
"跳过当前轮次，不生成任何语音回复。"
|
|
"仅当用户明确要求稍等、话还没说完，或输入可确认只是噪音时调用。"
|
|
),
|
|
},
|
|
"request_human_handoff": {
|
|
"id": "tool_request_human_handoff_default",
|
|
"name": "转接人工",
|
|
"function_name": "request_human_handoff",
|
|
"description": (
|
|
"提交人工接管请求。当用户明确要求人工服务、投诉升级或 AI 无法"
|
|
"解决时调用。该工具只提交请求，不代表人工已经接通；调用后继续"
|
|
"回复用户并说明正在等待人工响应。"
|
|
),
|
|
},
|
|
}
|
|
|
|
|
|
def _definition(kind: str) -> dict:
|
|
config: dict[str, object] = {"kind": kind}
|
|
if kind == "end_conversation":
|
|
config.update(
|
|
{
|
|
"message_type": "none",
|
|
"custom_message": "",
|
|
"capture_reason": True,
|
|
}
|
|
)
|
|
return {"schema_version": 1, "type": "system", "config": config}
|
|
|
|
|
|
def _copy_global_agent_config(data: dict, settings: dict) -> None:
|
|
"""Freeze an inherited Agent before moving its node-local System tools."""
|
|
key_pairs = {
|
|
"defaultLlmResourceId": "llmResourceId",
|
|
"defaultAsrResourceId": "asrResourceId",
|
|
"defaultTtsResourceId": "ttsResourceId",
|
|
}
|
|
for source, target in key_pairs.items():
|
|
data[target] = settings.get(source)
|
|
for key in (
|
|
"visionEnabled",
|
|
"visionModelResourceId",
|
|
"knowledgeBaseId",
|
|
"knowledgeMode",
|
|
"knowledgeTopN",
|
|
"knowledgeScoreThreshold",
|
|
"enableInterrupt",
|
|
"turnConfig",
|
|
):
|
|
if key in settings:
|
|
data[key] = settings[key]
|
|
data["toolIds"] = list(settings.get("toolIds") or [])
|
|
data["inheritGlobalConfig"] = False
|
|
|
|
|
|
def upgrade() -> None:
|
|
connection = op.get_bind()
|
|
for kind, spec in SYSTEM_TOOLS.items():
|
|
if kind == "end_conversation":
|
|
connection.execute(
|
|
sa.text(
|
|
"""
|
|
UPDATE tools
|
|
SET name = :name,
|
|
function_name = :function_name,
|
|
type = 'system',
|
|
description = :description,
|
|
definition = CAST(:definition AS jsonb),
|
|
status = 'active',
|
|
updated_at = now()
|
|
WHERE id = :id
|
|
"""
|
|
),
|
|
{**spec, "definition": json.dumps(_definition(kind))},
|
|
)
|
|
connection.execute(
|
|
sa.text(
|
|
"""
|
|
INSERT INTO tools
|
|
(id, name, function_name, type, description, definition, secrets, status)
|
|
VALUES
|
|
(:id, :name, :function_name, 'system', :description,
|
|
CAST(:definition AS jsonb), '{}'::jsonb, 'active')
|
|
ON CONFLICT (function_name) DO NOTHING
|
|
"""
|
|
),
|
|
{**spec, "definition": json.dumps(_definition(kind))},
|
|
)
|
|
|
|
assistants = connection.execute(
|
|
sa.text("SELECT id, system_tools, graph FROM assistants")
|
|
).mappings().all()
|
|
for assistant in assistants:
|
|
assistant_id = str(assistant["id"])
|
|
for kind in assistant["system_tools"] or []:
|
|
spec = SYSTEM_TOOLS.get(str(kind))
|
|
if not spec:
|
|
continue
|
|
connection.execute(
|
|
sa.text(
|
|
"""
|
|
INSERT INTO assistant_tool_bindings (assistant_id, tool_id)
|
|
VALUES (:assistant_id, :tool_id)
|
|
ON CONFLICT DO NOTHING
|
|
"""
|
|
),
|
|
{"assistant_id": assistant_id, "tool_id": spec["id"]},
|
|
)
|
|
|
|
graph = assistant["graph"] or {}
|
|
if not isinstance(graph, dict):
|
|
continue
|
|
settings = graph.get("settings") or {}
|
|
changed = False
|
|
for node in graph.get("nodes") or []:
|
|
if node.get("type") != "agent":
|
|
continue
|
|
data = node.get("data") or {}
|
|
had_legacy_selection = "systemTools" in data
|
|
kinds = list(dict.fromkeys(data.pop("systemTools", []) or []))
|
|
if had_legacy_selection:
|
|
node["data"] = data
|
|
changed = True
|
|
if not kinds:
|
|
continue
|
|
if data.get("inheritGlobalConfig", True):
|
|
_copy_global_agent_config(data, settings)
|
|
tool_ids = list(data.get("toolIds") or [])
|
|
for kind in kinds:
|
|
spec = SYSTEM_TOOLS.get(str(kind))
|
|
if spec and spec["id"] not in tool_ids:
|
|
tool_ids.append(spec["id"])
|
|
connection.execute(
|
|
sa.text(
|
|
"""
|
|
INSERT INTO assistant_tool_bindings (assistant_id, tool_id)
|
|
VALUES (:assistant_id, :tool_id)
|
|
ON CONFLICT DO NOTHING
|
|
"""
|
|
),
|
|
{"assistant_id": assistant_id, "tool_id": spec["id"]},
|
|
)
|
|
data["toolIds"] = tool_ids
|
|
node["data"] = data
|
|
if changed:
|
|
connection.execute(
|
|
sa.text("UPDATE assistants SET graph = CAST(:graph AS json) WHERE id = :id"),
|
|
{"id": assistant_id, "graph": json.dumps(graph)},
|
|
)
|
|
|
|
op.drop_column("assistants", "system_tools")
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.add_column(
|
|
"assistants",
|
|
sa.Column(
|
|
"system_tools",
|
|
sa.JSON(),
|
|
server_default=sa.text("'[]'"),
|
|
nullable=False,
|
|
),
|
|
)
|
|
op.execute(
|
|
"""
|
|
DELETE FROM assistant_tool_bindings
|
|
WHERE tool_id IN (
|
|
'tool_update_state_default',
|
|
'tool_skip_turn_default',
|
|
'tool_request_human_handoff_default'
|
|
)
|
|
"""
|
|
)
|
|
op.execute(
|
|
"""
|
|
DELETE FROM tools
|
|
WHERE id IN (
|
|
'tool_update_state_default',
|
|
'tool_skip_turn_default',
|
|
'tool_request_human_handoff_default'
|
|
)
|
|
"""
|
|
)
|
|
op.execute(
|
|
"""
|
|
UPDATE tools
|
|
SET function_name = 'end_call'
|
|
WHERE id = 'tool_end_call_default'
|
|
"""
|
|
)
|