fix update dataset failed when embedding model is not exist (#6920)
This commit is contained in:
@@ -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)
|
||||
|
Reference in New Issue
Block a user