langfun 0.1.2.dev202503100804__py3-none-any.whl → 0.1.2.dev202503110804__py3-none-any.whl

langfun/core/structured/querying.py

@@ -391,6 +391,13 @@ def query(
   if pg.MISSING_VALUE != prompt and not skip_lm:
     trackers = lf.context_value('__query_trackers__', [])
     if trackers:
+      # To minimize payload for serialization, we remove the result and usage
+      # fields from the metadata. They will be computed on the fly when the
+      # invocation is rendered.
+      metadata = dict(output_message.metadata)
+      metadata.pop('result', None)
+      metadata.pop('usage', None)
+
       invocation = QueryInvocation(
           input=pg.Ref(query_input),
           schema=(
@@ -399,7 +406,7 @@ def query(
           ),
           lm=pg.Ref(lm),
           examples=pg.Ref(examples) if examples else [],
-          lm_response=lf.AIMessage(output_message.text),
+          lm_response=lf.AIMessage(output_message.text, metadata=metadata),
           usage_summary=usage_summary,
           start_time=start_time,
           end_time=end_time,

langfun/core/structured/querying_test.py

@@ -1069,6 +1069,8 @@ class QueryInvocationTest(unittest.TestCase):
       querying.query('foo', Activity, lm=lm)
 
     self.assertIn('schema', queries[0].to_html_str())
+    self.assertEqual(queries[0].lm_response.score, 1.0)
+    self.assertFalse(queries[0].lm_response.is_cached)
 
 
 class TrackQueriesTest(unittest.TestCase):

{langfun-0.1.2.dev202503100804.dist-info → langfun-0.1.2.dev202503110804.dist-info}/METADATA

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

{langfun-0.1.2.dev202503100804.dist-info → langfun-0.1.2.dev202503110804.dist-info}/RECORD

@@ -131,8 +131,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=RgYsiiOu-a-mE87iDVJBrbkrfnCPXvoc6auIA8OrupQ,23098
-langfun/core/structured/querying_test.py,sha256=1l0032afpSC_ewBNOp17bKK11fHlxlE9A2eJumsz_CE,32380
+langfun/core/structured/querying.py,sha256=Uvg64HOCzi5J-Uk3TNpkCgtKe6I02mThwjGbnRY6M7U,23420
+langfun/core/structured/querying_test.py,sha256=H6LUgxaPOHqf37DCQ_PKpXUqgMc4aN211S34xpE-KcQ,32491
 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
@@ -150,8 +150,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.dev202503100804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202503100804.dist-info/METADATA,sha256=3_ZHDh4WGF6EsjobrgPNTRE6rUORuyoJ1YgOsZmRNsg,8172
-langfun-0.1.2.dev202503100804.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-langfun-0.1.2.dev202503100804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202503100804.dist-info/RECORD,,
+langfun-0.1.2.dev202503110804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202503110804.dist-info/METADATA,sha256=RAbqoIhzrV4vsEMV0kgFmA0WJPCWdnAmE--8bNuD81o,8172
+langfun-0.1.2.dev202503110804.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+langfun-0.1.2.dev202503110804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202503110804.dist-info/RECORD,,