From ecb74d491e3c878541e1e7e76669cfc240eae983 Mon Sep 17 00:00:00 2001 From: Aditya Aggarwal <42476079+aditya-786@users.noreply.github.com> Date: Tue, 4 Aug 2026 19:42:09 +0530 Subject: [PATCH] fix(custom-openai): align discovery annotations with staging --- litellm/llms/openai_like/chat/transformation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)