Spaces:
Sleeping
Sleeping
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" |