Eli Ribble
a68b8781e7
This is extremely useful for testing. In order to do this I needed to actually deploy the migration to a bob fork so I could start to add support for behaviors I really want. Specifically the ability to search for ids in a slice.
35 lines
636 B
Go
35 lines
636 B
Go
package llm
|
|
|
|
import (
|
|
"log"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog"
|
|
"go.mau.fi/util/exzerolog"
|
|
)
|
|
|
|
type Logger = zerolog.Logger
|
|
|
|
func linkLogger(logger *zerolog.Logger) {
|
|
exzerolog.SetupDefaults(logger)
|
|
}
|
|
|
|
type ZerologWriter struct {
|
|
zerologger zerolog.Logger
|
|
level zerolog.Level
|
|
}
|
|
|
|
func (w ZerologWriter) Write(p []byte) (n int, err error) {
|
|
msg := strings.TrimSuffix(string(p), "\n")
|
|
event := w.zerologger.WithLevel(w.level)
|
|
event.Msg(msg)
|
|
return len(p), nil
|
|
}
|
|
|
|
func LoggerShim(l zerolog.Logger) *log.Logger {
|
|
writer := &ZerologWriter{
|
|
zerologger: l,
|
|
level: zerolog.DebugLevel,
|
|
}
|
|
return log.New(writer, "", 0)
|
|
}
|