fix: set LITELLM_API_KEY env var for unified API key support
This commit is contained in:
@@ -191,6 +191,7 @@ async def warm_up_llm() -> None:
|
|||||||
api_key = os.getenv("LLM_API_KEY")
|
api_key = os.getenv("LLM_API_KEY")
|
||||||
|
|
||||||
if api_key:
|
if api_key:
|
||||||
|
os.environ.setdefault("LITELLM_API_KEY", api_key)
|
||||||
litellm.api_key = api_key
|
litellm.api_key = api_key
|
||||||
|
|
||||||
api_base = (
|
api_base = (
|
||||||
|
|||||||
@@ -27,6 +27,7 @@ logger = logging.getLogger(__name__)
|
|||||||
|
|
||||||
api_key = os.getenv("LLM_API_KEY")
|
api_key = os.getenv("LLM_API_KEY")
|
||||||
if api_key:
|
if api_key:
|
||||||
|
os.environ.setdefault("LITELLM_API_KEY", api_key)
|
||||||
litellm.api_key = api_key
|
litellm.api_key = api_key
|
||||||
|
|
||||||
api_base = (
|
api_base = (
|
||||||
|
|||||||
Reference in New Issue
Block a user