From 9e72afee3c9fb3671a2b703506cd70a2e1c43de4 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Fri, 16 May 2025 14:00:35 +0800
Subject: [PATCH] r2

---
 api/services/dataset_service.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/api/services/dataset_service.py b/api/services/dataset_service.py
index df31e0f7ca..ac45981ee5 100644
--- a/api/services/dataset_service.py
+++ b/api/services/dataset_service.py
@@ -244,18 +244,20 @@ class DatasetService:
         rag_pipeline_dataset_create_entity: RagPipelineDatasetCreateEntity,
     ):
         # check if dataset name already exists
-        if Dataset.query.filter_by(name=rag_pipeline_dataset_create_entity.name, tenant_id=tenant_id).first():
+        if db.session.query(Dataset).filter_by(name=rag_pipeline_dataset_create_entity.name, tenant_id=tenant_id).first():
             raise DatasetNameDuplicateError(
                 f"Dataset with name {rag_pipeline_dataset_create_entity.name} already exists."
             )
 
         dataset = Dataset(
+            tenant_id=tenant_id,
             name=rag_pipeline_dataset_create_entity.name,
             description=rag_pipeline_dataset_create_entity.description,
             permission=rag_pipeline_dataset_create_entity.permission,
             provider="vendor",
             runtime_mode="rag_pipeline",
             icon_info=rag_pipeline_dataset_create_entity.icon_info,
+            created_by=current_user.id
         )
         db.session.add(dataset)
         db.session.commit()
@@ -267,7 +269,7 @@ class DatasetService:
         rag_pipeline_dataset_create_entity: RagPipelineDatasetCreateEntity,
     ):
         # check if dataset name already exists
-        if Dataset.query.filter_by(name=rag_pipeline_dataset_create_entity.name, tenant_id=tenant_id).first():
+        if db.session.query(Dataset).filter_by(name=rag_pipeline_dataset_create_entity.name, tenant_id=tenant_id).first():
             raise DatasetNameDuplicateError(
                 f"Dataset with name {rag_pipeline_dataset_create_entity.name} already exists."
             )