feat: new gemini models

arthurbrenno · Feb 5, 2025 · 896fdde · 896fdde
1 parent ab00e6d
commit 896fdde
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/intellibricks/llms/integrations/google/google.py b/src/intellibricks/llms/integrations/google/google.py
@@ -124,6 +124,9 @@ class GoogleLanguageModel(LanguageModel, frozen=True):
         "gemini-1.5-pro-002",
         "gemini-flash-experimental",
         "gemini-pro-experimental",
+        "gemini-2.0-flash",
+        "gemini-2.0-flash-lite-preview-02-05",
+        "gemini-2.0-flash-thinking-exp-01-21"
     ]
 
     vertexai: Optional[bool]

diff --git a/src/intellibricks/llms/types.py b/src/intellibricks/llms/types.py
@@ -195,6 +195,9 @@
     "google/genai/gemini-flash-experimental",
     "google/genai/gemini-pro-experimental",
     "google/genai/gemini-2.0-flash-exp",
+    "google/genai/gemini-2.0-flash",
+    "google/genai/gemini-2.0-flash-lite-preview-02-05",
+    "google/genai/gemini-2.0-flash-thinking-exp-01-21"
 ]
 
 VertexAIModelType: TypeAlias = Literal[
@@ -209,6 +212,9 @@
     "google/vertexai/gemini-1.5-pro-002",
     "google/vertexai/gemini-flash-experimental",
     "google/vertexai/gemini-pro-experimental",
+    "google/vertexai/gemini-2.0-flash",
+    "google/vertexai/gemini-2.0-flash-lite-preview-02-05",
+    "google/vertexai/gemini-2.0-flash-thinking-exp-01-21"
 ]
 
 GoogleModelType: TypeAlias = Literal[