langfun 0.1.2.dev202503270804__tar.gz → 0.1.2.dev202503290803__tar.gz

{langfun-0.1.2.dev202503270804 → langfun-0.1.2.dev202503290803}/PKG-INFO

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

{langfun-0.1.2.dev202503270804 → langfun-0.1.2.dev202503290803}/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-0.1.2.dev202503270804 → langfun-0.1.2.dev202503290803}/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.dev202503270804 → langfun-0.1.2.dev202503290803}/langfun.egg-info/PKG-INFO

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