mirror of
https://github.com/langgenius/dify.git
synced 2026-02-15 13:00:50 -05:00
0495dc5085
- Added an `enabled` field to `DifyCliToolConfig` and `ToolDependency` to manage tool activation status. - Updated `DifyCliConfig` to handle tool dependencies more effectively, ensuring only enabled tools are processed. - Refactored `SkillCompiler` to utilize `tool_id` for better identification of tools and improved handling of disabled tools. - Introduced a new method `_extract_disabled_tools` in `LLMNode` to streamline the extraction of disabled tools from node data. - Enhanced metadata parsing to account for tool enablement, improving overall tool management.
59 lines
1.4 KiB
Python
59 lines
1.4 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}
|
|
|
|
|
|
def create_tool_id(provider: str, tool_name: str) -> str:
|
|
return f"{provider}.{tool_name}"
|
|
|
|
|
|
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
|
|
|
|
def reference_id(self) -> str:
|
|
return f"{self.provider}.{self.tool_name}.{self.uuid}"
|
|
|
|
def tool_id(self) -> str:
|
|
return f"{self.provider}.{self.tool_name}"
|
|
|
|
|
|
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)
|