langfun 0.1.2.dev202509010805__tar.gz → 0.1.2.dev202509030804__tar.gz

{langfun-0.1.2.dev202509010805 → langfun-0.1.2.dev202509030804}/PKG-INFO

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

{langfun-0.1.2.dev202509010805 → langfun-0.1.2.dev202509030804}/langfun/core/llms/__init__.py

@@ -61,6 +61,7 @@ from langfun.core.llms.google_genai import Gemini15Flash8B_001
 from langfun.core.llms.google_genai import Gemini2ProExp_20250205
 from langfun.core.llms.google_genai import Gemini2FlashThinkingExp_20250121
 from langfun.core.llms.google_genai import GeminiExp_20241206
+from langfun.core.llms.google_genai import Gemini25FlashImagePreview
 
 from langfun.core.llms.vertexai import VertexAIGemini
 from langfun.core.llms.vertexai import VertexAIGemini2Flash
@@ -86,6 +87,7 @@ from langfun.core.llms.vertexai import VertexAIGemini25FlashPreview_20250520
 from langfun.core.llms.vertexai import VertexAIGemini25ProPreview_20250605
 from langfun.core.llms.vertexai import VertexAIGemini25Pro
 from langfun.core.llms.vertexai import VertexAIGemini25Flash
+from langfun.core.llms.vertexai import VertexAIGemini25FlashImagePreview
 
 # For backward compatibility.
 GeminiPro1_5 = Gemini15Pro

{langfun-0.1.2.dev202509010805 → langfun-0.1.2.dev202509030804}/langfun/core/llms/gemini.py

@@ -195,7 +195,7 @@ SUPPORTED_MODELS = [
         rate_limits=lf.ModelInfo.RateLimits(
             max_requests_per_minute=2000,
             max_tokens_per_minute=4_000_000,
-        )
+        ),
     ),
     # Gemini 2.5 Pro 0605
     GeminiModelInfo(
@@ -218,7 +218,7 @@ SUPPORTED_MODELS = [
         rate_limits=lf.ModelInfo.RateLimits(
             max_requests_per_minute=2000,
             max_tokens_per_minute=4_000_000,
-        )
+        ),
     ),
     # Gemini 2.5 Flash Preview 0520
     GeminiModelInfo(
@@ -264,7 +264,7 @@ SUPPORTED_MODELS = [
         rate_limits=lf.ModelInfo.RateLimits(
             max_requests_per_minute=2000,
             max_tokens_per_minute=4_000_000,
-        )
+        ),
     ),
     # Gemini 2.5 Flash Preview
     GeminiModelInfo(
@@ -614,6 +614,21 @@ SUPPORTED_MODELS = [
     #
     # Experimental models.
     #
+    GeminiModelInfo(
+        model_id='gemini-2.5-flash-image-preview',
+        in_service=True,
+        experimental=True,
+        provider=pg.oneof(['Google GenAI', 'VertexAI']),
+        model_type='instruction-tuned',
+        description='Gemini 2.5 Flash Image Preview model.',
+        release_date=datetime.datetime(2025, 8, 17),
+        input_modalities=GeminiModelInfo.INPUT_IMAGE_TYPES
+        + GeminiModelInfo.INPUT_DOC_TYPES,
+        context_length=lf.ModelInfo.ContextLength(
+            max_input_tokens=32_768,
+            max_output_tokens=32_768,
+        ),
+    ),
     GeminiModelInfo(
         model_id='gemini-2.0-pro-exp-02-05',
         in_service=True,
@@ -690,6 +705,12 @@ class Gemini(rest.REST):
       'The name of the model to use.',
   ]
 
+  response_modalities: pg.typing.Annotated[
+      list[str] | None,
+      'Response modalities. It is needed for models whose response modalities '
+      + 'are more than plain text.',
+  ] = None
+
   @functools.cached_property
   def model_info(self) -> GeminiModelInfo:
     return _SUPPORTED_MODELS_BY_ID[self.model]
@@ -768,6 +789,11 @@ class Gemini(rest.REST):
           'thinkingBudget': options.max_thinking_tokens,
       }
 
+    if self.response_modalities:
+      config['responseModalities'] = self.response_modalities
+      if 'IMAGE' in self.response_modalities:
+        config.pop('responseLogprobs', None)
+        config.pop('logprobs', None)
     return config
 
   def result(self, json: dict[str, Any]) -> lf.LMSamplingResult:

{langfun-0.1.2.dev202509010805 → langfun-0.1.2.dev202509030804}/langfun/core/llms/google_genai.py

@@ -92,6 +92,11 @@ class GenAI(gemini.Gemini):
 #
 
 
+class Gemini25FlashImagePreview(GenAI):
+  """Gemini 2.5 Flash Image Preview model."""
+  model = 'gemini-2.5-flash-image-preview'
+
+
 class Gemini25Pro(GenAI):
   """Gemini 2.5 Pro GA model."""
 

{langfun-0.1.2.dev202509010805 → langfun-0.1.2.dev202509030804}/langfun/core/llms/vertexai.py

@@ -284,6 +284,14 @@ class VertexAIGemini15Flash8B_001(VertexAIGemini):  # pylint: disable=invalid-na
 #
 
 
+class VertexAIGemini25FlashImagePreview(VertexAIGemini):  # pylint: disable=invalid-name
+  """Gemini 2.5 Flash Image Preview model."""
+
+  model = 'gemini-2.5-flash-image-preview'
+  location = 'global'
+  response_modalities = ['TEXT', 'IMAGE']
+
+
 class VertexAIGemini2ProExp_20250205(VertexAIGemini):  # pylint: disable=invalid-name
   """Gemini Flash 2.0 Pro model launched on 02/05/2025."""
   model = 'gemini-2.0-pro-exp-02-05'

{langfun-0.1.2.dev202509010805 → langfun-0.1.2.dev202509030804}/langfun.egg-info/PKG-INFO

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