AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-15 09:06:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

r2

Browse Source
This commit is contained in:
jyong 2025-05-16 14:00:35 +08:00
parent 613b94a6e6
commit 9e72afee3c
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/services/dataset_service.py
View File

@ -244,18 +244,20 @@ class DatasetService:
rag_pipeline_dataset_create_entity: RagPipelineDatasetCreateEntity, rag_pipeline_dataset_create_entity: RagPipelineDatasetCreateEntity,
): ):
# check if dataset name already exists # 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( raise DatasetNameDuplicateError(
f"Dataset with name {rag_pipeline_dataset_create_entity.name} already exists." f"Dataset with name {rag_pipeline_dataset_create_entity.name} already exists."
) )
dataset = Dataset( dataset = Dataset(
tenant_id=tenant_id,
name=rag_pipeline_dataset_create_entity.name, name=rag_pipeline_dataset_create_entity.name,
description=rag_pipeline_dataset_create_entity.description, description=rag_pipeline_dataset_create_entity.description,
permission=rag_pipeline_dataset_create_entity.permission, permission=rag_pipeline_dataset_create_entity.permission,
provider="vendor", provider="vendor",
runtime_mode="rag_pipeline", runtime_mode="rag_pipeline",
icon_info=rag_pipeline_dataset_create_entity.icon_info, icon_info=rag_pipeline_dataset_create_entity.icon_info,
created_by=current_user.id
) )
db.session.add(dataset) db.session.add(dataset)
db.session.commit() db.session.commit()
@ -267,7 +269,7 @@ class DatasetService:
rag_pipeline_dataset_create_entity: RagPipelineDatasetCreateEntity, rag_pipeline_dataset_create_entity: RagPipelineDatasetCreateEntity,
): ):
# check if dataset name already exists # 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( raise DatasetNameDuplicateError(
f"Dataset with name {rag_pipeline_dataset_create_entity.name} already exists." f"Dataset with name {rag_pipeline_dataset_create_entity.name} already exists."
) )