78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
|
"""工具元数据和数据类型定义"""
|
|||
|
|
|
|||
|
|
from dataclasses import dataclass, field
|
|||
|
|
from enum import Enum
|
|||
|
|
from typing import Any
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ToolCategory(Enum):
|
|||
|
|
"""工具类别"""
|
|||
|
|
|
|||
|
|
READ = "read"
|
|||
|
|
WRITE = "write"
|
|||
|
|
EXTERNAL = "external"
|
|||
|
|
DB_WRITE = "db_write"
|
|||
|
|
NETWORK = "network"
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SideEffectScope(Enum):
|
|||
|
|
"""副作用范围"""
|
|||
|
|
|
|||
|
|
NONE = "none"
|
|||
|
|
LOCAL_STATE = "local_state"
|
|||
|
|
DB_WRITE = "db_write"
|
|||
|
|
NETWORK = "network"
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PermissionClass(Enum):
|
|||
|
|
"""权限级别"""
|
|||
|
|
|
|||
|
|
READ = "read"
|
|||
|
|
WRITE = "write"
|
|||
|
|
EXTERNAL = "external"
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class ToolManifest:
|
|||
|
|
"""工具元数据"""
|
|||
|
|
|
|||
|
|
name: str
|
|||
|
|
description: str
|
|||
|
|
category: ToolCategory
|
|||
|
|
parameters: dict[str, Any] # JSON Schema
|
|||
|
|
return_schema: dict[str, Any]
|
|||
|
|
permission_class: PermissionClass
|
|||
|
|
side_effect_scope: SideEffectScope
|
|||
|
|
requires_confirmation: bool = False
|
|||
|
|
is_streaming: bool = False
|
|||
|
|
tags: list[str] = field(default_factory=list)
|
|||
|
|
|
|||
|
|
def to_dict(self) -> dict[str, Any]:
|
|||
|
|
return {
|
|||
|
|
"name": self.name,
|
|||
|
|
"description": self.description,
|
|||
|
|
"category": self.category.value,
|
|||
|
|
"parameters": self.parameters,
|
|||
|
|
"return_schema": self.return_schema,
|
|||
|
|
"permission_class": self.permission_class.value,
|
|||
|
|
"side_effect_scope": self.side_effect_scope.value,
|
|||
|
|
"requires_confirmation": self.requires_confirmation,
|
|||
|
|
"is_streaming": self.is_streaming,
|
|||
|
|
"tags": self.tags,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class HookConfig:
|
|||
|
|
"""Hook 配置"""
|
|||
|
|
|
|||
|
|
name: str
|
|||
|
|
hook_type: str # "pre_tool_use", "post_tool_use", "tool_error", "tool_skip"
|
|||
|
|
filter_names: list[str] | None = None # 只对特定工具生效，None 表示全部
|
|||
|
|
|
|||
|
|
def matches_tool(self, tool_name: str) -> bool:
|
|||
|
|
"""检查 Hook 是否对指定工具生效"""
|
|||
|
|
if self.filter_names is None:
|
|||
|
|
return True
|
|||
|
|
return tool_name in self.filter_names
|