From 77febef064a89cedb706c3465049ba1accec4fbe Mon Sep 17 00:00:00 2001 From: hjlarry Date: Mon, 12 May 2025 11:12:01 +0800 Subject: [PATCH] fix CI --- api/commands.py | 2 +- .../nodes/knowledge_retrieval/knowledge_retrieval_node.py | 2 +- api/services/external_knowledge_service.py | 4 +++- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/api/commands.py b/api/commands.py index b3f77f0b36..c05ed786aa 100644 --- a/api/commands.py +++ b/api/commands.py @@ -594,7 +594,7 @@ def old_metadata_migration(): db.session.add(dataset_metadata_binding) else: dataset_metadata_binding = ( - db.session.query(DatasetMetadataBinding) + db.session.query(DatasetMetadataBinding) # type: ignore .filter( DatasetMetadataBinding.dataset_id == document.dataset_id, DatasetMetadataBinding.document_id == document.id, diff --git a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py index 808f9d06c2..962deba1fd 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py +++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py @@ -275,7 +275,7 @@ class KnowledgeRetrievalNode(LLMNode): if records: for record in records: segment = record.segment - dataset = db.session.query(Dataset).filter_by(id=segment.dataset_id).first() + dataset = db.session.query(Dataset).filter_by(id=segment.dataset_id).first() # type: ignore document = ( db.session.query(Document) .filter( diff --git a/api/services/external_knowledge_service.py b/api/services/external_knowledge_service.py index 1a63286099..eb50d79494 100644 --- a/api/services/external_knowledge_service.py +++ b/api/services/external_knowledge_service.py @@ -25,7 +25,9 @@ from services.errors.dataset import DatasetNameDuplicateError class ExternalDatasetService: @staticmethod - def get_external_knowledge_apis(page, per_page, tenant_id, search=None) -> tuple[list[ExternalKnowledgeApis], int]: + def get_external_knowledge_apis( + page, per_page, tenant_id, search=None + ) -> tuple[list[ExternalKnowledgeApis], int | None]: query = ( select(ExternalKnowledgeApis) .filter(ExternalKnowledgeApis.tenant_id == tenant_id)