23 lines
803 B
Python
23 lines
803 B
Python
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_names = {str(name).strip() for name in enabled_knowledges if str(name).strip()}
|
|
databases = [db for db in databases if str(db.get("name") or "").strip() in enabled_names]
|
|
|
|
setattr(context, "_visible_knowledge_bases", databases)
|
|
return databases
|