diff --git a/common/utils/model_utils.py b/common/utils/model_utils.py index a68418a..2f11300 100644 --- a/common/utils/model_utils.py +++ b/common/utils/model_utils.py @@ -59,7 +59,7 @@ def replace_variable_in_template(template: str, variable: str, value: str) -> st return template.replace(variable, value or "") -def get_embedding_model_and_class(tenant_id, catalog_id, full_embedding_name): +def get_embedding_model_and_class(tenant_id, catalog_id, full_embedding_name="mistral.mistral-embed"): """ Retrieve the embedding model and embedding model class to store Embeddings