Render content-block arrays fully

Use render_content_block for array items

Add tests for image and tool_use arrays

Author: ShlomoStept

src/claude_code_transcripts/__init__.py

@@ -108,21 +108,7 @@ def render_content_block_array(blocks):
     """
     parts = []
     for block in blocks:
-        if not isinstance(block, dict):
-            continue
-        block_type = block.get("type", "")
-        if block_type == "text":
-            text = block.get("text", "")
-            # Render as markdown
-            parts.append(render_markdown_text(text))
-        elif block_type == "thinking":
-            thinking = block.get("thinking", "")
-            parts.append(render_markdown_text(thinking))
-        else:
-            # For other types, just show as formatted text
-            text = block.get("text", block.get("content", ""))
-            if text:
-                parts.append(f"<pre>{html.escape(str(text))}</pre>")
+        parts.append(render_content_block(block))
     return "".join(parts) if parts else None
 
 

tests/__snapshots__/test_generate_html/TestRenderContentBlock.test_tool_result_content_block_array.html

@@ -1,3 +1,4 @@
-<div class="tool-result"><div class="truncatable"><div class="truncatable-content"><p>Here is the file content:</p>
+<div class="tool-result"><div class="truncatable"><div class="truncatable-content">
+<div class="assistant-text"><p>Here is the file content:</p>
 <p>Line 1
-Line 2</p></div><button class="expand-btn">Show more</button></div></div>
+Line 2</p></div></div><button class="expand-btn">Show more</button></div></div>

tests/__snapshots__/test_generate_html/TestRenderContentBlock.test_tool_result_content_block_array_with_image.html

@@ -0,0 +1,2 @@
+<div class="tool-result"><div class="truncatable"><div class="truncatable-content">
+<div class="image-block"><img src="data:image/gif;base64,R0lGODlhAQABAIAAAAUEBA==" style="max-width: 100%"></div></div><button class="expand-btn">Show more</button></div></div>

tests/__snapshots__/test_generate_html/TestRenderContentBlock.test_tool_result_content_block_array_with_tool_use.html

@@ -0,0 +1,5 @@
+<div class="tool-result"><div class="truncatable"><div class="truncatable-content">
+<div class="tool-use bash-tool" data-tool-id="toolu_123">
+<div class="tool-header"><span class="tool-icon">$</span> Bash</div>
+<div class="tool-description">List files</div><div class="truncatable"><div class="truncatable-content"><pre class="bash-command">ls -la</pre></div><button class="expand-btn">Show more</button></div>
+</div></div><button class="expand-btn">Show more</button></div></div>

tests/test_generate_html.py

@@ -335,6 +335,39 @@ def test_tool_result_content_block_array(self, snapshot_html):
         assert '"type": "text"' not in result
         assert result == snapshot_html
 
+    def test_tool_result_content_block_array_with_image(self, snapshot_html):
+        """Test that image blocks inside tool_result arrays render correctly."""
+        block = {
+            "type": "tool_result",
+            "content": (
+                '[{"type": "image", "source": {"type": "base64",'
+                ' "media_type": "image/gif", "data": "R0lGODlhAQABAIAAAAUEBA=="}}]'
+            ),
+            "is_error": False,
+        }
+        result = render_content_block(block)
+        assert 'src="data:image/gif;base64,' in result
+        assert "image-block" in result
+        assert '"type": "image"' not in result
+        assert result == snapshot_html
+
+    def test_tool_result_content_block_array_with_tool_use(self, snapshot_html):
+        """Test that tool_use blocks inside tool_result arrays render correctly."""
+        block = {
+            "type": "tool_result",
+            "content": (
+                '[{"type": "tool_use", "id": "toolu_123", "name": "Bash",'
+                ' "input": {"command": "ls -la", "description": "List files"}}]'
+            ),
+            "is_error": False,
+        }
+        result = render_content_block(block)
+        assert "tool-use" in result
+        assert "bash-tool" in result
+        assert "List files" in result
+        assert '"type": "tool_use"' not in result
+        assert result == snapshot_html
+
 
 class TestAnalyzeConversation:
     """Tests for conversation analysis."""