From 29b9e819afaccbaad26dcfad34637900c8c87ebf Mon Sep 17 00:00:00 2001 From: raspawar Date: Tue, 1 Apr 2025 11:04:54 +0000 Subject: [PATCH] fix _client in embedding --- llama_stack/providers/remote/inference/nvidia/nvidia.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/inference/nvidia/nvidia.py b/llama_stack/providers/remote/inference/nvidia/nvidia.py index 69e6335c6..99c8d0613 100644 --- a/llama_stack/providers/remote/inference/nvidia/nvidia.py +++ b/llama_stack/providers/remote/inference/nvidia/nvidia.py @@ -203,7 +203,7 @@ class NVIDIAInferenceAdapter(Inference, ModelRegistryHelper): extra_body["input_type"] = task_type_options[task_type] try: - response = await self._client.embeddings.create( + response = await self._get_client(model).embeddings.create( model=model, input=input, extra_body=extra_body,