OSSForks/ollama
1
0
Fork 0
mirror of https://github.com/zebrajr/ollama.git synced 2025-12-06 00:19:51 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

logutil: fix source field (#12279)

Browse Source
This commit is contained in:
Michael Yang 2025-09-16 16:18:07 -07:00 committed by GitHub
parent fa1c987a29
commit b225508c9b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
logutil/logutil.go
View File

@ -5,6 +5,8 @@ import (
"io" "io"
"log/slog" "log/slog"
"path/filepath" "path/filepath"
"runtime"
"time"
) )
const LevelTrace slog.Level = -8 const LevelTrace slog.Level = -8
@ -29,10 +31,18 @@ func NewLogger(w io.Writer, level slog.Level) *slog.Logger {
})) }))
} }
type key string
func Trace(msg string, args ...any) { func Trace(msg string, args ...any) {
slog.Log(context.TODO(), LevelTrace, msg, args...) TraceContext(context.WithValue(context.TODO(), key("skip"), 1), msg, args...)
} }
func TraceContext(ctx context.Context, msg string, args ...any) { func TraceContext(ctx context.Context, msg string, args ...any) {
slog.Log(ctx, LevelTrace, msg, args...) if logger := slog.Default(); logger.Enabled(ctx, LevelTrace) {
skip, _ := ctx.Value(key("skip")).(int)
pc, _, _, _ := runtime.Caller(1 + skip)
record := slog.NewRecord(time.Now(), LevelTrace, msg, pc)
record.Add(args...)
logger.Handler().Handle(ctx, record)
}
} }