diff --git a/labs/config.py b/labs/config.py index 9238cb8..34479af 100644 --- a/labs/config.py +++ b/labs/config.py @@ -31,19 +31,19 @@ REPORTS_DIR = PROJ_ROOT / "reports" FIGURES_DIR = REPORTS_DIR / "figures" -GITHUB_ACCESS_TOKEN = os.environ["GITHUB_ACCESS_TOKEN"] -GITHUB_OWNER = os.environ["GITHUB_OWNER"] -GITHUB_REPO = os.environ["GITHUB_REPO"] -GITHUB_USERNAME = os.environ["GITHUB_USERNAME"] +GITHUB_ACCESS_TOKEN = os.environ.get("GITHUB_ACCESS_TOKEN") +GITHUB_OWNER = os.environ.get("GITHUB_OWNER") +GITHUB_REPO = os.environ.get("GITHUB_REPO") +GITHUB_USERNAME = os.environ.get("GITHUB_USERNAME") GITHUB_API_BASE_URL = "https://api.github.com" LITELLM_MASTER_KEY = os.environ["LITELLM_MASTER_KEY"] LITELLM_API_KEY = os.environ["LITELLM_API_KEY"] -OPENAI_API_KEY = os.environ["OPENAI_API_KEY"] -GEMINI_API_KEY = os.environ["GEMINI_API_KEY"] -COHERE_API_KEY = os.environ["COHERE_API_KEY"] -GROQ_API_KEY = os.environ["GROQ_API_KEY"] +OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY") +GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY") +COHERE_API_KEY = os.environ.get("COHERE_API_KEY") +GROQ_API_KEY = os.environ.get("GROQ_API_KEY") ACTIVELOOP_TOKEN = os.environ["ACTIVELOOP_TOKEN"]