adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

【fix]Nested query of folder knowledge base

Browse Source
This commit is contained in:
lixiangcheng1
2026-03-19 08:17:58 +08:00
parent d674b48f7d
commit f52b681133
2 changed files with 133 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
api/app/repositories/knowledge_repository.py
View File

@@ -111,6 +111,20 @@ def get_knowledge_by_id(db: Session, knowledge_id: uuid.UUID) -> Knowledge | Non
raise
def get_knowledges_by_parent_id(db: Session, parent_id: uuid.UUID) -> list[Knowledge]:
db_logger.debug(f"Query knowledge bases based on parent ID: parent_id={parent_id}")
try:
knowledges = db.query(Knowledge).filter(Knowledge.parent_id == parent_id).all()
if knowledges:
db_logger.debug(f"Knowledge bases query successful: count={len(knowledges)} (parent_id: {parent_id})")
else:
db_logger.debug(f"No knowledge bases found for given parent: parent_id={parent_id}")
return knowledges
except Exception as e:
db_logger.error(f"Failed to query the knowledge bases based on parent ID: parent_id={parent_id} - {str(e)}")
raise
def get_knowledge_by_name(db: Session, name: str, workspace_id: uuid.UUID) -> Knowledge | None:
db_logger.debug(f"Query knowledge base based on name and workspace_id: name={name}, workspace_id={workspace_id}")