mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-07 14:26:44 +00:00
fixing type
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
parent
179399edf4
commit
6cffb6e871
1 changed files with 2 additions and 2 deletions
|
@ -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
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue