langfun 0.0.2.dev20240608__tar.gz → 0.0.2.dev20240610__tar.gz

{langfun-0.0.2.dev20240608 → langfun-0.0.2.dev20240610}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langfun
-Version: 0.0.2.dev20240608
+Version: 0.0.2.dev20240610
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.0.2.dev20240608 → langfun-0.0.2.dev20240610}/langfun/core/__init__.py

@@ -121,6 +121,9 @@ from langfun.core.memory import Memory
 # Utility for console output.
 from langfun.core import console
 
+# Helpers for implementing _repr_xxx_ methods.
+from langfun.core import repr_utils
+
 # Utility for event logging.
 from langfun.core import logging
 

langfun-0.0.2.dev20240610/langfun/core/repr_utils.py

@@ -0,0 +1,83 @@
+# Copyright 2024 The Langfun Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Helpers for implementing _repr_xxx_ methods."""
+
+import collections
+import contextlib
+import io
+from typing import Iterator
+
+from langfun.core import component
+
+
+@contextlib.contextmanager
+def share_parts() -> Iterator[dict[str, int]]:
+  """Context manager for defining the context (scope) of shared content.
+
+  Under the context manager, call to `lf.write_shared` with the same content
+  will be written only once. This is useful for writing shared content such as
+  shared style and script sections in the HTML.
+
+  Example:
+    ```
+    class Foo(pg.Object):
+      def _repr_html_(self) -> str:
+        s = io.StringIO()
+        lf.repr_utils.write_shared_part(s, '<style>..</style>')
+        lf.repr_utils.write_shared_part(s, '<script>..</script>')
+        return s.getvalue()
+
+    with lf.repr_utils.share_parts() as share_parts:
+      # The <style> and <script> section will be written only once.
+      lf.console.display(Foo())
+      lf.console.display(Foo())
+
+    # Assert that the shared content is attempted to be written twice.
+    assert share_parts['<style>..</style>'] == 2
+    ```
+
+  Yields:
+    A dictionary mapping the shared content to the number of times it is
+    attempted to be written.
+  """
+  context = component.context_value(
+      '__shared_parts__', collections.defaultdict(int)
+  )
+  with component.context(__shared_parts__=context):
+    try:
+      yield context
+    finally:
+      pass
+
+
+def write_maybe_shared(s: io.StringIO, content: str) -> bool:
+  """Writes a maybe shared part to an string stream.
+
+  Args:
+    s: The string stream to write to.
+    content: A maybe shared content to write.
+
+  Returns:
+    True if the content is written to the string. False if the content is
+    already written under the same share context.
+  """
+  context = component.context_value('__shared_parts__', None)
+  if context is None:
+    s.write(content)
+    return True
+  written = content in context
+  if not written:
+    s.write(content)
+  context[content] += 1
+  return not written

langfun-0.0.2.dev20240610/langfun/core/repr_utils_test.py

@@ -0,0 +1,58 @@
+# Copyright 2024 The Langfun Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Tests for langfun.core.repr_utils."""
+
+import io
+import unittest
+
+from langfun.core import repr_utils
+
+
+class SharingContentTest(unittest.TestCase):
+
+  def test_sharing(self):
+    s = io.StringIO()
+
+    self.assertTrue(repr_utils.write_maybe_shared(s, '<hr>'))
+    self.assertTrue(repr_utils.write_maybe_shared(s, '<hr>'))
+
+    with repr_utils.share_parts() as ctx1:
+      self.assertTrue(repr_utils.write_maybe_shared(s, '<style></style>'))
+      self.assertFalse(repr_utils.write_maybe_shared(s, '<style></style>'))
+
+      with repr_utils.share_parts() as ctx2:
+        self.assertIs(ctx2, ctx1)
+        self.assertFalse(repr_utils.write_maybe_shared(s, '<style></style>'))
+        self.assertTrue(repr_utils.write_maybe_shared(s, '<style>a</style>'))
+        self.assertFalse(repr_utils.write_maybe_shared(s, '<style>a</style>'))
+        self.assertTrue(repr_utils.write_maybe_shared(s, '<style>b</style>'))
+
+      with repr_utils.share_parts() as ctx3:
+        self.assertIs(ctx3, ctx1)
+        self.assertFalse(repr_utils.write_maybe_shared(s, '<style></style>'))
+        self.assertFalse(repr_utils.write_maybe_shared(s, '<style>a</style>'))
+        self.assertFalse(repr_utils.write_maybe_shared(s, '<style>a</style>'))
+        self.assertFalse(repr_utils.write_maybe_shared(s, '<style>b</style>'))
+
+    self.assertEqual(
+        s.getvalue(),
+        '<hr><hr><style></style><style>a</style><style>b</style>'
+    )
+    self.assertEqual(ctx1['<style></style>'], 4)
+    self.assertEqual(ctx1['<style>b</style>'], 2)
+    self.assertEqual(ctx1['<style>a</style>'], 4)
+
+
+if __name__ == '__main__':
+  unittest.main()

{langfun-0.0.2.dev20240608 → langfun-0.0.2.dev20240610}/langfun.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langfun
-Version: 0.0.2.dev20240608
+Version: 0.0.2.dev20240610
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.0.2.dev20240608 → langfun-0.0.2.dev20240610}/langfun.egg-info/SOURCES.txt

@@ -27,6 +27,8 @@ langfun/core/modality.py
 langfun/core/modality_test.py
 langfun/core/natural_language.py
 langfun/core/natural_language_test.py
+langfun/core/repr_utils.py
+langfun/core/repr_utils_test.py
 langfun/core/sampling.py
 langfun/core/sampling_test.py
 langfun/core/subscription.py