diff --git a/applogic/gensokyo.go b/applogic/gensokyo.go index 4467a0a..71bbd19 100644 --- a/applogic/gensokyo.go +++ b/applogic/gensokyo.go @@ -407,7 +407,9 @@ func (app *App) GensokyoHandler(w http.ResponseWriter, r *http.Request) { } // 处理接收到的数据 - fmtf.Printf("Received SSE data: %s", string(line)) + if !config.GetHideExtraLogs() { + fmtf.Printf("Received SSE data: %s", string(line)) + } // 去除"data: "前缀后进行JSON解析 jsonData := strings.TrimPrefix(string(line), "data: ") @@ -485,7 +487,9 @@ func (app *App) GensokyoHandler(w http.ResponseWriter, r *http.Request) { } } else { //发送信息 - fmtf.Printf("收到流数据,切割并发送信息: %s", string(line)) + if !config.GetHideExtraLogs() { + fmtf.Printf("收到流数据,切割并发送信息: %s", string(line)) + } splitAndSendMessages(message, string(line), newmsg) } } diff --git a/config/config.go b/config/config.go index 356bea8..b88bbb1 100644 --- a/config/config.go +++ b/config/config.go @@ -110,6 +110,7 @@ type Settings struct { RwkvSystemName string `yaml:"rwkvSystemName"` RwkvPreSystem bool `yaml:"rwkvPreSystem"` RwkvSseType int `yaml:"rwkvSseType"` + HideExtraLogs bool `yaml:"hideExtraLogs"` } // LoadConfig 从文件中加载配置并初始化单例配置 @@ -1130,3 +1131,13 @@ func GetRwkvPreSystem() bool { } return false } + +// 获取隐藏日志 +func GetHideExtraLogs() bool { + mu.Lock() + defer mu.Unlock() + if instance != nil { + return instance.Settings.HideExtraLogs + } + return false +} diff --git a/template/config_template.go b/template/config_template.go index b47ff46..0dfe20e 100644 --- a/template/config_template.go +++ b/template/config_template.go @@ -35,7 +35,8 @@ settings: promptkeyboard : [""] #临时的promptkeyboard超过3个则随机,后期会增加一个ai生成的方式,也会是ai-agent savelogs : false #本地落地日志. noContext : false #不开启上下文 - withdrawCommand : ["撤回"] #撤回指令 + withdrawCommand : ["撤回"] #撤回指令 + hideExtraLogs : false #忽略流信息的log,提高性能 functionMode : false #是否指定本agent使用func模式(目前仅支持千帆平台),效果不好,暂时不用. functionPath : "" #调用另一个启用了func模式的gsk-llm联合工作的/conversation地址,效果不好,暂时不用.