[Fix] embeddings llm engine

edenai · Feb 12, 2025 · 3654101 · 3654101
1 parent 2ee3710
commit 3654101
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/edenai_apis/llmengine/clients/litellm_client/litellm_client.py b/edenai_apis/llmengine/clients/litellm_client/litellm_client.py
@@ -57,7 +57,7 @@ def completion(
         n: Optional[int] = None,
         stream: Optional[bool] = None,
         stream_options: Optional[dict] = None,
-        stop: Optional[str]=None,
+        stop: Optional[str] = None,
         stop_sequences: Optional[any] = None,
         max_tokens: Optional[int] = None,
         presence_penalty: Optional[float] = None,
@@ -228,7 +228,7 @@ def embedding(
         call_params["input"] = input if isinstance(input, list) else [input]
         if model is not None:
             self.model_name = model
-        call_params["model"] = self.model_name
+        call_params["model"] = f"{self.provider_name}/{model}"
         call_params["timeout"] = timeout
         if dimensions is not None:
             call_params["dimensions"] = dimensions