feat: handle partial/malformed tool call detection for claude-code (#184)

Author: 35C4n0r

lib/msgfmt/format_tool_call.go

@@ -4,30 +4,51 @@ import (
 	"strings"
 )
 
+type toolCallRange struct {
+	start     int
+	end       int
+	malformed bool
+}
+
 func removeClaudeReportTaskToolCall(msg string) (string, []string) {
 	msg = "\n" + msg // This handles the case where the message starts with a tool call
 
 	// Remove all tool calls that start with `● coder - coder_report_task (MCP)`
 	lines := strings.Split(msg, "\n")
 
 	toolCallStartIdx := -1
+	newLineAfterToolCall := -1
 
 	// Store all tool call start and end indices [[start, end], ...]
-	var toolCallIdxs [][]int
+	var toolCallIdxs []toolCallRange
 
 	for i := 1; i < len(lines)-1; i++ {
 		prevLine := strings.TrimSpace(lines[i-1])
-		line := strings.TrimSpace(lines[i])
+		line := strings.Trim(strings.TrimSpace(lines[i]), "\n")
 		nextLine := strings.TrimSpace(lines[i+1])
 
 		if strings.Contains(line, "coder - coder_report_task (MCP)") {
 			toolCallStartIdx = i
-		} else if toolCallStartIdx != -1 && line == "\"message\": \"Thanks for reporting!\"" && nextLine == "}" && strings.HasSuffix(prevLine, "{") {
-			// Store [start, end] pair
-			toolCallIdxs = append(toolCallIdxs, []int{toolCallStartIdx, min(len(lines), i+2)})
+		} else if toolCallStartIdx != -1 {
+			if line == "\"message\": \"Thanks for reporting!\"" && nextLine == "}" && strings.HasSuffix(prevLine, "{") {
+				// Store [start, end] pair
+				toolCallIdxs = append(toolCallIdxs, toolCallRange{toolCallStartIdx, min(len(lines), i+2), false})
+
+				// Reset to find the next tool call
+				toolCallStartIdx = -1
+				newLineAfterToolCall = -1
+			} else if len(line) == 0 {
+				newLineAfterToolCall = i + 1
+			}
+		}
+	}
 
-			// Reset to find the next tool call
-			toolCallStartIdx = -1
+	// Handle the malformed/partially rendered tool_calls
+	if toolCallStartIdx != -1 {
+		if newLineAfterToolCall != -1 {
+			toolCallIdxs = append(toolCallIdxs, toolCallRange{toolCallStartIdx, newLineAfterToolCall, true})
+		} else {
+			toolCallIdxs = append(toolCallIdxs, toolCallRange{toolCallStartIdx, len(lines), true})
 		}
 	}
 
@@ -40,10 +61,12 @@ func removeClaudeReportTaskToolCall(msg string) (string, []string) {
 
 	// Remove tool calls from the message
 	for i := len(toolCallIdxs) - 1; i >= 0; i-- {
-		idxPair := toolCallIdxs[i]
-		start, end := idxPair[0], idxPair[1]
+		start, end := toolCallIdxs[i].start, toolCallIdxs[i].end
 
-		toolCallMessages = append(toolCallMessages, strings.Join(lines[start:end], "\n"))
+		// If the toolCall is malformed, we don't want to log it
+		if !toolCallIdxs[i].malformed {
+			toolCallMessages = append(toolCallMessages, strings.Join(lines[start:end], "\n"))
+		}
 
 		lines = append(lines[:start], lines[end:]...)
 	}

lib/msgfmt/testdata/format/claude/remove-task-tool-call/expected.txt

@@ -40,3 +40,6 @@
   Open the HTML file in your web browser and use the arrow keys
    to move the snake. Collect the red food to grow and increase
    your score!
+
+
+ Above is a malformed tool call message

lib/msgfmt/testdata/format/claude/remove-task-tool-call/msg.txt

@@ -72,6 +72,13 @@
        "message": "Thanks for reporting!"
      }
 
+ coder - coder_report_task (MCP)(summary: "Snake game created
+                                  successfully at snake-game.html",
+                                  link: "file:///home/coder/snake-ga
+                                  me.html", state: "working")
+
+ Above is a malformed tool call message
+
 
 ───────────────────────────────────────────────────────────────────
 ❯