diff --git a/README.md b/README.md index 68ef80b..7f56e4c 100644 --- a/README.md +++ b/README.md @@ -72,8 +72,6 @@ xiaomoinfo/wechatgpt:latest `警告:以下命令会让任何消息都会被机器人接管,微信主号不要用下面这个命令` ``` - -# linux amd64 docker run -d \ --name wechatgpt \ -e api_key="你的chatgpt api_key" \ @@ -85,7 +83,6 @@ xiaomoinfo/wechatgpt:latest 运行`telegram`智能机器人的话运行下面这段代码 ``` -# linux amd64 docker run -d \ --name wechatgpt \ -e api_key="你的chatgpt api_key" \ @@ -97,7 +94,6 @@ xiaomoinfo/wechatgpt:latest 如果运行`telegram`智能机器人时只希望指定的人使用,白名单以外的人发消息机器人不会回复 ``` -# linux amd64 docker run -d \ --name wechatgpt \ -e api_key="你的chatgpt api_key" \ @@ -110,7 +106,6 @@ xiaomoinfo/wechatgpt:latest 如果运行`telegram`智能机器人时希望在群里回复别人消息,可以指定一个关键字触发 ``` -# linux amd64 docker run -d \ --name wechatgpt \ -e api_key="你的chatgpt api_key" \ diff --git a/Taskfile.yaml b/Taskfile.yaml index 599b034..3a6384c 100644 --- a/Taskfile.yaml +++ b/Taskfile.yaml @@ -13,13 +13,4 @@ tasks: release: cmds: - docker build --platform linux/amd64 -t xiaomoinfo/wechatgpt-amd64:latest . - - docker push xiaomoinfo/wechatgpt-amd64:latest - - docker build -t xiaomoinfo/wechatgpt:latest . - - docker push xiaomoinfo/wechatgpt:latest - - version: - cmds: - - docker build --platform linux/amd64 -t xiaomoinfo/wechatgpt-amd64:2.9.2 . - - docker push xiaomoinfo/wechatgpt-amd64:2.9.2 - - docker build -t xiaomoinfo/wechatgpt:2.9.2 . - - docker push xiaomoinfo/wechatgpt:2.9.2 \ No newline at end of file + - docker push xiaomoinfo/wechatgpt-amd64:latest \ No newline at end of file diff --git a/config/config.go b/config/config.go index a002589..836f214 100644 --- a/config/config.go +++ b/config/config.go @@ -1,6 +1,7 @@ package config import ( + "fmt" "github.com/spf13/viper" "os" "strings" @@ -38,6 +39,10 @@ func LoadConfig() error { func GetWechat() *string { wechat := getEnv("wechat") + + if wechat != nil { + return wechat + } if config == nil { return nil } @@ -49,6 +54,10 @@ func GetWechat() *string { func GetWechatKeyword() *string { keyword := getEnv("wechat_keyword") + + if keyword != nil { + return keyword + } if config == nil { return nil } @@ -60,6 +69,10 @@ func GetWechatKeyword() *string { func GetTelegram() *string { tg := getEnv("telegram") + fmt.Println(tg) + if tg != nil { + return tg + } if config == nil { return nil } @@ -71,6 +84,10 @@ func GetTelegram() *string { func GetTelegramKeyword() *string { tgKeyword := getEnv("tg_keyword") + + if tgKeyword != nil { + return tgKeyword + } if config == nil { return nil } @@ -82,6 +99,10 @@ func GetTelegramKeyword() *string { func GetTelegramWhitelist() *string { tgWhitelist := getEnv("tg_whitelist") + + if tgWhitelist != nil { + return tgWhitelist + } if config == nil { return nil } @@ -93,6 +114,11 @@ func GetTelegramWhitelist() *string { func GetOpenAiApiKey() *string { apiKey := getEnv("api_key") + + if apiKey != nil { + return apiKey + } + if config == nil { return nil } @@ -104,11 +130,14 @@ func GetOpenAiApiKey() *string { func getEnv(key string) *string { value := os.Getenv(key) - if len(value) == 0 { value = os.Getenv(strings.ToUpper(key)) } + if len(value) > 0 { + return &value + } + if config == nil { return nil }