fix update dataset failed when embedding model is not exist (#6920)

This commit is contained in:
Jyong
2024-08-02 20:30:22 +08:00
committed by GitHub
parent 4d0a6cc382
commit 048bc4c06e
2 changed files with 29 additions and 2 deletions

View File

@@ -197,6 +197,28 @@ class DatasetService:
f"{ex.description}"
)
@staticmethod
def check_embedding_model_setting(tenant_id: str, embedding_model_provider: str, embedding_model:str):
try:
model_manager = ModelManager()
model_manager.get_model_instance(
tenant_id=tenant_id,
provider=embedding_model_provider,
model_type=ModelType.TEXT_EMBEDDING,
model=embedding_model
)
except LLMBadRequestError:
raise ValueError(
"No Embedding Model available. Please configure a valid provider "
"in the Settings -> Model Provider."
)
except ProviderTokenNotInitError as ex:
raise ValueError(
f"The dataset in unavailable, due to: "
f"{ex.description}"
)
@staticmethod
def update_dataset(dataset_id, data, user):
data.pop('partial_member_list', None)