Merge pull request #861 from p3ddd/refactor/modernize

refactor(modernize): apply safe modernize fixes
This commit is contained in:
Meng Zhuo 2026-03-01 15:38:59 +08:00 committed by GitHub
commit f7136b6a5d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
27 changed files with 88 additions and 119 deletions

View file

@ -7,6 +7,7 @@ import (
"os" "os"
"path/filepath" "path/filepath"
"runtime" "runtime"
"slices"
"strings" "strings"
"sync" "sync"
"time" "time"
@ -249,10 +250,8 @@ func (cb *ContextBuilder) sourceFilesChangedLocked() bool {
} }
// Check tracked source files (bootstrap + memory). // Check tracked source files (bootstrap + memory).
for _, p := range cb.sourcePaths() { if slices.ContainsFunc(cb.sourcePaths(), cb.fileChangedSince) {
if cb.fileChangedSince(p) { return true
return true
}
} }
// --- Skills directory (handled separately from sourcePaths) --- // --- Skills directory (handled separately from sourcePaths) ---

View file

@ -404,11 +404,11 @@ func TestConcurrentBuildSystemPromptWithCache(t *testing.T) {
var wg sync.WaitGroup var wg sync.WaitGroup
errs := make(chan string, goroutines*iterations) errs := make(chan string, goroutines*iterations)
for g := 0; g < goroutines; g++ { for g := range goroutines {
wg.Add(1) wg.Add(1)
go func(id int) { go func(id int) {
defer wg.Done() defer wg.Done()
for i := 0; i < iterations; i++ { for i := range iterations {
result := cb.BuildSystemPromptWithCache() result := cb.BuildSystemPromptWithCache()
if result == "" { if result == "" {
errs <- "empty prompt returned" errs <- "empty prompt returned"

View file

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"os" "os"
"path/filepath" "path/filepath"
"slices"
"testing" "testing"
"time" "time"
@ -187,13 +188,7 @@ func TestToolRegistry_ToolRegistration(t *testing.T) {
toolsList := toolsInfo["names"].([]string) toolsList := toolsInfo["names"].([]string)
// Check that our custom tool name is in the list // Check that our custom tool name is in the list
found := false found := slices.Contains(toolsList, "mock_custom")
for _, name := range toolsList {
if name == "mock_custom" {
found = true
break
}
}
if !found { if !found {
t.Error("Expected custom tool to be registered") t.Error("Expected custom tool to be registered")
} }
@ -262,13 +257,7 @@ func TestToolRegistry_GetDefinitions(t *testing.T) {
toolsList := toolsInfo["names"].([]string) toolsList := toolsInfo["names"].([]string)
// Check that our custom tool name is in the list // Check that our custom tool name is in the list
found := false found := slices.Contains(toolsList, "mock_custom")
for _, name := range toolsList {
if name == "mock_custom" {
found = true
break
}
}
if !found { if !found {
t.Error("Expected custom tool to be registered") t.Error("Expected custom tool to be registered")
} }

View file

@ -111,7 +111,7 @@ func (ms *MemoryStore) GetRecentDailyNotes(days int) string {
var sb strings.Builder var sb strings.Builder
first := true first := true
for i := 0; i < days; i++ { for i := range days {
date := time.Now().AddDate(0, 0, -i) date := time.Now().AddDate(0, 0, -i)
dateStr := date.Format("20060102") // YYYYMMDD dateStr := date.Format("20060102") // YYYYMMDD
monthDir := dateStr[:6] // YYYYMM monthDir := dateStr[:6] // YYYYMM

View file

@ -67,7 +67,7 @@ func TestPublishInbound_ContextCancel(t *testing.T) {
// Fill the buffer // Fill the buffer
ctx := context.Background() ctx := context.Background()
for i := 0; i < defaultBusBufferSize; i++ { for i := range defaultBusBufferSize {
if err := mb.PublishInbound(ctx, InboundMessage{Content: "fill"}); err != nil { if err := mb.PublishInbound(ctx, InboundMessage{Content: "fill"}); err != nil {
t.Fatalf("fill failed at %d: %v", i, err) t.Fatalf("fill failed at %d: %v", i, err)
} }
@ -154,7 +154,7 @@ func TestConcurrentPublishClose(t *testing.T) {
wg.Add(numGoroutines + 1) wg.Add(numGoroutines + 1)
// Spawn many goroutines trying to publish // Spawn many goroutines trying to publish
for i := 0; i < numGoroutines; i++ { for range numGoroutines {
go func() { go func() {
defer wg.Done() defer wg.Done()
// Use a short timeout context so we don't block forever after close // Use a short timeout context so we don't block forever after close
@ -194,7 +194,7 @@ func TestPublishInbound_FullBuffer(t *testing.T) {
ctx := context.Background() ctx := context.Background()
// Fill the buffer // Fill the buffer
for i := 0; i < defaultBusBufferSize; i++ { for i := range defaultBusBufferSize {
if err := mb.PublishInbound(ctx, InboundMessage{Content: "fill"}); err != nil { if err := mb.PublishInbound(ctx, InboundMessage{Content: "fill"}); err != nil {
t.Fatalf("fill failed at %d: %v", i, err) t.Fatalf("fill failed at %d: %v", i, err)
} }

View file

@ -274,13 +274,12 @@ func TestWorkerRateLimiter(t *testing.T) {
limiter: rate.NewLimiter(2, 1), limiter: rate.NewLimiter(2, 1),
} }
ctx, cancel := context.WithCancel(context.Background()) ctx := t.Context()
defer cancel()
go m.runWorker(ctx, "test", w) go m.runWorker(ctx, "test", w)
// Enqueue 4 messages // Enqueue 4 messages
for i := 0; i < 4; i++ { for i := range 4 {
w.queue <- bus.OutboundMessage{Channel: "test", ChatID: "1", Content: fmt.Sprintf("msg%d", i)} w.queue <- bus.OutboundMessage{Channel: "test", ChatID: "1", Content: fmt.Sprintf("msg%d", i)}
} }
@ -352,8 +351,7 @@ func TestRunWorker_MessageSplitting(t *testing.T) {
limiter: rate.NewLimiter(rate.Inf, 1), limiter: rate.NewLimiter(rate.Inf, 1),
} }
ctx, cancel := context.WithCancel(context.Background()) ctx := t.Context()
defer cancel()
go m.runWorker(ctx, "test", w) go m.runWorker(ctx, "test", w)
@ -576,7 +574,7 @@ func TestRecordPlaceholder_ConcurrentSafe(t *testing.T) {
m := newTestManager() m := newTestManager()
var wg sync.WaitGroup var wg sync.WaitGroup
for i := 0; i < 100; i++ { for i := range 100 {
wg.Add(1) wg.Add(1)
go func(i int) { go func(i int) {
defer wg.Done() defer wg.Done()
@ -591,7 +589,7 @@ func TestRecordTypingStop_ConcurrentSafe(t *testing.T) {
m := newTestManager() m := newTestManager()
var wg sync.WaitGroup var wg sync.WaitGroup
for i := 0; i < 100; i++ { for i := range 100 {
wg.Add(1) wg.Add(1)
go func(i int) { go func(i int) {
defer wg.Done() defer wg.Done()
@ -834,7 +832,7 @@ func TestLazyWorkerCreation(t *testing.T) {
func TestBuildMediaScope_FastIDUniqueness(t *testing.T) { func TestBuildMediaScope_FastIDUniqueness(t *testing.T) {
seen := make(map[string]bool) seen := make(map[string]bool)
for i := 0; i < 1000; i++ { for range 1000 {
scope := BuildMediaScope("test", "chat1", "") scope := BuildMediaScope("test", "chat1", "")
if seen[scope] { if seen[scope] {
t.Fatalf("duplicate scope generated: %s", scope) t.Fatalf("duplicate scope generated: %s", scope)

View file

@ -337,10 +337,7 @@ func (c *OneBotChannel) sendAPIRequest(action string, params any, timeout time.D
} }
func (c *OneBotChannel) reconnectLoop() { func (c *OneBotChannel) reconnectLoop() {
interval := time.Duration(c.config.ReconnectInterval) * time.Second interval := max(time.Duration(c.config.ReconnectInterval)*time.Second, 5*time.Second)
if interval < 5*time.Second {
interval = 5 * time.Second
}
for { for {
select { select {

View file

@ -292,8 +292,8 @@ func (c *PicoChannel) authenticate(r *http.Request) bool {
// Check Authorization header // Check Authorization header
auth := r.Header.Get("Authorization") auth := r.Header.Get("Authorization")
if strings.HasPrefix(auth, "Bearer ") { if after, ok := strings.CutPrefix(auth, "Bearer "); ok {
if strings.TrimPrefix(auth, "Bearer ") == token { if after == token {
return true return true
} }
} }

View file

@ -23,10 +23,7 @@ func SplitMessage(content string, maxLen int) []string {
var messages []string var messages []string
// Dynamic buffer: 10% of maxLen, but at least 50 chars if possible // Dynamic buffer: 10% of maxLen, but at least 50 chars if possible
codeBlockBuffer := maxLen / 10 codeBlockBuffer := max(maxLen/10, 50)
if codeBlockBuffer < 50 {
codeBlockBuffer = 50
}
if codeBlockBuffer > maxLen/2 { if codeBlockBuffer > maxLen/2 {
codeBlockBuffer = maxLen / 2 codeBlockBuffer = maxLen / 2
} }
@ -40,10 +37,7 @@ func SplitMessage(content string, maxLen int) []string {
} }
// Effective split point: maxLen minus buffer, to leave room for code blocks // Effective split point: maxLen minus buffer, to leave room for code blocks
effectiveLimit := maxLen - codeBlockBuffer effectiveLimit := max(maxLen-codeBlockBuffer, maxLen/2)
if effectiveLimit < maxLen/2 {
effectiveLimit = maxLen / 2
}
end := start + effectiveLimit end := start + effectiveLimit
@ -85,10 +79,9 @@ func SplitMessage(content string, maxLen int) []string {
// If we have a reasonable amount of content after the header, split inside // If we have a reasonable amount of content after the header, split inside
if msgEnd > headerEndIdx+20 { if msgEnd > headerEndIdx+20 {
// Find a better split point closer to maxLen // Find a better split point closer to maxLen
innerLimit := start + maxLen - 5 // Leave room for "\n```" innerLimit := min(
if innerLimit > totalLen { // Leave room for "\n```"
innerLimit = totalLen start+maxLen-5, totalLen)
}
betterEnd := findLastNewlineInRange(runes, start, innerLimit, 200) betterEnd := findLastNewlineInRange(runes, start, innerLimit, 200)
if betterEnd > headerEndIdx { if betterEnd > headerEndIdx {
msgEnd = betterEnd msgEnd = betterEnd
@ -117,10 +110,7 @@ func SplitMessage(content string, maxLen int) []string {
if unclosedIdx-start > 20 { if unclosedIdx-start > 20 {
msgEnd = unclosedIdx msgEnd = unclosedIdx
} else { } else {
splitAt := start + maxLen - 5 splitAt := min(start+maxLen-5, totalLen)
if splitAt > totalLen {
splitAt = totalLen
}
chunk := strings.TrimRight(string(runes[start:splitAt]), " \t\n\r") + "\n```" chunk := strings.TrimRight(string(runes[start:splitAt]), " \t\n\r") + "\n```"
messages = append(messages, chunk) messages = append(messages, chunk)
remaining := strings.TrimSpace(header + "\n" + string(runes[splitAt:totalLen])) remaining := strings.TrimSpace(header + "\n" + string(runes[splitAt:totalLen]))
@ -196,10 +186,7 @@ func findNewlineFrom(runes []rune, from int) int {
// findLastNewlineInRange finds the last newline within the last searchWindow runes // findLastNewlineInRange finds the last newline within the last searchWindow runes
// of the range runes[start:end]. Returns the absolute index or start-1 (indicating not found). // of the range runes[start:end]. Returns the absolute index or start-1 (indicating not found).
func findLastNewlineInRange(runes []rune, start, end, searchWindow int) int { func findLastNewlineInRange(runes []rune, start, end, searchWindow int) int {
searchStart := end - searchWindow searchStart := max(end-searchWindow, start)
if searchStart < start {
searchStart = start
}
for i := end - 1; i >= searchStart; i-- { for i := end - 1; i >= searchStart; i-- {
if runes[i] == '\n' { if runes[i] == '\n' {
return i return i
@ -211,10 +198,7 @@ func findLastNewlineInRange(runes []rune, start, end, searchWindow int) int {
// findLastSpaceInRange finds the last space/tab within the last searchWindow runes // findLastSpaceInRange finds the last space/tab within the last searchWindow runes
// of the range runes[start:end]. Returns the absolute index or start-1 (indicating not found). // of the range runes[start:end]. Returns the absolute index or start-1 (indicating not found).
func findLastSpaceInRange(runes []rune, start, end, searchWindow int) int { func findLastSpaceInRange(runes []rune, start, end, searchWindow int) int {
searchStart := end - searchWindow searchStart := max(end-searchWindow, start)
if searchStart < start {
searchStart = start
}
for i := end - 1; i >= searchStart; i-- { for i := end - 1; i >= searchStart; i-- {
if runes[i] == ' ' || runes[i] == '\t' { if runes[i] == ' ' || runes[i] == '\t' {
return i return i

View file

@ -43,7 +43,7 @@ func encryptTestMessageApp(message, aesKey string) (string, error) {
// Prepare message: random(16) + msg_len(4) + msg + corp_id // Prepare message: random(16) + msg_len(4) + msg + corp_id
random := make([]byte, 0, 16) random := make([]byte, 0, 16)
for i := 0; i < 16; i++ { for i := range 16 {
random = append(random, byte(i+1)) random = append(random, byte(i+1))
} }

View file

@ -42,7 +42,7 @@ func encryptTestMessage(message, aesKey string) (string, error) {
// Prepare message: random(16) + msg_len(4) + msg + receiveid // Prepare message: random(16) + msg_len(4) + msg + receiveid
random := make([]byte, 0, 16) random := make([]byte, 0, 16)
for i := 0; i < 16; i++ { for i := range 16 {
random = append(random, byte(i)) random = append(random, byte(i))
} }

View file

@ -125,7 +125,7 @@ func pkcs7Unpad(data []byte) ([]byte, error) {
return nil, fmt.Errorf("padding size larger than data") return nil, fmt.Errorf("padding size larger than data")
} }
// Verify all padding bytes // Verify all padding bytes
for i := 0; i < padding; i++ { for i := range padding {
if data[len(data)-1-i] != byte(padding) { if data[len(data)-1-i] != byte(padding) {
return nil, fmt.Errorf("invalid padding byte at position %d", i) return nil, fmt.Errorf("invalid padding byte at position %d", i)
} }

View file

@ -64,7 +64,7 @@ func TestGetModelConfig_RoundRobin(t *testing.T) {
// Test round-robin distribution // Test round-robin distribution
results := make(map[string]int) results := make(map[string]int)
for i := 0; i < 30; i++ { for range 30 {
result, err := cfg.GetModelConfig("lb-model") result, err := cfg.GetModelConfig("lb-model")
if err != nil { if err != nil {
t.Fatalf("GetModelConfig() error = %v", err) t.Fatalf("GetModelConfig() error = %v", err)
@ -94,17 +94,15 @@ func TestGetModelConfig_Concurrent(t *testing.T) {
var wg sync.WaitGroup var wg sync.WaitGroup
errors := make(chan error, goroutines*iterations) errors := make(chan error, goroutines*iterations)
for i := 0; i < goroutines; i++ { for range goroutines {
wg.Add(1) wg.Go(func() {
go func() { for range iterations {
defer wg.Done()
for j := 0; j < iterations; j++ {
_, err := cfg.GetModelConfig("concurrent-model") _, err := cfg.GetModelConfig("concurrent-model")
if err != nil { if err != nil {
errors <- err errors <- err
} }
} }
}() })
} }
wg.Wait() wg.Wait()

View file

@ -4,6 +4,7 @@ import (
"context" "context"
"encoding/json" "encoding/json"
"fmt" "fmt"
"maps"
"net/http" "net/http"
"sync" "sync"
"time" "time"
@ -122,9 +123,7 @@ func (s *Server) readyHandler(w http.ResponseWriter, r *http.Request) {
s.mu.RLock() s.mu.RLock()
ready := s.ready ready := s.ready
checks := make(map[string]Check) checks := make(map[string]Check)
for k, v := range s.checks { maps.Copy(checks, s.checks)
checks[k] = v
}
s.mu.RUnlock() s.mu.RUnlock()
if !ready { if !ready {

View file

@ -49,7 +49,7 @@ func TestReleaseAll(t *testing.T) {
paths := make([]string, 3) paths := make([]string, 3)
refs := make([]string, 3) refs := make([]string, 3)
for i := 0; i < 3; i++ { for i := range 3 {
paths[i] = createTempFile(t, dir, strings.Repeat("a", i+1)+".jpg") paths[i] = createTempFile(t, dir, strings.Repeat("a", i+1)+".jpg")
var err error var err error
refs[i], err = store.Store(paths[i], MediaMeta{Source: "test"}, "scope1") refs[i], err = store.Store(paths[i], MediaMeta{Source: "test"}, "scope1")
@ -228,12 +228,12 @@ func TestConcurrentSafety(t *testing.T) {
var wg sync.WaitGroup var wg sync.WaitGroup
wg.Add(goroutines) wg.Add(goroutines)
for g := 0; g < goroutines; g++ { for g := range goroutines {
go func(gIdx int) { go func(gIdx int) {
defer wg.Done() defer wg.Done()
scope := strings.Repeat("s", gIdx+1) scope := strings.Repeat("s", gIdx+1)
for i := 0; i < filesPerGoroutine; i++ { for i := range filesPerGoroutine {
path := createTempFile(t, dir, strings.Repeat("f", gIdx*filesPerGoroutine+i+1)+".tmp") path := createTempFile(t, dir, strings.Repeat("f", gIdx*filesPerGoroutine+i+1)+".tmp")
ref, err := store.Store(path, MediaMeta{Source: "test"}, scope) ref, err := store.Store(path, MediaMeta{Source: "test"}, scope)
if err != nil { if err != nil {
@ -448,11 +448,11 @@ func TestConcurrentCleanupSafety(t *testing.T) {
wg.Add(workers * 4) wg.Add(workers * 4)
// Store workers // Store workers
for w := 0; w < workers; w++ { for w := range workers {
go func(wIdx int) { go func(wIdx int) {
defer wg.Done() defer wg.Done()
scope := fmt.Sprintf("scope-%d", wIdx) scope := fmt.Sprintf("scope-%d", wIdx)
for i := 0; i < ops; i++ { for i := range ops {
p := createTempFile(t, dir, fmt.Sprintf("w%d-f%d.tmp", wIdx, i)) p := createTempFile(t, dir, fmt.Sprintf("w%d-f%d.tmp", wIdx, i))
store.Store(p, MediaMeta{Source: "test"}, scope) store.Store(p, MediaMeta{Source: "test"}, scope)
} }
@ -460,30 +460,30 @@ func TestConcurrentCleanupSafety(t *testing.T) {
} }
// Resolve workers // Resolve workers
for w := 0; w < workers; w++ { for range workers {
go func() { go func() {
defer wg.Done() defer wg.Done()
for i := 0; i < ops; i++ { for range ops {
store.Resolve("media://nonexistent") store.Resolve("media://nonexistent")
} }
}() }()
} }
// ReleaseAll workers // ReleaseAll workers
for w := 0; w < workers; w++ { for w := range workers {
go func(wIdx int) { go func(wIdx int) {
defer wg.Done() defer wg.Done()
for i := 0; i < ops; i++ { for range ops {
store.ReleaseAll(fmt.Sprintf("scope-%d", wIdx)) store.ReleaseAll(fmt.Sprintf("scope-%d", wIdx))
} }
}(w) }(w)
} }
// CleanExpired workers // CleanExpired workers
for w := 0; w < workers; w++ { for range workers {
go func() { go func() {
defer wg.Done() defer wg.Done()
for i := 0; i < ops; i++ { for range ops {
store.CleanExpired() store.CleanExpired()
} }
}() }()

View file

@ -212,14 +212,14 @@ func translateTools(tools []ToolDefinition) []anthropic.ToolUnionParam {
} }
func parseResponse(resp *anthropic.Message) *LLMResponse { func parseResponse(resp *anthropic.Message) *LLMResponse {
var content string var content strings.Builder
var toolCalls []ToolCall var toolCalls []ToolCall
for _, block := range resp.Content { for _, block := range resp.Content {
switch block.Type { switch block.Type {
case "text": case "text":
tb := block.AsText() tb := block.AsText()
content += tb.Text content.WriteString(tb.Text)
case "tool_use": case "tool_use":
tu := block.AsToolUse() tu := block.AsToolUse()
var args map[string]any var args map[string]any
@ -246,7 +246,7 @@ func parseResponse(resp *anthropic.Message) *LLMResponse {
} }
return &LLMResponse{ return &LLMResponse{
Content: content, Content: content.String(),
ToolCalls: toolCalls, ToolCalls: toolCalls,
FinishReason: finishReason, FinishReason: finishReason,
Usage: &UsageInfo{ Usage: &UsageInfo{
@ -264,8 +264,8 @@ func normalizeBaseURL(apiBase string) string {
} }
base = strings.TrimRight(base, "/") base = strings.TrimRight(base, "/")
if strings.HasSuffix(base, "/v1") { if before, ok := strings.CutSuffix(base, "/v1"); ok {
base = strings.TrimSuffix(base, "/v1") base = before
} }
if base == "" { if base == "" {
return defaultBaseURL return defaultBaseURL

View file

@ -163,8 +163,8 @@ func resolveCodexModel(model string) (string, string) {
return codexDefaultModel, "empty model" return codexDefaultModel, "empty model"
} }
if strings.HasPrefix(m, "openai/") { if after, ok := strings.CutPrefix(m, "openai/"); ok {
m = strings.TrimPrefix(m, "openai/") m = after
} else if strings.Contains(m, "/") { } else if strings.Contains(m, "/") {
return codexDefaultModel, "non-openai model namespace" return codexDefaultModel, "non-openai model namespace"
} }

View file

@ -138,7 +138,7 @@ func TestCooldown_FailureWindowReset(t *testing.T) {
ct, current := newTestTracker(now) ct, current := newTestTracker(now)
// 4 errors → 1h cooldown // 4 errors → 1h cooldown
for i := 0; i < 4; i++ { for range 4 {
ct.MarkFailure("openai", FailoverRateLimit) ct.MarkFailure("openai", FailoverRateLimit)
*current = current.Add(2 * time.Second) // small advance between errors *current = current.Add(2 * time.Second) // small advance between errors
} }
@ -230,7 +230,7 @@ func TestCooldown_ConcurrentAccess(t *testing.T) {
ct := NewCooldownTracker() ct := NewCooldownTracker()
var wg sync.WaitGroup var wg sync.WaitGroup
for i := 0; i < 100; i++ { for range 100 {
wg.Add(3) wg.Add(3)
go func() { go func() {
defer wg.Done() defer wg.Done()

View file

@ -312,8 +312,8 @@ func stripSystemParts(messages []Message) []openaiMessage {
} }
func normalizeModel(model, apiBase string) string { func normalizeModel(model, apiBase string) string {
idx := strings.Index(model, "/") before, after, ok := strings.Cut(model, "/")
if idx == -1 { if !ok {
return model return model
} }
@ -321,10 +321,10 @@ func normalizeModel(model, apiBase string) string {
return model return model
} }
prefix := strings.ToLower(model[:idx]) prefix := strings.ToLower(before)
switch prefix { switch prefix {
case "moonshot", "nvidia", "groq", "ollama", "deepseek", "google", "openrouter", "zhipu", "mistral": case "moonshot", "nvidia", "groq", "ollama", "deepseek", "google", "openrouter", "zhipu", "mistral":
return model[idx+1:] return after
default: default:
return model return model
} }

View file

@ -1,6 +1,9 @@
package routing package routing
import "testing" import (
"strings"
"testing"
)
func TestNormalizeAgentID_Empty(t *testing.T) { func TestNormalizeAgentID_Empty(t *testing.T) {
if got := NormalizeAgentID(""); got != DefaultAgentID { if got := NormalizeAgentID(""); got != DefaultAgentID {
@ -57,11 +60,11 @@ func TestNormalizeAgentID_AllInvalid(t *testing.T) {
} }
func TestNormalizeAgentID_TruncatesAt64(t *testing.T) { func TestNormalizeAgentID_TruncatesAt64(t *testing.T) {
long := "" var long strings.Builder
for i := 0; i < 100; i++ { for range 100 {
long += "a" long.WriteString("a")
} }
got := NormalizeAgentID(long) got := NormalizeAgentID(long.String())
if len(got) > MaxAgentIDLength { if len(got) > MaxAgentIDLength {
t.Errorf("length = %d, want <= %d", len(got), MaxAgentIDLength) t.Errorf("length = %d, want <= %d", len(got), MaxAgentIDLength)
} }

View file

@ -240,7 +240,7 @@ func (sl *SkillsLoader) parseSimpleYAML(content string) map[string]string {
normalized := strings.ReplaceAll(content, "\r\n", "\n") normalized := strings.ReplaceAll(content, "\r\n", "\n")
normalized = strings.ReplaceAll(normalized, "\r", "\n") normalized = strings.ReplaceAll(normalized, "\r", "\n")
for _, line := range strings.Split(normalized, "\n") { for line := range strings.SplitSeq(normalized, "\n") {
line = strings.TrimSpace(line) line = strings.TrimSpace(line)
if line == "" || strings.HasPrefix(line, "#") { if line == "" || strings.HasPrefix(line, "#") {
continue continue

View file

@ -1,7 +1,7 @@
package skills package skills
import ( import (
"sort" "slices"
"strings" "strings"
"sync" "sync"
"time" "time"
@ -183,7 +183,7 @@ func buildTrigrams(s string) []uint32 {
} }
// Sort and Deduplication // Sort and Deduplication
sort.Slice(trigrams, func(i, j int) bool { return trigrams[i] < trigrams[j] }) slices.Sort(trigrams)
n := 1 n := 1
for i := 1; i < len(trigrams); i++ { for i := 1; i < len(trigrams); i++ {
if trigrams[i] != trigrams[i-1] { if trigrams[i] != trigrams[i-1] {

View file

@ -153,7 +153,7 @@ func TestSearchCacheConcurrency(t *testing.T) {
// Concurrent writes // Concurrent writes
go func() { go func() {
for i := 0; i < 100; i++ { for i := range 100 {
cache.Put("query-write-"+string(rune('a'+i%26)), []SearchResult{{Slug: "x"}}) cache.Put("query-write-"+string(rune('a'+i%26)), []SearchResult{{Slug: "x"}})
} }
done <- struct{}{} done <- struct{}{}
@ -161,7 +161,7 @@ func TestSearchCacheConcurrency(t *testing.T) {
// Concurrent reads // Concurrent reads
go func() { go func() {
for i := 0; i < 100; i++ { for range 100 {
cache.Get("query-write-a") cache.Get("query-write-a")
} }
done <- struct{}{} done <- struct{}{}

View file

@ -135,7 +135,7 @@ func TestConcurrentAccess(t *testing.T) {
// Test concurrent writes // Test concurrent writes
done := make(chan bool, 10) done := make(chan bool, 10)
for i := 0; i < 10; i++ { for i := range 10 {
go func(idx int) { go func(idx int) {
channel := fmt.Sprintf("channel-%d", idx) channel := fmt.Sprintf("channel-%d", idx)
sm.SetLastChannel(channel) sm.SetLastChannel(channel)
@ -144,7 +144,7 @@ func TestConcurrentAccess(t *testing.T) {
} }
// Wait for all goroutines to complete // Wait for all goroutines to complete
for i := 0; i < 10; i++ { for range 10 {
<-done <-done
} }

View file

@ -3,6 +3,7 @@ package tools
import ( import (
"context" "context"
"fmt" "fmt"
"strings"
"sync" "sync"
"time" "time"
@ -222,7 +223,8 @@ func (t *CronTool) listJobs() *ToolResult {
return SilentResult("No scheduled jobs") return SilentResult("No scheduled jobs")
} }
result := "Scheduled jobs:\n" var result strings.Builder
result.WriteString("Scheduled jobs:\n")
for _, j := range jobs { for _, j := range jobs {
var scheduleInfo string var scheduleInfo string
if j.Schedule.Kind == "every" && j.Schedule.EveryMS != nil { if j.Schedule.Kind == "every" && j.Schedule.EveryMS != nil {
@ -234,10 +236,10 @@ func (t *CronTool) listJobs() *ToolResult {
} else { } else {
scheduleInfo = "unknown" scheduleInfo = "unknown"
} }
result += fmt.Sprintf("- %s (id: %s, %s)\n", j.Name, j.ID, scheduleInfo) result.WriteString(fmt.Sprintf("- %s (id: %s, %s)\n", j.Name, j.ID, scheduleInfo))
} }
return SilentResult(result) return SilentResult(result.String())
} }
func (t *CronTool) removeJob(args map[string]any) *ToolResult { func (t *CronTool) removeJob(args map[string]any) *ToolResult {

View file

@ -329,7 +329,7 @@ func TestToolRegistry_ConcurrentAccess(t *testing.T) {
r := NewToolRegistry() r := NewToolRegistry()
var wg sync.WaitGroup var wg sync.WaitGroup
for i := 0; i < 50; i++ { for i := range 50 {
wg.Add(1) wg.Add(1)
go func(n int) { go func(n int) {
defer wg.Done() defer wg.Done()

View file

@ -284,7 +284,7 @@ func (p *DuckDuckGoSearchProvider) extractResults(html string, count int, query
maxItems := min(len(matches), count) maxItems := min(len(matches), count)
for i := 0; i < maxItems; i++ { for i := range maxItems {
urlStr := matches[i][1] urlStr := matches[i][1]
title := stripTags(matches[i][2]) title := stripTags(matches[i][2])
title = strings.TrimSpace(title) title = strings.TrimSpace(title)
@ -292,9 +292,9 @@ func (p *DuckDuckGoSearchProvider) extractResults(html string, count int, query
// URL decoding if needed // URL decoding if needed
if strings.Contains(urlStr, "uddg=") { if strings.Contains(urlStr, "uddg=") {
if u, err := url.QueryUnescape(urlStr); err == nil { if u, err := url.QueryUnescape(urlStr); err == nil {
idx := strings.Index(u, "uddg=") _, after, ok := strings.Cut(u, "uddg=")
if idx != -1 { if ok {
urlStr = u[idx+5:] urlStr = after
} }
} }
} }