#  Copyright 2026 zhaoxi826
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

from pydantic import BaseModel
from typing import List, Literal, Dict
from pydantic import ConfigDict


class BaseToolData(BaseModel):
    """所有工具插件的基类：声明工具的作用域、是否系统级以及配置参数 schema。"""

    model_config = ConfigDict(extra="allow")
    is_system: bool
    action_scope: List[
        Literal[
            "control_node",
            "consciousness_node",
            "regulatory_node",
            "growth_node",
            "",
        ]
    ] = []
    config_args: Dict[str, str] = {}
    category: str = "other"
    """工具分类：system（系统内置）、search（搜索）、mcp（MCP 服务器）、other（其他）"""
    toolset: str = "other"
    """工具集：retrieval（检索）、generation（生成）、other（其他）。system_node 只能用 retrieval 集。"""