main.py

"""Entry point for the BigQuery SQL Agent."""

import os
import warnings
import logging

# Suppress Google ADK experimental feature warnings
warnings.filterwarnings("ignore", message=r".*EXPERIMENTAL.*feature.*", category=UserWarning)
warnings.filterwarnings("ignore", message=r".*non-text parts.*", category=UserWarning)
warnings.filterwarnings("ignore", message=r".*GOOGLE_API_KEY.*GEMINI_API_KEY.*", category=UserWarning)

# Suppress Google SDK info/warning logs
logging.getLogger("google.genai").setLevel(logging.ERROR)
logging.getLogger("google.adk").setLevel(logging.ERROR)

from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

# Remove GEMINI_API_KEY if GOOGLE_API_KEY is set to avoid duplicate key warning
if os.environ.get("GOOGLE_API_KEY") and os.environ.get("GEMINI_API_KEY"):
    del os.environ["GEMINI_API_KEY"]

from bigquery_agent.cli import main

if __name__ == "__main__":
    main()