import os | |
############################################################################### | |
# ENVIRONMENT & KEY MANAGEMENT # | |
############################################################################### | |
# OpenAI API Key | |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "") | |
# PubMed / NCBI usage | |
PUBMED_EMAIL = os.getenv("PUBMED_EMAIL", "[email protected]") | |
# Default LLM Model | |
OPENAI_DEFAULT_MODEL = "gpt-4o-mini" # Example; replace with real name if needed | |
# Hugging Face Image-Text Model for advanced medical imaging | |
HF_IMAGE_MODEL = "HuggingFaceTB/SmolVLM-500M-Instruct" | |
# Vector DB paths/config | |
VECTORDB_PATH = "vector_db" | |
EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2" | |
# Summaries | |
DEFAULT_SUMMARIZATION_CHUNK = 512 | |
# Misc settings | |
MAX_PUBMED_RESULTS = 5 | |