feat(extensions): add ZIP installation and unify action dialogs
This commit is contained in:
+24
-1
@@ -5,11 +5,13 @@ from contextlib import aclosing
|
||||
from datetime import datetime, timezone
|
||||
from uuid import uuid4
|
||||
|
||||
from fastapi import APIRouter, Header, Query
|
||||
from fastapi import APIRouter, Header, Query, Request
|
||||
from fastapi.responses import StreamingResponse
|
||||
|
||||
from app.agent import AgentCapacityError, AgentRunNotFoundError
|
||||
from app.container import container
|
||||
from app.config import get_settings
|
||||
from app.extensions.archive import MAX_ZIP_BYTES, install_zip
|
||||
from app.services.persona_settings import PersonaSettings, load_persona, save_persona
|
||||
from app.contracts import (
|
||||
AgentRun,
|
||||
@@ -659,6 +661,27 @@ async def install_skill(request: ExtensionInstallRequest) -> Skill:
|
||||
return extension_call(lambda: container.skills.install(request.package_path))
|
||||
|
||||
|
||||
async def read_extension_zip(request: Request) -> bytes:
|
||||
data = bytearray()
|
||||
async for chunk in request.stream():
|
||||
if len(data) + len(chunk) > MAX_ZIP_BYTES:
|
||||
raise ApiError(413, 'EXTENSION_ZIP_TOO_LARGE', 'ZIP 文件不能超过 10 MiB。')
|
||||
data.extend(chunk)
|
||||
return bytes(data)
|
||||
|
||||
|
||||
@router.post('/skills/install-zip', response_model=Skill, status_code=202, tags=['Skills'])
|
||||
async def install_skill_zip(request: Request) -> Skill:
|
||||
data = await read_extension_zip(request)
|
||||
return extension_call(lambda: install_zip(data, 'skill', get_settings().data_dir / 'extension-packages', container.skills.install))
|
||||
|
||||
|
||||
@router.post('/plugins/install-zip', response_model=Plugin, status_code=202, tags=['Plugins'])
|
||||
async def install_plugin_zip(request: Request) -> Plugin:
|
||||
data = await read_extension_zip(request)
|
||||
return extension_call(lambda: install_zip(data, 'plugin', get_settings().data_dir / 'extension-packages', container.plugins.install))
|
||||
|
||||
|
||||
@router.post(
|
||||
"/skills/{skill_id}/enable",
|
||||
response_model=Skill,
|
||||
|
||||
Reference in New Issue
Block a user