Fix wechatbot

This commit is contained in:
linghaihui 2023-03-23 11:41:42 +08:00
parent c70ea53d93
commit f93d65abe6
5 changed files with 14 additions and 6 deletions

View File

@ -6,4 +6,4 @@ RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o chatgpt
FROM alpine:3.17.2 FROM alpine:3.17.2
WORKDIR /app WORKDIR /app
COPY --from=builder /app/chatgpt . COPY --from=builder /app/chatgpt .
ENTRYPOINT ["./chatgpt", ">>", "chatgpt.log", "2>&1"] ENTRYPOINT ["./chatgpt"]

View File

@ -56,7 +56,7 @@ func BingSearch(msg string, nickName string) (string, error) {
return "非常抱歉😭,网络异常,请稍后重试", err return "非常抱歉😭,网络异常,请稍后重试", err
} }
if response.StatusCode != 200 { if response.StatusCode != 200 {
return "非常抱歉😭,网络异常,请稍后重试 [" + string(rune(response.StatusCode)) + "]", nil return "非常抱歉😭,网络异常,请稍后重试 [" + string(response.StatusCode) + "]", nil
} }
defer response.Body.Close() defer response.Body.Close()

View File

@ -49,9 +49,9 @@ func Completions(msg string, nickName string) (string, error) {
messageCache = []Message{message} messageCache = []Message{message}
} else { } else {
messageCache = append(messageCache, message) messageCache = append(messageCache, message)
// 只保留10条 // 只保留20条
if len(messageCache) > 10 { if len(messageCache) > 20 {
messageCache = messageCache[(len(messageCache) - 10):] messageCache = messageCache[(len(messageCache) - 20):]
} }
} }
MessageCacheRegistry[nickName] = messageCache MessageCacheRegistry[nickName] = messageCache

View File

@ -23,6 +23,9 @@ func (g *GroupMessageHandler) handle(msg *openwechat.Message) error {
msg.ReplyText("目前我只支持文字哦~") msg.ReplyText("目前我只支持文字哦~")
} }
} }
if msg.IsPaiYiPai() {
msg.ReplyText("我是机器人🤖️,会拍坏的哦~")
}
return nil return nil
} }

View File

@ -19,7 +19,12 @@ func (g *UserMessageHandler) handle(msg *openwechat.Message) error {
if msg.IsText() { if msg.IsText() {
return g.ReplyText(msg) return g.ReplyText(msg)
} }
if msg.IsSendByFriend() {
msg.ReplyText("目前我只支持文字哦~") msg.ReplyText("目前我只支持文字哦~")
}
if msg.IsPaiYiPai() {
msg.ReplyText("我是机器人🤖️,会拍坏的哦~")
}
return nil return nil
} }