diff --git a/Dockerfile b/Dockerfile index 8e5b32f..c8ff795 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,6 +2,7 @@ FROM golang:1.19-alpine ENV apiKey="" ENV telegram="" +ENV wechat="" RUN export GOPRIVATE=github.com/houko/wechatgpt diff --git a/handler/wechat/wechat_handler.go b/handler/wechat/wechat_handler.go index 8eceabb..4b42cdd 100644 --- a/handler/wechat/wechat_handler.go +++ b/handler/wechat/wechat_handler.go @@ -6,6 +6,7 @@ import ( "github.com/wechatgpt/wechatbot/config" "github.com/wechatgpt/wechatbot/openai" "log" + "os" "strings" ) @@ -31,10 +32,12 @@ func (gmh *GroupMessageHandler) ReplyText(msg *openwechat.Message) error { group := openwechat.Group{User: sender} log.Printf("Received Group %v Text Msg : %v", group.NickName, msg.Content) - keyword := "chatgpt" - appConfig := config.GetConfig() - if appConfig != nil { - keyword = appConfig.ChatGpt.Keyword + keyword := os.Getenv("wechat") + if len(keyword) == 0 { + appConfig := config.GetConfig() + if appConfig != nil { + keyword = appConfig.ChatGpt.Keyword + } } if !strings.Contains(msg.Content, keyword) {