phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-07 22:34:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixing type

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-06-13 05:57:09 -04:00
parent 179399edf4
commit 6cffb6e871
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/remote/inference/ollama/ollama.py
View file

@ -392,7 +392,7 @@ class OllamaInferenceAdapter(
if model_obj.model_type != ModelType.embedding: if model_obj.model_type != ModelType.embedding:
raise ValueError(f"Model {model} is not an embedding model") raise ValueError(f"Model {model} is not an embedding model")
params = { params: dict[str, Any] = {
"model": model_obj.provider_resource_id, "model": model_obj.provider_resource_id,
"input": input, "input": input,
} }
@ -401,7 +401,7 @@ class OllamaInferenceAdapter(
if encoding_format is not None: if encoding_format is not None:
params["encoding_format"] = encoding_format params["encoding_format"] = encoding_format
if dimensions is not None: if dimensions is not None:
params["dimensions"] = str(dimensions) params["dimensions"] = dimensions
if user is not None: if user is not None:
params["user"] = user params["user"] = user