diff --git a/app/server/appsmith-plugins/googleAiPlugin/src/main/java/com/external/plugins/constants/GoogleAIConstants.java b/app/server/appsmith-plugins/googleAiPlugin/src/main/java/com/external/plugins/constants/GoogleAIConstants.java
index b2f6bd2caf89..ec9714ebb9be 100644
--- a/app/server/appsmith-plugins/googleAiPlugin/src/main/java/com/external/plugins/constants/GoogleAIConstants.java
+++ b/app/server/appsmith-plugins/googleAiPlugin/src/main/java/com/external/plugins/constants/GoogleAIConstants.java
@@ -23,7 +23,12 @@ public class GoogleAIConstants {
     public static final String MESSAGES = "messages";
     public static final String LABEL = "label";
     public static final String VALUE = "value";
-    public static final List<String> GOOGLE_AI_MODELS = List.of("gemini-pro");
+    public static final List<String> GOOGLE_AI_MODELS = List.of(
+            "gemini-2.5-pro",
+            "gemini-2.5-flash",
+            "gemini-2.0-flash",
+            "gemini-flash-latest",
+            "gemini-flash-lite-latest");
     public static final ExchangeStrategies EXCHANGE_STRATEGIES = ExchangeStrategies.builder()
             .codecs(configurer -> configurer.defaultCodecs().maxInMemorySize(/* 10MB */ 10 * 1024 * 1024))
             .build();
diff --git a/app/server/appsmith-plugins/googleAiPlugin/src/test/java/com/external/plugins/GenerateContentCommandTest.java b/app/server/appsmith-plugins/googleAiPlugin/src/test/java/com/external/plugins/GenerateContentCommandTest.java
index a7178c683bc5..3668534cbc57 100644
--- a/app/server/appsmith-plugins/googleAiPlugin/src/test/java/com/external/plugins/GenerateContentCommandTest.java
+++ b/app/server/appsmith-plugins/googleAiPlugin/src/test/java/com/external/plugins/GenerateContentCommandTest.java
@@ -32,10 +32,11 @@ public void testCreateTriggerUri() {
     @Test
     public void testCreateExecutionUri() {
         ActionConfiguration actionConfiguration = new ActionConfiguration();
-        actionConfiguration.setFormData(Map.of(GENERATE_CONTENT_MODEL, Map.of(DATA, "gemini-pro")));
+        actionConfiguration.setFormData(Map.of(GENERATE_CONTENT_MODEL, Map.of(DATA, "gemini-2.5-pro")));
         URI uri = generateContentCommand.createExecutionUri(actionConfiguration);
         Assertions.assertEquals(
-                "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent", uri.toString());
+                "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro:generateContent",
+                uri.toString());
     }
 
     @Test
diff --git a/app/server/appsmith-plugins/googleAiPlugin/src/test/java/com/external/plugins/GoogleAiPluginTest.java b/app/server/appsmith-plugins/googleAiPlugin/src/test/java/com/external/plugins/GoogleAiPluginTest.java
index dd8b0b757feb..dde7dcb8bf2e 100644
--- a/app/server/appsmith-plugins/googleAiPlugin/src/test/java/com/external/plugins/GoogleAiPluginTest.java
+++ b/app/server/appsmith-plugins/googleAiPlugin/src/test/java/com/external/plugins/GoogleAiPluginTest.java
@@ -109,7 +109,7 @@ public void verifyDatasourceTriggerResultsForChatModels() {
         apiKeyAuth.setValue("apiKey");
         DatasourceConfiguration datasourceConfiguration = new DatasourceConfiguration();
         datasourceConfiguration.setAuthentication(apiKeyAuth);
-        String responseBody = "[\"gemini-pro\"]";
+        String responseBody = "[\"gemini-2.5-pro\"]";
         MockResponse mockResponse = new MockResponse().setBody(responseBody);
         mockResponse.setResponseCode(200);
         mockEndpoint.enqueue(mockResponse);
@@ -122,8 +122,15 @@ public void verifyDatasourceTriggerResultsForChatModels() {
         StepVerifier.create(datasourceTriggerResultMono)
                 .assertNext(result -> {
                     assertTrue(result.getTrigger() instanceof List<?>);
-                    assertEquals(((List) result.getTrigger()).size(), 1);
-                    assertEquals(result.getTrigger(), getDataToMap(List.of("gemini-pro")));
+                    assertEquals(((List) result.getTrigger()).size(), 5);
+                    assertEquals(
+                            result.getTrigger(),
+                            getDataToMap(List.of(
+                                    "gemini-2.5-pro",
+                                    "gemini-2.5-flash",
+                                    "gemini-2.0-flash",
+                                    "gemini-flash-latest",
+                                    "gemini-flash-lite-latest")));
                 })
                 .verifyComplete();
     }