chore: inject more db.session (#38045)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>

api/core/app/app_config/easy_ui_based_app/dataset/manager.py

@@ -9,6 +9,7 @@ from core.app.app_config.entities import (
 )
 from core.entities.agent_entities import PlanningStrategy
 from core.rag.data_post_processor.data_post_processor import RerankingModelDict, WeightsDict
+from extensions.ext_database import db
 from models.model import AppMode, AppModelConfigDict
 from services.dataset_service import DatasetService
 
@@ -256,7 +257,7 @@ class DatasetConfigManager:
     @classmethod
     def is_dataset_exists(cls, tenant_id: str, dataset_id: str) -> bool:
         # verify if the dataset ID exists
-        dataset = DatasetService.get_dataset(dataset_id)
+        dataset = DatasetService.get_dataset(dataset_id, db.session)
 
         if not dataset:
             return False

api/core/app/apps/pipeline/pipeline_generator.py

@@ -144,7 +144,7 @@ class PipelineGenerator(BaseAppGenerator):
             DocumentService.check_document_creation_limits(len(datasource_info_list), features)
 
             for datasource_info in datasource_info_list:
-                position = DocumentService.get_documents_position(dataset.id)
+                position = DocumentService.get_documents_position(dataset.id, session)
                 document = self._build_document(
                     tenant_id=pipeline.tenant_id,
                     dataset_id=dataset.id,

api/core/app/features/annotation_reply/annotation_reply.py

@@ -45,7 +45,7 @@ class AnnotationReplyFeature:
             embedding_model_name = collection_binding_detail.model_name
 
             dataset_collection_binding = DatasetCollectionBindingService.get_dataset_collection_binding(
-                embedding_provider_name, embedding_model_name, CollectionBindingType.ANNOTATION
+                embedding_provider_name, embedding_model_name, db.session, CollectionBindingType.ANNOTATION
             )
 
             dataset = Dataset(