mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
fix(custom-openai): align discovery annotations with staging
This commit is contained in:
parent
44062f382b
commit
ecb74d491e
1 changed files with 1 additions and 1 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue