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

fix(db): fix database connection handling

Browse Source
This commit is contained in:
mengyonghao
2025-12-24 12:23:16 +08:00
parent 0a8c1be084
commit 1253bedcde
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/app/core/workflow/nodes/knowledge/node.py
View File

@@ -20,7 +20,7 @@ class KnowledgeRetrievalNode(BaseNode):
async def execute(self, state: WorkflowState) -> Any: async def execute(self, state: WorkflowState) -> Any:
query = self._render_template(self.typed_config.query, state) query = self._render_template(self.typed_config.query, state)
with get_db_context(): with get_db_context() as db:
filters = [ filters = [
knowledge_model.Knowledge.id.in_(self.typed_config.kb_ids), knowledge_model.Knowledge.id.in_(self.typed_config.kb_ids),
knowledge_model.Knowledge.permission_id == knowledge_model.PermissionType.Private, knowledge_model.Knowledge.permission_id == knowledge_model.PermissionType.Private,