- Corrected old reference to catalog embedding model

2025-03-18 14:44:43 +01:00
parent cf2201a1f7
commit 09555ae8b0
1 changed files with 1 additions and 1 deletions
--- 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