internal/llm/logging.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

package llm

import (
    "fmt"
    "log"
)

// ANSI color utilities shared across LLM providers.
const (
    AnsiBgBlack = "\x1b[40m"
    AnsiGrey    = "\x1b[90m"
    AnsiCyan    = "\x1b[36m"
    AnsiGreen   = "\x1b[32m"
    AnsiRed     = "\x1b[31m"
    AnsiReset   = "\x1b[0m"
)

// AnsiBase is the default style: black background + grey foreground.
const AnsiBase = AnsiBgBlack + AnsiGrey

// LogPrintf wraps a formatted message with a base style and prints with a prefix.
func LogPrintf(logger *log.Logger, prefix, format string, args ...any) {
    if logger == nil {
        return
    }
    msg := fmt.Sprintf(format, args...)
    logger.Print(AnsiBase + prefix + msg + AnsiReset)
}

// Logging configuration for previews (shared)
var logPreviewLimit int // 0 means unlimited

// SetLogPreviewLimit sets the maximum number of characters to log for
// request/response previews. Set to 0 for unlimited.
func SetLogPreviewLimit(n int) { logPreviewLimit = n }

func previewForLog(s string) string {
    if logPreviewLimit > 0 {
        return trimPreview(s, logPreviewLimit)
    }
    return s
}