add google/gemini-2.0-flash-exp as default model for vertex (#1214)

livekit · Dec 12, 2024 · f169a0b · f169a0b
1 parent b45cf88
commit f169a0b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/.changeset/chilled-students-clap.md b/.changeset/chilled-students-clap.md
@@ -0,0 +1,5 @@
+---
+"livekit-plugins-openai": patch
+---
+
+add `google/gemini-2.0-flash-exp` as default model for vertex
diff --git a/livekit-plugins/livekit-plugins-openai/livekit/plugins/openai/llm.py b/livekit-plugins/livekit-plugins-openai/livekit/plugins/openai/llm.py
@@ -200,7 +200,7 @@ def with_cerebras(
     @staticmethod
     def with_vertex(
         *,
-        model: str | VertexModels = "google/gemini-1.5-pro",
+        model: str | VertexModels = "google/gemini-2.0-flash-exp",
         project_id: str | None = None,
         location: str = "us-central1",
         user: str | None = None,

diff --git a/livekit-plugins/livekit-plugins-openai/livekit/plugins/openai/models.py b/livekit-plugins/livekit-plugins-openai/livekit/plugins/openai/models.py
@@ -80,6 +80,7 @@
 ]
 
 VertexModels = Literal[
+    "google/gemini-2.0-flash-exp",
     "google/gemini-1.5-flash",
     "google/gemini-1.5-pro",
     "google/gemini-1.0-pro-vision",