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
50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
from core.sandbox.entities.files import SandboxFileDownloadTicket, SandboxFileNode
|
|
from core.sandbox.inspector import SandboxFileBrowser
|
|
from extensions.ext_storage import storage
|
|
from extensions.storage.cached_presign_storage import CachedPresignStorage
|
|
from extensions.storage.file_presign_storage import FilePresignStorage
|
|
|
|
|
|
class SandboxFileService:
|
|
@staticmethod
|
|
def get_storage() -> CachedPresignStorage:
|
|
"""Get a lazily-initialized storage instance for sandbox files.
|
|
|
|
Returns a CachedPresignStorage wrapping FilePresignStorage,
|
|
providing presign fallback and URL caching.
|
|
"""
|
|
return CachedPresignStorage(
|
|
storage=FilePresignStorage(storage.storage_runner),
|
|
cache_key_prefix="sandbox_files",
|
|
)
|
|
|
|
@classmethod
|
|
def exists(cls, *, tenant_id: str, app_id: str, sandbox_id: str) -> bool:
|
|
"""Check if the sandbox source exists and is available."""
|
|
browser = SandboxFileBrowser(tenant_id=tenant_id, app_id=app_id, sandbox_id=sandbox_id)
|
|
return browser.exists()
|
|
|
|
@classmethod
|
|
def list_files(
|
|
cls,
|
|
*,
|
|
tenant_id: str,
|
|
app_id: str,
|
|
sandbox_id: str,
|
|
path: str | None = None,
|
|
recursive: bool = False,
|
|
) -> list[SandboxFileNode]:
|
|
browser = SandboxFileBrowser(tenant_id=tenant_id, app_id=app_id, sandbox_id=sandbox_id)
|
|
if not browser.exists():
|
|
return []
|
|
return browser.list_files(path=path, recursive=recursive)
|
|
|
|
@classmethod
|
|
def download_file(cls, *, tenant_id: str, app_id: str, sandbox_id: str, path: str) -> SandboxFileDownloadTicket:
|
|
browser = SandboxFileBrowser(tenant_id=tenant_id, app_id=app_id, sandbox_id=sandbox_id)
|
|
if not browser.exists():
|
|
raise ValueError("Sandbox source not found")
|
|
return browser.download_file(path=path)
|