Add pretty_print and disable_escape_html to tool_feedback defaults

tool_feedback.go

@@ -0,0 +1,86 @@
+package utils
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"strings"
+)
+
+const ToolFeedbackContinuationHint = "Continuing the current task."
+
+func FormatArgsJSON(args map[string]any, prettyPrint, disableEscapeHTML bool) string {
+	var buf bytes.Buffer
+	enc := json.NewEncoder(&buf)
+	if prettyPrint {
+		enc.SetIndent("", "  ")
+	}
+	if disableEscapeHTML {
+		enc.SetEscapeHTML(false)
+	}
+	if err := enc.Encode(args); err != nil {
+		return "{}"
+	}
+	return strings.TrimSpace(buf.String())
+}
+
+func FormatToolFeedbackMessage(toolName, explanation, argsPreview string) string {
+	toolName = strings.TrimSpace(toolName)
+	explanation = strings.TrimSpace(explanation)
+	argsPreview = strings.TrimSpace(argsPreview)
+
+	bodyLines := make([]string, 0, 2)
+	if explanation != "" {
+		bodyLines = append(bodyLines, explanation)
+	}
+	if argsPreview != "" {
+		bodyLines = append(bodyLines, "```json\n"+argsPreview+"\n```")
+	}
+	body := strings.Join(bodyLines, "\n")
+
+	if toolName == "" {
+		return body
+	}
+	if body == "" {
+		return fmt.Sprintf("\U0001f527 `%s`", toolName)
+	}
+
+	return fmt.Sprintf("\U0001f527 `%s`\n%s", toolName, body)
+}
+
+func FitToolFeedbackMessage(content string, maxLen int) string {
+	content = strings.TrimSpace(content)
+	if content == "" || maxLen <= 0 {
+		return ""
+	}
+	if len([]rune(content)) <= maxLen {
+		return content
+	}
+
+	firstLine, rest, hasRest := strings.Cut(content, "\n")
+	firstLine = strings.TrimSpace(firstLine)
+	rest = strings.TrimSpace(rest)
+
+	if !hasRest || rest == "" {
+		return Truncate(firstLine, maxLen)
+	}
+
+	if len([]rune(firstLine)) >= maxLen {
+		return Truncate(firstLine, maxLen)
+	}
+
+	remaining := maxLen - len([]rune(firstLine)) - 1
+	if remaining <= 0 {
+		return Truncate(firstLine, maxLen)
+	}
+
+	return firstLine + "\n" + Truncate(rest, remaining)
+}
+
+func Truncate(s string, maxLen int) string {
+	runes := []rune(s)
+	if len(runes) <= maxLen {
+		return s
+	}
+	return string(runes[:maxLen])
+}