|
@@ -136,7 +136,9 @@ class DatasetService:
|
|
|
return datasets.items, datasets.total
|
|
|
|
|
|
@staticmethod
|
|
|
- def create_empty_dataset(tenant_id: str, name: str, indexing_technique: Optional[str], account: Account):
|
|
|
+ def create_empty_dataset(
|
|
|
+ tenant_id: str, name: str, indexing_technique: Optional[str], account: Account, permission: Optional[str]
|
|
|
+ ):
|
|
|
# check if dataset name already exists
|
|
|
if Dataset.query.filter_by(name=name, tenant_id=tenant_id).first():
|
|
|
raise DatasetNameDuplicateError(f"Dataset with name {name} already exists.")
|
|
@@ -153,6 +155,7 @@ class DatasetService:
|
|
|
dataset.tenant_id = tenant_id
|
|
|
dataset.embedding_model_provider = embedding_model.provider if embedding_model else None
|
|
|
dataset.embedding_model = embedding_model.model if embedding_model else None
|
|
|
+ dataset.permission = permission if permission else DatasetPermissionEnum.ONLY_ME
|
|
|
db.session.add(dataset)
|
|
|
db.session.commit()
|
|
|
return dataset
|