File size: 1,197 Bytes
cd55ee8 |
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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
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",
) |