File size: 1,127 Bytes
05e3517
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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"