langfun 0.1.2.dev202502080803__py3-none-any.whl → 0.1.2.dev202502100804__py3-none-any.whl

langfun/core/structured/querying.py

@@ -579,7 +579,7 @@ class QueryInvocation(pg.Object, pg.views.HtmlTreeView.Extension):
 
   @functools.cached_property
   def lm_request(self) -> lf.Message:
-    return query_prompt(self.input, self.schema)
+    return query_prompt(self.input, self.schema, examples=self.examples or None)
 
   @functools.cached_property
   def output(self) -> Any:

langfun/core/structured/querying_test.py

@@ -1081,9 +1081,12 @@ class TrackQueriesTest(unittest.TestCase):
     with querying.track_queries() as queries:
       querying.query('foo', lm=lm)
       with querying.track_queries() as child_queries:
-        querying.query('give me an activity', Activity, lm=lm)
+        querying.query('give me an activity', Activity, lm=lm, examples=[
+            mapping.MappingExample(input='2 + 2', schema=int, output=4)
+        ])
 
     self.assertEqual(len(queries), 2)
+    self.assertEqual(queries[0].lm_response, 'bar')
     self.assertTrue(pg.eq(queries[0].input, lf.Template('foo')))
     self.assertIsNone(queries[0].schema)
     self.assertEqual(queries[0].output, 'bar')
@@ -1091,7 +1094,11 @@ class TrackQueriesTest(unittest.TestCase):
 
     self.assertTrue(pg.eq(queries[1].input, lf.Template('give me an activity')))
     self.assertEqual(queries[1].schema.spec.cls, Activity)
+    self.assertIn('2 + 2', queries[1].lm_request.text)
     self.assertTrue(pg.eq(queries[1].output, Activity(description='hi')))
+    self.assertTrue(pg.eq(queries[1].examples, [
+        mapping.MappingExample(input='2 + 2', schema=int, output=4)
+    ]))
     self.assertIs(queries[1].lm, lm)
     self.assertGreater(queries[0].elapse, 0)
     self.assertGreater(queries[0].usage_summary.total.total_tokens, 0)

{langfun-0.1.2.dev202502080803.dist-info → langfun-0.1.2.dev202502100804.dist-info}/METADATA

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

{langfun-0.1.2.dev202502080803.dist-info → langfun-0.1.2.dev202502100804.dist-info}/RECORD

@@ -129,8 +129,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=nqvsfMS_KLv5EvO0_VAGEHwY4pHy4S0CvJmeV0HBXlM,23066
-langfun/core/structured/querying_test.py,sha256=YlC4s9LVChfhGZzaXGW1UYlcBnAjNOunu4SLl5_p7PQ,32054
+langfun/core/structured/querying.py,sha256=RgYsiiOu-a-mE87iDVJBrbkrfnCPXvoc6auIA8OrupQ,23098
+langfun/core/structured/querying_test.py,sha256=1l0032afpSC_ewBNOp17bKK11fHlxlE9A2eJumsz_CE,32380
 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
@@ -148,8 +148,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.dev202502080803.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202502080803.dist-info/METADATA,sha256=HwxvNwOrL6oliIMpZ9vm2dAU8E7BY5FGU6-8JpOmbkg,8172
-langfun-0.1.2.dev202502080803.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-langfun-0.1.2.dev202502080803.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202502080803.dist-info/RECORD,,
+langfun-0.1.2.dev202502100804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202502100804.dist-info/METADATA,sha256=j1-_QWWv17zKz4eGuqDiQrlbOdLv_X6jo2YZneQ9v60,8172
+langfun-0.1.2.dev202502100804.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+langfun-0.1.2.dev202502100804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202502100804.dist-info/RECORD,,