init v1.0.0

ageer

2024-02-27 a079ef44e53acd9e8df51dbb31cf5aea4f9be5bd

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package com.xmzs.system.handler;
 
import com.xmzs.common.chat.domain.request.ChatRequest;
import com.xmzs.common.chat.domain.request.Dall3Request;
import com.xmzs.common.chat.entity.chat.ChatCompletion;
import com.xmzs.common.chat.entity.images.Item;
import com.xmzs.common.wechat.api.MessageTools;
import com.xmzs.common.wechat.beans.BaseMsg;
import com.xmzs.common.wechat.core.Core;
import com.xmzs.common.wechat.face.IMsgHandlerFace;
import com.xmzs.system.cofing.KeywordConfig;
import com.xmzs.system.service.ISseService;
 
import java.util.List;
 
/**
 * 微信消息处理程序
 *
 * @author ashinnotfound
 * @date 2023/03/19
 */
public class WechatMessageHandler implements IMsgHandlerFace {
    private final ISseService sseService;
    private final KeywordConfig keywordConfig;
 
    public WechatMessageHandler(ISseService sseService, KeywordConfig keywordConfig) {
        this.sseService = sseService;
        this.keywordConfig = keywordConfig;
    }
 
    @Override
    public String textMsgHandle(BaseMsg baseMsg) {
        //如果是在群聊
        if (baseMsg.isGroupMsg()){
            //存在@机器人的消息就向ChatGPT提问
            if (baseMsg.getText().contains("@"+ Core.getInstance().getNickName())){
                //去除@再提问
                String prompt = baseMsg.getText().replace("@"+ Core.getInstance().getNickName() + " ", "").trim();
                return textResponse(baseMsg.getFromUserName(), prompt);
            }
        }else {
            ChatRequest chatBO = new ChatRequest();
            chatBO.setPrompt(baseMsg.getText());
            chatBO.setModel(ChatCompletion.Model.GPT_3_5_TURBO.getName());
            return sseService.chat(chatBO);
        }
        return null;
    }
 
 
    private String textResponse(String userName, String content) {
        if (keywordConfig.getReset().equals(content)){
            return "重置会话成功";
        }else {
            ChatRequest chatBO = new ChatRequest();
            chatBO.setPrompt(content);
            chatBO.setUserId(userName);
            if (content.startsWith(keywordConfig.getImage())) {
                Dall3Request dall3Request = new Dall3Request();
                dall3Request.setPrompt(content.replaceFirst(keywordConfig.getImage() + " ", ""));
                List<Item> items = sseService.dall3(dall3Request);
                MessageTools.sendPicMsgByUserId(userName, items.get(0).getUrl());
            } else {
                chatBO.setPrompt(content);
            }
        }
        return "这个问题我还没学会呢";
    }
 
    @Override
    public String picMsgHandle(BaseMsg baseMsg) {
        return null;
    }
 
    @Override
    public String voiceMsgHandle(BaseMsg baseMsg) {
        return null;
    }
 
    @Override
    public String viedoMsgHandle(BaseMsg baseMsg) {
        return null;
    }
 
    @Override
    public String nameCardMsgHandle(BaseMsg baseMsg) {
        return null;
    }
 
    @Override
    public void sysMsgHandle(BaseMsg baseMsg) {
    }
 
    @Override
    public String verifyAddFriendMsgHandle(BaseMsg baseMsg) {
        return null;
    }
 
    @Override
    public String mediaMsgHandle(BaseMsg baseMsg) {
        return null;
    }
}