import os
from dotenv import load_dotenv
from util import ErrorBot

# Load environment variables
load_dotenv()
GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
# if not GOOGLE_API_KEY:
#     raise ValueError("Set GOOGLE_API_KEY in your environment variables")

# EMBEDDING_MODEL = "BAAI/bge-base-en-v1.5"
# LLM_MODEL = "gemini-2.5-flash"  # Gemini model

# # Initialize singleton bot
# bot = ErrorBot(
#     embedding_model_name=EMBEDDING_MODEL,
#     llm_model_name=LLM_MODEL,
#     google_api_key=GOOGLE_API_KEY,
# )

# Ingest MongoDB
# bot.ingest_from_mongodb(
#     mongo_uri="mongodb+srv://dhaval:Dhaval15@cluster0.rwu1ze6.mongodb.net/prontoDB?retryWrites=true&w=majority&appName=Cluster0",
#     db_name="prontoDB",
# )

GROQ_API_KEY = os.getenv("GROQ_API_KEY")

EMBEDDING_MODEL = "BAAI/bge-base-en-v1.5"

# --- Gemini Bot ---
gemini_bot = ErrorBot(
    embedding_model_name=EMBEDDING_MODEL,
    llm_model_name="gemini-2.5-flash",
    google_api_key=GOOGLE_API_KEY,
    llm_provider="gemini",
)

# --- Groq Bot (LLaMA) ---
llama_bot = ErrorBot(
    embedding_model_name=EMBEDDING_MODEL,
    llm_model_name="llama-3.3-70b-versatile",
    groq_api_key=GROQ_API_KEY,
    llm_provider="groq",
)