langfun 0.1.2.dev202505150805__py3-none-any.whl → 0.1.2.dev202505170804__py3-none-any.whl

langfun/core/agentic/action.py

@@ -225,7 +225,7 @@ class Action(pg.Object):
         result = self.call(session=session, **kwargs)
         self._invocation.end(result)
       except BaseException as e:
-        error = pg.utils.ErrorInfo.from_exception(e)
+        error = pg.ErrorInfo.from_exception(e)
         self._invocation.end(result=None, error=error)
         if self._session is not None:
           self._session.end(result=None, error=error)
@@ -819,7 +819,7 @@ class ActionInvocation(pg.Object, pg.views.html.HtmlTreeView.Extension):
   ] = {}
 
   error: Annotated[
-      pg.utils.ErrorInfo | None,
+      pg.ErrorInfo | None,
       'Error from the action if failed.'
   ] = None
 
@@ -910,7 +910,7 @@ class ActionInvocation(pg.Object, pg.views.html.HtmlTreeView.Extension):
   def end(
       self,
       result: Any,
-      error: pg.utils.ErrorInfo | None = None,
+      error: pg.ErrorInfo | None = None,
       metadata: dict[str, Any] | None = None,
   ) -> None:
     """Ends the execution of the action with result and metadata."""
@@ -1138,7 +1138,7 @@ class Session(pg.Object, pg.views.html.HtmlTreeView.Extension):
     return self.root.result
 
   @property
-  def final_error(self) -> pg.utils.ErrorInfo | None:
+  def final_error(self) -> pg.ErrorInfo | None:
     """Returns the error of the session."""
     return self.root.error
 
@@ -1186,7 +1186,7 @@ class Session(pg.Object, pg.views.html.HtmlTreeView.Extension):
   def end(
       self,
       result: Any,
-      error: pg.utils.ErrorInfo | None = None,
+      error: pg.ErrorInfo | None = None,
       metadata: dict[str, Any] | None = None,
   ) -> None:
     """Ends the session."""
@@ -1220,7 +1220,7 @@ class Session(pg.Object, pg.views.html.HtmlTreeView.Extension):
 
     if exc_val is not None:
       result, metadata = None, None
-      error = pg.utils.ErrorInfo.from_exception(exc_val)
+      error = pg.ErrorInfo.from_exception(exc_val)
     else:
       actions = self.root.actions
       if actions:

langfun/core/agentic/action_test.py

@@ -308,7 +308,7 @@ class SessionTest(unittest.TestCase):
     self.assertTrue(session.has_started)
     self.assertTrue(session.has_stopped)
     self.assertTrue(session.has_error)
-    self.assertIsInstance(session.final_error, pg.utils.ErrorInfo)
+    self.assertIsInstance(session.final_error, pg.ErrorInfo)
     self.assertIn('Bar error', str(session.root.error))
 
   def test_succeeded_with_explicit_session(self):
@@ -375,7 +375,7 @@ class SessionTest(unittest.TestCase):
     self.assertTrue(session.has_stopped)
     self.assertTrue(session.has_error)
     self.assertIsNone(session.final_result)
-    self.assertIsInstance(session.root.error, pg.utils.ErrorInfo)
+    self.assertIsInstance(session.root.error, pg.ErrorInfo)
     self.assertIn('Bar error', str(session.root.error))
 
   def test_failed_with_explicit_session_without_start(self):
@@ -412,7 +412,7 @@ class SessionTest(unittest.TestCase):
     self.assertTrue(session.has_started)
     self.assertTrue(session.has_stopped)
     self.assertTrue(session.has_error)
-    self.assertIsInstance(session.root.error, pg.utils.ErrorInfo)
+    self.assertIsInstance(session.root.error, pg.ErrorInfo)
     self.assertEqual(len(session.root.execution), 3)
     self.assertEqual(len(session.root.actions), 2)
     self.assertEqual(len(session.root.logs), 1)

langfun/core/eval/v2/evaluation.py

@@ -255,7 +255,7 @@ class Evaluation(experiment_lib.Experiment):
       except BaseException as e:  # pylint: disable=broad-except
         if raise_if_has_error:
           raise
-        example.error = pg.object_utils.ErrorInfo.from_exception(e)
+        example.error = pg.ErrorInfo.from_exception(e)
 
   #
   # Handling evaluation scheduling.

langfun/core/eval/v2/example.py

@@ -39,7 +39,7 @@ class Example(pg.JSONConvertible, pg.views.HtmlTreeView.Extension):
   id: int
   input: Any = pg.MISSING_VALUE
   output: Any = pg.MISSING_VALUE
-  error: pg.object_utils.ErrorInfo | None = None
+  error: pg.ErrorInfo | None = None
   metadata: dict[str, Any] = dataclasses.field(default_factory=dict)
   metric_metadata: dict[str, Any] | None = None
   # Execution information.
@@ -47,12 +47,13 @@ class Example(pg.JSONConvertible, pg.views.HtmlTreeView.Extension):
   start_time: float | None = None
   end_time: float | None = None
   usage_summary: lf.UsageSummary | None = None
-  execution_status: dict[str, pg.object_utils.TimeIt.Status] | None = None
+  execution_status: dict[str, pg.utils.TimeIt.Status] | None = None
 
   def __post_init__(self):
     if self.execution_status is not None:
       for status in self.execution_status.values():
         if status.has_error:
+          assert isinstance(status.error, pg.ErrorInfo)
           self.error = status.error
           break
 
@@ -188,7 +189,7 @@ class Example(pg.JSONConvertible, pg.views.HtmlTreeView.Extension):
         text = f'{key}:{value}'
       return pg.views.html.controls.Badge(
           text,
-          css_classes=[pg.object_utils.camel_to_snake(key, '-')],
+          css_classes=[pg.utils.camel_to_snake(key, '-')],
       )
 
     def _render_header():

langfun/core/eval/v2/example_test.py

@@ -22,13 +22,13 @@ Example = example_lib.Example
 class ExampleTest(unittest.TestCase):
 
   def test_basic(self):
-    error = pg.object_utils.ErrorInfo(
+    error = pg.ErrorInfo(
         tag='ValueError',
         description='Bad input',
         stacktrace='...',
     )
     ex = Example(id=1, execution_status={
-        'evaluate': pg.object_utils.TimeIt.Status(
+        'evaluate': pg.utils.TimeIt.Status(
             name='evaluation', elapse=1.0, error=error
         )
     })

langfun/core/eval/v2/metrics_test.py

@@ -37,7 +37,7 @@ class MatchTest(unittest.TestCase):
             Example(
                 id=3,
                 input=pg.Dict(groundtruth=1),
-                error=pg.object_utils.ErrorInfo(
+                error=pg.symbolic.ErrorInfo(
                     tag='ValueError',
                     description='Bad input.',
                     stacktrace='...',
@@ -51,7 +51,7 @@ class MatchTest(unittest.TestCase):
             Example(
                 id=3,
                 input=pg.Dict(groundtruth=1),
-                error=pg.object_utils.ErrorInfo(
+                error=pg.symbolic.ErrorInfo(
                     tag='MappingError.CodeError',
                     description='Bad input.',
                     stacktrace='...',
@@ -130,7 +130,7 @@ class ScoreTest(unittest.TestCase):
             Example(
                 id=3,
                 input=pg.Dict(x=1),
-                error=pg.object_utils.ErrorInfo(
+                error=pg.symbolic.ErrorInfo(
                     tag='ValueError',
                     description='Bad input.',
                     stacktrace='...',
@@ -144,7 +144,7 @@ class ScoreTest(unittest.TestCase):
             Example(
                 id=3,
                 input=pg.Dict(x=1),
-                error=pg.object_utils.ErrorInfo(
+                error=pg.symbolic.ErrorInfo(
                     tag='MappingError.CodeError',
                     description='Bad input.',
                     stacktrace='...',

langfun/core/logging.py

@@ -276,7 +276,7 @@ def log(level: LogLevel,
       console_lib.write(entry)
   elif not console:
     if kwargs:
-      message = f'{message} (metadata: {pg.format(kwargs)})'
+      message = f'{message} (metadata: {pg.format(kwargs, verbose=False)})'
     system_log_func(level)(message)
   return entry
 

langfun/core/structured/querying.py

@@ -649,7 +649,7 @@ def query(
       _mark_query_completed(output_message, None, usage_summary)
     except mapping.MappingError as e:
       _mark_query_completed(
-          e.lm_response, pg.utils.ErrorInfo.from_exception(e), usage_summary
+          e.lm_response, pg.ErrorInfo.from_exception(e), usage_summary
       )
       if lf.RAISE_IF_HAS_ERROR == default:
         raise e
@@ -657,7 +657,7 @@ def query(
       output_message.result = default
     except BaseException as e:
       _mark_query_completed(
-          None, pg.utils.ErrorInfo.from_exception(e), usage_summary
+          None, pg.ErrorInfo.from_exception(e), usage_summary
       )
       raise e
 
@@ -867,7 +867,7 @@ class QueryInvocation(pg.Object, pg.views.HtmlTreeView.Extension):
   ] = None
 
   error: Annotated[
-      pg.utils.ErrorInfo | None,
+      pg.ErrorInfo | None,
       'Error info if the query failed.'
   ] = None
 
@@ -946,7 +946,7 @@ class QueryInvocation(pg.Object, pg.views.HtmlTreeView.Extension):
   def mark_completed(
       self,
       lm_response: lf.Message | None,
-      error: pg.utils.ErrorInfo | None = None,
+      error: pg.ErrorInfo | None = None,
       usage_summary: lf.UsageSummary | None = None) -> None:
     """Marks the query as completed."""
     assert self.end_time is None, 'Query is already completed.'
@@ -965,7 +965,7 @@ class QueryInvocation(pg.Object, pg.views.HtmlTreeView.Extension):
           )
         except mapping.MappingError as e:
           output = None
-          error = pg.utils.ErrorInfo.from_exception(e)
+          error = pg.ErrorInfo.from_exception(e)
         self._output = output
       else:
         assert lm_response is not None

langfun/core/structured/querying_test.py

@@ -1333,7 +1333,7 @@ class QueryInvocationTest(unittest.TestCase):
 
     self.assertEqual(queries[0].id, '123')
     self.assertTrue(queries[0].has_error)
-    self.assertIsInstance(queries[0].error, pg.utils.ErrorInfo)
+    self.assertIsInstance(queries[0].error, pg.ErrorInfo)
     self.assertIn('MappingError', queries[0].error.tag)
     self.assertIsNone(queries[0].output)
 

{langfun-0.1.2.dev202505150805.dist-info → langfun-0.1.2.dev202505170804.dist-info}/METADATA

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

{langfun-0.1.2.dev202505150805.dist-info → langfun-0.1.2.dev202505170804.dist-info}/RECORD

@@ -10,7 +10,7 @@ langfun/core/langfunc.py,sha256=G50YgoVZ0y1GFw2ev41MlOqr6qa8YakbvNC0h_E0PiA,1114
 langfun/core/langfunc_test.py,sha256=CDn-gJCa5EnjN7cotAVCfSCbuzddq2o-HzEt7kV8HbY,8882
 langfun/core/language_model.py,sha256=GCvbu749TviZyLQSNunO0rKeDAb7E_6G4rzqQJerN_E,47913
 langfun/core/language_model_test.py,sha256=iA5uo7rIj2jAtCYzMzhyNg1fWqE2Onn60bOO58q72C0,36454
-langfun/core/logging.py,sha256=jwPamwFYozrs3Tsds8xXG23PhfqODTZ2Au7poTH3yxE,9121
+langfun/core/logging.py,sha256=7IGAhp7mGokZxxqtL-XZvFLKaZ5k3F5_Xp2NUtR4GwE,9136
 langfun/core/logging_test.py,sha256=vbVGOQxwMmVSiFfbt2897gUt-8nqDpV64jCAeUG_q5U,6924
 langfun/core/memory.py,sha256=vyXVvfvSdLLJAzdIupnbn3k26OgclCx-OJ7gddS5e1Y,2070
 langfun/core/message.py,sha256=qUJZ9NfrlYG3aU_K2ud236gdTnvZ7Qw2T4pv4hI9ivg,32817
@@ -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=WEKCaqtGVYKHARvY0Nqichc_hMKkJUYarrUky-DsfHk,50709
+langfun/core/agentic/action.py,sha256=ngBN__OUjMrw4eg8ljHKQ7VY-EABW8T7gF1nw_ZkQmM,50673
 langfun/core/agentic/action_eval.py,sha256=YTilyUEkJl_8FVMgdfO17PurWWaEJ6oA15CuefJJRLk,4887
 langfun/core/agentic/action_eval_test.py,sha256=7AkOwNbUX-ZgR1R0a7bvUZ5abNTUV7blf_8Mnrwb-II,2811
-langfun/core/agentic/action_test.py,sha256=FO42Ab_riHHJxD6iJensskDCMx1-j7blzPr3fFaIfwc,15885
+langfun/core/agentic/action_test.py,sha256=46piw5J7vGwmJGjmVG13HQRBmp5TLgkV_ymJ2EQVEIc,15867
 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=4PD76Xfv36Xrm8Ji3-GgGDNImtcDqWfMw3z6ircJMlM,7285
@@ -63,16 +63,16 @@ langfun/core/eval/v2/__init__.py,sha256=9lNKJwbvl0lcFblAXYT_OHI8fOubJsTOdSkxEqsP
 langfun/core/eval/v2/checkpointing.py,sha256=t47rBfzGZYgIqWW1N1Ak9yQnNtHd-IRbEO0cZjG2VRo,11755
 langfun/core/eval/v2/checkpointing_test.py,sha256=NggOSJ_6XSa4cNP6nGIu9wLsK59dUwe8SPWDiXtGGDE,9197
 langfun/core/eval/v2/eval_test_helper.py,sha256=sKFi_wPYCNmr96WyTduuXY0KnxjFxcJyEhXey-_nGX8,3962
-langfun/core/eval/v2/evaluation.py,sha256=BijbrYufnlRZLryxezDU32gHjFCCZvLAZMLF6kQ90eA,27050
+langfun/core/eval/v2/evaluation.py,sha256=ihT5dljnUkHM97XS9OwE2wOnYC-oYnHYgG5KN1hmiaU,27037
 langfun/core/eval/v2/evaluation_test.py,sha256=QNp_HEvRTupvNuLEeYTvylykh1Ut2jpMqHQ-gCUZQ10,6919
-langfun/core/eval/v2/example.py,sha256=Jegt-viQSNYzPVkOZE_M19GON2TYGTct4Cp9HnJ7DGo,10861
-langfun/core/eval/v2/example_test.py,sha256=1DNm6EuyZOq827DKvf3oTRVFkMNM_qTnLUpvOjpgz5I,3419
+langfun/core/eval/v2/example.py,sha256=v1dIz89pccIqujt7utrk0EbqMWM9kBn-2fYGRTKe358,10890
+langfun/core/eval/v2/example_test.py,sha256=wsHQD6te7ghROmxe3Xg_NK4TU0xS2MkNfnpo-H0H8xM,3399
 langfun/core/eval/v2/experiment.py,sha256=fb3RHNOSRftV7ZTBfYVV50iEevqdPwRHCt3mgtLzuFw,33408
 langfun/core/eval/v2/experiment_test.py,sha256=UmCobeS6ifPcaGkTJp0WPISolXrVFbeFCBiyJeA0Lt4,13666
 langfun/core/eval/v2/metric_values.py,sha256=_B905bC-jxrYPLSEcP2M8MaHZOVMz_bVrUw8YC4arCE,4660
 langfun/core/eval/v2/metric_values_test.py,sha256=ab2oF_HsIwrSy459108ggyjgefHSPn8UVILR4dRwx14,2634
 langfun/core/eval/v2/metrics.py,sha256=bl8i6u-ZHRBz4hAc3LzsZ2Dc7ZRQcuTYeUhhH-GxfF0,10628
-langfun/core/eval/v2/metrics_test.py,sha256=p4FzLJsE8XAzAQuyP9hfEf9YeKWZ__PO_ue8a9P0-cc,6082
+langfun/core/eval/v2/metrics_test.py,sha256=LibZXvWEJDVRY-Mza_bQT-SbmbXCHUnFhL7Ztlzhzw4,6066
 langfun/core/eval/v2/progress.py,sha256=azZgssQgNdv3IgjKEaQBuGI5ucFDNbdi02P4z_nQ8GE,10292
 langfun/core/eval/v2/progress_test.py,sha256=YU7VHzmy5knPZwj9vpBN3rQQH2tukj9eKHkuBCI62h8,2540
 langfun/core/eval/v2/progress_tracking.py,sha256=zNhNPGlnJnHELEfFpbTMCSXFn8d1IJ57OOYkfFaBFfM,6097
@@ -137,8 +137,8 @@ langfun/core/structured/mapping.py,sha256=iraHpcEeF_iuEX2eoTsLGwTHHvxqp2gNDjoMf9
 langfun/core/structured/mapping_test.py,sha256=OntYvfDitAf0tAnzQty3YS90vyEn6FY1Mi93r_ViEk8,9594
 langfun/core/structured/parsing.py,sha256=MGvI7ypXlwfzr5XB8_TFU9Ei0_5reYqkWkv64eAy0EA,12015
 langfun/core/structured/parsing_test.py,sha256=V8Cj1tJK4Lxv_b0YQj6-2hzXZgnYNBa2JR7rOLRBKoQ,22346
-langfun/core/structured/querying.py,sha256=7QvM4TS57O5kDCty6H8leWXfZxcN5ugWV-4bmdDABC0,38006
-langfun/core/structured/querying_test.py,sha256=AXIVamBeKvby10zogjZl_wqtBopbpud-NNEZlqfLyqY,49766
+langfun/core/structured/querying.py,sha256=vE_NOLNlIe4A0DueQfyiBEUh3AsSD8Hhx2dSDHNYpYk,37976
+langfun/core/structured/querying_test.py,sha256=pYWXqAnzp5eOCjU4yEPtE73iLNqxHISb3y3FaSbI7vs,49760
 langfun/core/structured/schema.py,sha256=r_ewdRMsALVOdnvGSeYBcz2-VJ_3_nMxY4GtzUHCYUU,29111
 langfun/core/structured/schema_generation.py,sha256=pEWeTd8tQWYnEHukas6GVl4uGerLsQ2aNybtnm4Qgxc,5352
 langfun/core/structured/schema_generation_test.py,sha256=RM9s71kMNg2jTePwInkiW9fK1ACN37eyPeF8OII-0zw,2950
@@ -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.dev202505150805.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202505150805.dist-info/METADATA,sha256=O9L1gbEi6xg41c0yHT45Pla-2HS9u_5f4f04flDUcIw,8178
-langfun-0.1.2.dev202505150805.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-langfun-0.1.2.dev202505150805.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202505150805.dist-info/RECORD,,
+langfun-0.1.2.dev202505170804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202505170804.dist-info/METADATA,sha256=U2rauYCb-_LY_r_xxtlTa61nIxkMWL4icBLDu5SrW8Y,8178
+langfun-0.1.2.dev202505170804.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+langfun-0.1.2.dev202505170804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202505170804.dist-info/RECORD,,