config.py

"""
Configuration file for medical chatbot
"""
import os
from dotenv import load_dotenv

load_dotenv()

# API Keys
PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")

# Pinecone Configuration
INDEX_NAME = "medical-chatbot-index"
NAMESPACE = "medical-data"

# Model Configuration
EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
# Try different model names based on API version
LLM_MODEL = "models/gemini-pro"  # Updated model name format

# Retrieval Configuration
TOP_K = 5  # Number of relevant chunks to retrieve
SIMILARITY_THRESHOLD = 0.5  # Minimum similarity score

# Hugging Face Dataset
DATASET_NAME = "medical"