add Vertex embedding and Gemini flash models

2026-09-07 22:34:02 -07:00
parent 4a504a17e1
commit 3efbd32e8f
2 changed files with 15 additions and 0 deletions
@@ -12,5 +12,17 @@ model_list:
      vertex_project: os.environ/VERTEXAI_PROJECT
      vertex_location: os.environ/VERTEXAI_LOCATION

+  - model_name: google/gemini-embedding-2
+    litellm_params:
+      model: vertex_ai/gemini-embedding-2
+      vertex_project: os.environ/VERTEXAI_PROJECT
+      vertex_location: os.environ/VERTEXAI_LOCATION
+
+  - model_name: google/gemini-3.8-flash
+    litellm_params:
+      model: vertex_ai/gemini-3.8-flash
+      vertex_project: os.environ/VERTEXAI_PROJECT
+      vertex_location: os.environ/VERTEXAI_LOCATION
+
 general_settings:
  master_key: os.environ/LITELLM_MASTER_KEY