phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 13:00:39 +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:
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}")