Medapp / config.py
mgbam's picture
Update config.py
834aa30 verified
raw
history blame
833 Bytes
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