phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-13 21:29:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix input_type values

Browse source
This commit is contained in:
Matthew Farrellee 2025-02-21 15:47:44 -06:00
parent 610dea84c2
commit e4037d03c2
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/nvidia/nvidia.py
View file

@ -163,8 +163,8 @@ class NVIDIAInferenceAdapter(Inference, ModelRegistryHelper):
if task_type is not None: if task_type is not None:
task_type_options = { task_type_options = {
EmbeddingTaskType.document: "DOCUMENT", EmbeddingTaskType.document: "passage",
EmbeddingTaskType.query: "QUERY", EmbeddingTaskType.query: "query",
} }
if task_type not in task_type_options: if task_type not in task_type_options:
raise ValueError(f"Invalid task_type: {task_type}") raise ValueError(f"Invalid task_type: {task_type}")