mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
26 lines
595 B
Python
26 lines
595 B
Python
import logging
|
|
import os
|
|
from typing import Optional
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def load_api_key() -> Optional[str]:
|
|
"""
|
|
Load the NVIDIA NIM API key from environment variables.
|
|
|
|
Returns:
|
|
The API key from environment variables, or None if not found
|
|
"""
|
|
api_key = os.environ.get("NVIDIA_NIM_API_KEY")
|
|
if not api_key:
|
|
logger.error(
|
|
"NVIDIA_NIM_API_KEY not found in environment variables. "
|
|
"Please set it in your .env file."
|
|
)
|
|
return None
|
|
|
|
return api_key
|