kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix(kb): 修复 worker 容器 获取不到 知识库实例的问题

Browse Source
This commit is contained in:
肖泽涛 2026-02-25 19:09:53 +08:00
parent b0f21309f7
commit 6768fa9ed3
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/knowledge/manager.py
View File

@ -175,12 +175,24 @@ class KnowledgeBaseManager:
kb_repo = KnowledgeBaseRepository() kb_repo = KnowledgeBaseRepository()
rows = await kb_repo.get_all() rows = await kb_repo.get_all()
all_databases = [] all_databases = []
metadata_reloaded_types: set[str] = set()
for row in rows: for row in rows:
kb_instance = self._get_or_create_kb_instance(row.kb_type or "lightrag") kb_type = row.kb_type or "lightrag"
kb_instance = self._get_or_create_kb_instance(kb_type)
db_info = kb_instance.get_database_info(row.db_id) db_info = kb_instance.get_database_info(row.db_id)
if db_info: if not db_info and kb_type not in metadata_reloaded_types:
try:
await kb_instance._load_metadata()
metadata_reloaded_types.add(kb_type)
except Exception as e:
logger.warning(f"Failed to reload metadata for kb_type={kb_type}: {e}")
db_info = kb_instance.get_database_info(row.db_id)
if not db_info:
logger.warning(f"Skip database due to missing metadata: db_id={row.db_id}, kb_type={kb_type}")
continue
# 补充 share_config 和 additional_params # 补充 share_config 和 additional_params
db_info["share_config"] = row.share_config or {"is_shared": True, "accessible_departments": []} db_info["share_config"] = row.share_config or {"is_shared": True, "accessible_departments": []}
db_info["additional_params"] = ensure_chunk_defaults_in_additional_params(row.additional_params) db_info["additional_params"] = ensure_chunk_defaults_in_additional_params(row.additional_params)