fix(custom-openai): align discovery annotations with staging

This commit is contained in:
Aditya Aggarwal 2026-08-04 19:42:09 +05:30
parent 44062f382b
commit ecb74d491e

View file

@ -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)