diff --git a/.changeset/chilled-students-clap.md b/.changeset/chilled-students-clap.md new file mode 100644 index 000000000..bdd90d12d --- /dev/null +++ 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 index 796939b5f..22d73e73b 100644 --- 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 index c2667665d..7000d1239 100644 --- 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",