langfun 0.1.2.dev202410250804__py3-none-any.whl → 0.1.2.dev202410260804__py3-none-any.whl

langfun/core/eval/base.py

@@ -531,19 +531,14 @@ class Evaluable(lf.Component):
     self._render_message(self.dryrun_output, s)
 
   def _render_message(self, message: lf.Message, s: io.StringIO) -> None:
-    for m in message.trace():
-      if 'lm-input' in m.tags:
-        color = 'green'
-      elif 'lm-response' in m.tags:
-        color = 'blue'
-      else:
-        continue
-
-      s.write(
-          f'<div style="color: {color}; border: 1px solid; margin-top: 10px">'
-      )
-      s.write(m._repr_html_())    # pylint: disable=protected-access
-      s.write('</div>')
+    s.write(
+        message.to_html_str(
+            extra_flags=dict(
+                include_message_metadata=False,
+                source_tag=['lm-input', 'lm-response'],
+            )
+        )
+    )
 
   @classmethod
   def from_dir(

{langfun-0.1.2.dev202410250804.dist-info → langfun-0.1.2.dev202410260804.dist-info}/METADATA

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

{langfun-0.1.2.dev202410250804.dist-info → langfun-0.1.2.dev202410260804.dist-info}/RECORD

@@ -44,7 +44,7 @@ langfun/core/coding/python/parsing_test.py,sha256=9vAWF484kWIm6JZq8NFiMgKUDhXV-d
 langfun/core/coding/python/permissions.py,sha256=1QWGHvzL8MM0Ok_auQ9tURqZHtdOfJaDpBzZ29GUE-c,2544
 langfun/core/coding/python/permissions_test.py,sha256=w5EDb8QxpxgJyZkojyzVWQvDfg366zn99-g__6TbPQ0,2699
 langfun/core/eval/__init__.py,sha256=Ogdr9OtTywhhLPHi3AZzOD2mXX2oyaHWflrSTMm96uA,1899
-langfun/core/eval/base.py,sha256=hCblGSBirslPtthi_iz98boyu46UaOBSQ99STyw7L28,75898
+langfun/core/eval/base.py,sha256=7CPWykFZB4-1a8rMBrE4XNAtRQ9w_O1T3bKeg7ykkw4,75735
 langfun/core/eval/base_test.py,sha256=gAxamZKvHzT3wVr3xVtXLiwBBi5VFX1I29_QuKSR4Wg,26958
 langfun/core/eval/matching.py,sha256=UnjdM_ebPqXKJamY4lvL3AYxrMIz3LqkjRTnHJ5xsYc,9349
 langfun/core/eval/matching_test.py,sha256=QCoYEuf4b_1bkHqUCuRzKMbXHrV3AB2FCOBivo1stC4,5249
@@ -119,8 +119,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.dev202410250804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202410250804.dist-info/METADATA,sha256=tYKkpmIIXhUO6f3i6uUr5FfkVYrme8O3WtbNCD6YkqY,8890
-langfun-0.1.2.dev202410250804.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-langfun-0.1.2.dev202410250804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202410250804.dist-info/RECORD,,
+langfun-0.1.2.dev202410260804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202410260804.dist-info/METADATA,sha256=bthuWyjoDbeg9CyvHeaqZaEqeK-xJeIFgBwQP9sKNEk,8890
+langfun-0.1.2.dev202410260804.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+langfun-0.1.2.dev202410260804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202410260804.dist-info/RECORD,,