add synthetic child calls

Author: zbirenbaum

weave/integrations/claude_agent_sdk/claude_agent_sdk_integration.py

@@ -9,6 +9,8 @@
 from typing import Any
 
 from weave.integrations.claude_agent_sdk.display_utils import (
+    text_display_name,
+    thinking_display_name,
     tool_use_display_name,
     turn_display_name,
 )
@@ -66,6 +68,7 @@ def _process_message_inline(
     from claude_agent_sdk import (
         AssistantMessage,
         SystemMessage,
+        TextBlock,
         ThinkingBlock,
         ToolResultBlock,
         ToolUseBlock,
@@ -104,6 +107,34 @@ def _is_thinking_only(m: Any) -> bool:
         history = list(accumulated)
         accumulated.append(serialized)
 
+        # Create child calls for thinking blocks
+        thinking_blocks = [b for b in msg.content if isinstance(b, ThinkingBlock)]
+        if thinking_blocks:
+            thinking_text = "\n".join(b.thinking for b in thinking_blocks)
+            thinking_call = wc.create_call(
+                op="claude_agent_sdk.thinking",
+                inputs={},
+                display_name=thinking_display_name(thinking_text),
+                parent=root_call,
+                attributes={"kind": "llm"},
+                use_stack=False,
+            )
+            wc.finish_call(thinking_call, output={"thinking": thinking_text})
+
+        # Create child calls for text blocks
+        text_blocks = [b for b in msg.content if isinstance(b, TextBlock)]
+        if text_blocks:
+            text_content = "\n".join(b.text for b in text_blocks)
+            text_call = wc.create_call(
+                op="claude_agent_sdk.text",
+                inputs={},
+                display_name=text_display_name(text_content),
+                parent=root_call,
+                attributes={"kind": "llm"},
+                use_stack=False,
+            )
+            wc.finish_call(text_call, output={"text": text_content, "model": msg.model})
+
         tool_uses = [b for b in msg.content if isinstance(b, ToolUseBlock)]
 
         # Open tool calls

weave/integrations/claude_agent_sdk/display_utils.py

@@ -54,6 +54,25 @@ def tool_use_display_name(tool_name: str, tool_input: dict[str, Any]) -> str:
     return f"{tool_name}({params})"
 
 
+def _abbreviate(text: str, max_words: int = 8) -> str:
+    """Abbreviate text to the first few words."""
+    words = text.split()[:max_words]
+    name = " ".join(words)
+    if len(text.split()) > max_words:
+        name += "..."
+    return name
+
+
+def thinking_display_name(thinking: str) -> str:
+    """Generate display name for a thinking block child call."""
+    return f"Thinking: {_abbreviate(thinking)}"
+
+
+def text_display_name(text: str) -> str:
+    """Generate display name for a text block child call."""
+    return f"Text: {_abbreviate(text)}"
+
+
 def response_display_name(model: str | None) -> str:
     """Generate display name for a response call."""
     if model: