35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
"""
|
|
LLM mode 空壳注册表：按 mode 名取 invoke_shell。
|
|
|
|
本文件只做分发到各 mode_* 模块；ordered_actions 走 routing/DeepSeekA。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Callable
|
|
|
|
from agent.llm import mode_attachment_text, mode_extract_fields, mode_group_quote, mode_sheet_map, mode_vision
|
|
from agent.llm.modes import LlmMode
|
|
|
|
_SHELLS: dict[str, Callable[..., dict[str, Any]]] = {
|
|
LlmMode.EXTRACT_FIELDS.value: mode_extract_fields.invoke_shell,
|
|
LlmMode.ATTACHMENT_TEXT.value: mode_attachment_text.invoke_shell,
|
|
LlmMode.VISION.value: mode_vision.invoke_shell,
|
|
LlmMode.GROUP_QUOTE.value: mode_group_quote.invoke_shell,
|
|
LlmMode.SHEET_MAP.value: mode_sheet_map.invoke_shell,
|
|
}
|
|
|
|
|
|
def invoke_mode_shell(mode: str | LlmMode, *, payload: dict[str, Any]) -> dict[str, Any]:
|
|
"""调用对应 mode 空壳；未知 mode 返回错误结构。"""
|
|
key = mode.value if isinstance(mode, LlmMode) else str(mode)
|
|
fn = _SHELLS.get(key)
|
|
if fn is None:
|
|
return {"ok": False, "stub": True, "mode": key, "error": "unknown_or_unregistered_mode"}
|
|
return fn(payload=payload)
|
|
|
|
|
|
def list_mode_shells() -> list[str]:
|
|
"""已注册空壳 mode 名。"""
|
|
return sorted(_SHELLS.keys())
|