chore: make all remote inference provider configs RemoteInferenceProviderConfigs

This commit is contained in:
Matthew Farrellee 2025-10-02 18:13:50 -04:00
parent 4dfbe46954
commit 71d67a983e
37 changed files with 65 additions and 26 deletions

View file

@ -7,13 +7,14 @@
import os
from typing import Any
from pydantic import BaseModel, Field, SecretStr
from pydantic import Field, SecretStr
from llama_stack.providers.utils.inference.model_registry import RemoteInferenceProviderConfig
from llama_stack.schema_utils import json_schema_type
@json_schema_type
class NVIDIAConfig(BaseModel):
class NVIDIAConfig(RemoteInferenceProviderConfig):
"""
Configuration for the NVIDIA NIM inference endpoint.