Research_Agent / config.py
Ashmit003's picture
Uploading modified local folder struture
be5f49d
raw
history blame contribute delete
906 Bytes
import os
from dotenv import load_dotenv
def set_environment(verbose=True) -> dict:
env_path = os.path.abspath(".env")
# Load .env only if it exists
if os.path.exists(env_path):
load_dotenv(dotenv_path=env_path, override=True)
if verbose:
print(f"✅ Loaded .env from: {env_path}")
else:
if verbose:
print("⚠️ No .env file found. Relying on system environment variables.")
required_keys = [
"OPENAI_API_KEY",
"OPENROUTER_API_KEY",
"GEMINI_API_KEY",
"GROQ_API_KEY"
]
missing_keys = [key for key in required_keys if not os.getenv(key)]
if verbose:
if missing_keys:
print(f"⚠️ Missing environment variables: {missing_keys}")
else:
print("✅ All required API keys loaded successfully!")
return {key: os.getenv(key) for key in required_keys}