langfun 0.1.2.dev202507140805__tar.gz → 0.1.2.dev202507160024__tar.gz

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfun
-Version: 0.1.2.dev202507140805
+Version: 0.1.2.dev202507160024
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors
@@ -21,30 +21,28 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: Software Development :: Libraries
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: pyglove>=0.4.5.dev202409110000
 Requires-Dist: jinja2>=3.1.2
+Requires-Dist: puremagic>=1.20
+Requires-Dist: pyglove>=0.4.5.dev202507140812
 Requires-Dist: requests>=2.31.0
 Provides-Extra: all
-Requires-Dist: pyglove>=0.4.5.dev202409110000; extra == "all"
 Requires-Dist: jinja2>=3.1.2; extra == "all"
+Requires-Dist: puremagic>=1.20; extra == "all"
+Requires-Dist: pyglove>=0.4.5.dev202507140812; extra == "all"
 Requires-Dist: requests>=2.31.0; extra == "all"
-Requires-Dist: termcolor==1.1.0; extra == "all"
-Requires-Dist: tqdm>=4.64.1; extra == "all"
 Requires-Dist: google-auth>=2.16.0; extra == "all"
-Requires-Dist: python-magic>=0.4.27; extra == "all"
 Requires-Dist: pillow>=10.0.0; extra == "all"
-Provides-Extra: ui
-Requires-Dist: termcolor==1.1.0; extra == "ui"
-Requires-Dist: tqdm>=4.64.1; extra == "ui"
+Requires-Dist: termcolor==1.1.0; extra == "all"
+Requires-Dist: tqdm>=4.64.1; extra == "all"
 Provides-Extra: vertexai
 Requires-Dist: google-auth>=2.16.0; extra == "vertexai"
 Provides-Extra: mime
-Requires-Dist: python-magic>=0.4.27; extra == "mime"
 Requires-Dist: pillow>=10.0.0; extra == "mime"
-Provides-Extra: mime-auto
-Requires-Dist: python-magic>=0.4.27; extra == "mime-auto"
 Provides-Extra: mime-pil
 Requires-Dist: pillow>=10.0.0; extra == "mime-pil"
+Provides-Extra: ui
+Requires-Dist: termcolor==1.1.0; extra == "ui"
+Requires-Dist: tqdm>=4.64.1; extra == "ui"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -202,37 +200,14 @@ a tag from the list below:
 | all                 | All Langfun features.                    |
 | vertexai            | VertexAI access.                         |
 | mime                | All MIME supports.                       |
-| mime-auto           | Automatic MIME type detection.           |
 | mime-pil            | Image support for PIL.                   |
 | ui                  | UI enhancements                          |
 
 For example, to install a nightly build that includes VertexAI access, full
 modality support, and UI enhancements, use:
-```
-pip install langfun[vertexai,mime,ui] --pre
-```
-
-### Solving import issue with `libmagic`
-
-Langfun utilizes `libmagic` for automatic MIME type detection to support
-multi-modal functionalities. However, `pip install libmagic` may not work
-out-of-the-box on all operation systems, sometimes leading to an
-`'ImportError: failed to find libmagic.'` error after Langfun installation.
-
-If you encounter this error, you will need to follow the recommendations below
-to fix the installation of `libmagic` library.
 
-#### OSX
-
-```
-conda install conda-forge::libmagic
 ```
-
-#### Windows:
-```
-pip install python-magic
-pip uninstall python-magic-bin
-pip install python-magic-bin
+pip install langfun[vertexai,mime,ui] --pre
 ```
 
 *Disclaimer: this is not an officially supported Google product.*

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/README.md

@@ -142,37 +142,14 @@ a tag from the list below:
 | all                 | All Langfun features.                    |
 | vertexai            | VertexAI access.                         |
 | mime                | All MIME supports.                       |
-| mime-auto           | Automatic MIME type detection.           |
 | mime-pil            | Image support for PIL.                   |
 | ui                  | UI enhancements                          |
 
 For example, to install a nightly build that includes VertexAI access, full
 modality support, and UI enhancements, use:
-```
-pip install langfun[vertexai,mime,ui] --pre
-```
-
-### Solving import issue with `libmagic`
-
-Langfun utilizes `libmagic` for automatic MIME type detection to support
-multi-modal functionalities. However, `pip install libmagic` may not work
-out-of-the-box on all operation systems, sometimes leading to an
-`'ImportError: failed to find libmagic.'` error after Langfun installation.
-
-If you encounter this error, you will need to follow the recommendations below
-to fix the installation of `libmagic` library.
 
-#### OSX
-
-```
-conda install conda-forge::libmagic
 ```
-
-#### Windows:
-```
-pip install python-magic
-pip uninstall python-magic-bin
-pip install python-magic-bin
+pip install langfun[vertexai,mime,ui] --pre
 ```
 
 *Disclaimer: this is not an officially supported Google product.*

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/concurrent_test.py

@@ -17,6 +17,7 @@ import collections
 from concurrent import futures
 import contextlib
 import io
+import sys
 import time
 import unittest
 from langfun.core import component
@@ -330,6 +331,8 @@ class ProgressBarTest(unittest.TestCase):
       with self.assertRaisesRegex(ValueError, 'Unsupported status'):
         concurrent.ProgressBar.update(bar_id, 0, status=1)
       concurrent.ProgressBar.uninstall(bar_id)
+      sys.stderr.flush()
+      time.sleep(1)
     self.assertIn('1/4', string_io.getvalue())
     self.assertIn('2/4', string_io.getvalue())
     self.assertIn('hello', string_io.getvalue())

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/eval/v2/checkpointing_test.py

@@ -28,7 +28,7 @@ Example = example_lib.Example
 class SequenceWriterTest(unittest.TestCase):
 
   def test_basic(self):
-    file = os.path.join(tempfile.gettempdir(), 'test.jsonl')
+    file = os.path.join(tempfile.mkdtemp(), 'test.jsonl')
     writer = checkpointing.SequenceWriter(file)
     example = Example(id=1, input=pg.Dict(x=1), output=2)
     writer.add(example)
@@ -36,7 +36,7 @@ class SequenceWriterTest(unittest.TestCase):
     self.assertTrue(pg.io.path_exists(file))
 
   def test_error_handling(self):
-    file = os.path.join(tempfile.gettempdir(), 'test_error_handling.jsonl')
+    file = os.path.join(tempfile.mkdtemp(), 'test_error_handling.jsonl')
     writer = checkpointing.SequenceWriter(file)
     writer.add(Example(id=1, input=pg.Dict(x=1), output=2))
 
@@ -87,7 +87,7 @@ class CheckpointerTest(unittest.TestCase):
 class PerExampleCheckpointerTest(CheckpointerTest):
 
   def test_checkpointing(self):
-    root_dir = os.path.join(tempfile.gettempdir(), 'per_example_checkpointer')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'per_example_checkpointer')
     experiment = eval_test_helper.test_experiment()
     checkpoint_filename = 'checkpoint.jsonl'
     checkpointer = checkpointing.PerExampleCheckpointer(checkpoint_filename)
@@ -119,7 +119,7 @@ class PerExampleCheckpointerTest(CheckpointerTest):
       self.assertEqual(leaf.progress.num_skipped, num_processed[leaf.id])
 
     # Test warm start without reprocess.
-    root_dir = os.path.join(tempfile.gettempdir(), 'per_example_checkpointer2')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'per_example_checkpointer2')
     experiment = eval_test_helper.test_experiment()
     _ = experiment.run(
         root_dir, 'new', runner='sequential', plugins=[checkpointer],
@@ -129,7 +129,7 @@ class PerExampleCheckpointerTest(CheckpointerTest):
       self.assertEqual(leaf.progress.num_skipped, num_processed[leaf.id])
 
     # Test warm start with reprocess.
-    root_dir = os.path.join(tempfile.gettempdir(), 'per_example_checkpointer3')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'per_example_checkpointer3')
     experiment = eval_test_helper.test_experiment()
     _ = experiment.run(
         root_dir, 'new', runner='sequential', plugins=[checkpointer],
@@ -139,7 +139,7 @@ class PerExampleCheckpointerTest(CheckpointerTest):
     for leaf in experiment.leaf_nodes:
       self.assertEqual(leaf.progress.num_skipped, 0)
 
-    root_dir = os.path.join(tempfile.gettempdir(), 'per_example_checkpointer4')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'per_example_checkpointer4')
     experiment = eval_test_helper.test_experiment()
     _ = experiment.run(
         root_dir, 'new', runner='sequential', plugins=[checkpointer],
@@ -151,7 +151,7 @@ class PerExampleCheckpointerTest(CheckpointerTest):
 
   def test_loading_corrupted_checkpoint(self):
     root_dir = os.path.join(
-        tempfile.gettempdir(),
+        tempfile.mkdtemp(),
         'per_example_checkpointer_with_corrupted_checkpoint'
     )
     experiment = eval_test_helper.TestEvaluation()
@@ -178,7 +178,7 @@ class PerExampleCheckpointerTest(CheckpointerTest):
         num_processed[example.id] = i + 1
 
     root_dir = os.path.join(
-        tempfile.gettempdir(),
+        tempfile.mkdtemp(),
         'per_example_checkpointer_with_corrupted_checkpoint_warm_start'
     )
     experiment = eval_test_helper.TestEvaluation()
@@ -192,7 +192,7 @@ class PerExampleCheckpointerTest(CheckpointerTest):
 
   def test_checkpointing_error(self):
     root_dir = os.path.join(
-        tempfile.gettempdir(),
+        tempfile.mkdtemp(),
         'per_example_checkpointer_with_checkpointing_error'
     )
     experiment = (eval_test_helper
@@ -207,7 +207,7 @@ class PerExampleCheckpointerTest(CheckpointerTest):
 class BulkCheckpointerTest(CheckpointerTest):
 
   def test_checkpointing(self):
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_bulk_checkpointer')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_bulk_checkpointer')
     experiment = eval_test_helper.test_experiment()
     checkpoint_filename = 'checkpoint.jsonl'
     checkpointer = checkpointing.BulkCheckpointer(checkpoint_filename)
@@ -238,7 +238,7 @@ class BulkCheckpointerTest(CheckpointerTest):
 
   def test_checkpointing_error(self):
     root_dir = os.path.join(
-        tempfile.gettempdir(),
+        tempfile.mkdtemp(),
         'bulk_checkpointer_with_checkpointing_error'
     )
     experiment = (eval_test_helper

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/eval/v2/evaluation_test.py

@@ -116,7 +116,7 @@ class EvaluationTest(unittest.TestCase):
     self.assertEqual(example.metric_metadata, dict(error='ValueError'))
 
   def test_evaluate_withstate(self):
-    eval_dir = os.path.join(tempfile.gettempdir(), 'test_eval')
+    eval_dir = os.path.join(tempfile.mkdtemp(), 'test_eval')
     pg.io.mkdirs(eval_dir, exist_ok=True)
     state_file = os.path.join(eval_dir, 'state.jsonl')
     with pg.io.open_sequence(state_file, 'w') as f:

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/eval/v2/experiment_test.py

@@ -145,7 +145,7 @@ class RunIdTest(unittest.TestCase):
     )
 
   def test_get_latest(self):
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_eval')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_eval')
     pg.io.mkdirs(os.path.join(root_dir, 'run_20241102_0'))
     pg.io.mkdirs(os.path.join(root_dir, 'run_20241101_0'))
     self.assertEqual(
@@ -153,15 +153,15 @@ class RunIdTest(unittest.TestCase):
         RunId.from_id('20241102_0')
     )
     self.assertIsNone(RunId.get_latest('/notexist'))
-    self.assertIsNone(RunId.get_latest(tempfile.gettempdir()))
+    self.assertIsNone(RunId.get_latest(tempfile.mkdtemp()))
 
   def test_new(self):
     rid = RunId(date=datetime.date.today(), number=1)
     self.assertEqual(
-        RunId.new(root_dir=os.path.join(tempfile.gettempdir(), 'test_new')),
+        RunId.new(root_dir=os.path.join(tempfile.mkdtemp(), 'test_new')),
         rid
     )
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_eval2')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_eval2')
     pg.io.mkdirs(rid.dirname(root_dir))
     self.assertEqual(RunId.new(root_dir), rid.next())
 
@@ -185,13 +185,13 @@ class RunIdTest(unittest.TestCase):
     with self.assertRaisesRegex(
         ValueError, '.* no previous runs'
     ):
-      RunId.from_id('latest', root_dir=tempfile.gettempdir())
+      RunId.from_id('latest', root_dir=tempfile.mkdtemp())
 
     self.assertEqual(
         RunId.from_id('20241102_1'),
         RunId(date=datetime.date(2024, 11, 2), number=1)
     )
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_eval3')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_eval3')
     rid = RunId.from_id('20241102_1')
     pg.io.mkdirs(rid.dirname(root_dir))
     self.assertEqual(
@@ -413,7 +413,7 @@ class RunnerTest(unittest.TestCase):
         ),
         TestRunner
     )
-    root_dir = os.path.join(tempfile.gettempdir(), 'my_eval')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'my_eval')
 
     # Test standard run.
     MyEvaluation(replica_id=0).run(

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/eval/v2/progress_tracking_test.py

@@ -34,7 +34,7 @@ class HtmlProgressTrackerTest(unittest.TestCase):
     lf_console._notebook = pg.Dict(
         display=display
     )
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_html_progress_tracker')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_html_progress_tracker')
     experiment = eval_test_helper.test_experiment()
     _ = experiment.run(root_dir, 'new', plugins=[])
     self.assertIsInstance(result['view'], pg.Html)
@@ -44,7 +44,7 @@ class HtmlProgressTrackerTest(unittest.TestCase):
 class TqdmProgressTrackerTest(unittest.TestCase):
 
   def test_basic(self):
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_tqdm_progress_tracker')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_tqdm_progress_tracker')
     experiment = eval_test_helper.test_experiment()
     string_io = io.StringIO()
     with contextlib.redirect_stderr(string_io):
@@ -53,7 +53,7 @@ class TqdmProgressTrackerTest(unittest.TestCase):
 
   def test_with_example_ids(self):
     root_dir = os.path.join(
-        tempfile.gettempdir(), 'test_tqdm_progress_tracker_with_example_ids'
+        tempfile.mkdtemp(), 'test_tqdm_progress_tracker_with_example_ids'
     )
     experiment = eval_test_helper.test_experiment()
     string_io = io.StringIO()

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/eval/v2/reporting_test.py

@@ -25,7 +25,7 @@ import pyglove as pg
 class ReportingTest(unittest.TestCase):
 
   def test_reporting(self):
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_reporting')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_reporting')
     experiment = eval_test_helper.test_experiment()
     checkpointer = checkpointing.BulkCheckpointer('checkpoint.jsonl')
     reporter = reporting.HtmlReporter()
@@ -49,7 +49,7 @@ class ReportingTest(unittest.TestCase):
       self.assertTrue(found_generation_log)
 
     # Test warm start.
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_reporting2')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_reporting2')
     experiment = eval_test_helper.test_experiment()
     run = experiment.run(
         root_dir, 'new', plugins=[checkpointer, reporter],
@@ -75,7 +75,7 @@ class ReportingTest(unittest.TestCase):
 
   def test_index_html_generation_error(self):
     root_dir = os.path.join(
-        tempfile.gettempdir(),
+        tempfile.mkdtemp(),
         'test_reporting_with_index_html_generation_error'
     )
     experiment = (eval_test_helper
@@ -98,7 +98,7 @@ class ReportingTest(unittest.TestCase):
 
   def test_example_html_generation_error(self):
     root_dir = os.path.join(
-        tempfile.gettempdir(),
+        tempfile.mkdtemp(),
         'test_reporting_with_example_html_generation_error'
     )
     experiment = (eval_test_helper
@@ -126,7 +126,7 @@ class ReportingTest(unittest.TestCase):
 
     # Test warm start.
     root_dir = os.path.join(
-        tempfile.gettempdir(),
+        tempfile.mkdtemp(),
         'test_reporting_with_example_html_generation_error2'
     )
     experiment = (eval_test_helper

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/eval/v2/runners_test.py

@@ -103,7 +103,7 @@ class RunnerTest(unittest.TestCase):
   def test_basic(self):
     plugin = TestPlugin()
     exp = eval_test_helper.test_experiment()
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_sequential_runner')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_sequential_runner')
     run = exp.run(root_dir, runner='sequential', plugins=[plugin])
 
     self.assertIsNotNone(plugin.start_time)
@@ -143,7 +143,7 @@ class RunnerTest(unittest.TestCase):
         self.assertEqual(node.progress.num_processed, node.progress.num_total)
 
   def test_raise_if_has_error(self):
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_raise_if_has_error')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_raise_if_has_error')
     exp = eval_test_helper.TestEvaluation()
     with self.assertRaisesRegex(ValueError, 'x should not be 5'):
       exp.run(
@@ -154,7 +154,7 @@ class RunnerTest(unittest.TestCase):
       exp.run(root_dir, runner='parallel', plugins=[], raise_if_has_error=True)
 
   def test_example_ids(self):
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_example_ids')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_example_ids')
     exp = eval_test_helper.test_experiment()
     plugin = TestPlugin()
     _ = exp.run(
@@ -164,7 +164,7 @@ class RunnerTest(unittest.TestCase):
     self.assertEqual(plugin.completed_example_ids, [5, 7, 9] * 6)
 
   def test_shuffle_inputs(self):
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_shuffle_inputs')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_shuffle_inputs')
     exp = eval_test_helper.test_experiment()
     plugin = TestPlugin()
     run = exp.run(
@@ -175,7 +175,7 @@ class RunnerTest(unittest.TestCase):
   def test_filter(self):
     plugin = TestPlugin()
     exp = eval_test_helper.test_experiment()
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_filter')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_filter')
 
     _ = exp.run(
         root_dir, runner='sequential', plugins=[plugin],
@@ -207,7 +207,7 @@ class RunnerTest(unittest.TestCase):
         inputs=test_inputs(num_examples=pg.oneof([2, 4]))
     )
     # Global cache.
-    root_dir = os.path.join(tempfile.gettempdir(), 'global_cache')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'global_cache')
     run = exp.run(
         root_dir, 'new', runner='sequential', use_cache='global', plugins=[]
     )
@@ -216,7 +216,7 @@ class RunnerTest(unittest.TestCase):
     self.assertEqual(exp.usage_summary.uncached.total.num_requests, 2)
 
     # Per-dataset cache.
-    root_dir = os.path.join(tempfile.gettempdir(), 'per_dataset')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'per_dataset')
     run = exp.run(
         root_dir, 'new', runner='sequential',
         use_cache='per_dataset', plugins=[]
@@ -229,7 +229,7 @@ class RunnerTest(unittest.TestCase):
     self.assertEqual(exp.usage_summary.uncached.total.num_requests, 3)
 
     # No cache.
-    root_dir = os.path.join(tempfile.gettempdir(), 'no')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'no')
     run = exp.run(root_dir, runner='sequential', use_cache='no', plugins=[])
     self.assertFalse(pg.io.path_exists(run.output_path_for(exp, 'cache.json')))
     for leaf in exp.leaf_nodes:
@@ -245,7 +245,7 @@ class ParallelRunnerTest(RunnerTest):
   def test_parallel_runner(self):
     plugin = TestPlugin()
     exp = eval_test_helper.test_experiment()
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_parallel_runner')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_parallel_runner')
     run = exp.run(root_dir, runner='parallel', plugins=[plugin])
 
     self.assertIsNotNone(plugin.start_time)
@@ -286,7 +286,7 @@ class ParallelRunnerTest(RunnerTest):
     plugin = TestPlugin()
     exp = eval_test_helper.test_experiment()
     root_dir = os.path.join(
-        tempfile.gettempdir(), 'test_concurrent_startup_delay'
+        tempfile.mkdtemp(), 'test_concurrent_startup_delay'
     )
     _ = exp.run(
         root_dir,
@@ -301,7 +301,7 @@ class DebugRunnerTest(RunnerTest):
   def test_debug_runner(self):
     plugin = TestPlugin()
     exp = eval_test_helper.test_experiment()
-    root_dir = os.path.join(tempfile.gettempdir(), 'test_debug_runner')
+    root_dir = os.path.join(tempfile.mkdtemp(), 'test_debug_runner')
     run = exp.run(root_dir, runner='debug', plugins=[plugin])
 
     self.assertIsNotNone(plugin.start_time)

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/language_model.py

@@ -667,10 +667,7 @@ class LMDebugMode(enum.IntFlag):
   PROMPT = enum.auto()
   RESPONSE = enum.auto()
 
-  @classmethod
-  @property
-  def ALL(cls) -> 'LMDebugMode':  # pylint: disable=invalid-name
-    return LMDebugMode.INFO | LMDebugMode.PROMPT | LMDebugMode.RESPONSE
+  ALL = INFO | PROMPT | RESPONSE
 
 
 class LanguageModel(component.Component):
@@ -1101,7 +1098,10 @@ class LanguageModel(component.Component):
     return [job.result for job in executed_jobs]
 
   def __call__(
-      self, prompt: message_lib.Message, *, cache_seed: int = 0, **kwargs
+      self,
+      prompt: str | message_lib.Message,
+      *,
+      cache_seed: int = 0, **kwargs
   ) -> message_lib.Message:
     """Returns the first candidate."""
     prompt = message_lib.UserMessage.from_value(prompt)

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/language_model_test.py

@@ -683,16 +683,14 @@ class LanguageModelTest(unittest.TestCase):
 
       debug_info = string_io.getvalue()
       expected_included = [
-          debug_prints[f]
-          for f in lm_lib.LMDebugMode
-          if f != lm_lib.LMDebugMode.NONE and f in debug_mode
+          debug_prints[f] for f in (info_flag, prompt_flag, response_flag)
+          if f in debug_mode
       ]
       expected_excluded = [
           debug_prints[f]
-          for f in lm_lib.LMDebugMode
-          if f != lm_lib.LMDebugMode.NONE and f not in debug_mode
+          for f in (info_flag, prompt_flag, response_flag)
+          if f not in debug_mode
       ]
-
       for expected_include in expected_included:
         self.assertIn('[0] ' + expected_include, debug_info)
       for expected_exclude in expected_excluded:
@@ -750,13 +748,13 @@ class LanguageModelTest(unittest.TestCase):
       debug_info = string_io.getvalue()
       expected_included = [
           debug_prints[f]
-          for f in lm_lib.LMDebugMode
-          if f != lm_lib.LMDebugMode.NONE and f in debug_mode
+          for f in (info_flag, prompt_flag, response_flag)
+          if f in debug_mode
       ]
       expected_excluded = [
           debug_prints[f]
-          for f in lm_lib.LMDebugMode
-          if f != lm_lib.LMDebugMode.NONE and f not in debug_mode
+          for f in (info_flag, prompt_flag, response_flag)
+          if f not in debug_mode
       ]
 
       for expected_include in expected_included:
@@ -813,13 +811,13 @@ class LanguageModelTest(unittest.TestCase):
       debug_info = string_io.getvalue()
       expected_included = [
           debug_prints[f]
-          for f in lm_lib.LMDebugMode
-          if f != lm_lib.LMDebugMode.NONE and f in debug_mode
+          for f in (info_flag, prompt_flag, response_flag)
+          if f in debug_mode
       ]
       expected_excluded = [
           debug_prints[f]
-          for f in lm_lib.LMDebugMode
-          if f != lm_lib.LMDebugMode.NONE and f not in debug_mode
+          for f in (info_flag, prompt_flag, response_flag)
+          if f not in debug_mode
       ]
 
       for expected_include in expected_included:

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/llms/compositional.py

@@ -78,7 +78,7 @@ class RandomChoice(lf.LanguageModel):
     )
 
   def __call__(
-      self, prompt: lf.Message, *, cache_seed: int = 0, **kwargs
+      self, prompt: str | lf.Message, *, cache_seed: int = 0, **kwargs
   ) -> lf.Message:
     return self._select_lm()(prompt, cache_seed=cache_seed, **kwargs)
 

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/message.py

@@ -225,13 +225,11 @@ class Message(
     return MessageConverter.get(format_or_type, **kwargs).to_value(self)
 
   @classmethod
-  @property
   def convertible_formats(cls) -> list[str]:
     """Returns supported format for message conversion."""
     return MessageConverter.convertible_formats()
 
   @classmethod
-  @property
   def convertible_types(cls) -> list[str]:
     """Returns supported types for message conversion."""
     return MessageConverter.convertible_types()
@@ -938,8 +936,8 @@ class MessageConverter(pg.Object):
     """Converts a Langfun message to other formats."""
 
   @abc.abstractmethod
-  def from_value(self, value: Message) -> Message:
-    """Returns a MessageConverter from a Langfun message."""
+  def from_value(self, value: Any) -> Message:
+    """Returns a Langfun message from other formats."""
 
   @classmethod
   def _safe_read(

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/message_test.py

@@ -521,12 +521,12 @@ class MessageConverterTest(unittest.TestCase):
       def from_value(self, value: tuple[int, ...]) -> message.Message:
         return message.UserMessage(','.join(str(x) for x in value))
 
-    self.assertIn('test_format1', message.Message.convertible_formats)
-    self.assertIn('test_format2', message.Message.convertible_formats)
-    self.assertIn('test_format3', message.Message.convertible_formats)
+    self.assertIn('test_format1', message.Message.convertible_formats())
+    self.assertIn('test_format2', message.Message.convertible_formats())
+    self.assertIn('test_format3', message.Message.convertible_formats())
 
-    self.assertIn(int, message.Message.convertible_types)
-    self.assertIn(tuple, message.Message.convertible_types)
+    self.assertIn(int, message.Message.convertible_types())
+    self.assertIn(tuple, message.Message.convertible_types())
     self.assertEqual(
         message.Message.from_value(1, format='test_format1'),
         message.UserMessage('1')

{langfun-0.1.2.dev202507140805 → langfun-0.1.2.dev202507160024}/langfun/core/modalities/audio_test.py

@@ -34,8 +34,8 @@ class AudioTest(unittest.TestCase):
 
   def test_audio_content(self):
     audio = audio_lib.Audio.from_bytes(content_bytes)
-    self.assertEqual(audio.mime_type, 'audio/x-wav')
-    self.assertEqual(audio.audio_format, 'x-wav')
+    self.assertIn(audio.mime_type, ('audio/wave', 'audio/x-wav'))
+    self.assertIn(audio.audio_format, ('wave', 'x-wav'))
     self.assertEqual(audio.to_bytes(), content_bytes)
 
   def test_bad_audio(self):
@@ -50,8 +50,8 @@ class AudioFileTest(unittest.TestCase):
     audio = audio_lib.Audio.from_uri('http://mock/web/a.wav')
     with mock.patch('requests.get') as mock_requests_get:
       mock_requests_get.side_effect = mock_request
-      self.assertEqual(audio.audio_format, 'x-wav')
-      self.assertEqual(audio.mime_type, 'audio/x-wav')
+      self.assertIn(audio.audio_format, ('wave', 'x-wav'))
+      self.assertIn(audio.mime_type, ('audio/wave', 'audio/x-wav'))
       self.assertEqual(
           audio._raw_html(),
           '<audio controls> <source src="http://mock/web/a.wav"> </audio>',