Merge pull request #9369 from graysonchen/feature/add_openrouter_api_base

feat: Add support for custom OPENROUTER_API_BASE via get_secret in co…
This commit is contained in:
Krish Dholakia 2025-03-20 13:52:03 -07:00 committed by GitHub
commit cb4155fb16
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -2246,7 +2246,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