langfun 0.1.2.dev202503280804__py3-none-any.whl → 0.1.2.dev202503300803__py3-none-any.whl

langfun/core/structured/querying.py

@@ -19,7 +19,6 @@ import time
 from typing import Annotated, Any, Callable, Iterator, Type, Union
 
 import langfun.core as lf
-from langfun.core.llms import fake
 from langfun.core.structured import mapping
 from langfun.core.structured import schema as schema_lib
 import pyglove as pg
@@ -477,6 +476,10 @@ def query_output(
     **kwargs,
 ) -> Any:
   """Returns the final output of `lf.query` from a provided LLM response."""
+  # Delay import to avoid circular dependency in Colab.
+  # llms > data/conversion > structured > querying
+  from langfun.core.llms import fake  # pylint: disable=g-import-not-at-top
+
   kwargs.pop('prompt', None)
   kwargs.pop('lm', None)
   return query(
@@ -606,6 +609,18 @@ class QueryInvocation(pg.Object, pg.views.HtmlTreeView.Extension):
     """Returns query elapse in seconds."""
     return self.end_time - self.start_time
 
+  def as_mapping_example(
+      self,
+      metadata: dict[str, Any] | None = None
+  ) -> mapping.MappingExample:
+    """Returns a `MappingExample` object for this query invocation."""
+    return mapping.MappingExample(
+        input=self.input,
+        schema=self.schema,
+        output=self.output,
+        metadata=metadata or {},
+    )
+
   def _on_bound(self):
     super()._on_bound()
     self.__dict__.pop('lm_request', None)

langfun/core/structured/querying_test.py

@@ -1072,6 +1072,29 @@ class QueryInvocationTest(unittest.TestCase):
     self.assertEqual(queries[0].lm_response.score, 1.0)
     self.assertFalse(queries[0].lm_response.is_cached)
 
+  def test_as_mapping_example(self):
+    lm = fake.StaticSequence([
+        'Activity(description="hi")',
+    ])
+    with querying.track_queries() as queries:
+      querying.query(lf.Template('foo {{x}}', x=1), Activity, lm=lm)
+    mapping_example = queries[0].as_mapping_example()
+    self.assertTrue(pg.eq(mapping_example.input, lf.Template('foo {{x}}', x=1)))
+    self.assertEqual(mapping_example.schema.spec.cls, Activity)
+    self.assertEqual(mapping_example.output, Activity(description='hi'))
+
+    lm = fake.StaticSequence(['Activity(description="hi"'])
+    with querying.track_queries() as queries:
+      querying.query(
+          lf.Template('foo {{x}}', x=1), Activity, default=None, lm=lm
+      )
+
+    self.assertTrue(queries[0].has_error)
+    mapping_example = queries[0].as_mapping_example()
+    self.assertTrue(pg.eq(mapping_example.input, lf.Template('foo {{x}}', x=1)))
+    self.assertEqual(mapping_example.schema.spec.cls, Activity)
+    self.assertIsInstance(mapping_example.output, mapping.MappingError)
+
 
 class TrackQueriesTest(unittest.TestCase):
 

{langfun-0.1.2.dev202503280804.dist-info → langfun-0.1.2.dev202503300803.dist-info}/METADATA

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

{langfun-0.1.2.dev202503280804.dist-info → langfun-0.1.2.dev202503300803.dist-info}/RECORD

@@ -137,8 +137,8 @@ langfun/core/structured/mapping.py,sha256=of-EeBq0RgmkiUaSk2rVEDVCzgn_wXU8tRke7N
 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=kNPrhpdPY3iWhUld0TFYU-Zgn44wC0d6YuQ9XdVbQ8o,22346
-langfun/core/structured/querying.py,sha256=Uvg64HOCzi5J-Uk3TNpkCgtKe6I02mThwjGbnRY6M7U,23420
-langfun/core/structured/querying_test.py,sha256=H6LUgxaPOHqf37DCQ_PKpXUqgMc4aN211S34xpE-KcQ,32491
+langfun/core/structured/querying.py,sha256=6dE4bRHpyE8H-YNDVFokDmTLgS1G1YdrE9n2TEuyw_I,23911
+langfun/core/structured/querying_test.py,sha256=OZKIh9-P0W5FFiDeDUrQikjYEwcIQOzLSm3u81XZceQ,33514
 langfun/core/structured/schema.py,sha256=_iqhHEGDQsHk0AsybWnK44sOspTWkKJjci781PWD7x0,27988
 langfun/core/structured/schema_generation.py,sha256=3AcuKvv3VOtKY5zMVqODrxfOuDxzoZtGeBxHlOWDOWw,5308
 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.dev202503280804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202503280804.dist-info/METADATA,sha256=FEdEIi7kFzDHZBLPrEcAuoEtUf6Htt7Y4GOyL9QL0n0,7692
-langfun-0.1.2.dev202503280804.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-langfun-0.1.2.dev202503280804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202503280804.dist-info/RECORD,,
+langfun-0.1.2.dev202503300803.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202503300803.dist-info/METADATA,sha256=cK9l4c5VJZyhL315GgLbAVSCTtMmE2mJKVxyhfXNoVI,7692
+langfun-0.1.2.dev202503300803.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+langfun-0.1.2.dev202503300803.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202503300803.dist-info/RECORD,,