ForcePilot/backend/package/yuxi/agents/backends/knowledge_base_backend.py

from __future__ import annotations

from typing import Any

from yuxi import knowledge_base


async def resolve_visible_knowledge_bases_for_context(context) -> list[dict[str, Any]]:
    uid = getattr(context, "uid", None)
    if not uid:
        setattr(context, "_visible_knowledge_bases", [])
        return []

    result = await knowledge_base.get_databases_by_uid(str(uid))
    databases = result.get("databases") or []
    enabled_knowledges = getattr(context, "knowledges", None)
    if enabled_knowledges is not None:
        enabled_ids = {str(value).strip() for value in enabled_knowledges if str(value).strip()}
        databases = [db for db in databases if str(db.get("kb_id") or "").strip() in enabled_ids]

    setattr(context, "_visible_knowledge_bases", databases)
    return databases