From 6cffb6e871f038189e106c2d655f040abe6f5523 Mon Sep 17 00:00:00 2001 From: Francisco Javier Arceo Date: Fri, 13 Jun 2025 05:57:09 -0400 Subject: [PATCH] fixing type Signed-off-by: Francisco Javier Arceo --- llama_stack/providers/remote/inference/ollama/ollama.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/remote/inference/ollama/ollama.py b/llama_stack/providers/remote/inference/ollama/ollama.py index 89dc5d093..3979fe074 100644 --- a/llama_stack/providers/remote/inference/ollama/ollama.py +++ b/llama_stack/providers/remote/inference/ollama/ollama.py @@ -392,7 +392,7 @@ class OllamaInferenceAdapter( if model_obj.model_type != ModelType.embedding: raise ValueError(f"Model {model} is not an embedding model") - params = { + params: dict[str, Any] = { "model": model_obj.provider_resource_id, "input": input, } @@ -401,7 +401,7 @@ class OllamaInferenceAdapter( if encoding_format is not None: params["encoding_format"] = encoding_format if dimensions is not None: - params["dimensions"] = str(dimensions) + params["dimensions"] = dimensions if user is not None: params["user"] = user