14e297a488
- 新增 pam_deploy_graph 包，包含 Agent runtime、ActionRouter、脚本/MCP/fake runner - 支持 hybrid_node_mcp 策略：PAM_HOME 走脚本 action，PAM_NODE 走 MCP - 支持 script_only 离线策略，全部 action 走现有脚本 action - 新增 LLM structured output 骨架和规则 fallback，支持意图识别、参数抽取、计划生成 - 新增 LangGraph StateGraph 工厂和 MCP client adapter - 新增 CLI：preview、analyze、run-global、run-deploy - 增加 fake 完整部署流程、单 IP 失败待回滚确认状态和报告输出 - 增加单元测试覆盖路由、parser、runner、Skill 加载、LLM 输出、MCP adapter 和 LangGraph 图 - 更新 README，记录当前代码骨架、进度、使用方式和下一步计划
103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
"""Shared dataclasses for the PAM deploy agent."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass, field
|
|
from typing import Any, Literal
|
|
|
|
BackendName = Literal["mcp", "script", "fake"]
|
|
ExecutionStrategy = Literal["hybrid_node_mcp", "script_only", "fake"]
|
|
IntentName = Literal["deploy", "show_usage", "preview", "query_node_ips", "rollback"]
|
|
ModePreference = Literal["MCP", "API脚本", "未指定"]
|
|
StrategyPreference = Literal["hybrid_node_mcp", "script_only", "fake", "未指定"]
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class ActionResult:
|
|
action: str
|
|
backend: BackendName
|
|
ok: bool
|
|
values: dict[str, Any] = field(default_factory=dict)
|
|
exit_code: int = 0
|
|
tool_name: str = ""
|
|
stdout: str = ""
|
|
stderr: str = ""
|
|
raw_output: str = ""
|
|
error_summary: str = ""
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class SkillPolicy:
|
|
name: str
|
|
source_path: str
|
|
description: str = ""
|
|
allowed_modes: tuple[str, ...] = ("MCP", "API脚本")
|
|
allowed_actions: tuple[str, ...] = ()
|
|
required_confirmations: tuple[str, ...] = (
|
|
"params",
|
|
"target_scope",
|
|
"rollback",
|
|
)
|
|
required_params: tuple[str, ...] = ()
|
|
optional_params: dict[str, Any] = field(default_factory=dict)
|
|
action_sequence: tuple[str, ...] = ()
|
|
ip_action_sequence: tuple[str, ...] = ()
|
|
forbidden_actions: tuple[str, ...] = (
|
|
"script-main-flow",
|
|
"auto-rollback",
|
|
"modify-deploy-scripts",
|
|
)
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class LlmIntentResult:
|
|
intent: IntentName
|
|
mode_preference: ModePreference = "未指定"
|
|
strategy_preference: StrategyPreference = "未指定"
|
|
confidence: float = 0.0
|
|
reasons: list[str] = field(default_factory=list)
|
|
needs_clarification: bool = False
|
|
clarification_questions: list[str] = field(default_factory=list)
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class LlmParamResult:
|
|
extracted_params: dict[str, Any] = field(default_factory=dict)
|
|
extracted_control: dict[str, Any] = field(default_factory=dict)
|
|
missing_required_params: list[str] = field(default_factory=list)
|
|
ambiguous_fields: list[str] = field(default_factory=list)
|
|
sensitive_fields_present: list[str] = field(default_factory=list)
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class LlmDeployPlan:
|
|
summary: str
|
|
risk_notes: list[str] = field(default_factory=list)
|
|
planned_actions: list[str] = field(default_factory=list)
|
|
requires_confirmation: bool = True
|
|
execution_strategy: StrategyPreference = "未指定"
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class AgentState:
|
|
run_id: str
|
|
params: dict[str, Any]
|
|
execution_strategy: ExecutionStrategy
|
|
action_backends: dict[str, BackendName]
|
|
script_entry: str = ""
|
|
script_base_dir: str = "."
|
|
config_path: str = ""
|
|
trace_file_path: str = ""
|
|
node_mcp_server_name: str = ""
|
|
node_mcp_tool_names: dict[str, str] = field(default_factory=dict)
|
|
completed_global_steps: list[str] = field(default_factory=list)
|
|
hash_code: str = ""
|
|
node_url: str = ""
|
|
online_ips: list[str] = field(default_factory=list)
|
|
target_ips: list[str] = field(default_factory=list)
|
|
ip_states: dict[str, dict[str, Any]] = field(default_factory=dict)
|
|
pending_confirmation: str = ""
|
|
last_success_step: str = ""
|
|
last_failed_step: str = ""
|
|
events: list[dict[str, Any]] = field(default_factory=list)
|