langfun 0.1.2.dev202502090804__py3-none-any.whl → 0.1.2.dev202502110804__py3-none-any.whl

langfun/core/llms/gemini.py

@@ -94,12 +94,12 @@ SUPPORTED_MODELS_AND_SETTINGS = {
         tpm_free=40_000_000,
         rpm_paid=0,
         tpm_paid=0,
-        cost_per_1m_input_tokens_up_to_128k=0,
-        cost_per_1m_output_tokens_up_to_128k=0,
-        cost_per_1m_cached_tokens_up_to_128k=0,
-        cost_per_1m_input_tokens_longer_than_128k=0,
-        cost_per_1m_output_tokens_longer_than_128k=0,
-        cost_per_1m_cached_tokens_longer_than_128k=0,
+        cost_per_1m_input_tokens_up_to_128k=0.1,
+        cost_per_1m_output_tokens_up_to_128k=0.4,
+        cost_per_1m_cached_tokens_up_to_128k=0.025,
+        cost_per_1m_input_tokens_longer_than_128k=0.1,
+        cost_per_1m_output_tokens_longer_than_128k=0.4,
+        cost_per_1m_cached_tokens_longer_than_128k=0.025,
     ),
     'gemini-2.0-flash': pg.Dict(
         latest_update='2025-02-05',

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.dev202502090804.dist-info → langfun-0.1.2.dev202502110804.dist-info}/METADATA

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

{langfun-0.1.2.dev202502090804.dist-info → langfun-0.1.2.dev202502110804.dist-info}/RECORD

@@ -82,7 +82,7 @@ langfun/core/llms/deepseek.py,sha256=9EbuNrngd5BwpsvsEfkW2XQdq3K23S3-jlI8RbMNZF4
 langfun/core/llms/deepseek_test.py,sha256=dS72i52bwMpCN4dJDvpJI59AnNChpwxS5eYYFrhGh90,1843
 langfun/core/llms/fake.py,sha256=gCHBYBLvBCsC78HI1hpoqXCS-p1FMTgY1P1qh_sGBPk,3070
 langfun/core/llms/fake_test.py,sha256=2h13qkwEz_JR0mtUDPxdAhQo7MueXaFSwsD2DIRDW9g,7653
-langfun/core/llms/gemini.py,sha256=jSwj9TQMDHkByr09hftq3CQozifjqwPwHfn-A64MIBg,19736
+langfun/core/llms/gemini.py,sha256=7qxdj9tDsG68dstkXAcvnqAMYv0tYkLW8bVLjxLNYqk,19752
 langfun/core/llms/gemini_test.py,sha256=2ERhYWCJwnfDTQbCaZHFuB1TdWJFrOBS7yyCBInIdQk,6129
 langfun/core/llms/google_genai.py,sha256=WbVZAIDhh7mKy15LZMXOEniQ8Obd-hjiy2KviUCfmXQ,4261
 langfun/core/llms/google_genai_test.py,sha256=JZf_cbQ4GGGpwiQCLjFJn7V4jxBBqgZhIx91AzbGKVo,1250
@@ -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.dev202502090804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202502090804.dist-info/METADATA,sha256=ybjKkW3vBDDWkzVoSa7LrSPdqM9IRhCKJkFZXPwrg48,8172
-langfun-0.1.2.dev202502090804.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-langfun-0.1.2.dev202502090804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202502090804.dist-info/RECORD,,
+langfun-0.1.2.dev202502110804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202502110804.dist-info/METADATA,sha256=vNl3CaNv4v8tHGItFTnZlRgk0hM9cmZBxmO0A8RRdIg,8172
+langfun-0.1.2.dev202502110804.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+langfun-0.1.2.dev202502110804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202502110804.dist-info/RECORD,,