66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
"""Capability gates for Prompt tools in speech-to-speech runtimes."""
|
||
|
||
from __future__ import annotations
|
||
|
||
from collections.abc import Mapping
|
||
|
||
from services.system_tools import RESERVED_TOOL_NAMES
|
||
|
||
|
||
QWEN_PROMPT_TOOL_FEATURES = {
|
||
"externalTypes": ["http", "mcp", "client"],
|
||
"builtIns": [
|
||
"endConversation",
|
||
"updateState",
|
||
"requestHumanHandoff",
|
||
],
|
||
}
|
||
|
||
|
||
def realtime_prompt_tool_features(interface_type: str) -> dict[str, list[str]]:
|
||
"""Return the server-trusted runtime capabilities for one provider."""
|
||
|
||
if interface_type == "qwen-audio-realtime":
|
||
return {
|
||
key: list(values) for key, values in QWEN_PROMPT_TOOL_FEATURES.items()
|
||
}
|
||
return {"externalTypes": [], "builtIns": []}
|
||
|
||
|
||
def validate_runtime_prompt_tools(
|
||
*,
|
||
interface_type: str,
|
||
external_types: set[str],
|
||
built_in_tools: Mapping[str, object] | None,
|
||
external_function_names: set[str] | None = None,
|
||
) -> None:
|
||
"""Reject stale or inline runtime configurations before opening a socket."""
|
||
|
||
features = realtime_prompt_tool_features(interface_type)
|
||
unsupported_external = external_types - set(features["externalTypes"])
|
||
reserved_external = set(external_function_names or set()) & RESERVED_TOOL_NAMES
|
||
aliases = {
|
||
"end_conversation": "endConversation",
|
||
"update_state": "updateState",
|
||
"skip_turn": "skipTurn",
|
||
"request_human_handoff": "requestHumanHandoff",
|
||
}
|
||
requested_built_ins = {
|
||
aliases.get(str(name), str(name))
|
||
for name, config in (built_in_tools or {}).items()
|
||
if config is not None
|
||
}
|
||
unsupported_built_ins = requested_built_ins - set(features["builtIns"])
|
||
if unsupported_external or unsupported_built_ins or reserved_external:
|
||
details = []
|
||
if unsupported_external:
|
||
details.append("外部工具:" + ",".join(sorted(unsupported_external)))
|
||
if unsupported_built_ins:
|
||
details.append("内建工具:" + ",".join(sorted(unsupported_built_ins)))
|
||
if reserved_external:
|
||
details.append("外部工具使用平台保留名:" + ",".join(sorted(reserved_external)))
|
||
raise ValueError(
|
||
f"Realtime 接口 {interface_type or 'unknown'} 不支持 "
|
||
+ "；".join(details)
|
||
)
|