From 2d4df68719d9285dcd4ff74ed15f6ccc6ed2fb67 Mon Sep 17 00:00:00 2001
From: Sarah Wooders <sarahwooders@gmail.com>
Date: Thu, 13 Feb 2025 16:45:21 -0800
Subject: [PATCH] fix: add tests for vertex and add file  (#994)

---
 letta/llm_api/google_constants.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 letta/llm_api/google_constants.py

diff --git a/letta/llm_api/google_constants.py b/letta/llm_api/google_constants.py
new file mode 100644
index 00000000..f7a11fc7
--- /dev/null
+++ b/letta/llm_api/google_constants.py
@@ -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}