mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-09-22 20:46:20 +00:00
42334f26d7
* feat(capabilities): unify catalog, plugin configuration and agent selection * fix(capabilities): address review isolation, validation and demo issues * fix(capabilities): preserve concurrent selections and guide launcher repair
45 lines
2.0 KiB
Python
45 lines
2.0 KiB
Python
"""Capability discovery is public to authenticated users; mutations reuse existing policy."""
|
|
|
|
import asyncio
|
|
from typing import Any
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Request
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from app.gateway.capabilities import AdapterContext, InstallationList, list_installations, registry
|
|
from app.gateway.deps import get_config, require_admin_user
|
|
from deerflow.capabilities.catalog import PluginManifest, load_catalog
|
|
from deerflow.config.app_config import AppConfig
|
|
from deerflow.runtime.user_context import get_effective_user_id
|
|
|
|
router = APIRouter(prefix="/api/capabilities", tags=["capabilities"])
|
|
|
|
|
|
class InstallRequest(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
plugin_id: str
|
|
name: str = Field(default="", max_length=128)
|
|
configuration: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
|
|
@router.get("/catalog", response_model=list[PluginManifest])
|
|
async def catalog() -> list[PluginManifest]:
|
|
return await asyncio.to_thread(load_catalog)
|
|
|
|
|
|
@router.get("/installations/{adapter}", response_model=InstallationList)
|
|
async def installations(adapter: str, request: Request, config: AppConfig = Depends(get_config)) -> InstallationList:
|
|
return await list_installations(adapter, request, config)
|
|
|
|
|
|
@router.post("/installations", response_model=InstallationList)
|
|
async def install(body: InstallRequest, request: Request, config: AppConfig = Depends(get_config)) -> InstallationList:
|
|
await require_admin_user(request, detail="Admin privileges required to install capabilities.")
|
|
entries = await asyncio.to_thread(load_catalog)
|
|
manifest = next((entry for entry in entries if entry.id == body.plugin_id), None)
|
|
if manifest is None:
|
|
raise HTTPException(404, "Plugin not found")
|
|
context = AdapterContext(request, config, get_effective_user_id())
|
|
await registry.get(manifest.adapter).install(context, manifest, body.name, body.configuration)
|
|
return await list_installations(manifest.adapter, request, config)
|