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

@ -8,6 +8,7 @@ from typing import Any
from pydantic import BaseModel, Field
from llama_stack.providers.utils.inference.model_registry import RemoteInferenceProviderConfig
from llama_stack.schema_utils import json_schema_type
@ -19,7 +20,7 @@ class GeminiProviderDataValidator(BaseModel):
@json_schema_type
class GeminiConfig(BaseModel):
class GeminiConfig(RemoteInferenceProviderConfig):
api_key: str | None = Field(
default=None,
description="API key for Gemini models",