22 lines
733 B
Python
22 lines
733 B
Python
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
|
|
class ToolGroupConfig(BaseModel):
|
|
"""Config section for a tool group"""
|
|
|
|
name: str = Field(..., description="Unique name for the tool group")
|
|
model_config = ConfigDict(extra="allow")
|
|
|
|
|
|
class ToolConfig(BaseModel):
|
|
"""Config section for a tool"""
|
|
|
|
name: str = Field(..., description="Unique name for the tool")
|
|
group: str = Field(..., description="Group name for the tool")
|
|
enabled: bool = Field(default=True, description="Whether the tool is registered with the agent")
|
|
use: str = Field(
|
|
...,
|
|
description="Variable name of the tool provider(e.g. deerflow.sandbox.tools:bash_tool)",
|
|
)
|
|
model_config = ConfigDict(extra="allow")
|