mirror of
https://github.com/langgenius/dify.git
synced 2026-03-19 05:37:42 +08:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from core.app_assets.entities import ToolReference
|
|
from core.tools.entities.tool_entities import ToolProviderType
|
|
|
|
|
|
class ToolDependency(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
type: ToolProviderType
|
|
provider: str
|
|
tool_name: str
|
|
|
|
|
|
class ToolArtifact(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
dependencies: list[ToolDependency] = Field(default_factory=list, description="List of tool dependencies")
|
|
|
|
references: list[ToolReference] = Field(default_factory=list, description="List of tool references")
|
|
|
|
"""
|
|
Filter the tool artifact to only include the given tools
|
|
|
|
:param tools: Tuple of (provider, tool_name)
|
|
:return: Filtered tool artifact
|
|
"""
|
|
|
|
def is_empty(self) -> bool:
|
|
return not self.dependencies and not self.references
|
|
|
|
def filter(self, tools: list[tuple[str, str]]) -> "ToolArtifact":
|
|
tool_names = {f"{provider}.{tool_name}" for provider, tool_name in tools}
|
|
return ToolArtifact(
|
|
dependencies=[
|
|
dependency
|
|
for dependency in self.dependencies
|
|
if f"{dependency.provider}.{dependency.tool_name}" in tool_names
|
|
],
|
|
references=[
|
|
reference
|
|
for reference in self.references
|
|
if f"{reference.provider}.{reference.tool_name}" in tool_names
|
|
],
|
|
)
|