mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-17 16:56:38 +08:00
feat: add env var to set log level
This commit is contained in:
parent
e7ac26ff5a
commit
9eafd3e6ca
@ -57,7 +57,7 @@ func (h *ConfigHandler) Update(c *gin.Context) {
|
|||||||
resp.ERROR(c, "Failed to update config cache: "+err.Error())
|
resp.ERROR(c, "Failed to update config cache: "+err.Error())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
logger.Infof("Update AppServer's config successfully: %v", config.Config)
|
logger.Debugf("Update AppServer's config successfully: %v", config.Config)
|
||||||
}
|
}
|
||||||
|
|
||||||
resp.SUCCESS(c, config)
|
resp.SUCCESS(c, config)
|
||||||
|
@ -205,9 +205,7 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if h.App.Debug { // 调试打印聊天上下文
|
logger.Debugf("聊天上下文:%+v", chatCtx)
|
||||||
logger.Info("聊天上下文:", chatCtx)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
reqMgs := make([]interface{}, 0)
|
reqMgs := make([]interface{}, 0)
|
||||||
for _, m := range chatCtx {
|
for _, m := range chatCtx {
|
||||||
@ -312,7 +310,6 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
|
|||||||
logger.Info(functionName)
|
logger.Info(functionName)
|
||||||
logger.Info(arguments)
|
logger.Info(arguments)
|
||||||
f := h.App.Functions[functionName]
|
f := h.App.Functions[functionName]
|
||||||
// TODO 调用函数完成任务
|
|
||||||
data, err := f.Invoke(arguments)
|
data, err := f.Invoke(arguments)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
msg := "调用函数出错:" + err.Error()
|
msg := "调用函数出错:" + err.Error()
|
||||||
|
@ -5,6 +5,7 @@ import (
|
|||||||
"go.uber.org/zap/zapcore"
|
"go.uber.org/zap/zapcore"
|
||||||
"gopkg.in/natefinch/lumberjack.v2"
|
"gopkg.in/natefinch/lumberjack.v2"
|
||||||
"os"
|
"os"
|
||||||
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
var logger *zap.Logger
|
var logger *zap.Logger
|
||||||
@ -15,7 +16,7 @@ func GetLogger() *zap.SugaredLogger {
|
|||||||
return sugarLogger
|
return sugarLogger
|
||||||
}
|
}
|
||||||
|
|
||||||
logLevel := zap.NewAtomicLevelAt(zapcore.InfoLevel)
|
logLevel := zap.NewAtomicLevelAt(getLogLevel(os.Getenv("LOG_LEVEL")))
|
||||||
encoder := getEncoder()
|
encoder := getEncoder()
|
||||||
writerSyncer := getLogWriter()
|
writerSyncer := getLogWriter()
|
||||||
fileCore := zapcore.NewCore(encoder, writerSyncer, logLevel)
|
fileCore := zapcore.NewCore(encoder, writerSyncer, logLevel)
|
||||||
@ -58,3 +59,16 @@ func getLogWriter() zapcore.WriteSyncer {
|
|||||||
}
|
}
|
||||||
return zapcore.AddSync(lumberJackLogger)
|
return zapcore.AddSync(lumberJackLogger)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func getLogLevel(level string) zapcore.Level {
|
||||||
|
switch strings.ToUpper(level) {
|
||||||
|
case "DEBUG":
|
||||||
|
return zapcore.DebugLevel
|
||||||
|
case "WARN":
|
||||||
|
return zapcore.WarnLevel
|
||||||
|
case "ERROR":
|
||||||
|
return zapcore.ErrorLevel
|
||||||
|
default:
|
||||||
|
return zapcore.InfoLevel
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user