From 25dfe7f474ece01eebc1894f714d6a6fc329f46b Mon Sep 17 00:00:00 2001
From: Raphael Valdetaro <79842132+raphaelchristi@users.noreply.github.com>
Date: Wed, 5 Feb 2025 15:35:34 -0300
Subject: [PATCH] chore: update Google AI models list (#6013)

* chore: update Google AI models list

* [autofix.ci] apply automated fixes

* chore: update Gemini model name from exp to standard version

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
---
 .../models/google_generative_ai_constants.py  | 21 ++++++++-----------
 1 file changed, 9 insertions(+), 12 deletions(-)

diff --git a/src/backend/base/langflow/base/models/google_generative_ai_constants.py b/src/backend/base/langflow/base/models/google_generative_ai_constants.py
index 9b4a71c9c..957f1c51c 100644
--- a/src/backend/base/langflow/base/models/google_generative_ai_constants.py
+++ b/src/backend/base/langflow/base/models/google_generative_ai_constants.py
@@ -1,18 +1,15 @@
 GOOGLE_GENERATIVE_AI_MODELS = [
+    # GEMINI 1.5
     "gemini-1.5-pro",
     "gemini-1.5-flash",
     "gemini-1.5-flash-8b",
-    "gemini-1.0-pro",
-    "gemini-1.0-pro-vision",
-    "gemini-1.5-pro-001",
-    "gemini-1.5-pro-002",
-    "gemini-1.5-flash-001",
-    "gemini-1.5-flash-002",
-    "gemini-1.5-pro-exp-0827",
-    "gemini-1.5-flash-exp-0827",
-    "gemini-1.5-flash-8b-exp-0827",
-    "gemini-1.5-flash-8b-exp-0924",
-    "gemini-exp-1114",
-    "gemini-2.0-flash-exp",
+    # PREVIEW
+    "gemini-2.0-flash",
     "gemini-exp-1206",
+    "gemini-2.0-flash-thinking-exp-01-21",
+    "learnlm-1.5-pro-experimental",
+    # GEMMA
+    "gemma-2-2b",
+    "gemma-2-9b",
+    "gemma-2-27b",
 ]