From 039b747318d7e70107891ad4656aa5a6f49fd3d8 Mon Sep 17 00:00:00 2001 From: redhelix Date: Fri, 27 Mar 2026 18:41:13 -0400 Subject: [PATCH] fix: strip ollama/ prefix from model name in embedding requests --- litellm/llms/ollama/completion/handler.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/llms/ollama/completion/handler.py b/litellm/llms/ollama/completion/handler.py index 97e4f13b560..869e9c5f64a 100644 --- a/litellm/llms/ollama/completion/handler.py +++ b/litellm/llms/ollama/completion/handler.py @@ -13,6 +13,8 @@ from litellm.types.utils import EmbeddingResponse def _prepare_ollama_embedding_payload( model: str, prompts: List[str], optional_params: Dict[str, Any] ) -> Dict[str, Any]: + if model.startswith("ollama/"): + model = model[len("ollama/"):] data: Dict[str, Any] = {"model": model, "input": prompts} special_optional_params = ["truncate", "options", "keep_alive", "dimensions"]