mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
fix: add tests for vertex and add file (#994)
This commit is contained in:
parent
a734f99d8d
commit
014a07f6fa
12
letta/llm_api/google_constants.py
Normal file
12
letta/llm_api/google_constants.py
Normal 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}
|
Loading…
Reference in New Issue
Block a user