scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

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 01c6cbd270
commit f3a0261bb4
1 changed files with 6 additions and 1 deletions
Show stats 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},
)
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