From 2366f673767fea26363dfdccd6c4af3961fb9b3c Mon Sep 17 00:00:00 2001
From: Luis Pater <webmaster@idotorg.org>
Date: Fri, 17 Jul 2026 16:28:29 +0800
Subject: [PATCH] fix(registry): update Gemini 3.1 Flash Lite model ID and add
 test for validation

Closes: #4391
---
 internal/registry/model_definitions_test.go | 19 +++++++++++++++++++
 internal/registry/models/models.json        |  6 +++---
 2 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/internal/registry/model_definitions_test.go b/internal/registry/model_definitions_test.go
index 461d0c14..196c2653 100644
--- a/internal/registry/model_definitions_test.go
+++ b/internal/registry/model_definitions_test.go
@@ -16,6 +16,25 @@ func TestModelOverrideHeadersFromEmbeddedModels(t *testing.T) {
 	}
 }
 
+func TestGeminiVertexModelsUseFlashLiteReleaseID(t *testing.T) {
+	const releaseID = "gemini-3.1-flash-lite"
+	const previewID = releaseID + "-preview"
+
+	for _, model := range GetGeminiVertexModels() {
+		if model == nil {
+			continue
+		}
+		if model.ID == previewID {
+			t.Fatalf("Vertex model ID = %q, want release ID %q", model.ID, releaseID)
+		}
+		if model.ID == releaseID {
+			return
+		}
+	}
+
+	t.Fatalf("Vertex models do not contain %q", releaseID)
+}
+
 func TestWithXAIBuiltinsIncludesVideoPreviewModel(t *testing.T) {
 	models := WithXAIBuiltins(nil)
 
diff --git a/internal/registry/models/models.json b/internal/registry/models/models.json
index 10784301..cbeaee13 100644
--- a/internal/registry/models/models.json
+++ b/internal/registry/models/models.json
@@ -735,13 +735,13 @@
       }
     },
     {
-      "id": "gemini-3.1-flash-lite-preview",
+      "id": "gemini-3.1-flash-lite",
       "object": "model",
       "created": 1776288000,
       "owned_by": "google",
       "type": "gemini",
-      "display_name": "Gemini 3.1 Flash Lite Preview",
-      "name": "models/gemini-3.1-flash-lite-preview",
+      "display_name": "Gemini 3.1 Flash Lite",
+      "name": "models/gemini-3.1-flash-lite",
       "version": "3.1",
       "description": "Our smallest and most cost effective model, built for at scale usage.",
       "inputTokenLimit": 1048576,