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
104 lines
3.5 KiB
Python
104 lines
3.5 KiB
Python
from __future__ import annotations
|
|
|
|
from fastapi.encoders import jsonable_encoder
|
|
from flask import request
|
|
from flask_restx import Resource, fields
|
|
from pydantic import BaseModel, Field
|
|
|
|
from controllers.console import console_ns
|
|
from controllers.console.wraps import account_initialization_required, setup_required
|
|
from libs.login import current_account_with_tenant, login_required
|
|
from services.sandbox.sandbox_file_service import SandboxFileService
|
|
|
|
DEFAULT_REF_TEMPLATE_SWAGGER_2_0 = "#/definitions/{model}"
|
|
|
|
|
|
class SandboxFileListQuery(BaseModel):
|
|
path: str | None = Field(default=None, description="Workspace relative path")
|
|
recursive: bool = Field(default=False, description="List recursively")
|
|
|
|
|
|
class SandboxFileDownloadRequest(BaseModel):
|
|
path: str = Field(..., description="Workspace relative file path")
|
|
|
|
|
|
console_ns.schema_model(
|
|
SandboxFileListQuery.__name__,
|
|
SandboxFileListQuery.model_json_schema(ref_template=DEFAULT_REF_TEMPLATE_SWAGGER_2_0),
|
|
)
|
|
console_ns.schema_model(
|
|
SandboxFileDownloadRequest.__name__,
|
|
SandboxFileDownloadRequest.model_json_schema(ref_template=DEFAULT_REF_TEMPLATE_SWAGGER_2_0),
|
|
)
|
|
|
|
|
|
SANDBOX_FILE_NODE_FIELDS = {
|
|
"path": fields.String,
|
|
"is_dir": fields.Boolean,
|
|
"size": fields.Raw,
|
|
"mtime": fields.Raw,
|
|
"extension": fields.String,
|
|
}
|
|
|
|
|
|
SANDBOX_FILE_DOWNLOAD_TICKET_FIELDS = {
|
|
"download_url": fields.String,
|
|
"expires_in": fields.Integer,
|
|
"export_id": fields.String,
|
|
}
|
|
|
|
|
|
sandbox_file_node_model = console_ns.model("SandboxFileNode", SANDBOX_FILE_NODE_FIELDS)
|
|
sandbox_file_download_ticket_model = console_ns.model("SandboxFileDownloadTicket", SANDBOX_FILE_DOWNLOAD_TICKET_FIELDS)
|
|
|
|
|
|
@console_ns.route("/apps/<string:app_id>/sandbox/files")
|
|
class SandboxFilesApi(Resource):
|
|
"""List sandbox files for the current user.
|
|
|
|
The sandbox_id is derived from the current user's ID, as each user has
|
|
their own sandbox workspace per app.
|
|
"""
|
|
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@console_ns.expect(console_ns.models[SandboxFileListQuery.__name__])
|
|
@console_ns.marshal_list_with(sandbox_file_node_model)
|
|
def get(self, app_id: str):
|
|
args = SandboxFileListQuery.model_validate(request.args.to_dict(flat=True)) # type: ignore[arg-type]
|
|
account, tenant_id = current_account_with_tenant()
|
|
sandbox_id = account.id
|
|
return jsonable_encoder(
|
|
SandboxFileService.list_files(
|
|
tenant_id=tenant_id,
|
|
app_id=app_id,
|
|
sandbox_id=sandbox_id,
|
|
path=args.path,
|
|
recursive=args.recursive,
|
|
)
|
|
)
|
|
|
|
|
|
@console_ns.route("/apps/<string:app_id>/sandbox/files/download")
|
|
class SandboxFileDownloadApi(Resource):
|
|
"""Download a sandbox file for the current user.
|
|
|
|
The sandbox_id is derived from the current user's ID, as each user has
|
|
their own sandbox workspace per app.
|
|
"""
|
|
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@console_ns.expect(console_ns.models[SandboxFileDownloadRequest.__name__])
|
|
@console_ns.marshal_with(sandbox_file_download_ticket_model)
|
|
def post(self, app_id: str):
|
|
payload = SandboxFileDownloadRequest.model_validate(console_ns.payload or {})
|
|
account, tenant_id = current_account_with_tenant()
|
|
sandbox_id = account.id
|
|
res = SandboxFileService.download_file(
|
|
tenant_id=tenant_id, app_id=app_id, sandbox_id=sandbox_id, path=payload.path
|
|
)
|
|
return jsonable_encoder(res)
|