langfun 0.1.2.dev202507140805__py3-none-any.whl → 0.1.2.dev202507150805__py3-none-any.whl

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/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/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/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/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/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/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/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/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/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/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/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/core/structured/mapping.py

@@ -433,8 +433,18 @@ class Mapping(lf.LangFunc):
     schema = self.mapping_request.schema
     if schema is None:
       return None
+    response_text = lm_output.text
+
+    # For Gemini, we might have tool calls in the metadata, use tool call codes
+    # to construct the response text if it's present.
+    # NOTE(daiyip): This logic is subject to change.
+    if 'tool_calls' in lm_output.metadata:
+      assert lm_output.metadata['tool_calls'], lm_output.metadata
+      response_text = '\n'.join(
+          tc.text for tc in lm_output.metadata['tool_calls']
+      )
     return schema.parse(
-        lm_output.text,
+        response_text,
         protocol=self.protocol,
         additional_context=self.globals(),
         autofix=self.autofix,

langfun/core/structured/querying_test.py

@@ -114,6 +114,21 @@ class QueryTest(unittest.TestCase):
         ),
         'The answer is one.',
     )
+    # Testing tool calls in the response.
+    self.assertEqual(
+        querying.query(
+            'abc',
+            Activity,
+            lm=fake.StaticResponse(
+                lf.AIMessage(
+                    'Here is the answer.',
+                    tool_calls=[lf.AIMessage('Activity(description="hello")')],
+                    ),
+            ),
+        ),
+        Activity(description='hello'),
+    )
+    # Test completing a partial object.
     self.assertEqual(
         querying.query(
             Activity.partial(),

langfun/core/structured/schema.py

@@ -17,6 +17,7 @@ import abc
 import inspect
 import io
 import re
+import sys
 import textwrap
 import typing
 from typing import Any, Literal, Sequence, Type, Union
@@ -451,6 +452,15 @@ def class_definition(
       out.write(f'  """{cls.__doc__}"""\n')
     else:
       out.write('  """')
+
+      # Since Python 3.13, the indentation of docstring lines is removed.
+      # Therefore, we add two spaces to each non-empty line to keep the
+      # indentation consistent with the class definition.
+      if sys.version_info >= (3, 13):
+        for i in range(1, len(doc_lines)):
+          if doc_lines[i]:
+            doc_lines[i] = ' ' * 2 + doc_lines[i]
+
       for line in doc_lines:
         out.write(line)
         out.write('\n')

{langfun-0.1.2.dev202507140805.dist-info → langfun-0.1.2.dev202507150805.dist-info}/METADATA

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

{langfun-0.1.2.dev202507140805.dist-info → langfun-0.1.2.dev202507150805.dist-info}/RECORD

@@ -3,18 +3,18 @@ langfun/core/__init__.py,sha256=pW4prpiyWNkRbtWBGYF1thn7_0F_TgDVfAIZPvGn6HA,4758
 langfun/core/component.py,sha256=g1kQM0bryYYYWVDrSMnHfc74wIBbpfe5_B3s-UIP5GE,3028
 langfun/core/component_test.py,sha256=0CxTgjAud3aj8wBauFhG2FHDqrxCTl4OI4gzQTad-40,9254
 langfun/core/concurrent.py,sha256=zY-pXqlGqss_GI20tM1gXvyW8QepVPUuFNmutcIdhbI,32760
-langfun/core/concurrent_test.py,sha256=roMFze0EKuyPbmG6DZzz8K8VGsZwWzc0F1uJZTFROC4,17572
+langfun/core/concurrent_test.py,sha256=fjVcxD_OSH9fBqBEpDpuIVfcfoKZWDtwmkoM2ZMHqy8,17628
 langfun/core/console.py,sha256=cLQEf84aDxItA9fStJV22xJch0TqFLNf9hLqwJ0RHmU,2652
 langfun/core/console_test.py,sha256=pBOcuNMJdVELywvroptfcRtJMsegMm3wSlHAL2TdxVk,1679
 langfun/core/langfunc.py,sha256=G50YgoVZ0y1GFw2ev41MlOqr6qa8YakbvNC0h_E0PiA,11140
 langfun/core/langfunc_test.py,sha256=CDn-gJCa5EnjN7cotAVCfSCbuzddq2o-HzEt7kV8HbY,8882
-langfun/core/language_model.py,sha256=fJeYDz_TD1feiUvSysXXeo2bV-cq5T34HWeYgsTICP4,49680
-langfun/core/language_model_test.py,sha256=VyiHwrUtJGkoLyzsjhVqawijwtwoRqsYOvQD57n8Iv8,37413
+langfun/core/language_model.py,sha256=5i0Je5526JO2YY6qExi6Yf7VQVgSVeZIKOjt3I8kxqQ,49573
+langfun/core/language_model_test.py,sha256=9EofP3_gTH28SNWiOKzTUYMHH0EYtbi9xGuT1KZT1XU,37330
 langfun/core/logging.py,sha256=7IGAhp7mGokZxxqtL-XZvFLKaZ5k3F5_Xp2NUtR4GwE,9136
 langfun/core/logging_test.py,sha256=vbVGOQxwMmVSiFfbt2897gUt-8nqDpV64jCAeUG_q5U,6924
 langfun/core/memory.py,sha256=vyXVvfvSdLLJAzdIupnbn3k26OgclCx-OJ7gddS5e1Y,2070
-langfun/core/message.py,sha256=qUJZ9NfrlYG3aU_K2ud236gdTnvZ7Qw2T4pv4hI9ivg,32817
-langfun/core/message_test.py,sha256=XE7SaYJUK8TCJBdih4XtnpnB6NhcU-vLxJoaP67WbSU,40793
+langfun/core/message.py,sha256=Nx9SqEIkPMS5I1RyMQFlWUjZCsdlGamv_wTze2-3R4M,32784
+langfun/core/message_test.py,sha256=dAA_ZzI5MGyFfXyejxPrB90SbR066mkIgmRtdZ5ZbL4,40803
 langfun/core/modality.py,sha256=K8pUGuMpfWcOtVcXC_OqVjro1-RhHF6ddQni61DuYzM,4166
 langfun/core/modality_test.py,sha256=0WL_yd3B4K-FviWdSpDnOwj0f9TQI0v9t6X0vWvvJbo,2415
 langfun/core/natural_language.py,sha256=3ynSnaYQnjE60LIPK5fyMgdIjubnPYZwzGq4rWPeloE,1177
@@ -61,14 +61,14 @@ langfun/core/eval/scoring.py,sha256=_DvnlgI1SdRVaOojao_AkV3pnenfCPOqyhvlg-Sw-5M,
 langfun/core/eval/scoring_test.py,sha256=UcBH0R6vAovZ0A4yM22s5cBHL1qVKASubrbu1t8dYBw,4529
 langfun/core/eval/v2/__init__.py,sha256=9lNKJwbvl0lcFblAXYT_OHI8fOubJsTOdSkxEqsP1xU,1726
 langfun/core/eval/v2/checkpointing.py,sha256=t47rBfzGZYgIqWW1N1Ak9yQnNtHd-IRbEO0cZjG2VRo,11755
-langfun/core/eval/v2/checkpointing_test.py,sha256=NggOSJ_6XSa4cNP6nGIu9wLsK59dUwe8SPWDiXtGGDE,9197
+langfun/core/eval/v2/checkpointing_test.py,sha256=cuQ1zom5DMXIebxYW6L3N5XRyhfoEEDrs7XQcAxg8Nc,9164
 langfun/core/eval/v2/eval_test_helper.py,sha256=sKFi_wPYCNmr96WyTduuXY0KnxjFxcJyEhXey-_nGX8,3962
 langfun/core/eval/v2/evaluation.py,sha256=ihT5dljnUkHM97XS9OwE2wOnYC-oYnHYgG5KN1hmiaU,27037
-langfun/core/eval/v2/evaluation_test.py,sha256=QNp_HEvRTupvNuLEeYTvylykh1Ut2jpMqHQ-gCUZQ10,6919
+langfun/core/eval/v2/evaluation_test.py,sha256=46bGjNZmd57NXcJSoaC17DO9B74rpVBOVTEln_4W61c,6916
 langfun/core/eval/v2/example.py,sha256=v1dIz89pccIqujt7utrk0EbqMWM9kBn-2fYGRTKe358,10890
 langfun/core/eval/v2/example_test.py,sha256=wsHQD6te7ghROmxe3Xg_NK4TU0xS2MkNfnpo-H0H8xM,3399
 langfun/core/eval/v2/experiment.py,sha256=fb3RHNOSRftV7ZTBfYVV50iEevqdPwRHCt3mgtLzuFw,33408
-langfun/core/eval/v2/experiment_test.py,sha256=UmCobeS6ifPcaGkTJp0WPISolXrVFbeFCBiyJeA0Lt4,13666
+langfun/core/eval/v2/experiment_test.py,sha256=BYrPYfQfU2jDfAlZcHDT0KUaXOnCnyTWyUEKhDoqXfw,13645
 langfun/core/eval/v2/metric_values.py,sha256=_B905bC-jxrYPLSEcP2M8MaHZOVMz_bVrUw8YC4arCE,4660
 langfun/core/eval/v2/metric_values_test.py,sha256=ab2oF_HsIwrSy459108ggyjgefHSPn8UVILR4dRwx14,2634
 langfun/core/eval/v2/metrics.py,sha256=bl8i6u-ZHRBz4hAc3LzsZ2Dc7ZRQcuTYeUhhH-GxfF0,10628
@@ -76,17 +76,17 @@ langfun/core/eval/v2/metrics_test.py,sha256=LibZXvWEJDVRY-Mza_bQT-SbmbXCHUnFhL7Z
 langfun/core/eval/v2/progress.py,sha256=azZgssQgNdv3IgjKEaQBuGI5ucFDNbdi02P4z_nQ8GE,10292
 langfun/core/eval/v2/progress_test.py,sha256=YU7VHzmy5knPZwj9vpBN3rQQH2tukj9eKHkuBCI62h8,2540
 langfun/core/eval/v2/progress_tracking.py,sha256=zNhNPGlnJnHELEfFpbTMCSXFn8d1IJ57OOYkfFaBFfM,6097
-langfun/core/eval/v2/progress_tracking_test.py,sha256=fouMVJkFJqHjbhQJngGLGCmA9x3n0dU4USI2dY163mg,2291
+langfun/core/eval/v2/progress_tracking_test.py,sha256=sJhlVfinGsg3Kf2wQ_hT7VMcpQfaI4ZkqyW9ujElkwA,2282
 langfun/core/eval/v2/reporting.py,sha256=yUIPCAMnp7InIzpv1DDWrcLO-75iiOUTpscj7smkfrA,8335
-langfun/core/eval/v2/reporting_test.py,sha256=hcPJJaMtPulqERvHYTpId83WXdqDKnnexmULtK7WKwk,5686
+langfun/core/eval/v2/reporting_test.py,sha256=CMK-vwho8cNRJwlbkCqm_v5fykE7Y3V6SaIOCY0CDyA,5671
 langfun/core/eval/v2/runners.py,sha256=iqbH4jMtnNMhfuv1eHaxJmk1Vvsrz-sAJJFP8U44-tA,16758
-langfun/core/eval/v2/runners_test.py,sha256=DO3xV0sBNB6n65j41xx2i7gqUCJcPF37DFZLEjrmISg,11987
+langfun/core/eval/v2/runners_test.py,sha256=spjkmqlls_vyERdZMdjv6dhIN9ZfxsDDvIQAWTj2kMk,11954
 langfun/core/llms/__init__.py,sha256=CtxUdXohQ8AQk1DqBT6MBy2zdAoPSggNo00SYrj9-AY,9521
 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
 langfun/core/llms/azure_openai_test.py,sha256=lkMZkQdJBV97fTM4C4z8qNfvr6spgiN5G4hvVUIVr0M,1735
-langfun/core/llms/compositional.py,sha256=csW_FLlgL-tpeyCOTVvfUQkMa_zCN5Y2I-YbSNuK27U,2872
+langfun/core/llms/compositional.py,sha256=W_Fe2BdbkjwTzWW-paCWcEeG9oOR3-IcBG8oc73taSM,2878
 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
@@ -133,13 +133,13 @@ langfun/core/structured/description.py,sha256=6BztYOiucPkF4CrTQtPLPJo1gN2dwnKmaJ
 langfun/core/structured/description_test.py,sha256=UxaXnKKP7TnyPDPUyf3U-zPE0TvLlIP6DGr8thjcePw,7365
 langfun/core/structured/function_generation.py,sha256=g7AOR_e8HxFU6n6Df750aGkgMgV1KExLZMAz0yd5Agg,8555
 langfun/core/structured/function_generation_test.py,sha256=LaXYDXf9GlqUrR6v_gtmK_H4kxzonmU7SYbn7XXMgjU,12128
-langfun/core/structured/mapping.py,sha256=iraHpcEeF_iuEX2eoTsLGwTHHvxqp2gNDjoMf98l0Kk,13941
+langfun/core/structured/mapping.py,sha256=1YBW8PKpJKXS7DKukfzKNioL84PrKUcB4KOUudrQ20w,14374
 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=V8Cj1tJK4Lxv_b0YQj6-2hzXZgnYNBa2JR7rOLRBKoQ,22346
 langfun/core/structured/querying.py,sha256=vE_NOLNlIe4A0DueQfyiBEUh3AsSD8Hhx2dSDHNYpYk,37976
-langfun/core/structured/querying_test.py,sha256=pYWXqAnzp5eOCjU4yEPtE73iLNqxHISb3y3FaSbI7vs,49760
-langfun/core/structured/schema.py,sha256=r_ewdRMsALVOdnvGSeYBcz2-VJ_3_nMxY4GtzUHCYUU,29111
+langfun/core/structured/querying_test.py,sha256=Q0HwmbUI9BqMaeN8vgn_EvX29CzfcomGIKVqKJ6dZyY,50212
+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
 langfun/core/structured/schema_test.py,sha256=H42ZZdPi8CIv7WzrnXwMwQQaPQxlmDSY31pfqQs-Xqw,26567
@@ -156,8 +156,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.dev202507140805.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202507140805.dist-info/METADATA,sha256=PTIyICp15xRxYrwgOSZMqa95ZM-bB3aQ5vlOc8Ggw_8,8178
-langfun-0.1.2.dev202507140805.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-langfun-0.1.2.dev202507140805.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202507140805.dist-info/RECORD,,
+langfun-0.1.2.dev202507150805.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202507150805.dist-info/METADATA,sha256=sO0bfLTiiYrI_s4AbIEF_5v0li1JRZ1IjH492KCQrGU,8178
+langfun-0.1.2.dev202507150805.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+langfun-0.1.2.dev202507150805.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202507150805.dist-info/RECORD,,