langfun 0.1.2.dev202509050804__py3-none-any.whl → 0.1.2.dev202509070803__py3-none-any.whl

langfun/core/llms/__init__.py

@@ -20,6 +20,7 @@
 # LMs for testing.
 from langfun.core.llms.fake import Fake
 from langfun.core.llms.fake import Echo
+from langfun.core.llms.fake import Pseudo
 from langfun.core.llms.fake import StaticMapping
 from langfun.core.llms.fake import StaticResponse
 from langfun.core.llms.fake import StaticSequence

langfun/core/llms/fake.py

@@ -62,6 +62,13 @@ class Echo(Fake):
     return lf.AIMessage(prompt.text)
 
 
+class Pseudo(Fake):
+  """A pseudo language model that should never be called."""
+
+  def _response_from(self, prompt: lf.Message) -> lf.Message:
+    raise ValueError('Pseudo language model should never be called.')
+
+
 @lf.use_init_args(['response'])
 class StaticResponse(Fake):
   """Language model that always gives the same canned response."""

langfun/core/llms/fake_test.py

@@ -20,6 +20,15 @@ import langfun.core as lf
 from langfun.core.llms import fake as fakelm
 
 
+class PseudoTest(unittest.TestCase):
+
+  def test_sample(self):
+    lm = fakelm.Pseudo()
+    self.assertEqual(lm.model_id, 'Pseudo')
+    with self.assertRaises(ValueError):
+      _ = lm.sample(['hi'])
+
+
 class EchoTest(unittest.TestCase):
 
   def test_sample(self):

langfun/core/structured/querying.py

@@ -278,7 +278,7 @@ def query(
     schema: schema_lib.SchemaType |  None = None,
     default: Any = lf.RAISE_IF_HAS_ERROR,
     *,
-    lm: lf.LanguageModel | list[lf.LanguageModel] | None = None,
+    lm: lf.LanguageModel | list[lf.LanguageModel],
     num_samples: int | list[int] = 1,
     system_message: str | lf.Template | None = None,
     examples: list[mapping.MappingExample] | None = None,
@@ -414,7 +414,6 @@ def query(
     lm: The language model(s) to query. Can be:
       - A single `LanguageModel`,
       - A list of `LanguageModel`s for multi-model fan-out.
-      If `None`, the LM from `lf.context` will be used.
     num_samples: Number of samples to generate. If a list is provided, its
       length must match the number of models in `lm`.
     system_message: System instructions to guide the model output. If None,
@@ -766,9 +765,20 @@ def query_prompt(
     **kwargs,
 ) -> lf.Message:
   """Returns the final prompt sent to LLM for `lf.query`."""
+  # 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('returns_message', None)
   kwargs.pop('skip_lm', None)
-  return query(prompt, schema, skip_lm=True, returns_message=True, **kwargs)
+  return query(
+      prompt, schema,
+      # The LLM will never be used, it's just a placeholder.
+      lm=fake.Pseudo(),
+      skip_lm=True,
+      returns_message=True,
+      **kwargs
+  )
 
 
 def query_output(

langfun/core/structured/querying_test.py

@@ -991,15 +991,11 @@ class LfQueryPythonV2Test(unittest.TestCase):
     )
 
   def test_bad_response(self):
-    with lf.context(
-        lm=fake.StaticSequence(['a2']),
-        override_attrs=True,
+    with self.assertRaisesRegex(
+        mapping.MappingError,
+        'name .* is not defined',
     ):
-      with self.assertRaisesRegex(
-          mapping.MappingError,
-          'name .* is not defined',
-      ):
-        querying.query('Compute 1 + 2', int)
+      querying.query('Compute 1 + 2', int, lm=fake.StaticSequence(['a2']))
 
   def test_not_allowed_code(self):
     lm = fake.StaticResponse(
@@ -1026,21 +1022,20 @@ class LfQueryPythonV2Test(unittest.TestCase):
     self.assertEqual(querying.query('what is 1 + 0', int, lm=lm, autofix=3), 1)
 
   def test_response_postprocess(self):
-    with lf.context(
-        lm=fake.StaticResponse('<!-- some comment-->\n3'),
-        override_attrs=True,
-    ):
-      self.assertEqual(
-          querying.query(
-              'Compute 1 + 2', response_postprocess=lambda x: x.split('\n')[1]),
-          '3'
-      )
-      self.assertEqual(
-          querying.query(
-              'Compute 1 + 2', int,
-              response_postprocess=lambda x: x.split('\n')[1]),
-          3
-      )
+    self.assertEqual(
+        querying.query(
+            'Compute 1 + 2',
+            lm=fake.StaticResponse('<!-- some comment-->\n3'),
+            response_postprocess=lambda x: x.split('\n')[1]),
+        '3'
+    )
+    self.assertEqual(
+        querying.query(
+            'Compute 1 + 2', int,
+            lm=fake.StaticResponse('<!-- some comment-->\n3'),
+            response_postprocess=lambda x: x.split('\n')[1]),
+        3
+    )
 
   def test_render(self):
     l = querying.LfQuery.from_protocol('python:2.0')(
@@ -1312,15 +1307,15 @@ class LfQueryJsonV1Test(unittest.TestCase):
 
   def test_bad_transform(self):
     with in_memory.lm_cache() as cache:
-      with lf.context(
-          lm=fake.StaticSequence(['3']),
-          override_attrs=True,
+      with self.assertRaisesRegex(
+          mapping.MappingError,
+          'No JSON dict in the output',
       ):
-        with self.assertRaisesRegex(
-            mapping.MappingError,
-            'No JSON dict in the output',
-        ):
-          querying.query('Compute 1 + 2', int, protocol='json', cache_seed=1)
+        querying.query(
+            'Compute 1 + 2', int, 
+            lm=fake.StaticSequence(['3']),
+            protocol='json', cache_seed=1
+        )
       # Make sure bad mapping does not impact cache.
       self.assertEqual(len(cache), 0)
 

{langfun-0.1.2.dev202509050804.dist-info → langfun-0.1.2.dev202509070803.dist-info}/METADATA

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

{langfun-0.1.2.dev202509050804.dist-info → langfun-0.1.2.dev202509070803.dist-info}/RECORD

@@ -90,7 +90,7 @@ langfun/core/eval/v2/reporting.py,sha256=yUIPCAMnp7InIzpv1DDWrcLO-75iiOUTpscj7sm
 langfun/core/eval/v2/reporting_test.py,sha256=CMK-vwho8cNRJwlbkCqm_v5fykE7Y3V6SaIOCY0CDyA,5671
 langfun/core/eval/v2/runners.py,sha256=bEniZDNu44AQgvqpwLsvBU4V_7WltAe-NPhYgIsLj1E,16848
 langfun/core/eval/v2/runners_test.py,sha256=spjkmqlls_vyERdZMdjv6dhIN9ZfxsDDvIQAWTj2kMk,11954
-langfun/core/llms/__init__.py,sha256=VrVwI9gwPp4KEZdWbg-uQESg41L4VXPXMyWiGDkZNQo,9663
+langfun/core/llms/__init__.py,sha256=u__rpDCoJz88P2vHnPtNwdvr4Em3SmmHgUhx-hHnDcc,9705
 langfun/core/llms/anthropic.py,sha256=YcQ2VG8iOfXtry_tTpAukmiwXa2hK_9LkpkmXk41Nm0,26226
 langfun/core/llms/anthropic_test.py,sha256=qA9vByp_cwwXNlXzcwHpPWFnO9lfFo8NKfDi5nBNqgI,9052
 langfun/core/llms/azure_openai.py,sha256=-KkSLaR54MlsIqz_XIwv0TnsBnvNTAxnjA2Q2O2u5KM,2733
@@ -99,8 +99,8 @@ langfun/core/llms/compositional.py,sha256=W_Fe2BdbkjwTzWW-paCWcEeG9oOR3-IcBG8oc7
 langfun/core/llms/compositional_test.py,sha256=4eTnOer-DncRKGaIJW2ZQQMLnt5r2R0UIx_DYOvGAQo,2027
 langfun/core/llms/deepseek.py,sha256=jvTxdXPr-vH6HNakn_Ootx1heDg8Fen2FUkUW36bpCs,5247
 langfun/core/llms/deepseek_test.py,sha256=DvROWPlDuow5E1lfoSkhyGt_ELA19JoQoDsTnRgDtTg,1847
-langfun/core/llms/fake.py,sha256=xmgCkk9y0I4x0IT32SZ9_OT27aLadXH8PRiYNo5VTd4,3265
-langfun/core/llms/fake_test.py,sha256=2h13qkwEz_JR0mtUDPxdAhQo7MueXaFSwsD2DIRDW9g,7653
+langfun/core/llms/fake.py,sha256=bDk_4u7V2LmYUotyOaicwzi0-lnWOIIBbR3-Bil1P3o,3481
+langfun/core/llms/fake_test.py,sha256=lC-C2TpEsnf2kmZpa3OiH2H944I4hMWTAaHEXzRj1DU,7855
 langfun/core/llms/gemini.py,sha256=r4WVWMWHVtyR7tf4NdJ_eAL8Uq6OyQU_2-yBbzrx5mQ,30065
 langfun/core/llms/gemini_test.py,sha256=y1s0W65SrdepbZxzgIeoTB2MI7sXnfBDf1NsGn57LbM,7617
 langfun/core/llms/google_genai.py,sha256=ogyoOUK4s1OcSFKun0YK5xBRDVyxmvz9WsYNKAwuB0g,5918
@@ -146,8 +146,8 @@ langfun/core/structured/mapping.py,sha256=1YBW8PKpJKXS7DKukfzKNioL84PrKUcB4KOUud
 langfun/core/structured/mapping_test.py,sha256=OntYvfDitAf0tAnzQty3YS90vyEn6FY1Mi93r_ViEk8,9594
 langfun/core/structured/parsing.py,sha256=bLi7o1AdyDWc6TwxmYg70t_oTgmLkJWBafakdF8n2RI,14195
 langfun/core/structured/parsing_test.py,sha256=vRfCSzA9q7C1cElkAnDvbRepULEa_vclqDIv-heypDw,22745
-langfun/core/structured/querying.py,sha256=LV7ZCiVx2ef1Qx_ONwexzSvCSBSq1V41E8RdxHDGP-g,39225
-langfun/core/structured/querying_test.py,sha256=klj9OmLKzJbsTYSil58Lex_PVaIlt3ngWQziibg9i4g,50464
+langfun/core/structured/querying.py,sha256=DKp3HCJRetbfC2hnrb680aQBr2AIkXoVA9pd8r6nuPI,39452
+langfun/core/structured/querying_test.py,sha256=iDaqDJCJpdILj2b6I7fTLmbYzAU4EFjQacTRDxx2b0g,50353
 langfun/core/structured/schema.py,sha256=xtgrr3t5tcYQ2gi_fkTKz2IgDMf84gpiykmBdfnV6Io,29486
 langfun/core/structured/schema_generation.py,sha256=pEWeTd8tQWYnEHukas6GVl4uGerLsQ2aNybtnm4Qgxc,5352
 langfun/core/structured/schema_generation_test.py,sha256=RM9s71kMNg2jTePwInkiW9fK1ACN37eyPeF8OII-0zw,2950
@@ -165,8 +165,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.dev202509050804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202509050804.dist-info/METADATA,sha256=t_Pos__ufFg2RXCDwVMc9js6nh1d0NfPq5bzP2M9PzU,7380
-langfun-0.1.2.dev202509050804.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-langfun-0.1.2.dev202509050804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202509050804.dist-info/RECORD,,
+langfun-0.1.2.dev202509070803.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202509070803.dist-info/METADATA,sha256=tBuxpz6mToS9KtFj1l_HUiJJJMlt0nqtpxlOuBuOdu4,7380
+langfun-0.1.2.dev202509070803.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+langfun-0.1.2.dev202509070803.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202509070803.dist-info/RECORD,,