fix: add tests for vertex and add file (#994)

This commit is contained in:
Sarah Wooders
2025-02-13 16:45:21 -08:00
committed by GitHub
parent 8d765f286e
commit cfb61e2fd5

View File

@@ -0,0 +1,12 @@
GOOGLE_MODEL_TO_CONTEXT_LENGTH = {
"gemini-2.0-flash-001": 1048576,
"gemini-2.0-pro-exp-02-05": 2097152,
"gemini-2.0-flash-lite-preview-02-05": 1048576,
"gemini-2.0-flash-thinking-exp-01-21": 1048576,
"gemini-1.5-flash": 1048576,
"gemini-1.5-pro": 2097152,
"gemini-1.0-pro": 32760,
"gemini-1.0-pro-vision": 16384,
}
GOOGLE_EMBEDING_MODEL_TO_DIM = {"text-embedding-005": 768, "text-multilingual-embedding-002": 768}