mirror of
https://github.com/langgenius/dify.git
synced 2026-02-13 07:01:23 -05:00
48 lines
1.1 KiB
Python
48 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
|
|
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)
|