mirror of
https://github.com/langgenius/dify.git
synced 2026-05-13 08:57:28 +08:00
144089d3ed
- Introduced a new regex pattern for tool groups to support multiple tool placeholders. - Updated the DefaultToolResolver to format outputs for specific built-in tools (bash, python). - Enhanced the SkillCompiler to filter out disabled tools in tool groups, ensuring only enabled tools are rendered. - Added tests to verify the correct behavior of tool group filtering and rendering.
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from core.tools.entities.tool_entities import ToolProviderType
|
|
|
|
|
|
class ToolFieldConfig(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
id: str
|
|
value: Any
|
|
auto: bool = False
|
|
|
|
|
|
class ToolConfiguration(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
fields: list[ToolFieldConfig] = Field(default_factory=list)
|
|
|
|
def default_values(self) -> dict[str, Any]:
|
|
return {field.id: field.value for field in self.fields if field.value is not None}
|
|
|
|
|
|
class ToolReference(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
uuid: str
|
|
type: ToolProviderType
|
|
provider: str
|
|
tool_name: str
|
|
enabled: bool = True
|
|
credential_id: str | None = None
|
|
configuration: ToolConfiguration | None = None
|
|
|
|
|
|
class FileReference(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
source: str
|
|
asset_id: str
|
|
|
|
|
|
class SkillMetadata(BaseModel):
|
|
model_config = ConfigDict(extra="allow")
|
|
|
|
tools: dict[str, ToolReference] = Field(default_factory=dict)
|
|
files: list[FileReference] = Field(default_factory=list)
|