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
105 lines
3.8 KiB
Python
105 lines
3.8 KiB
Python
import logging
|
|
|
|
from flask import request
|
|
from flask_restx import Resource, fields
|
|
from pydantic import BaseModel
|
|
|
|
from controllers.console import console_ns
|
|
from controllers.console.wraps import account_initialization_required, setup_required
|
|
from graphon.model_runtime.utils.encoders import jsonable_encoder
|
|
from libs.login import current_account_with_tenant, login_required
|
|
from services.sandbox.sandbox_provider_service import SandboxProviderService
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SandboxProviderConfigRequest(BaseModel):
|
|
config: dict
|
|
activate: bool = False
|
|
|
|
|
|
class SandboxProviderActivateRequest(BaseModel):
|
|
type: str
|
|
|
|
|
|
@console_ns.route("/workspaces/current/sandbox-providers")
|
|
class SandboxProviderListApi(Resource):
|
|
@console_ns.doc("list_sandbox_providers")
|
|
@console_ns.doc(description="Get list of available sandbox providers with configuration status")
|
|
@console_ns.response(200, "Success", fields.List(fields.Raw(description="Sandbox provider information")))
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def get(self):
|
|
_, current_tenant_id = current_account_with_tenant()
|
|
providers = SandboxProviderService.list_providers(current_tenant_id)
|
|
return jsonable_encoder([p.model_dump() for p in providers])
|
|
|
|
|
|
@console_ns.route("/workspaces/current/sandbox-provider/<string:provider_type>/config")
|
|
class SandboxProviderConfigApi(Resource):
|
|
@console_ns.doc("save_sandbox_provider_config")
|
|
@console_ns.doc(description="Save or update configuration for a sandbox provider")
|
|
@console_ns.response(200, "Success")
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def post(self, provider_type: str):
|
|
_, current_tenant_id = current_account_with_tenant()
|
|
args = SandboxProviderConfigRequest.model_validate(request.get_json())
|
|
|
|
try:
|
|
result = SandboxProviderService.save_config(
|
|
tenant_id=current_tenant_id,
|
|
provider_type=provider_type,
|
|
config=args.config,
|
|
activate=args.activate,
|
|
)
|
|
return result
|
|
except ValueError as e:
|
|
return {"message": str(e)}, 400
|
|
|
|
@console_ns.doc("delete_sandbox_provider_config")
|
|
@console_ns.doc(description="Delete configuration for a sandbox provider")
|
|
@console_ns.response(200, "Success")
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def delete(self, provider_type: str):
|
|
_, current_tenant_id = current_account_with_tenant()
|
|
|
|
try:
|
|
result = SandboxProviderService.delete_config(
|
|
tenant_id=current_tenant_id,
|
|
provider_type=provider_type,
|
|
)
|
|
return result
|
|
except ValueError as e:
|
|
return {"message": str(e)}, 400
|
|
|
|
|
|
@console_ns.route("/workspaces/current/sandbox-provider/<string:provider_type>/activate")
|
|
class SandboxProviderActivateApi(Resource):
|
|
"""Activate a sandbox provider."""
|
|
|
|
@console_ns.doc("activate_sandbox_provider")
|
|
@console_ns.doc(description="Activate a sandbox provider for the current workspace")
|
|
@console_ns.response(200, "Success")
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def post(self, provider_type: str):
|
|
"""Activate a sandbox provider."""
|
|
_, current_tenant_id = current_account_with_tenant()
|
|
|
|
try:
|
|
args = SandboxProviderActivateRequest.model_validate(request.get_json())
|
|
result = SandboxProviderService.activate_provider(
|
|
tenant_id=current_tenant_id,
|
|
provider_type=provider_type,
|
|
type=args.type,
|
|
)
|
|
return result
|
|
except ValueError as e:
|
|
return {"message": str(e)}, 400
|