TrialPath / trialpath /config.py
yakilee's picture
feat(medgemma): switch to text_generation with Gemma chat template
cd8cf7c
raw
history blame contribute delete
937 Bytes
"""Centralized configuration for TrialPath services."""
import os
MEDGEMMA_ENDPOINT_URL = os.environ.get(
"MEDGEMMA_ENDPOINT_URL",
"https://pcmy7bkqtqesrrzd.us-east-1.aws.endpoints.huggingface.cloud",
)
HF_TOKEN = os.environ.get("HF_TOKEN", "")
GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
GEMINI_MODEL = os.environ.get("GEMINI_MODEL", "gemini-3-pro")
MCP_URL = os.environ.get("MCP_URL", "http://localhost:3000")
PARLANT_URL = os.environ.get("PARLANT_URL", "http://localhost:8800")
SESSION_COST_BUDGET = float(os.environ.get("SESSION_COST_BUDGET", "0.50"))
# MedGemma retry / cold-start settings
MEDGEMMA_MAX_RETRIES = int(os.environ.get("MEDGEMMA_MAX_RETRIES", "6"))
MEDGEMMA_RETRY_BACKOFF = float(os.environ.get("MEDGEMMA_RETRY_BACKOFF", "2.0"))
MEDGEMMA_MAX_WAIT = float(os.environ.get("MEDGEMMA_MAX_WAIT", "60.0"))
MEDGEMMA_COLD_START_TIMEOUT = float(os.environ.get("MEDGEMMA_COLD_START_TIMEOUT", "300.0"))