langfun 0.1.2.dev202505190805__py3-none-any.whl → 0.1.2.dev202505210804__py3-none-any.whl

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/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/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/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.dist-info → langfun-0.1.2.dev202505210804.dist-info}/METADATA

@@ -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.dist-info → langfun-0.1.2.dev202505210804.dist-info}/RECORD

@@ -81,7 +81,7 @@ langfun/core/eval/v2/reporting.py,sha256=yUIPCAMnp7InIzpv1DDWrcLO-75iiOUTpscj7sm
 langfun/core/eval/v2/reporting_test.py,sha256=hcPJJaMtPulqERvHYTpId83WXdqDKnnexmULtK7WKwk,5686
 langfun/core/eval/v2/runners.py,sha256=iqbH4jMtnNMhfuv1eHaxJmk1Vvsrz-sAJJFP8U44-tA,16758
 langfun/core/eval/v2/runners_test.py,sha256=DO3xV0sBNB6n65j41xx2i7gqUCJcPF37DFZLEjrmISg,11987
-langfun/core/llms/__init__.py,sha256=QWxRhzVn_vgJvdmW_xs5PcPuDbHsUxTU94YyV4Ofl34,8684
+langfun/core/llms/__init__.py,sha256=tYMsdEiCJCDGQK8fF2PTZgOQylVTHyPnqP8ytonh2YY,8834
 langfun/core/llms/anthropic.py,sha256=dCkTYvwb_wWWeKQNzMtRfA6kno1E_bO9PBHSNqL0wiw,22080
 langfun/core/llms/anthropic_test.py,sha256=qA9vByp_cwwXNlXzcwHpPWFnO9lfFo8NKfDi5nBNqgI,9052
 langfun/core/llms/azure_openai.py,sha256=-KkSLaR54MlsIqz_XIwv0TnsBnvNTAxnjA2Q2O2u5KM,2733
@@ -92,9 +92,9 @@ langfun/core/llms/deepseek.py,sha256=jvTxdXPr-vH6HNakn_Ootx1heDg8Fen2FUkUW36bpCs
 langfun/core/llms/deepseek_test.py,sha256=DvROWPlDuow5E1lfoSkhyGt_ELA19JoQoDsTnRgDtTg,1847
 langfun/core/llms/fake.py,sha256=xmgCkk9y0I4x0IT32SZ9_OT27aLadXH8PRiYNo5VTd4,3265
 langfun/core/llms/fake_test.py,sha256=2h13qkwEz_JR0mtUDPxdAhQo7MueXaFSwsD2DIRDW9g,7653
-langfun/core/llms/gemini.py,sha256=JCxwTDdOBBhtME_s-wxV3-fOREIkQ-uMOVYFg09RmfQ,25589
+langfun/core/llms/gemini.py,sha256=kAhW6KfGhjk6S6_n915Tqdxr6Xv9Z2BGvTNcWNahP6E,26459
 langfun/core/llms/gemini_test.py,sha256=Ve9X2Wvwu9wVFHpKZDP-qoM1_hzB4kgt6_HR9wxtNkg,7592
-langfun/core/llms/google_genai.py,sha256=YjvyasibmXYPFr_sxdrIdHDPHfaNzMCiJjWJs0iQyms,5188
+langfun/core/llms/google_genai.py,sha256=L7cevWaD69xMpo5GtaVP0yEomRwbapKBrUPTClrEuPw,5332
 langfun/core/llms/google_genai_test.py,sha256=NKNtpebArQ9ZR7Qsnhd2prFIpMjleojy6o6VMXkJ1zY,1502
 langfun/core/llms/groq.py,sha256=S9V10kFo3cgX89qPgt_umq-SpRnxEDLTt_hJmpERfbo,12066
 langfun/core/llms/groq_test.py,sha256=P4EgexCqsh4K2x11w0UL_vz-YYNaPdQU0WsDAdnTRQ8,2045
@@ -106,7 +106,7 @@ langfun/core/llms/openai_compatible_test.py,sha256=KwOMA7tsmOxFBjezltkBDSU77AvOQ
 langfun/core/llms/openai_test.py,sha256=gwuO6aoa296iM2welWV9ua4KF8gEVGsEPakgbtkWkFQ,2687
 langfun/core/llms/rest.py,sha256=MCybcHApJcf49lubLnDzScN9Oc2IWY_JnMHIGdbDOuU,4474
 langfun/core/llms/rest_test.py,sha256=_zM7nV8DEVyoXNiQOnuwJ917mWjki0614H88rNmDboE,5020
-langfun/core/llms/vertexai.py,sha256=nPWwv1C-wVpWpE4Vyl4vJoAEMB519-XNzPjyLY4N3tM,18816
+langfun/core/llms/vertexai.py,sha256=PVUMcxLNgYfbKOS2pw4M8aqb0Jq7bn-bOw-VsxQqOf0,19009
 langfun/core/llms/vertexai_test.py,sha256=_e-acnNBAf9C3WO6i1b2J_mhRzdDdYQTorD9hIVZKOg,5034
 langfun/core/llms/cache/__init__.py,sha256=QAo3InUMDM_YpteNnVCSejI4zOsnjSMWKJKzkb3VY64,993
 langfun/core/llms/cache/base.py,sha256=rt3zwmyw0y9jsSGW-ZbV1vAfLxQ7_3AVk0l2EySlse4,3918
@@ -156,8 +156,8 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
 langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
 langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
 langfun/core/templates/selfplay_test.py,sha256=Ot__1P1M8oJfoTp-M9-PQ6HUXqZKyMwvZ5f7yQ3yfyM,2326
-langfun-0.1.2.dev202505190805.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202505190805.dist-info/METADATA,sha256=Od2m1ntzT0dDBzmbZXxRrKNJIBgt8ztF8uvXl1KQ2lU,8178
-langfun-0.1.2.dev202505190805.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-langfun-0.1.2.dev202505190805.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202505190805.dist-info/RECORD,,
+langfun-0.1.2.dev202505210804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202505210804.dist-info/METADATA,sha256=A7O0C4eWmx6GYJILxp3F9_p2s12WCiqcRlIVqDr8Mbc,8178
+langfun-0.1.2.dev202505210804.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+langfun-0.1.2.dev202505210804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202505210804.dist-info/RECORD,,

{langfun-0.1.2.dev202505190805.dist-info → langfun-0.1.2.dev202505210804.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any