langfun 0.1.2.dev202409100804__py3-none-any.whl → 0.1.2.dev202409130804__py3-none-any.whl

langfun/core/component.py

@@ -73,7 +73,7 @@ class Component(pg.Object):
         field.value.set_default(attr_value)
         additional_fields.append(field)
     if additional_fields:
-      pg.symbolic.update_schema(cls, additional_fields)
+      cls.update_schema(additional_fields)
 
   def _on_bound(self):
     super()._on_bound()

langfun/core/concurrent.py

@@ -469,7 +469,9 @@ class ProgressBar:
       # Process uninstall requests.
       if cls._uninstall_requests:
         for bar_id in cls._uninstall_requests:
-          cls._progress_bars.pop(bar_id, None)
+          bar = cls._progress_bars.pop(bar_id, None)
+          if bar is not None:
+            bar.close()
         cls._uninstall_requests.clear()
 
 
@@ -765,6 +767,10 @@ class _ProgressControl(pg.Object):
   def refresh(self) -> None:
     """Refresh progress bar."""
 
+  @abc.abstractmethod
+  def close(self) -> None:
+    """Close progress bar."""
+
 
 class _TqdmProgressControl(_ProgressControl):
   """Tqdm-based progress control."""
@@ -791,6 +797,9 @@ class _TqdmProgressControl(_ProgressControl):
     self._tqdm.colour = self.color
     self._tqdm.refresh()
 
+  def close(self):
+    self._tqdm.close()
+
 
 class _ConsoleProgressControl(_ProgressControl):
   """Simple progress control by printing the status to the console."""
@@ -824,6 +833,9 @@ class _ConsoleProgressControl(_ProgressControl):
       s.write(f' : {status}')
     sys.stderr.write(s.getvalue() + '\n')
 
+  def close(self):
+    sys.stderr.flush()
+
 
 class _NoopProgressControl(_ProgressControl):
   """No-op progress control."""
@@ -834,6 +846,9 @@ class _NoopProgressControl(_ProgressControl):
   def refresh(self) -> None:
     pass
 
+  def close(self) -> None:
+    pass
+
 
 def _progress_control(
     total: int,

langfun/core/concurrent_test.py

@@ -576,8 +576,8 @@ class ConcurrentMapTest(unittest.TestCase):
             (3, pg.MISSING_VALUE),
         ],
     )
-    self.assertIn('100%', string_io.getvalue())
     concurrent.ProgressBar.uninstall(bar_id)
+    self.assertIn('100%', string_io.getvalue())
 
 
 class ExecutorPoolTest(unittest.TestCase):

langfun/core/eval/base.py

@@ -941,7 +941,7 @@ class Evaluation(Evaluable):
 
     fields = list(cls.__schema__.values())
     fields.insert(0, (self.completion_prompt_field, pg.typing.Str()))
-    pg.symbolic.update_schema(cls, fields, extend=False)
+    cls.update_schema(fields, extend=False)
 
   def _maybe_adjust_examples_for_completion(
       self,

langfun/core/llms/__init__.py

@@ -39,6 +39,11 @@ from langfun.core.llms.google_genai import Palm2_IT
 # OpenAI models.
 from langfun.core.llms.openai import OpenAI
 
+from langfun.core.llms.openai import GptO1Preview
+from langfun.core.llms.openai import GptO1Preview_20240912
+from langfun.core.llms.openai import GptO1Mini
+from langfun.core.llms.openai import GptO1Mini_20240912
+
 from langfun.core.llms.openai import Gpt4oMini
 from langfun.core.llms.openai import Gpt4oMini_20240718
 from langfun.core.llms.openai import Gpt4o

langfun/core/llms/google_genai.py

@@ -27,18 +27,18 @@ try:
   import google.generativeai as genai   # pylint: disable=g-import-not-at-top
   BlobDict = genai.types.BlobDict
   GenerativeModel = genai.GenerativeModel
-  Completion = genai.types.Completion
+  Completion = getattr(genai.types, 'Completion', Any)
+  ChatResponse = getattr(genai.types, 'ChatResponse', Any)
+  GenerateContentResponse = getattr(genai.types, 'GenerateContentResponse', Any)
   GenerationConfig = genai.GenerationConfig
-  GenerateContentResponse = genai.types.GenerateContentResponse
-  ChatResponse = genai.types.ChatResponse
 except ImportError:
   genai = None
   BlobDict = Any
   GenerativeModel = Any
   Completion = Any
+  ChatResponse = Any
   GenerationConfig = Any
   GenerateContentResponse = Any
-  ChatResponse = Any
 
 
 @lf.use_init_args(['model'])

langfun/core/llms/google_genai_test.py

@@ -192,39 +192,41 @@ class GenAITest(unittest.TestCase):
   def test_call_with_legacy_completion_model(self):
     orig_get_model = genai.get_model
     genai.get_model = mock_get_model
-    orig_generate_text = genai.generate_text
-    genai.generate_text = mock_generate_text
+    orig_generate_text = getattr(genai, 'generate_text', None)
+    if orig_generate_text is not None:
+      genai.generate_text = mock_generate_text
 
-    lm = google_genai.Palm2(api_key='test_key')
-    self.maxDiff = None
-    self.assertEqual(
-        lm('hello', temperature=2.0, top_k=20).text,
-        (
-            "hello to models/text-bison-001 with {'temperature': 2.0, "
-            "'top_k': 20, 'top_p': None, 'candidate_count': 1, "
-            "'max_output_tokens': None, 'stop_sequences': None}"
-        ),
-    )
+      lm = google_genai.Palm2(api_key='test_key')
+      self.maxDiff = None
+      self.assertEqual(
+          lm('hello', temperature=2.0, top_k=20).text,
+          (
+              "hello to models/text-bison-001 with {'temperature': 2.0, "
+              "'top_k': 20, 'top_p': None, 'candidate_count': 1, "
+              "'max_output_tokens': None, 'stop_sequences': None}"
+          ),
+      )
+      genai.generate_text = orig_generate_text
     genai.get_model = orig_get_model
-    genai.generate_text = orig_generate_text
 
   def test_call_with_legacy_chat_model(self):
     orig_get_model = genai.get_model
     genai.get_model = mock_get_model
-    orig_chat = genai.chat
-    genai.chat = mock_chat
+    orig_chat = getattr(genai, 'chat', None)
+    if orig_chat is not None:
+      genai.chat = mock_chat
 
-    lm = google_genai.Palm2_IT(api_key='test_key')
-    self.maxDiff = None
-    self.assertEqual(
-        lm('hello', temperature=2.0, top_k=20).text,
-        (
-            "hello to models/chat-bison-001 with {'temperature': 2.0, "
-            "'top_k': 20, 'top_p': None, 'candidate_count': 1}"
-        ),
-    )
+      lm = google_genai.Palm2_IT(api_key='test_key')
+      self.maxDiff = None
+      self.assertEqual(
+          lm('hello', temperature=2.0, top_k=20).text,
+          (
+              "hello to models/chat-bison-001 with {'temperature': 2.0, "
+              "'top_k': 20, 'top_p': None, 'candidate_count': 1}"
+          ),
+      )
+      genai.chat = orig_chat
     genai.get_model = orig_get_model
-    genai.chat = orig_chat
 
 
 if __name__ == '__main__':

langfun/core/llms/openai.py

@@ -49,6 +49,11 @@ _DEFAULT_RPM = 3000
 SUPPORTED_MODELS_AND_SETTINGS = {
     # Models from https://platform.openai.com/docs/models
     # RPM is from https://platform.openai.com/docs/guides/rate-limits
+    # o1 (preview) models.
+    'o1-preview': pg.Dict(rpm=10000, tpm=5000000),
+    'o1-preview-2024-09-12': pg.Dict(rpm=10000, tpm=5000000),
+    'o1-mini': pg.Dict(rpm=10000, tpm=5000000),
+    'o1-mini-2024-09-12': pg.Dict(rpm=10000, tpm=5000000),
     # GPT-4o models
     'gpt-4o-mini': pg.Dict(rpm=10000, tpm=5000000),
     'gpt-4o-mini-2024-07-18': pg.Dict(rpm=10000, tpm=5000000),
@@ -175,7 +180,7 @@ class OpenAI(lf.LanguageModel):
   @property
   def is_chat_model(self):
     """Returns True if the model is a chat model."""
-    return self.model.startswith(('gpt-4', 'gpt-3.5-turbo'))
+    return self.model.startswith(('o1', 'gpt-4', 'gpt-3.5-turbo'))
 
   def _get_request_args(
       self, options: lf.LMSamplingOptions) -> dict[str, Any]:
@@ -186,9 +191,14 @@ class OpenAI(lf.LanguageModel):
         n=options.n,
         stream=False,
         timeout=self.timeout,
-        logprobs=options.logprobs,
         top_logprobs=options.top_logprobs,
     )
+    if options.logprobs:
+      # Reasoning models (o1 series) does not support `logprobs` by 2024/09/12.
+      if self.model.startswith('o1-'):
+        raise RuntimeError('`logprobs` is not supported on {self.model!r}.')
+      args['logprobs'] = options.logprobs
+
     # Completion and ChatCompletion uses different parameter name for model.
     args['model' if self.is_chat_model else 'engine'] = self.model
 
@@ -316,14 +326,15 @@ class OpenAI(lf.LanguageModel):
       samples = []
       for choice in response.choices:
         logprobs = None
-        if choice.logprobs:
+        choice_logprobs = getattr(choice, 'logprobs', None)
+        if choice_logprobs:
           logprobs = [
               (
                   t.token,
                   t.logprob,
                   [(tt.token, tt.logprob) for tt in t.top_logprobs],
               )
-              for t in choice.logprobs.content
+              for t in choice_logprobs.content
           ]
         samples.append(
             lf.LMSample(
@@ -353,6 +364,26 @@ class OpenAI(lf.LanguageModel):
     )
 
 
+class GptO1Preview(OpenAI):
+  """GPT-O1."""
+  model = 'o1-preview'
+
+
+class GptO1Preview_20240912(OpenAI):   # pylint: disable=invalid-name
+  """GPT O1."""
+  model = 'o1-preview-2024-09-12'
+
+
+class GptO1Mini(OpenAI):
+  """GPT O1-mini."""
+  model = 'o1-mini'
+
+
+class GptO1Mini_20240912(OpenAI):   # pylint: disable=invalid-name
+  """GPT O1-mini."""
+  model = 'o1-mini-2024-09-12'
+
+
 class Gpt4(OpenAI):
   """GPT-4."""
   model = 'gpt-4'

langfun/core/llms/openai_test.py

@@ -117,12 +117,13 @@ class OpenAITest(unittest.TestCase):
         openai.Gpt35(api_key='test_key', timeout=90.0)._get_request_args(
             lf.LMSamplingOptions(
                 temperature=2.0,
+                logprobs=True,
                 n=2,
                 max_tokens=4096,
                 top_p=1.0)),
         dict(
             engine='text-davinci-003',
-            logprobs=False,
+            logprobs=True,
             top_logprobs=None,
             n=2,
             temperature=2.0,
@@ -140,7 +141,6 @@ class OpenAITest(unittest.TestCase):
         ),
         dict(
             model='gpt-4',
-            logprobs=False,
             top_logprobs=None,
             n=1,
             temperature=1.0,
@@ -150,6 +150,12 @@ class OpenAITest(unittest.TestCase):
             seed=123,
         ),
     )
+    with self.assertRaisesRegex(RuntimeError, '`logprobs` is not supported.*'):
+      openai.GptO1Preview(api_key='test_key')._get_request_args(
+          lf.LMSamplingOptions(
+              temperature=1.0, logprobs=True
+          )
+      )
 
   def test_call_completion(self):
     with mock.patch('openai.Completion.create') as mock_completion:

langfun/core/structured/__init__.py

@@ -68,6 +68,7 @@ from langfun.core.structured.prompting import QueryStructurePython
 from langfun.core.structured.prompting import query
 from langfun.core.structured.prompting import query_prompt
 from langfun.core.structured.prompting import query_output
+from langfun.core.structured.prompting import query_reward
 
 from langfun.core.structured.description import DescribeStructure
 from langfun.core.structured.description import describe

langfun/core/structured/prompting.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 """Symbolic query."""
 
+import functools
 from typing import Any, Callable, Type, Union
 
 import langfun.core as lf
@@ -265,3 +266,57 @@ def query_output(
   return query(
       'Unused prompt', schema, lm=fake.StaticResponse(response), **kwargs
   )
+
+
+def query_reward(
+    mapping_example: Union[str, mapping.MappingExample],
+    response: Union[str, lf.Message],
+) -> float | None:
+  """Returns the reward of an LLM response based on an mapping example."""
+  if isinstance(mapping_example, str):
+    mapping_example = pg.from_json_str(mapping_example)
+    assert isinstance(mapping_example, mapping.MappingExample), mapping_example
+  schema = mapping_example.schema
+
+  if schema and isinstance(schema.spec, pg.typing.Object):
+    output_cls = schema.spec.cls
+  elif schema is None and isinstance(mapping_example.output, pg.Object):
+    output_cls = mapping_example.output.__class__
+  else:
+    output_cls = None
+
+  reward_fn = _reward_fn(output_cls)
+  if reward_fn is None:
+    return None
+
+  return reward_fn(
+      query_output(response, output_cls),
+      mapping_example.input,
+      mapping_example.output,
+      mapping_example.metadata,
+  )
+
+
+@functools.cache
+def _reward_fn(cls) -> Callable[
+    [
+        pg.Object,    # Actual output object.
+        Any,          # Input object.
+        pg.Object,    # Expected output object.
+        pg.Dict       # User metadata.
+    ], float] | None:
+  """Returns the reward function for a class that is being queried."""
+  if not callable(getattr(cls, '__reward__', None)):
+    return None
+
+  signature = pg.typing.signature(cls.__reward__)
+  num_args = len(signature.args)
+  if num_args < 2 or num_args > 4:
+    raise TypeError(
+        f'`{cls.__type_name__}.__reward__` should have signature: '
+        '`__reward__(self, input, [expected_output], [expected_metadata])`.'
+    )
+  def _reward(self, input, expected_output, metadata):  # pylint: disable=redefined-builtin
+    args = [self, input, expected_output, metadata]
+    return cls.__reward__(*args[:num_args])
+  return _reward

langfun/core/structured/prompting_test.py

@@ -14,6 +14,8 @@
 """Tests for structured prompting."""
 
 import inspect
+import math
+from typing import Any
 import unittest
 
 import langfun.core as lf
@@ -382,6 +384,128 @@ class QueryTest(unittest.TestCase):
         1,
     )
 
+  def test_query_reward(self):
+
+    class Answer(pg.Object):
+      final_answer: int
+
+      def __reward__(self, inputs: lf.Template) -> None:
+        diff = abs(self.final_answer - (inputs.x + inputs.y))
+        # Center screwed sigmoid scaled to [-1.0 and 1.0].
+        return 4 / (1 + math.exp(diff)) - 1.0
+
+    # Case 1: Reward function based on input and output.
+    self.assertEqual(
+        prompting.query_reward(
+            mapping.MappingExample(
+                input=lf.Template('{{x}} + {{y}}', x=1, y=1),
+                schema=Answer,
+                output=Answer(final_answer=2),
+            ),
+            'Answer(2)'
+        ),
+        1.0
+    )
+    self.assertEqual(
+        prompting.query_reward(
+            mapping.MappingExample(
+                input=lf.Template('{{x}} + {{y}}', x=2, y=3),
+                output=Answer(final_answer=2),
+            ).to_json_str(),
+            'Answer(5)'
+        ),
+        1.0
+    )
+
+    # Case 2: Reward function based on input, result and expected output.
+    class Answer2(pg.Object):
+      final_answer: int
+
+      def __reward__(self, inputs: lf.Template, expected_output: 'Answer2'):
+        return (
+            1.0 if self.final_answer == expected_output.final_answer else -1.0
+        )
+
+    self.assertEqual(
+        prompting.query_reward(
+            mapping.MappingExample(
+                input=lf.Template('{{x}} + {{y}}', x=1, y=1),
+                output=Answer2(final_answer=2),
+            ),
+            'Answer2(3)'
+        ),
+        -1.0
+    )
+
+    # Case 3: Reward function based on input, result, expected output
+    # and metadata.
+    class Answer3(pg.Object):
+      final_answer: int
+
+      def __reward__(self,
+                     inputs: lf.Template,
+                     expected_output: 'Answer3',
+                     metadata: dict[str, Any]):
+        del inputs
+        return (
+            1.0 if self.final_answer == expected_output.final_answer else -1.0
+        ) * metadata['weight']
+
+    self.assertEqual(
+        prompting.query_reward(
+            mapping.MappingExample(
+                input=lf.Template('{{x}} + {{y}}', x=1, y=1),
+                output=Answer3(final_answer=2),
+                metadata=dict(weight=0.5)
+            ),
+            'Answer3(3)'
+        ),
+        -0.5
+    )
+
+    # Case 4: No reward function is provided.
+    class Answer4(pg.Object):
+      final_answer: int
+
+    self.assertIsNone(
+        prompting.query_reward(
+            mapping.MappingExample(
+                input=lf.Template('{{x}} + {{y}}', x=1, y=1),
+                output=Answer4(final_answer=2),
+            ),
+            'Answer2(2)'
+        )
+    )
+
+    # Case 5: Not a structured output.
+    self.assertIsNone(
+        prompting.query_reward(
+            mapping.MappingExample(
+                input=lf.Template('{{x}} + {{y}}', x=1, y=1),
+                output='2',
+            ),
+            '2'
+        )
+    )
+
+    # Case 6: Bad reward function.
+    class Answer5(pg.Object):
+      final_answer: int
+
+      def __reward__(self):
+        return 0.0
+
+    with self.assertRaisesRegex(
+        TypeError, '.*Answer5.__reward__` should have signature'
+    ):
+      prompting.query_reward(
+          mapping.MappingExample(
+              input=lf.Template('{{x}} + {{y}}', x=1, y=1),
+              output=Answer5(final_answer=2),
+          ),
+          'Answer5(2)'
+      )
+
 
 class QueryStructurePythonTest(unittest.TestCase):
 

langfun/core/structured/schema.py

@@ -262,7 +262,7 @@ def class_dependencies(
               )
 
         # Add members as dependencies.
-        for field in _pg_schema(vs.cls).values():
+        for field in pg.schema(vs.cls).values():
           _fill_dependencies(field.value, include_subclasses)
       _add_dependency(vs.cls)
 
@@ -390,7 +390,7 @@ def class_definition(
 ) -> str:
   """Returns the Python class definition."""
   out = io.StringIO()
-  schema = _pg_schema(cls)
+  schema = pg.schema(cls)
   eligible_bases = []
   for base_cls in cls.__bases__:
     if base_cls is not object:
@@ -913,13 +913,3 @@ class Unknown(pg.Object, pg.typing.CustomTyping):
 
 
 UNKNOWN = Unknown()
-
-
-def _pg_schema(cls: Type[Any]) -> pg.Schema:
-  """Returns PyGlove schema for the constructor of a class."""
-  schema = getattr(cls, '__schema__', None)
-  if schema is None:
-    schema = pg.symbolic.callable_schema(
-        cls.__init__, auto_typing=True, auto_doc=True, remove_self=True
-    )
-  return schema

{langfun-0.1.2.dev202409100804.dist-info → langfun-0.1.2.dev202409130804.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langfun
-Version: 0.1.2.dev202409100804
+Version: 0.1.2.dev202409130804
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors
@@ -21,11 +21,11 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: Software Development :: Libraries
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: pyglove>=0.4.5.dev20240824
+Requires-Dist: pyglove>=0.4.5.dev202409110000
 Requires-Dist: jinja2>=3.1.2
 Requires-Dist: requests>=2.31.0
 Provides-Extra: all
-Requires-Dist: pyglove>=0.4.5.dev20240824; extra == "all"
+Requires-Dist: pyglove>=0.4.5.dev202409110000; extra == "all"
 Requires-Dist: jinja2>=3.1.2; extra == "all"
 Requires-Dist: requests>=2.31.0; extra == "all"
 Requires-Dist: termcolor==1.1.0; extra == "all"

{langfun-0.1.2.dev202409100804.dist-info → langfun-0.1.2.dev202409130804.dist-info}/RECORD

@@ -1,9 +1,9 @@
 langfun/__init__.py,sha256=mCES7t3R7Z-ZQYvG38-yrVqZubrXNfGCa8tI5HGB7mE,2274
 langfun/core/__init__.py,sha256=r86kuy-BiJIveqnXx5OklUUXtMG3q79nWRBum6zFOCQ,4835
-langfun/core/component.py,sha256=Icyoj9ICoJoK2r2PHbrFXbxnseOr9QZZOvKWklLWNo8,10276
+langfun/core/component.py,sha256=nxF4UD7NLduPjrf4IAtysMAIg3IChfoQHjY77vC8f_E,10263
 langfun/core/component_test.py,sha256=q15Xn51cVTu2RKxZ9U5VQgT3bm6RQ4638bKhWBtvW5o,8220
-langfun/core/concurrent.py,sha256=qZteOkA7jzLt7a5AVmNqDc5ZEvj_hwiLECqGMVYdzSU,27756
-langfun/core/concurrent_test.py,sha256=ULVrE6JGzsClRBLHAlqLF1A2mfxMO2X0245qYTYQ76o,16943
+langfun/core/concurrent.py,sha256=D_zoLGFREGfm0G93V27wzOlFh3MjaDxUZKb9g6Z69d4,28019
+langfun/core/concurrent_test.py,sha256=-FO_VfzNtMvdCv-qKU5StVSStDHldVLzN3xBqkAnw3U,16943
 langfun/core/console.py,sha256=bk5rNPNm9rMGW5YT2HixxU04p2umnoabn5SDz6Dqe88,2317
 langfun/core/console_test.py,sha256=5SYJdxpJGLgdSSQqqMPoA1X6jpsLD8rgcyk-EgI65oE,1077
 langfun/core/langfunc.py,sha256=G50YgoVZ0y1GFw2ev41MlOqr6qa8YakbvNC0h_E0PiA,11140
@@ -44,7 +44,7 @@ langfun/core/coding/python/parsing_test.py,sha256=9vAWF484kWIm6JZq8NFiMgKUDhXV-d
 langfun/core/coding/python/permissions.py,sha256=1QWGHvzL8MM0Ok_auQ9tURqZHtdOfJaDpBzZ29GUE-c,2544
 langfun/core/coding/python/permissions_test.py,sha256=w5EDb8QxpxgJyZkojyzVWQvDfg366zn99-g__6TbPQ0,2699
 langfun/core/eval/__init__.py,sha256=Ogdr9OtTywhhLPHi3AZzOD2mXX2oyaHWflrSTMm96uA,1899
-langfun/core/eval/base.py,sha256=YU_R9pKITKWyWa5-dSY8Q5O9pK8xioWX1ZdqYVOZcK8,74751
+langfun/core/eval/base.py,sha256=xDZQ3lu5oJaPDZCE5-QbBEajq--HRU64GVKb3xB64aI,74738
 langfun/core/eval/base_test.py,sha256=VEraWaRybSxOCOcZrZouNkiroDEPR6uyFBJoAz-1pQg,26930
 langfun/core/eval/matching.py,sha256=9GX8HfO9jKxgNLAivgy5K88Xhoh6Z7Pptq65pe7vht8,9762
 langfun/core/eval/matching_test.py,sha256=f7iVyXH5KGJBWt4Wp14Bt9J3X59A6Ayfog9MbuFvPew,5532
@@ -52,19 +52,19 @@ langfun/core/eval/patching.py,sha256=R0s2eAd1m97exQt06dmUL0V_MBG0W2Hxg7fhNB7cXW0
 langfun/core/eval/patching_test.py,sha256=8kCd54Egjju22FMgtJuxEsrXkW8ifs-UUBHtrCG1L6w,4775
 langfun/core/eval/scoring.py,sha256=AlCwEVrU6nvURDB1aPxA2XBUmOjWxuNJDXJoS4-6VbU,6386
 langfun/core/eval/scoring_test.py,sha256=O8olHbrUEg60gMxwOkWzKBJZpZoUlmVnBANX5Se2SXM,4546
-langfun/core/llms/__init__.py,sha256=a1AV3XWi2gY4UvmmaPP1GapaQxygA6xzQJvVQRp6EPA,4818
+langfun/core/llms/__init__.py,sha256=ZgnzZSjI37LhUVNvgPGNSLS-HMCqPjaCGdwPd2Ij5Rs,5031
 langfun/core/llms/anthropic.py,sha256=Gon3fOi31RhZFgNd0ijyTnKnUdp9hrWrCoSXyO4UaLw,7316
 langfun/core/llms/anthropic_test.py,sha256=T-swuMkfnlgs8Fpif4rtXs579exGk0TsbLMirXDZCkg,5533
 langfun/core/llms/fake.py,sha256=gCHBYBLvBCsC78HI1hpoqXCS-p1FMTgY1P1qh_sGBPk,3070
 langfun/core/llms/fake_test.py,sha256=sIl_Mg7nFVjaN7AJhYCpA_qzDJpSnJzkazepGXpfQQg,7338
-langfun/core/llms/google_genai.py,sha256=OYrlKIEY57CFWYwndvEdhXLwffNvDwKXyy9ULZTNeYI,10885
-langfun/core/llms/google_genai_test.py,sha256=iTISk3tJ4-3gjWmzcKQhEbH3ke4AkEiCu8rAGtB7SvU,7535
+langfun/core/llms/google_genai.py,sha256=btUIfWteBoj8Jl0j8d3e8hyI6p3Biq4rldlQYctVQfg,10936
+langfun/core/llms/google_genai_test.py,sha256=zw14sgWmk0P_irHyb7vpPy1WAuLEE0PmyfiFElu03sA,7686
 langfun/core/llms/groq.py,sha256=pqtyOZ_1_OJMOg8xATWT_B_SVbuT9nMRf4VkH9GzW8g,6308
 langfun/core/llms/groq_test.py,sha256=GYF_Qtq5S1H1TrKH38t6_lkdroqT7v-joYLDKnmS9e0,5274
 langfun/core/llms/llama_cpp.py,sha256=9tXQntSCDtjTF3bnyJrAPCr4N6wycy5nXYvp9uduygE,2843
 langfun/core/llms/llama_cpp_test.py,sha256=MWO_qaOeKjRniGjcaWPDScd7HPaIJemqUZoslrt4FPs,1806
-langfun/core/llms/openai.py,sha256=aZ5RgRBomVhBtzJtGYu9EM-qm07WEDcJr127VGpc_Sg,16161
-langfun/core/llms/openai_test.py,sha256=02KeysRppXcAwA4SBUow8hKolFiEU9_lTSdlVHZletM,17518
+langfun/core/llms/openai.py,sha256=vnDrKuD-pli0AtDIDq_TmlltOk7z7_PQ-xpU4K1ARdU,17083
+langfun/core/llms/openai_test.py,sha256=UcBFW_7RkkMEo47Tn5RuVRK_DryTN7bb9ITphlzthE8,17762
 langfun/core/llms/rest.py,sha256=laopuq-zD8V-3Y6eFDngftHEbE66VlUkCD2-rvvRaLU,3388
 langfun/core/llms/rest_test.py,sha256=NZ3Nf0XQVpT9kLP5cBVo_yBHLI7vWTYhWQxYEJVMGs4,3472
 langfun/core/llms/vertexai.py,sha256=mEQVwO3Kf3rGRmsI-qKrV6vg0hYy6OH1lEVOM81cb3U,15134
@@ -89,7 +89,7 @@ langfun/core/modalities/pdf.py,sha256=mfaeCbUA4JslFVTARiJh8hW7imvL4tLVw9gUhO5bAZ
 langfun/core/modalities/pdf_test.py,sha256=KE40zJD3Whe6ty2OULkp1J8jwLmB4ZjGXlGekluTP48,1952
 langfun/core/modalities/video.py,sha256=sKcXxbx9S1ERjH8yEzkbtySpcRJD40QiPIQiIBy-U5I,955
 langfun/core/modalities/video_test.py,sha256=GbsoefSeO7y8kCYhTtp4s9E3ah_eYrb6Z-MXpS01RFc,2046
-langfun/core/structured/__init__.py,sha256=VeB0_yV8ZEkey8kizTmB0GdkLs_aag7D9bclP8Nntac,3835
+langfun/core/structured/__init__.py,sha256=7EgI6pQIWnSNxhcIawBcSRDR8GPq5ytcmGxgPEjbxeA,3894
 langfun/core/structured/completion.py,sha256=cS2PjG7sqzDu5x0xoTk8RmNcoeX55iVwH38NTefkMHg,8108
 langfun/core/structured/completion_test.py,sha256=2mUzDMKGF_WGfTtsnfmfMDx97dkJ-98y8leen__qWLA,19281
 langfun/core/structured/description.py,sha256=SXW4MJvshFjbR-0gw6rE21o6WXq12UlRXawvDBXMZFA,5211
@@ -100,9 +100,9 @@ langfun/core/structured/mapping.py,sha256=dKOCvIA_kCQ88KoCnP5k0iOe9xRt8WLC2kbT6q
 langfun/core/structured/mapping_test.py,sha256=zQoVx3kAD5oSm_OJAQA6q41NXLLyn8qs6CIVJgAoP_w,4489
 langfun/core/structured/parsing.py,sha256=keoVqEfzAbdULh6GawWFsTQzU91MzJXYFZjXGXLaD8g,11492
 langfun/core/structured/parsing_test.py,sha256=34wDrXaQ-EYhJLfDL8mX9K53oQMSzh5pVYdKjnESmK8,20895
-langfun/core/structured/prompting.py,sha256=_U6Z65AwXvVvfaQFCY9GawB_QV9S3u7P7BOU2URABmw,8873
-langfun/core/structured/prompting_test.py,sha256=SyyBMmw-gwnrnr1MrwkjZiQaMq85kc9MHoYwzCITa10,23266
-langfun/core/structured/schema.py,sha256=oiT4P4Q9pG-QOnFzxETN2EQZqNln8nG4zAJHxcmeX9U,27729
+langfun/core/structured/prompting.py,sha256=huwwh01AQQCwPBQESOMI_V1V5PZkVQ8C89Yjk67_4Uw,10677
+langfun/core/structured/prompting_test.py,sha256=P4SsnWKC1MuslxUVPd8rJQQrBbMD_WrItNWqqSc_EV8,26689
+langfun/core/structured/schema.py,sha256=zpK3UqHPgxaHWptMr4I4zd_f22Yn4IFxF655d4O_jmQ,27416
 langfun/core/structured/schema_generation.py,sha256=U3nRQsqmMZg_qIVDh2fiY3K4JLfsAL1LcKzIFP1iXFg,5316
 langfun/core/structured/schema_generation_test.py,sha256=RM9s71kMNg2jTePwInkiW9fK1ACN37eyPeF8OII-0zw,2950
 langfun/core/structured/schema_test.py,sha256=RjYhwTgktQgyqAjzLvo967nTiIK9KWgP-aNGg4e7ihE,25258
@@ -119,8 +119,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=rBW2Qr8yi-aWYwoTwRR-n1peKyMX9QXPZXURjLgoiRs,2264
-langfun-0.1.2.dev202409100804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202409100804.dist-info/METADATA,sha256=jVTONStsTvQ-wV-qcsu8gMAmMPjX6WUvJ8nHYjgTYG4,8882
-langfun-0.1.2.dev202409100804.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-langfun-0.1.2.dev202409100804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202409100804.dist-info/RECORD,,
+langfun-0.1.2.dev202409130804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202409130804.dist-info/METADATA,sha256=ILLD9AJGZDwWiYQXBUKIVMPWWGXdIdqaQn6hOCumo9I,8890
+langfun-0.1.2.dev202409130804.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+langfun-0.1.2.dev202409130804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202409130804.dist-info/RECORD,,