Skill 是 Agent 的具体技能：模块化 AI 能力设计

from abc import ABC, abstractmethod
from typing import Any

class Skill(ABC):
    """Skill 基类"""

    @property
    @abstractmethod
    def name(self) -> str:
        """技能名称"""
        pass

    @property
    @abstractmethod
    def description(self) -> str:
        """技能描述（给 AI 看的）"""
        pass

    @property
    @abstractmethod
    def triggers(self) -> list[str]:
        """触发条件（用户说什么时激活）"""
        pass

    @property
    def tools_required(self) -> list[str]:
        """需要的工具列表"""
        return []

    @abstractmethod
    def execute(self, context: dict) -> dict:
        """执行技能，返回结果"""
        pass

class SkillRegistry:
    """技能注册中心"""

    def __init__(self):
        self.skills: dict[str, Skill] = {}

    def register(self, skill: Skill):
        self.skills[skill.name] = skill

    def find_by_trigger(self, user_message: str) -> Skill | None:
        """根据用户消息找到匹配的技能"""
        for skill in self.skills.values():
            for trigger in skill.triggers:
                if trigger in user_message:
                    return skill
        return None

    def list_skills(self) -> list[dict]:
        """列出所有可用技能"""
        return [
            {
                "name": s.name,
                "description": s.description,
                "triggers": s.triggers,
            }
            for s in self.skills.values()
        ]

class CodeReviewSkill(Skill):
    """代码审查技能"""

    @property
    def name(self) -> str:
        return "code_review"

    @property
    def description(self) -> str:
        return "对代码变更进行全面审查，包括逻辑、安全、性能和风格"

    @property
    def triggers(self) -> list[str]:
        return ["代码审查", "code review", "帮我看看代码", "review"]

    @property
    def tools_required(self) -> list[str]:
        return ["read_file", "search_code", "run_command"]

    def execute(self, context: dict) -> dict:
        diff = context.get("diff", "")

        # 1. 读取相关代码
        files = self._extract_changed_files(diff)

        # 2. 并行分析
        results = {
            "logic": self._check_logic(files),
            "security": self._check_security(files),
            "performance": self._check_performance(files),
            "style": self._check_style(files),
        }

        # 3. 生成报告
        return self._generate_report(results)

from langchain.agents import AgentExecutor, create_react_agent
from langchain.tools import Tool

# 定义 Skill 为一组 Tool 的组合
def code_review_skill(inputs):
    """代码审查 Skill"""
    diff = inputs["diff"]

    # Step 1: 分析
    analysis = llm.invoke(f"分析以下代码变更的潜在问题：\n{diff}")

    # Step 2: 安全检查
    security = llm.invoke(f"检查以下代码的安全隐患：\n{diff}")

    # Step 3: 生成报告
    report = llm.invoke(f"基于以下分析生成审查报告：\n分析：{analysis}\n安全：{security}")

    return report

def execute(self, context: dict) -> dict:
    try:
        # 主逻辑
        result = self._do_work(context)
        return {"status": "success", "result": result}
    except ToolTimeoutError:
        return {"status": "error", "message": "工具执行超时，请稍后重试"}
    except FileNotFoundError:
        return {"status": "error", "message": f"文件不存在: {context['file_path']}"}
    except Exception as e:
        return {"status": "error", "message": f"未知错误: {str(e)}"}

# 每个 Skill 都应该能独立测试
def test_code_review_skill():
    skill = CodeReviewSkill()
    context = {
        "diff": "def add(a, b): return a + b  # 简单测试用例"
    }
    result = skill.execute(context)

    assert result["status"] == "success"
    assert "review" in result