Files
Saurav Panda 81f5c49854 fix(tokens): map gemini-3.1-flash-lite to litellm preview slug
LiteLLM only recognizes gemini/gemini-3.1-flash-lite-preview today; the GA
slug fails the model lookup. Keeps the user-facing name unchanged.
2026-05-22 11:03:14 -07:00

7 lines
267 B
Python

# Mapping from model_name to LiteLLM model name
MODEL_TO_LITELLM: dict[str, str] = {
'gemini-flash-latest': 'gemini/gemini-flash-latest',
'gemini-3-flash-preview': 'gemini/gemini-3-flash-preview',
'gemini-3.1-flash-lite': 'gemini/gemini-3.1-flash-lite-preview',
}