add support for GeminiExp_20241206.

PiperOrigin-RevId: 704360193
google · Dec 9, 2024 · aa8c79c · aa8c79c
1 parent bcf7697
commit aa8c79c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/langfun/core/llms/__init__.py b/langfun/core/llms/__init__.py
@@ -33,6 +33,7 @@
 # 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 GeminiExp_20241206
 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

diff --git a/langfun/core/llms/google_genai.py b/langfun/core/llms/google_genai.py
@@ -54,6 +54,7 @@ class GenAI(lf.LanguageModel):
           'gemini-1.5-pro-latest',
           'gemini-1.5-flash-latest',
           'gemini-exp-1114',
+          'gemini-exp-1206',
       ],
       'Model name.',
   ]
@@ -345,6 +346,13 @@ def get(
 ]
 
 
+class GeminiExp_20241206(GenAI):  # pylint: disable=invalid-name
+  """Gemini Experimental model launched on 12/06/2024."""
+
+  model = 'gemini-exp-1206'
+  supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
+
+
 class GeminiExp_20241114(GenAI):  # pylint: disable=invalid-name
   """Gemini Experimental model launched on 11/14/2024."""