@@ -128,7 +128,9 @@ def _transform_input_messages(
             continue
 
         text_parts = [
-            part for part in content if isinstance(part, dict) and "text" in part
+            part
+            for part in content
+            if isinstance(part, dict) and part.get("type") == "text" and "text" in part
         ]
         input_messages.append(
             {
Original file line number	Diff line number	Diff line change
`@@ -128,7 +128,9 @@ def _transform_input_messages(`
`128`	`128`	`continue`
`129`	`129`
`130`	`130`	`text_parts = [`
`131`		`- part for part in content if isinstance(part, dict) and "text" in part`
	`131`	`+ part`
	`132`	`+ for part in content`
	`133`	`+ if isinstance(part, dict) and part.get("type") == "text" and "text" in part`
`132`	`134`	`]`
`133`	`135`	`input_messages.append(`
`134`	`136`	`{`