Merge PR google#340: Add Gemini-exp-20241114.

PiperOrigin-RevId: 698247741
tanwargenairesearch · Nov 20, 2024 · 6d55d1d · 6d55d1d
1 parent ee0917c
commit 6d55d1d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/langfun/core/llms/__init__.py b/langfun/core/llms/__init__.py
@@ -32,13 +32,13 @@
 
 # Gemini models.
 from langfun.core.llms.google_genai import GenAI
+from langfun.core.llms.google_genai import GeminiExp_20241114
 from langfun.core.llms.google_genai import GeminiFlash1_5
 from langfun.core.llms.google_genai import GeminiPro
 from langfun.core.llms.google_genai import GeminiPro1_5
 from langfun.core.llms.google_genai import GeminiProVision
 from langfun.core.llms.google_genai import Palm2
 from langfun.core.llms.google_genai import Palm2_IT
-from langfun.core.llms.google_genai import GeminiExp_20241114
 
 # OpenAI models.
 from langfun.core.llms.openai import OpenAI

diff --git a/langfun/core/llms/google_genai.py b/langfun/core/llms/google_genai.py
@@ -52,7 +52,8 @@ class GenAI(lf.LanguageModel):
           'text-bison-001',
           'chat-bison-001',
           'gemini-1.5-pro-latest',
-          'gemini-1.5-flash-latest'
+          'gemini-1.5-flash-latest',
+          'gemini-exp-1114',
       ],
       'Model name.',
   ]
@@ -344,6 +345,13 @@ def get(
 ]
 
 
+class GeminiExp_20241114(GenAI):  # pylint: disable=invalid-name
+  """Gemini Experimental model launched on 11/14/2024."""
+
+  model = 'gemini-exp-1114'
+  supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
+
+
 class GeminiPro1_5(GenAI):  # pylint: disable=invalid-name
   """Gemini Pro latest model."""