mirror of
https://github.com/langgenius/dify.git
synced 2026-02-13 16:00:55 -05:00
a43efef9f0
- Replaced SkillArtifactSet with SkillBundle across various components, enhancing the organization of skill dependencies and references. - Updated SkillManager methods to load and save bundles instead of artifacts, improving clarity in asset management. - Refactored SkillCompiler to compile skills into bundles, streamlining the dependency resolution process. - Adjusted DifyCli and SandboxBashSession to utilize ToolDependencies, ensuring consistent handling of tool references. - Introduced AssetReferences for better management of file dependencies within skill bundles.
98 lines
3.6 KiB
Python
98 lines
3.6 KiB
Python
from collections.abc import Iterable
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from core.skill.entities.skill_bundle_entry import SkillBundleEntry
|
|
from core.skill.entities.skill_metadata import ToolReference
|
|
from core.skill.entities.tool_dependencies import ToolDependencies, ToolDependency
|
|
|
|
|
|
class SkillBundle(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
assets_id: str = Field(description="Assets ID this bundle belongs to")
|
|
schema_version: int = Field(default=1, description="Schema version for forward compatibility")
|
|
built_at: datetime | None = Field(default=None, description="Build timestamp")
|
|
|
|
entries: dict[str, SkillBundleEntry] = Field(default_factory=dict, description="skill_id -> SkillBundleEntry")
|
|
|
|
dependency_graph: dict[str, list[str]] = Field(
|
|
default_factory=dict,
|
|
description="skill_id -> list of skill_ids it depends on",
|
|
)
|
|
|
|
reverse_graph: dict[str, list[str]] = Field(
|
|
default_factory=dict,
|
|
description="skill_id -> list of skill_ids that depend on it",
|
|
)
|
|
|
|
def get(self, skill_id: str) -> SkillBundleEntry | None:
|
|
return self.entries.get(skill_id)
|
|
|
|
def upsert(self, entry: SkillBundleEntry) -> None:
|
|
self.entries[entry.skill_id] = entry
|
|
|
|
def remove(self, skill_id: str) -> None:
|
|
self.entries.pop(skill_id, None)
|
|
self.dependency_graph.pop(skill_id, None)
|
|
self.reverse_graph.pop(skill_id, None)
|
|
for deps in self.reverse_graph.values():
|
|
if skill_id in deps:
|
|
deps.remove(skill_id)
|
|
for deps in self.dependency_graph.values():
|
|
if skill_id in deps:
|
|
deps.remove(skill_id)
|
|
|
|
def referenced_skill_ids(self, skill_id: str) -> set[str]:
|
|
return set(self.dependency_graph.get(skill_id, []))
|
|
|
|
def recompile_group_ids(self, skill_id: str) -> set[str]:
|
|
result: set[str] = {skill_id}
|
|
queue = [skill_id]
|
|
while queue:
|
|
current = queue.pop()
|
|
for dependent in self.reverse_graph.get(current, []):
|
|
if dependent not in result:
|
|
result.add(dependent)
|
|
queue.append(dependent)
|
|
return result
|
|
|
|
def subset(self, skill_ids: Iterable[str]) -> "SkillBundle":
|
|
skill_id_set = set(skill_ids)
|
|
return SkillBundle(
|
|
assets_id=self.assets_id,
|
|
schema_version=self.schema_version,
|
|
built_at=self.built_at,
|
|
entries={sid: self.entries[sid] for sid in skill_id_set if sid in self.entries},
|
|
dependency_graph={
|
|
sid: [dep for dep in deps if dep in skill_id_set]
|
|
for sid, deps in self.dependency_graph.items()
|
|
if sid in skill_id_set
|
|
},
|
|
reverse_graph={
|
|
sid: [dep for dep in deps if dep in skill_id_set]
|
|
for sid, deps in self.reverse_graph.items()
|
|
if sid in skill_id_set
|
|
},
|
|
)
|
|
|
|
def get_tool_dependencies(self) -> ToolDependencies:
|
|
dependencies: dict[str, ToolDependency] = {}
|
|
references: dict[str, ToolReference] = {}
|
|
|
|
for entry in self.entries.values():
|
|
for dep in entry.tools.dependencies:
|
|
key = f"{dep.provider}.{dep.tool_name}"
|
|
if key not in dependencies:
|
|
dependencies[key] = dep
|
|
|
|
for ref in entry.tools.references:
|
|
if ref.uuid not in references:
|
|
references[ref.uuid] = ref
|
|
|
|
return ToolDependencies(
|
|
dependencies=list(dependencies.values()),
|
|
references=list(references.values()),
|
|
)
|