api/config.py

import os
import sys
import logging
from api.vectorstore import VectorStoreManager
from dotenv import load_dotenv
from google import genai

logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s [%(levelname)s] %(message)s",
    handlers=[logging.StreamHandler(sys.stdout)],
)

logger = logging.getLogger(__name__)

load_dotenv()

GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY", "")
PINECONE_API_KEY = os.environ.get("PINECONE_API_KEY", "")
INDEX_NAME = "ragbot"


def log_critical(api_key: str):
    logger.critical(f"CRITICAL: {api_key} is missing from .env")
    sys.exit(1)


if not GOOGLE_API_KEY:
    log_critical("GOOGLE_API_KEY")

if not PINECONE_API_KEY:
    log_critical("PINECONE_API_KEY")


def system_instruction(url: str, context: str):
    return f"""
    INSTRUCTION:
    You are a helpful assistant for the website: {url}.  
    Answer only using the provided context below.  
    If the information is missing, reply: "Sorry, I haven't read that information yet."  
    Limit every answer to five sentences.

    CONTEXT:
    {context}
    """


try:
    vs = VectorStoreManager(api_key=PINECONE_API_KEY, index_name=INDEX_NAME)
    flash = genai.Client()
    logger.info("✅ System initialized: Pinecone & Gemini ready.")
except Exception as e:
    logger.critical(f"Startup Failure: {e}")
    sys.exit(1)