@@ -182,18 +182,25 @@ func GetMessageItems(sty *styles.Styles, msg *message.Message, toolResults map[s
182182 return []MessageItem{}
183183}
184184
185−func GetToolMessageItem(sty *styles.Styles, tc message.ToolCall, result *message.ToolResult, canceled bool) MessageItem {
185+// GetToolRenderer returns the appropriate ToolRenderFunc for a given tool call.
186+// this should be used for nested tools as well.
187+func GetToolRenderer(tc message.ToolCall) ToolRenderFunc {
186188 renderFunc := DefaultToolRenderer
187− // we only do full width for diffs (as far as I know)
188− cappedWidth := true
189189 switch tc.Name {
190190 case tools.BashToolName:
191191 renderFunc = BashToolRenderer
192192 }
193+ return renderFunc
194+}
195+
196+// GetToolMessageItem creates a MessageItem for a tool call and its result.
197+func GetToolMessageItem(sty *styles.Styles, tc message.ToolCall, result *message.ToolResult, canceled bool) MessageItem {
198+ // we only do full width for diffs (as far as I know)
199+ cappedWidth := tc.Name != tools.EditToolName && tc.Name != tools.MultiEditToolName
193200
194201 return NewToolMessageItem(
195202 sty,
196− renderFunc,
203+ GetToolRenderer(tc),
197204 tc,
198205 result,
199206 canceled,