Gonalb's picture
init commit
05e3517
raw
history blame
1.13 kB
import os
from dotenv import load_dotenv
# Load environment variables from .env file if it exists
load_dotenv()
# Check for required environment variables and provide guidance if missing
required_vars = ["OPENAI_API_KEY", "PROJECT_ID", "DATASET_ID", "BIGQUERY_KEY_PATH"]
missing_vars = [var for var in required_vars if not os.environ.get(var)]
if missing_vars:
print("ERROR: Missing required environment variables:")
for var in missing_vars:
print(f"- {var}")
print("\nPlease set these variables in your environment or in a .env file.")
print("Example .env file:")
print("""
OPENAI_API_KEY=your_openai_api_key
PROJECT_ID=your_gcp_project_id
DATASET_ID=your_bigquery_dataset_id
BIGQUERY_KEY_PATH=path/to/your/bigquery/credentials.json
""")
# Set Google Application Credentials environment variable if provided
if os.environ.get("BIGQUERY_KEY_PATH"):
os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = os.environ.get("BIGQUERY_KEY_PATH")
# BigQuery configuration
BIGQUERY_KEY_PATH = "xepelin-411205-1ba8a63c9c96.json"
PROJECT_ID = "xepelin-411205"
DATASET_ID = "ecommerceaie5"