| 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", | |
| ) |