git.iko.soy/crush
on main help sign in
fix(sessions): tag removal, handle multibyte
c49b4feChristian Rocha
internal/agent/agent.go+5−3
compare with main
unifiedsplit
@@ -16,6 +16,7 @@ import (
16 "fmt"
16 "fmt"
17 "log/slog"
17 "log/slog"
18 "os"
18 "os"
19+ "regexp"
19 "strconv"
20 "strconv"
20 "strings"
21 "strings"
21 "sync"
22 "sync"
@@ -45,6 +46,9 @@ var titlePrompt []byte
45//go:embed templates/summary.md
46//go:embed templates/summary.md
46var summaryPrompt []byte
47var summaryPrompt []byte
47
48
49+// Used to remove <think> tags from generated titles.
50+var thinkTagRegex = regexp.MustCompile(`<think>.*?</think>`)
51+
48type SessionAgentCall struct {
52type SessionAgentCall struct {
49 SessionID string
53 SessionID string
50 Prompt string
54 Prompt string
@@ -760,9 +764,7 @@ func (a *sessionAgent) generateTitle(ctx context.Context, sessionID string, prom
760 title = strings.ReplaceAll(title, "\n", " ")
764 title = strings.ReplaceAll(title, "\n", " ")
761
765
762 // Remove thinking tags if present.
766 // Remove thinking tags if present.
763− if idx := strings.Index(title, "</think>"); idx > 0 {
767+ title = thinkTagRegex.ReplaceAllString(title, "")
764− title = title[idx+len("</think>"):]
765− }
766
768
767 title = strings.TrimSpace(title)
769 title = strings.TrimSpace(title)
768 if title == "" {
770 if title == "" {