diff --git a/langfun/core/llms/__init__.py b/langfun/core/llms/__init__.py index dd5259fc..a0f1c48d 100644 --- 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 index b617a51c..2d24ec68 100644 --- 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."""