mirror of
https://github.com/langgenius/dify.git
synced 2026-05-13 08:57:28 +08:00
0c7e7e0c4e
Port the complete infrastructure for agent sandbox execution and skill system: Sandbox & Virtual Environment (core/sandbox/, core/virtual_environment/): - Sandbox entity with lifecycle management (ready/failed/cancelled states) - SandboxBuilder with fluent API for configuring providers - 5 VM providers: Local, SSH, Docker, E2B, AWS CodeInterpreter - VirtualEnvironment base with command execution, file transfer, transport layers - Channel transport: pipe, queue, socket implementations - Bash session management and DifyCli binary integration - Storage: archive storage, file storage, noop storage, presign storage - Initializers: DifyCli, AppAssets, DraftAppAssets, Skills - Inspector: file browser, archive/runtime source, script utils - Security: encryption utils, debug helpers Skill & App Assets (core/skill/, core/app_assets/, core/app_bundle/): - Skill entity and manager - App asset accessor, builder pipeline (file, skill builders) - App bundle source zip extractor - Storage and converter utilities API Endpoints: - CLI API blueprint (controllers/cli_api/) for sandbox callback - Sandbox provider management (workspace/sandbox_providers) - Sandbox file browser (console/sandbox_files) - App asset management (console/app/app_asset) - Skill management (console/app/skills) - Storage file endpoints (controllers/files/storage_files) Services: - Sandbox service, provider service, file service - App asset service, app bundle service Config: - CliApiConfig, CreatorsPlatformConfig, CollaborationConfig - FILES_API_URL for sandbox file access Note: Controller route registration temporarily commented out (marked TODO) pending resolution of deep dependency chains (socketio, workflow_comment, command node, etc.). Core sandbox modules are fully ported and syntax-validated. 110 files changed, 10,549 insertions. Made-with: Cursor
81 lines
2.9 KiB
Python
81 lines
2.9 KiB
Python
from collections.abc import Mapping
|
|
|
|
from core.app.entities.app_asset_entities import AppAssetFileTree
|
|
from core.skill.assembler.common import (
|
|
build_skill_graph,
|
|
compute_transitive_dependance,
|
|
expand_referenced_skill_ids,
|
|
get_metadata,
|
|
process_skill_content,
|
|
)
|
|
from core.skill.entities.skill_bundle import Skill, SkillBundle, SkillDependance
|
|
from core.skill.entities.skill_document import SkillDocument
|
|
|
|
|
|
class SkillBundleAssembler:
|
|
_file_tree: AppAssetFileTree
|
|
|
|
def __init__(self, file_tree: AppAssetFileTree) -> None:
|
|
self._file_tree = file_tree
|
|
|
|
def assemble_bundle(
|
|
self,
|
|
documents: Mapping[str, SkillDocument],
|
|
assets_id: str,
|
|
) -> SkillBundle:
|
|
direct_skills: dict[str, Skill] = {}
|
|
for skill_id, doc in documents.items():
|
|
metadata = get_metadata(doc.content, doc.metadata)
|
|
direct_dependance = SkillDependance.from_metadata(metadata)
|
|
direct_skills[skill_id] = Skill(
|
|
skill_id=skill_id,
|
|
direct_dependance=direct_dependance,
|
|
dependance=direct_dependance,
|
|
content=process_skill_content(doc.content, metadata, self._file_tree, skill_id),
|
|
)
|
|
|
|
graph = build_skill_graph(direct_skills, self._file_tree)
|
|
transitive_map = compute_transitive_dependance(direct_skills, graph)
|
|
|
|
compiled_skills: dict[str, Skill] = {}
|
|
for skill_id, skill in direct_skills.items():
|
|
compiled_skills[skill_id] = skill.model_copy(update={"dependance": transitive_map[skill_id]})
|
|
|
|
return SkillBundle(asset_tree=self._file_tree, assets_id=assets_id, skills=compiled_skills)
|
|
|
|
|
|
class SkillDocumentAssembler:
|
|
_bundle: SkillBundle
|
|
|
|
def __init__(self, bundle: SkillBundle) -> None:
|
|
self._bundle = bundle
|
|
|
|
def assemble_document(self, document: SkillDocument, base_path: str = "") -> Skill:
|
|
metadata = get_metadata(document.content, document.metadata)
|
|
direct_dependance = SkillDependance.from_metadata(metadata)
|
|
resolved_content = process_skill_content(
|
|
document.content,
|
|
metadata,
|
|
self._bundle.asset_tree,
|
|
document.skill_id,
|
|
base_path,
|
|
)
|
|
|
|
transitive_dependance = direct_dependance
|
|
known_skill_ids = set(self._bundle.skills.keys())
|
|
referenced_skill_ids = expand_referenced_skill_ids(
|
|
direct_dependance.files, known_skill_ids, self._bundle.asset_tree
|
|
)
|
|
for skill_id in sorted(referenced_skill_ids):
|
|
referenced_skill = self._bundle.get(skill_id)
|
|
if referenced_skill is None:
|
|
continue
|
|
transitive_dependance = transitive_dependance | referenced_skill.dependance
|
|
|
|
return Skill(
|
|
skill_id=document.skill_id,
|
|
direct_dependance=direct_dependance,
|
|
dependance=transitive_dependance,
|
|
content=resolved_content,
|
|
)
|