mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 13:16:24 +00:00
bd62f96732
* feat(knowledge): add optional platform connections and hierarchical spaces * fix(knowledge): return actionable errors for unavailable policy storage * perf(knowledge): resolve space policy once per snapshot
39 lines
1.7 KiB
Python
39 lines
1.7 KiB
Python
"""Optional management backend. No UI or retrieval installation dependency."""
|
|
|
|
from importlib import import_module
|
|
from typing import Any
|
|
|
|
from deerflow_extension_api import extension
|
|
from deerflow_extension_api.plugins import BackendAction, PluginContribution
|
|
from pydantic import Field
|
|
|
|
from .contracts import Model
|
|
from .router import NAMESPACE, build_router
|
|
|
|
|
|
class Settings(Model):
|
|
enabled: bool = Field(default=False, strict=True)
|
|
routes_enabled: bool = Field(default=True, strict=True)
|
|
provider_factory: str = "deerflow_extension_knowledge_management.weknora:WeKnoraManagement"
|
|
provider_config: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
|
|
async def describe_contract(payload, context):
|
|
if not context.principal.is_admin or context.principal.is_internal:
|
|
raise PermissionError("Management requires an administrator")
|
|
return {"version": 1, "authorization": "explicit-admin-resource-grants", "execution": "provider-owned"}
|
|
|
|
|
|
@extension(api="0.2.7", name="knowledge-management")
|
|
def install(registry, config):
|
|
settings = Settings.model_validate(dict(config))
|
|
if not settings.enabled:
|
|
return
|
|
if not registry.plugin(PluginContribution(namespace=NAMESPACE, title="Knowledge management", enabled=True, backend=(BackendAction("contract", describe_contract),))):
|
|
raise RuntimeError("Host does not support knowledge management plugin admission")
|
|
if settings.routes_enabled:
|
|
# Executable factory paths are operator-owned configuration, never HTTP input.
|
|
module, attribute = settings.provider_factory.split(":", 1)
|
|
provider = getattr(import_module(module), attribute)(settings.provider_config)
|
|
registry.routers((build_router(provider),))
|