langfun 0.1.2.dev202505050805__py3-none-any.whl → 0.1.2.dev202505070805__py3-none-any.whl

langfun/core/agentic/action.py

@@ -194,6 +194,11 @@ class Action(pg.Object):
     """Returns the metadata associated with the result from previous call."""
     return self._invocation.metadata if self._invocation else None
 
+  @property
+  def invocation(self) -> Optional['ActionInvocation']:
+    """Returns last invocation. None if the action is not executed."""
+    return self._invocation
+
   def __call__(
       self,
       session: Optional['Session'] = None,
@@ -825,6 +830,7 @@ class ActionInvocation(pg.Object, pg.views.html.HtmlTreeView.Extension):
   def _on_bound(self):
     super()._on_bound()
     self._tab_control = None
+    self.action._invocation = self    # pylint: disable=protected-access
 
   def _on_parent_change(self, *args, **kwargs):
     super()._on_parent_change(*args, **kwargs)
@@ -1154,7 +1160,6 @@ class Session(pg.Object, pg.views.html.HtmlTreeView.Extension):
       )
 
     invocation = ActionInvocation(pg.maybe_ref(action))
-    action._invocation = invocation  # pylint: disable=protected-access
     parent_action = self._current_action
     parent_execution = self._current_execution
     parent_execution.append(invocation)
@@ -1412,38 +1417,103 @@ class Session(pg.Object, pg.views.html.HtmlTreeView.Extension):
       level: lf.logging.LogLevel,
       message: str,
       keep: bool,
+      *,
+      for_action: Action | ActionInvocation | None = None,
       **kwargs
   ) -> None:
-    """Logs a message to the session."""
-    action_name = self._current_action.action.__class__.__name__
+    """Logs a message to the session.
+
+    Args:
+      level: The logging level.
+      message: The message to log.
+      keep: Whether to keep the log entry in the execution trace.
+      for_action: The action to log the message for. If not provided, the
+        current action will be used.
+      **kwargs: Additional keyword arguments to pass to `lf.logging.log` as
+        metadata to show.
+    """
     execution = self._current_execution
+    if for_action is None:
+      for_action = self._current_action
+    elif isinstance(for_action, Action):
+      for_action = for_action.invocation
+      assert for_action is not None
+
     log_entry = lf.logging.log(
         level,
-        f'[{execution.id} ({action_name})]: {message}',
+        (
+            f'[{for_action.id} ({for_action.action.__class__.__name__})]: '
+            f'{message}'
+        ),
         **kwargs
     )
     if keep:
       execution.append(log_entry)
 
-  def debug(self, message: str, keep: bool = True, **kwargs):
+  def debug(
+      self,
+      message: str,
+      keep: bool = True,
+      *,
+      for_action: Action | ActionInvocation | None = None,
+      **kwargs
+  ) -> None:
     """Logs a debug message to the session."""
-    self._log('debug', message, keep=keep, **kwargs)
+    self._log(
+        'debug', message, keep=keep, for_action=for_action, **kwargs
+    )
 
-  def info(self, message: str, keep: bool = True, **kwargs):
+  def info(
+      self,
+      message: str,
+      keep: bool = True,
+      *,
+      for_action: Action | ActionInvocation | None = None,
+      **kwargs
+  ) -> None:
     """Logs an info message to the session."""
-    self._log('info', message, keep=keep, **kwargs)
+    self._log(
+        'info', message, keep=keep, for_action=for_action, **kwargs
+    )
 
-  def warning(self, message: str, keep: bool = True, **kwargs):
+  def warning(
+      self,
+      message: str,
+      keep: bool = True,
+      *,
+      for_action: Action | ActionInvocation | None = None,
+      **kwargs
+  ) -> None:
     """Logs a warning message to the session."""
-    self._log('warning', message, keep=keep, **kwargs)
+    self._log(
+        'warning', message, keep=keep, for_action=for_action, **kwargs
+    )
 
-  def error(self, message: str, keep: bool = True, **kwargs):
+  def error(
+      self,
+      message: str,
+      keep: bool = True,
+      *,
+      for_action: Action | ActionInvocation | None = None,
+      **kwargs
+  ):
     """Logs an error message to the session."""
-    self._log('error', message, keep=keep, **kwargs)
+    self._log(
+        'error', message, keep=keep, for_action=for_action, **kwargs
+    )
 
-  def fatal(self, message: str, **kwargs):
+  def fatal(
+      self,
+      message: str,
+      keep: bool = True,
+      *,
+      for_action: Action | ActionInvocation | None = None,
+      **kwargs
+  ):
     """Logs a fatal message to the session."""
-    self._log('fatal', message, keep=True, **kwargs)
+    self._log(
+        'fatal', message, keep=keep, for_action=for_action, **kwargs
+    )
 
   def as_message(self) -> lf.AIMessage:
     """Returns the session as a message."""

langfun/core/agentic/action_test.py

@@ -107,6 +107,7 @@ class SessionTest(unittest.TestCase):
     lm = fake.StaticResponse('lm response')
     foo = Foo(1)
     self.assertIsNone(foo.session)
+    self.assertIsNone(foo.invocation)
     self.assertIsNone(foo.result)
     self.assertIsNone(foo.metadata)
 
@@ -167,6 +168,7 @@ class SessionTest(unittest.TestCase):
 
     # Inspecting the top-level action (Foo)
     foo_invocation = root.execution[0]
+    self.assertIs(foo.invocation, foo_invocation)
     self.assertIs(foo_invocation.parent_action, root)
     self.assertEqual(foo_invocation.id, 'agent@1:/a1')
     self.assertEqual(foo_invocation.execution.id, 'agent@1:/a1')
@@ -414,8 +416,8 @@ class SessionTest(unittest.TestCase):
   def test_log(self):
     session = action_lib.Session()
     session.debug('hi', x=1, y=2)
-    session.info('hi', x=1, y=2)
-    session.warning('hi', x=1, y=2)
+    session.info('hi', x=1, y=2, for_action=session.root)
+    session.warning('hi', x=1, y=2, for_action=session.root.action)
     session.error('hi', x=1, y=2)
     session.fatal('hi', x=1, y=2)
 

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 Gemini25ProPreview_20250506
 from langfun.core.llms.google_genai import Gemini25FlashPreview_20250417
 from langfun.core.llms.google_genai import Gemini25ProPreview_20250325
 from langfun.core.llms.google_genai import Gemini2Flash
@@ -77,6 +78,7 @@ from langfun.core.llms.vertexai import VertexAIGeminiExp_20241206
 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
 
 # For backward compatibility.
 GeminiPro1_5 = Gemini15Pro
@@ -87,6 +89,7 @@ VertexAIGeminiFlash1_5 = VertexAIGemini15Flash
 # OpenAI models.
 from langfun.core.llms.openai import OpenAI
 
+from langfun.core.llms.openai import Gpt41
 from langfun.core.llms.openai import GptO3
 from langfun.core.llms.openai import GptO4Mini
 from langfun.core.llms.openai import Gpt45Preview_20250227

langfun/core/llms/gemini.py

@@ -151,6 +151,29 @@ SUPPORTED_MODELS = [
     #
     # Production models.
     #
+    # Gemini 2.5 Pro Preview
+    GeminiModelInfo(
+        model_id='gemini-2.5-pro-preview-05-06',
+        in_service=True,
+        provider=pg.oneof(['Google GenAI', ' VertexAI']),
+        model_type='instruction-tuned',
+        description='Gemini 2.5 Pro.',
+        release_date=datetime.datetime(2025, 5, 6),
+        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=1.25,
+            cost_per_1m_input_tokens=1.25,
+            cost_per_1m_output_tokens=10.0,
+        ),
+        rate_limits=lf.ModelInfo.RateLimits(
+            max_requests_per_minute=2000,
+            max_tokens_per_minute=4_000_000,
+        )
+    ),
     # Gemini 2.5 Flash Preview
     GeminiModelInfo(
         model_id='gemini-2.5-flash-preview-04-17',
@@ -657,7 +680,7 @@ class Gemini(rest.REST):
   def result(self, json: dict[str, Any]) -> lf.LMSamplingResult:
     messages = [
         lf.Message.from_value(candidate['content'], format='gemini')
-        for candidate in json['candidates']
+        for candidate in json.get('candidates', [])
     ]
     usage = json['usageMetadata']
     input_tokens = usage['promptTokenCount']

langfun/core/llms/google_genai.py

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

langfun/core/llms/openai.py

@@ -49,6 +49,30 @@ class OpenAIModelInfo(lf.ModelInfo):
 #
 
 SUPPORTED_MODELS = [
+    # GPT-4.1 models
+    OpenAIModelInfo(
+        model_id='gpt-4.1',
+        alias_for='gpt-4.1-2025-04-14',
+        in_service=True,
+        model_type='instruction-tuned',
+        description='GPT 4.1 model (latest stable).',
+        url='https://platform.openai.com/docs/models/gpt-4.1',
+        input_modalities=OpenAIModelInfo.INPUT_IMAGE_TYPES,
+        context_length=lf.ModelInfo.ContextLength(
+            max_input_tokens=1_047_576,
+            max_output_tokens=32_768,
+        ),
+        pricing=lf.ModelInfo.Pricing(
+            cost_per_1m_cached_input_tokens=0.50,
+            cost_per_1m_input_tokens=2.0,
+            cost_per_1m_output_tokens=8.0,
+        ),
+        # Tier 5 rate limits.
+        rate_limits=lf.ModelInfo.RateLimits(
+            max_requests_per_minute=10_000,
+            max_tokens_per_minute=30_000_000,
+        ),
+    ),
     # o3 models.
     OpenAIModelInfo(
         model_id='o3',
@@ -1045,6 +1069,11 @@ class OpenAI(openai_compatible.OpenAICompatible):
     return super()._request_args(options)
 
 
+class Gpt41(OpenAI):
+  """GPT-4.1."""
+  model = 'gpt-4.1'
+
+
 class GptO4Mini(OpenAI):
   """GPT O4 mini."""
 

langfun/core/llms/vertexai.py

@@ -173,9 +173,13 @@ class VertexAIGemini(VertexAI, gemini.Gemini):
   def model_info(self) -> gemini.GeminiModelInfo:
     return super().model_info.clone(override=dict(provider='VertexAI'))
 
+
 #
 # Production models.
 #
+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'
 
 
 class VertexAIGemini25FlashPreview_20250417(VertexAIGemini):  # pylint: disable=invalid-name

langfun/core/memories/conversation_history.py

@@ -43,19 +43,19 @@ class ConversationHistory(lf.Memory):
       messages.append(r[1])
     return messages
 
-  def _recollect(self, **kwargs) -> lf.MemoryRecord:
-    return lf.MemoryRecord(
-        '\n'.join([f'{m.sender}: {m.text}' for m in self.messages]))
+  def _recollect(self, **kwargs) -> str:
+    return '\n'.join([f'{m.sender}: {m.text}' for m in self.messages])
 
   def _remember(
       self,
-      input_message: lf.Message,
-      output_message: lf.Message | None = None,
+      value: tuple[str | lf.Message, str | lf.Message],
       **kwargs
   ) -> None:
     if self.max_turns and self.max_turns == len(self._turns):
       self._turns.pop(0)
-    self._turns.append((input_message, output_message))
+    self._turns.append(
+        (lf.UserMessage.from_value(value[0]), lf.AIMessage.from_value(value[1]))
+    )
 
   def _reset(self, **kwargs) -> None:
     self._turns = []

langfun/core/memories/conversation_history_test.py

@@ -13,58 +13,62 @@
 # limitations under the License.
 """Base memory test."""
 
-import inspect
 import unittest
 import langfun.core as lf
 from langfun.core.memories.conversation_history import ConversationHistory
+import pyglove as pg
 
 
 class ConversationHistoryTest(unittest.TestCase):
 
   def test_history_with_no_round_limit(self):
     m = ConversationHistory()
-    m.remember('hi', 'hello')
+    m.remember(('hi', 'hello'))
     m.remember(
-        'how are you',
-        'Fine, thank you. Anything I can help with?',
+        ('how are you', 'Fine, thank you. Anything I can help with?'),
     )
     m.remember(
-        'Not for now, bye.',
-        'Okay, bye!',
+        ('Not for now, bye.', 'Okay, bye!'),
     )
     self.assertEqual(len(m.turns), 3)
     self.assertEqual(len(m.messages), 6)
-    self.assertEqual(
-        m.recollect(),
-        lf.MemoryRecord(inspect.cleandoc("""
-            User: hi
-            AI: hello
-            User: how are you
-            AI: Fine, thank you. Anything I can help with?
-            User: Not for now, bye.
-            AI: Okay, bye!
-            """)),
+    self.assertTrue(
+        pg.eq(
+            m.recollect(),
+            lf.Template(
+                """
+                User: hi
+                AI: hello
+                User: how are you
+                AI: Fine, thank you. Anything I can help with?
+                User: Not for now, bye.
+                AI: Okay, bye!
+                """
+            )
+        )
     )
 
   def test_history_with_round_limit(self):
     m = ConversationHistory(max_turns=1)
-    m.remember('hi', 'hello')
+    m.remember(('hi', 'hello'))
     m.remember(
-        'how are you',
-        'Fine, thank you. Anything I can help with?',
+        ('how are you', 'Fine, thank you. Anything I can help with?'),
     )
     m.remember(
-        'Not for now, bye.',
-        'Okay, bye!',
+        ('Not for now, bye.', 'Okay, bye!'),
     )
     self.assertEqual(len(m.turns), 1)
     self.assertEqual(len(m.messages), 2)
-    self.assertEqual(
-        m.recollect(),
-        lf.MemoryRecord(inspect.cleandoc("""
-            User: Not for now, bye.
-            AI: Okay, bye!
-            """)),
+    self.assertTrue(
+        pg.eq(
+            m.recollect(),
+            lf.Template(
+                """
+                User: Not for now, bye.
+                AI: Okay, bye!
+                """
+            )
+        )
     )
 
 

langfun/core/memory.py

@@ -14,7 +14,9 @@
 """Interface for memories."""
 
 import abc
-from langfun.core import message as message_lib
+from typing import Any
+
+from langfun.core import template as template_lib
 from langfun.core.component import Component
 from langfun.core.natural_language import NaturalLanguageFormattable
 
@@ -30,40 +32,29 @@ class Memory(NaturalLanguageFormattable, Component):
   """
 
   def natural_language_format(self) -> str:
-    return self.recollect().text
+    return self.recollect().render().text
 
-  def recollect(self, **kwargs) -> message_lib.MemoryRecord:
+  def recollect(self, **kwargs) -> template_lib.Template:
     """Recollects a message from the memory."""
-    return self._recollect(**kwargs)
+    return template_lib.Template.from_value(self._recollect(**kwargs))
 
   def remember(
       self,
-      input_message: str | message_lib.Message,
-      output_message: str | message_lib.Message,
+      value: Any,
       **kwargs
-  ) -> str:
-    if isinstance(input_message, str):
-      input_message = message_lib.UserMessage(input_message)
-    if isinstance(output_message, str):
-      output_message = message_lib.AIMessage(output_message)
-    self._remember(input_message, output_message, **kwargs)
-    return output_message.text
+  ) -> None:
+    self._remember(value, **kwargs)
 
   def reset(self, **kwargs) -> None:
     """Resets the memory."""
-    self._reset()
+    self._reset(**kwargs)
 
   @abc.abstractmethod
-  def _recollect(self, **kwargs) -> message_lib.MemoryRecord:
+  def _recollect(self, **kwargs) -> str | template_lib.Template:
     """Recollects a message from the memory."""
 
   @abc.abstractmethod
-  def _remember(
-      self,
-      input_message: message_lib.Message,
-      output_message: message_lib.Message,
-      **kwargs
-  ) -> None:
+  def _remember(self, value: Any, **kwargs) -> None:
     """Remembers the input message and the output message."""
 
   @abc.abstractmethod

langfun/core/structured/schema.py

@@ -185,10 +185,14 @@ class Schema(
   def class_dependencies(
       self,
       include_base_classes: bool = True,
-      include_subclasses: bool = True) -> list[Type[Any]]:
+      include_subclasses: bool = True,
+      include_generated_subclasses: bool = False) -> list[Type[Any]]:
     """Returns a list of class dependencies for current schema."""
     return class_dependencies(
-        self.spec, include_base_classes, include_subclasses
+        self.spec,
+        include_base_classes,
+        include_subclasses,
+        include_generated_subclasses
     )
 
   @classmethod
@@ -247,6 +251,7 @@ def class_dependencies(
     ],
     include_base_classes: bool = True,
     include_subclasses: bool = True,
+    include_generated_subclasses: bool = False,
 ) -> list[Type[Any]]:
   """Returns a list of class dependencies from a value or specs."""
   if isinstance(value_or_spec, Schema):
@@ -301,7 +306,12 @@ def class_dependencies(
       # Check subclasses if available.
       if include_subclasses:
         for cls in vs.cls.__subclasses__():
-          if cls not in dependencies:
+          # NOTE(daiyip): To prevent LLM-generated "hallucinated" classes from
+          # polluting the generation space, classes dynamically created by
+          # 'eval' (which have __module__ == 'builtins') are excluded from
+          # dependencies by default.
+          if ((include_generated_subclasses or cls.__module__ != 'builtins')
+              and cls not in dependencies):
             _fill_dependencies(pg.typing.Object(cls), include_subclasses=True)
 
     if isinstance(vs, pg.typing.List):

langfun/core/structured/schema_test.py

@@ -144,6 +144,14 @@ class SchemaTest(unittest.TestCase):
       foo2: Foo | X
 
     schema = schema_lib.Schema([B])
+    v = schema_lib.structure_from_python(
+        """
+        class C(B):
+          pass
+        """,
+        global_vars=dict(B=B)
+    )
+    self.assertEqual(v.__module__, 'builtins')
     self.assertEqual(schema.class_dependencies(), [Foo, A, Bar, X, B])
 
   def test_class_dependencies_non_pyglove(self):
@@ -162,6 +170,14 @@ class SchemaTest(unittest.TestCase):
     class BB(AA):
       foo2: Baz | XX
 
+    v = schema_lib.structure_from_python(
+        """
+        class CC(BB):
+          pass
+        """,
+        global_vars=dict(BB=BB)
+    )
+    self.assertEqual(v.__module__, 'builtins')
     schema = schema_lib.Schema([AA])
     self.assertEqual(schema.class_dependencies(), [Baz, AA, XX, BB])
 

langfun/core/templates/conversation.py

@@ -90,7 +90,7 @@ class Conversation(Completion):
       self, input_message: lf.Message | str, lm_response: lf.Message | str
   ) -> None:
     """Adds a turn."""
-    self.memory.remember(input_message, lm_response)
+    self.memory.remember((input_message, lm_response))
 
   def reset(self) -> None:
     """Resets current conversation."""

{langfun-0.1.2.dev202505050805.dist-info → langfun-0.1.2.dev202505070805.dist-info}/METADATA

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

{langfun-0.1.2.dev202505050805.dist-info → langfun-0.1.2.dev202505070805.dist-info}/RECORD

@@ -12,7 +12,7 @@ langfun/core/language_model.py,sha256=c6Vds9qXuC0BiHeZstxCTTsoU8ZL9K4w4co_hXA3-R
 langfun/core/language_model_test.py,sha256=iA5uo7rIj2jAtCYzMzhyNg1fWqE2Onn60bOO58q72C0,36454
 langfun/core/logging.py,sha256=ypjSwGzQpIhadewEKkIH5VWS55rfU4e6mlni41PMHlw,9116
 langfun/core/logging_test.py,sha256=vbVGOQxwMmVSiFfbt2897gUt-8nqDpV64jCAeUG_q5U,6924
-langfun/core/memory.py,sha256=f-asN1F7Vehgdn_fK84v73GrEUOxRtaW934keutTKjk,2416
+langfun/core/memory.py,sha256=vyXVvfvSdLLJAzdIupnbn3k26OgclCx-OJ7gddS5e1Y,2070
 langfun/core/message.py,sha256=qUJZ9NfrlYG3aU_K2ud236gdTnvZ7Qw2T4pv4hI9ivg,32817
 langfun/core/message_test.py,sha256=XE7SaYJUK8TCJBdih4XtnpnB6NhcU-vLxJoaP67WbSU,40793
 langfun/core/modality.py,sha256=K8pUGuMpfWcOtVcXC_OqVjro1-RhHF6ddQni61DuYzM,4166
@@ -26,10 +26,10 @@ langfun/core/subscription_test.py,sha256=Y4ZdbZEwm83YNZBxHff0QR4QUa4rdaNXA3_jfIc
 langfun/core/template.py,sha256=jNhYSrbLIn9kZOa03w5QZbyjgfnzJzE_ZrrMvvWY4t4,24929
 langfun/core/template_test.py,sha256=AQv_m9qE93WxhEhSlm1xaBgB4hu0UVtA53dljngkUW0,17090
 langfun/core/agentic/__init__.py,sha256=qR3jlfUO4rhIoYdRDLz-d22YZf3FvU4FW88vsjiGDQQ,1224
-langfun/core/agentic/action.py,sha256=3m2-k07Zz8qrzOdOa7xPl7fRH3I0c3VsMLR86_JBCcU,45359
+langfun/core/agentic/action.py,sha256=jxjW4vkJs3p-AYb83P0AVnmE93Vx7z0QoFsm40NQ65k,46853
 langfun/core/agentic/action_eval.py,sha256=JXhS5qEjWu9EZ0chDsjWxCqPAV26PUCBijtUYxiDeO4,4975
 langfun/core/agentic/action_eval_test.py,sha256=7AkOwNbUX-ZgR1R0a7bvUZ5abNTUV7blf_8Mnrwb-II,2811
-langfun/core/agentic/action_test.py,sha256=ezqg3tKlVwgLMnHKUmOdtxpnntuL8YIvhcTCSSdb8oc,15468
+langfun/core/agentic/action_test.py,sha256=wrx7_JUyL6EXhJeUqmPkPyIwOAh_KwTOWhSRgv2wYWo,15613
 langfun/core/coding/__init__.py,sha256=5utju_fwEsImaiftx4oXKl9FAM8p281k8-Esdh_-m1w,835
 langfun/core/coding/python/__init__.py,sha256=4ByknuoNU-mOIHwHKnTtmo6oD64oMFtlqPlYWmA5Wic,1736
 langfun/core/coding/python/correction.py,sha256=7zBedlhQKMPA4cfchUMxAOFl6Zl5RqCyllRHGWys40s,7092
@@ -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=8PGNlpqSGCW4O-z0E9SHcLklZNceAzwbS7letstljPE,16652
 langfun/core/eval/v2/runners_test.py,sha256=A37fKK2MvAVTiShsg_laluJzJ9AuAQn52k7HPbfD0Ks,11666
-langfun/core/llms/__init__.py,sha256=0XyDya5doAIK_8Fjw_a78El-eBf63ggJMY8vICtQMtw,8495
+langfun/core/llms/__init__.py,sha256=QWxRhzVn_vgJvdmW_xs5PcPuDbHsUxTU94YyV4Ofl34,8684
 langfun/core/llms/anthropic.py,sha256=qaclpfX3qeHoZMDxU3Gn-638Vi4IyCbxdow3zgGUHK4,22195
 langfun/core/llms/anthropic_test.py,sha256=dFnNvrgwCYUseDuiuWCBoQ5jloYX9RIlZQf7cCLPNU4,8282
 langfun/core/llms/azure_openai.py,sha256=-KkSLaR54MlsIqz_XIwv0TnsBnvNTAxnjA2Q2O2u5KM,2733
@@ -92,29 +92,29 @@ 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=54C_rbAOgMXHe_X9MrZCMCaLHEdrikjX2mq6zRkJ3as,24491
+langfun/core/llms/gemini.py,sha256=UGI89WjBTZHBAmWbGh4FUCryw6OW3yuiN7htY-Xv_zM,25358
 langfun/core/llms/gemini_test.py,sha256=Ve9X2Wvwu9wVFHpKZDP-qoM1_hzB4kgt6_HR9wxtNkg,7592
-langfun/core/llms/google_genai.py,sha256=j8W22WFvkT80Fw-r7Rg-e7MKhcSwljZkmtuufwSEn5s,5051
+langfun/core/llms/google_genai.py,sha256=YjvyasibmXYPFr_sxdrIdHDPHfaNzMCiJjWJs0iQyms,5188
 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
 langfun/core/llms/llama_cpp.py,sha256=Z7P3gc4xeIjc2bX0Ey1y5EUYJVMnMa2Q67PZ9iye9sE,1409
 langfun/core/llms/llama_cpp_test.py,sha256=wfTO7nmUwL65U2kK9P9fcMt92JjNDuVia4G1E7znf_4,1086
-langfun/core/llms/openai.py,sha256=F3vQkLVKeKhm34e2akqryuhjl7uOefAANUN3_i4SVhM,41414
+langfun/core/llms/openai.py,sha256=LxENwsuGh-h1LXMu2Ub8Rjp1g1WNpm1NHBrwhkD-M4Q,42333
 langfun/core/llms/openai_compatible.py,sha256=JlFUTiK4e3ox2DGeGBcAD-cXkxmBdx5g6LrYkyMIaps,5777
 langfun/core/llms/openai_compatible_test.py,sha256=KwOMA7tsmOxFBjezltkBDSU77AvOQkI23dO2nHLAlB4,17689
 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=4t_Noj7cqzLNmESYCYzz9Ndodd_K4I4zxVLmljJ7r3E,18630
+langfun/core/llms/vertexai.py,sha256=nPWwv1C-wVpWpE4Vyl4vJoAEMB519-XNzPjyLY4N3tM,18816
 langfun/core/llms/vertexai_test.py,sha256=0M4jsPOXGagdzPfEdJixmyLdhmmERePZWSFfTwnaYCQ,4875
 langfun/core/llms/cache/__init__.py,sha256=QAo3InUMDM_YpteNnVCSejI4zOsnjSMWKJKzkb3VY64,993
 langfun/core/llms/cache/base.py,sha256=rt3zwmyw0y9jsSGW-ZbV1vAfLxQ7_3AVk0l2EySlse4,3918
 langfun/core/llms/cache/in_memory.py,sha256=i58oiQL28RDsq37dwqgVpC2mBETJjIEFS20yHiV5MKU,5185
 langfun/core/llms/cache/in_memory_test.py,sha256=V2UPeu5co5vUwSkjekCH1B1iLm9qQKPaacvP6VW3GTg,10388
 langfun/core/memories/__init__.py,sha256=HpghfZ-w1NQqzJXBx8Lz0daRhB2rcy2r9Xm491SBhC4,773
-langfun/core/memories/conversation_history.py,sha256=c9amD8hCxGFiZuVAzkP0dOMWSp8L90uvwkOejjuBqO0,1835
-langfun/core/memories/conversation_history_test.py,sha256=AaW8aNoFjxNusanwJDV0r3384Mg0eAweGmPx5DIkM0Y,2052
+langfun/core/memories/conversation_history.py,sha256=KR78PurXTSeqsRK9QG9xM7-f245rs20EvWO7Mm2n2Vw,1827
+langfun/core/memories/conversation_history_test.py,sha256=2kzAq2pUrbR01Z9jhxviIao52JK4JVjr5iGP8pwGxlU,2156
 langfun/core/modalities/__init__.py,sha256=rh74vS_oL3XzD83ZJrDzwmW-8jnJHJ3s2_Prmp3ndYQ,1116
 langfun/core/modalities/audio.py,sha256=qCrVCX690SG0ps-ZfOtNWvHn_CmdJsmxF7GySScWUqY,964
 langfun/core/modalities/audio_test.py,sha256=yyGEBYqMXmNs_F2dEtj-PX8HE040vqh-YQppsvdxPw4,2025
@@ -139,10 +139,10 @@ langfun/core/structured/parsing.py,sha256=MGvI7ypXlwfzr5XB8_TFU9Ei0_5reYqkWkv64e
 langfun/core/structured/parsing_test.py,sha256=V8Cj1tJK4Lxv_b0YQj6-2hzXZgnYNBa2JR7rOLRBKoQ,22346
 langfun/core/structured/querying.py,sha256=lHSPQ4UKRfhLXRjofG1geIGpQ_ImZf0dIvjXIwr_jNk,30480
 langfun/core/structured/querying_test.py,sha256=_npZ3ztaZc6VerP7nU_QTJscWGgBqiwTE02z_S3Ahd4,40197
-langfun/core/structured/schema.py,sha256=pGiAjez-ON2nKLUSeAls27gJsMto5aJnCXLVwH3pUKM,28296
+langfun/core/structured/schema.py,sha256=UTonddBx3hVr0Zhm_38jqd8khTSXWy-bkk8l1YOUdLA,28797
 langfun/core/structured/schema_generation.py,sha256=3AcuKvv3VOtKY5zMVqODrxfOuDxzoZtGeBxHlOWDOWw,5308
 langfun/core/structured/schema_generation_test.py,sha256=RM9s71kMNg2jTePwInkiW9fK1ACN37eyPeF8OII-0zw,2950
-langfun/core/structured/schema_test.py,sha256=N8qNVA2hrHlxmHIoKtpzhBASa2RMVoPowF_noAfgPME,26035
+langfun/core/structured/schema_test.py,sha256=pW0uoriwE_F6VH2DlAopeXlltcLBP_xnf8DA7QhVroc,26409
 langfun/core/structured/scoring.py,sha256=Y7Jqs5VVjUQLF_9Z1uIY_dw5zasv2FF52Cz-cxGMsro,5857
 langfun/core/structured/scoring_test.py,sha256=QvlwDAzwuamKL5tCotm1L3Sx0cs3idoNK4aIEhaO4Yk,2272
 langfun/core/structured/tokenization.py,sha256=-b4_693quHeYn2AqndwucuXNmhd5NVXVTU3mmDane98,2189
@@ -150,14 +150,14 @@ langfun/core/structured/tokenization_test.py,sha256=dVW30kGYkX2HNtiRZe1oTmXFP7iI
 langfun/core/templates/__init__.py,sha256=bO0eMsVJbi7sxEB2YlInKRQ2EVP-RyyKUwcD-8msuN4,927
 langfun/core/templates/completion.py,sha256=mUqZHOEV3ag6-A08XghpeEltcrBvCDxXP004eDDfeag,1931
 langfun/core/templates/completion_test.py,sha256=vGnjnM38UHyVDUyaUYtmp20s9KBGOdbPVsX-H-ET11E,1636
-langfun/core/templates/conversation.py,sha256=dTy_fs3g92d3vP09bACGJ_zJ3b8-OzA8vsKgUDoB76o,2866
+langfun/core/templates/conversation.py,sha256=iURikG7JEvXGsFSZfdzj6PyGQCAPAEejrs8hXEh7Sc0,2868
 langfun/core/templates/conversation_test.py,sha256=HCD5f1sgmSGqL3OGe7hEctjRdcFjcmkz3NV1ArexNYI,3942
 langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fikKhwhzwhpKI,1460
 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.dev202505050805.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202505050805.dist-info/METADATA,sha256=bhWHxi4ysmA7AbYsj4OQlHgfQ_cTQ1GCflqdlICgoD4,8178
-langfun-0.1.2.dev202505050805.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-langfun-0.1.2.dev202505050805.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202505050805.dist-info/RECORD,,
+langfun-0.1.2.dev202505070805.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202505070805.dist-info/METADATA,sha256=AaKPpm66YjxOfH--Ct9d_I3L_0h5ngugLiU2RwI1CRE,8178
+langfun-0.1.2.dev202505070805.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+langfun-0.1.2.dev202505070805.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202505070805.dist-info/RECORD,,