langfun 0.1.2.dev202503220803__py3-none-any.whl → 0.1.2.dev202503240804__py3-none-any.whl

langfun/core/eval/v2/progress_tracking.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 """Tracking evaluation run progress."""
 
+import os
 import langfun.core as lf
 from langfun.core.eval.v2 import example as example_lib
 from langfun.core.eval.v2 import experiment as experiment_lib
@@ -95,7 +96,7 @@ class _TqdmProgressTracker(experiment_lib.Plugin):
         for i, leaf in enumerate(root.leaf_nodes)
     }
     summary_link = Experiment.link(
-        runner.current_run.output_path_for(root, 'summary.html')
+        os.path.join(runner.current_run.output_root, 'summary.html')
     )
     lf.console.write(f'Summary: {summary_link}.', color='green')
 

langfun/core/modalities/mime.py

@@ -152,12 +152,33 @@ class Mime(lf.Modality):
 
   @classmethod
   def from_uri(cls, uri: str, **kwargs) -> 'Mime':
+    if uri.startswith('data:'):
+      mime_type, content = cls._parse_data_uri(uri)
+      return cls.class_from_mime_type(mime_type).from_bytes(content, **kwargs)
+
     if cls is Mime:
       content = cls.download(uri)
       mime = from_buffer(content, mime=True).lower()
       return cls.class_from_mime_type(mime)(uri=uri, content=content, **kwargs)
     return cls(uri=uri, content=None, **kwargs)
 
+  @classmethod
+  def _parse_data_uri(cls, uri: str) -> tuple[str, bytes]:
+    """Returns the MIME type and content from the given data URI."""
+    assert uri.startswith('data:'), uri
+    mime_end_pos = uri.find(';', 0)
+    if mime_end_pos == -1:
+      raise ValueError(f'Invalid data URI: {uri!r}.')
+    mime_type = uri[5: mime_end_pos].strip().lower()
+    encoding_end_pos = uri.find(',', mime_end_pos + 1)
+    if encoding_end_pos == -1:
+      raise ValueError(f'Invalid data URI: {uri!r}.')
+    encoding = uri[mime_end_pos + 1: encoding_end_pos].strip().lower()
+    if encoding != 'base64':
+      raise ValueError(f'Unsupported encoding: {encoding!r}.')
+    base64_content = uri[encoding_end_pos + 1:].strip().encode()
+    return mime_type, base64.b64decode(base64_content)
+
   @classmethod
   def from_bytes(cls, content: bytes | str, **kwargs) -> 'Mime':
     if cls is Mime:

langfun/core/modalities/mime_test.py

@@ -93,6 +93,22 @@ class CustomMimeTest(unittest.TestCase):
       self.assertEqual(content.to_bytes(), b'bar')
       self.assertEqual(content.mime_type, 'text/plain')
 
+    content = mime.Mime.from_uri('data:text/plain;base64,Zm9v')
+    self.assertIsNone(content.uri)
+    self.assertEqual(content.mime_type, 'text/plain')
+    self.assertEqual(content.content, b'foo')
+    self.assertEqual(content.content_uri, 'data:text/plain;base64,Zm9v')
+    self.assertEqual(content.embeddable_uri, 'data:text/plain;base64,Zm9v')
+
+    with self.assertRaisesRegex(ValueError, 'Invalid data URI'):
+      mime.Mime.from_uri('data:text/plain')
+
+    with self.assertRaisesRegex(ValueError, 'Invalid data URI'):
+      mime.Mime.from_uri('data:text/plain;abcd')
+
+    with self.assertRaisesRegex(ValueError, 'Unsupported encoding'):
+      mime.Mime.from_uri('data:text/plain;base16,abcd')
+
   def assert_html_content(self, html, expected):
     expected = inspect.cleandoc(expected).strip()
     actual = html.content.strip()

{langfun-0.1.2.dev202503220803.dist-info → langfun-0.1.2.dev202503240804.dist-info}/METADATA

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

{langfun-0.1.2.dev202503220803.dist-info → langfun-0.1.2.dev202503240804.dist-info}/RECORD

@@ -67,7 +67,7 @@ langfun/core/eval/v2/metrics.py,sha256=bl8i6u-ZHRBz4hAc3LzsZ2Dc7ZRQcuTYeUhhH-Gxf
 langfun/core/eval/v2/metrics_test.py,sha256=p4FzLJsE8XAzAQuyP9hfEf9YeKWZ__PO_ue8a9P0-cc,6082
 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=l9fEkz4oP5McpZzf72Ua7PYm3lAWtRru7gRWNf8H0ms,6083
+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/reporting.py,sha256=yUIPCAMnp7InIzpv1DDWrcLO-75iiOUTpscj7smkfrA,8335
 langfun/core/eval/v2/reporting_test.py,sha256=hcPJJaMtPulqERvHYTpId83WXdqDKnnexmULtK7WKwk,5686
@@ -112,8 +112,8 @@ langfun/core/modalities/audio.py,sha256=qCrVCX690SG0ps-ZfOtNWvHn_CmdJsmxF7GySScW
 langfun/core/modalities/audio_test.py,sha256=yyGEBYqMXmNs_F2dEtj-PX8HE040vqh-YQppsvdxPw4,2025
 langfun/core/modalities/image.py,sha256=ovcX8NLBNv8WzxAdhQ-u4VQfHVBkWijRj_oiNwhE9lk,1768
 langfun/core/modalities/image_test.py,sha256=XMgtJXY75R5eo0CZ222D1QUy57_hESnttmCGWwDLt7k,3824
-langfun/core/modalities/mime.py,sha256=T00qHntL3uewF_DBVA-sJiV9mPK6bLAuBsg0bWSYKUQ,7919
-langfun/core/modalities/mime_test.py,sha256=Lg21nKFi--a884gnI7tFS4cHnwByEEhWg4Ugsr-UCbw,5277
+langfun/core/modalities/mime.py,sha256=Hz6MnxQO_ccgrEryUf7dwyUGYj616I68NshLDK9PC_k,8860
+langfun/core/modalities/mime_test.py,sha256=n084tOkeKHKMOVblCmi5s8nw4o7VYn3ynqvcrz8ww7c,5977
 langfun/core/modalities/pdf.py,sha256=mfaeCbUA4JslFVTARiJh8hW7imvL4tLVw9gUhO5bAZA,727
 langfun/core/modalities/pdf_test.py,sha256=ulZ0FbnlsU0wkrdckJ4ONZPTYRyMPO9Aob1UO6FXygk,1950
 langfun/core/modalities/video.py,sha256=vI9apcHIHGyp90i34Srg7S3G6IBDtDCk8qiXhwRQmkw,967
@@ -148,8 +148,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.dev202503220803.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202503220803.dist-info/METADATA,sha256=z8c-QQRstr1soNbtJeSrzitjRg0XSYkaDXeF01SoRpw,7692
-langfun-0.1.2.dev202503220803.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-langfun-0.1.2.dev202503220803.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202503220803.dist-info/RECORD,,
+langfun-0.1.2.dev202503240804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202503240804.dist-info/METADATA,sha256=hHWiQe8QXhgWGGzK6_DDK7_Sl7Bm99b6TTkLIJI6GV8,7692
+langfun-0.1.2.dev202503240804.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+langfun-0.1.2.dev202503240804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202503240804.dist-info/RECORD,,