功能: 支持设置log level

logs.go

@@ -12,6 +12,7 @@ import (
 
 type Logs struct {
 	Dir string
+	Level string
 
 	//
 	accessLogFile *os.File
@@ -57,5 +58,7 @@ func (l *Logs) Setup() (err error) {
 		}),
 	})
 
+	logger.SetLevel(l.Level)
+
 	return nil
 }

main.go

@@ -98,6 +98,12 @@ func main() {
 				EnvVars: []string{"LOGS_DIR"},
 				Value:   "/tmp/chatgpt-for-chatbot-feishu",
 			},
+			&cli.StringFlag{
+				Name:    "logs-level",
+				Usage:   "The logs level",
+				EnvVars: []string{"LOGS_LEVEL"},
+				Value:   "INFO",
+			},
 			&cli.StringFlag{
 				Name:    "offline-message",
 				Usage:   "The message to use for offline status",
@@ -145,6 +151,7 @@ func main() {
 	app.Command(func(ctx *cli.Context) (err error) {
 		return ServeFeishuBot(&FeishuBotConfig{
 			LogsDir:              ctx.String("logs-dir"),
+			LogsLevel:            ctx.String("logs-level"),
 			Port:                 ctx.Int64("port"),
 			APIPath:              ctx.String("api-path"),
 			OpenAIAPIKey:         ctx.String("openai-api-key"),

server.go

@@ -50,6 +50,7 @@ type FeishuBotConfig struct {
 	ConversationLanguage string
 	//
 	LogsDir string
+	LogsLevel string
 	//
 	OfflineMessage string
 	//
@@ -90,6 +91,7 @@ func ServeFeishuBot(cfg *FeishuBotConfig) (err error) {
 
 	logs := &Logs{
 		Dir: cfg.LogsDir,
+		Level: cfg.LogsLevel,
 	}
 	if err := logs.Setup(); err != nil {
 		return fmt.Errorf("failed to setup logs: %v", err)