From e4037d03c2119fcc2fe3431255bf8370d0741b29 Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Fri, 21 Feb 2025 15:47:44 -0600 Subject: [PATCH] fix input_type values --- llama_stack/providers/remote/inference/nvidia/nvidia.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/remote/inference/nvidia/nvidia.py b/llama_stack/providers/remote/inference/nvidia/nvidia.py index 4c731a365..9384928aa 100644 --- a/llama_stack/providers/remote/inference/nvidia/nvidia.py +++ b/llama_stack/providers/remote/inference/nvidia/nvidia.py @@ -163,8 +163,8 @@ class NVIDIAInferenceAdapter(Inference, ModelRegistryHelper): if task_type is not None: task_type_options = { - EmbeddingTaskType.document: "DOCUMENT", - EmbeddingTaskType.query: "QUERY", + EmbeddingTaskType.document: "passage", + EmbeddingTaskType.query: "query", } if task_type not in task_type_options: raise ValueError(f"Invalid task_type: {task_type}")