langfun 0.1.2.dev202505190805__tar.gz → 0.1.2.dev202505210804__tar.gz

{langfun-0.1.2.dev202505190805 → langfun-0.1.2.dev202505210804}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfun
-Version: 0.1.2.dev202505190805
+Version: 0.1.2.dev202505210804
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.1.2.dev202505190805 → langfun-0.1.2.dev202505210804}/langfun/core/llms/__init__.py

@@ -40,6 +40,7 @@ from langfun.core.llms.azure_openai import AzureOpenAI
 
 # Gemini models.
 from langfun.core.llms.google_genai import GenAI
+from langfun.core.llms.google_genai import Gemini25FlashPreview_20250520
 from langfun.core.llms.google_genai import Gemini25ProPreview_20250506
 from langfun.core.llms.google_genai import Gemini25FlashPreview_20250417
 from langfun.core.llms.google_genai import Gemini25ProPreview_20250325
@@ -79,6 +80,7 @@ from langfun.core.llms.vertexai import VertexAIGemini25ProExp_20250325
 from langfun.core.llms.vertexai import VertexAIGemini25ProPreview_20250325
 from langfun.core.llms.vertexai import VertexAIGemini25FlashPreview_20250417
 from langfun.core.llms.vertexai import VertexAIGemini25ProPreview_20250506
+from langfun.core.llms.vertexai import VertexAIGemini25FlashPreview_20250520
 
 # For backward compatibility.
 GeminiPro1_5 = Gemini15Pro

{langfun-0.1.2.dev202505190805 → langfun-0.1.2.dev202505210804}/langfun/core/llms/gemini.py

@@ -151,6 +151,29 @@ SUPPORTED_MODELS = [
     #
     # Production models.
     #
+    # Gemini 2.5 Flash Preview 0520
+    GeminiModelInfo(
+        model_id='gemini-2.5-flash-preview-05-20',
+        in_service=True,
+        provider=pg.oneof(['Google GenAI', 'VertexAI']),
+        model_type='instruction-tuned',
+        description='Gemini 2.5 Flash.',
+        release_date=datetime.datetime(2025, 5, 20),
+        input_modalities=GeminiModelInfo.ALL_SUPPORTED_INPUT_TYPES,
+        context_length=lf.ModelInfo.ContextLength(
+            max_input_tokens=1_048_576,
+            max_output_tokens=65_536,
+        ),
+        pricing=GeminiModelInfo.Pricing(
+            cost_per_1m_cached_input_tokens=0.15,
+            cost_per_1m_input_tokens=0.15,
+            cost_per_1m_output_tokens=0.6,
+        ),
+        rate_limits=lf.ModelInfo.RateLimits(
+            max_requests_per_minute=2000,
+            max_tokens_per_minute=4_000_000,
+        ),
+    ),
     # Gemini 2.5 Pro Preview
     GeminiModelInfo(
         model_id='gemini-2.5-pro-preview-05-06',

{langfun-0.1.2.dev202505190805 → langfun-0.1.2.dev202505210804}/langfun/core/llms/google_genai.py

@@ -106,6 +106,11 @@ class GeminiExp_20241206(GenAI):
 #
 # Production models.
 #
+class Gemini25FlashPreview_20250520(GenAI):
+  """Gemini 2.5 Flash model launched on 05/20/2025."""
+  model = 'gemini-2.5-flash-preview-05-20'
+
+
 class Gemini25ProPreview_20250506(GenAI):
   """Gemini 2.5 Pro model launched on 05/06/2025."""
 

{langfun-0.1.2.dev202505190805 → langfun-0.1.2.dev202505210804}/langfun/core/llms/vertexai.py

@@ -177,6 +177,11 @@ class VertexAIGemini(VertexAI, gemini.Gemini):
 #
 # Production models.
 #
+class VertexAIGemini25FlashPreview_20250520(VertexAIGemini):  # pylint: disable=invalid-name
+  """Gemini 2.5 Flash model launched on 05/20/2025."""
+  model = 'gemini-2.5-flash-preview-05-20'
+
+
 class VertexAIGemini25ProPreview_20250506(VertexAIGemini):  # pylint: disable=invalid-name
   """Gemini 2.5 Pro model launched on 05/06/2025."""
   model = 'gemini-2.5-pro-preview-05-06'

{langfun-0.1.2.dev202505190805 → langfun-0.1.2.dev202505210804}/langfun.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfun
-Version: 0.1.2.dev202505190805
+Version: 0.1.2.dev202505210804
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors