fix CI

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,

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(

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)