diff --git a/litellm/llms/openai_like/chat/transformation.py b/litellm/llms/openai_like/chat/transformation.py index cea1713ca61..36b78a74347 100644 --- a/litellm/llms/openai_like/chat/transformation.py +++ b/litellm/llms/openai_like/chat/transformation.py @@ -169,7 +169,7 @@ class OpenAILikeChatConfig(OpenAIGPTConfig): class CustomOpenAIChatConfig(OpenAILikeChatConfig): - def get_models(self, api_key: Optional[str] = None, api_base: Optional[str] = None) -> list[str]: + def get_models(self, api_key: str | None = None, api_base: str | None = None) -> list[str]: if api_base is None: raise ValueError("api_base must be set to discover models for the custom_openai provider") models = super().get_models(api_key=api_key or "", api_base=api_base)