myrmidon / python /src /server /config /model_ssot.py
tek Atrust
chore(deploy): build monolithic server for Hugging Face
d5ef46f
Raw
History Blame Contribute Delete
1.15 kB
"""
Model SSOT (Single Source of Truth) for Archon.
Physically aligns system-wide model identifiers with Google GenAI SDK requirements.
(Updated: 2026-04-21 based on Official Deprecation Docs)
Models optimized for Free Tier compatibility.
"""
SYSTEM_MODELS = {
# Gemini 3.1 Flash-Lite: Best Free Tier model (15 RPM / 1000 RPD)
"DEFAULT_TEXT": "models/gemini-3.1-flash-lite",
# Gemini 3 Flash: Best Free Tier model for reasoning
"DEFAULT_PRO": "models/gemini-3.1-flash-lite",
# Image Models (Requires Billing for API, MarketingService uses internal fallback)
"IMAGE_GEN": "models/gemini-3.1-flash-image-preview",
# TTS Model
"TTS_MODEL": "models/gemini-3.1-flash-tts-preview",
# Legacy Alignment (Kept per User Request - Deprecates July 2026)
"EMBEDDING": "models/gemini-embedding-001",
}
def get_model_path(key: str, default: str = "DEFAULT_TEXT") -> str:
"""Returns the full physical path (e.g. models/...) for a given model key."""
# Logic: Fallback to dictionary indexing to satisfy MyPy's str return guarantee
path = SYSTEM_MODELS.get(key) or SYSTEM_MODELS[default]
return str(path)