phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add support for custom OPENROUTER_API_BASE via get_secret in completion function

Browse source
This commit is contained in:
Grayson Chen 2025-03-19 21:09:03 +08:00
parent 887648a364
commit b186210cea
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/main.py
View file

@ -2245,7 +2245,12 @@ def completion( # type: ignore # noqa: PLR0915
additional_args={"headers": headers}, additional_args={"headers": headers},
) )
elif custom_llm_provider == "openrouter": elif custom_llm_provider == "openrouter":
api_base = api_base or litellm.api_base or "https://openrouter.ai/api/v1" api_base = (
api_base
or litellm.api_base
or get_secret("OPENROUTER_API_BASE")
or "https://openrouter.ai/api/v1"
)
api_key = ( api_key = (
api_key api_key