langfun 0.1.2.dev202509040804__tar.gz → 0.1.2.dev202509060804__tar.gz

{langfun-0.1.2.dev202509040804 → langfun-0.1.2.dev202509060804}/PKG-INFO

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

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

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