diff --git a/mallchat-common/src/main/java/com/abin/mallchat/common/common/constant/RedisKey.java b/mallchat-common/src/main/java/com/abin/mallchat/common/common/constant/RedisKey.java index faa6bd2..0cc9bd0 100644 --- a/mallchat-common/src/main/java/com/abin/mallchat/common/common/constant/RedisKey.java +++ b/mallchat-common/src/main/java/com/abin/mallchat/common/common/constant/RedisKey.java @@ -36,6 +36,16 @@ public class RedisKey { */ public static final String USER_SUMMARY_STRING = "userSummary:uid_%d"; + /** + * 用户GPT聊天次数 + */ + public static final String USER_CHAT_NUM = "useChatGPTNum:uid_%d"; + + /** + * 用户上次使用GLM使用时间 + */ + public static final String USER_GLM2_TIME_LAST = "userGLM2UseTime:uid_%d"; + public static String getKey(String key, Object... objects) { return BASE_KEY + String.format(key, objects); } diff --git a/mallchat-common/src/main/java/com/abin/mallchat/common/common/utils/DateUtils.java b/mallchat-common/src/main/java/com/abin/mallchat/common/common/utils/DateUtils.java new file mode 100644 index 0000000..ecc2a59 --- /dev/null +++ b/mallchat-common/src/main/java/com/abin/mallchat/common/common/utils/DateUtils.java @@ -0,0 +1,16 @@ +package com.abin.mallchat.common.common.utils; + +import java.util.Calendar; +import java.util.Date; + +public class DateUtils extends org.apache.commons.lang3.time.DateUtils { + public static Long getEndTimeByToday() { + Calendar instance = Calendar.getInstance(); + Date now = new Date(); + instance.setTime(now); + instance.set(Calendar.HOUR_OF_DAY, 23); + instance.set(Calendar.MINUTE, 59); + instance.set(Calendar.SECOND, 59); + return instance.getTime().getTime() - now.getTime(); + } +} diff --git a/mallchat-common/src/main/resources/application.yml b/mallchat-common/src/main/resources/application.yml index 53b6b9b..6f13a61 100644 --- a/mallchat-common/src/main/resources/application.yml +++ b/mallchat-common/src/main/resources/application.yml @@ -12,7 +12,7 @@ mybatis-plus: spring: profiles: #运行的环境 - active: my-prod + active: test application: name: mallchat datasource: @@ -37,8 +37,8 @@ spring: database: 0 # 连接超时时间 timeout: 1800000 - # 设置密码 - password: ${mallchat.redis.password} +# # 设置密码 +# password: ${mallchat.redis.password} lettuce: pool: # 最大阻塞等待时间，负数表示没有限制 @@ -62,4 +62,14 @@ wx: - appId: ${mallchat.wx.appId} # 第一个公众号的appid secret: ${mallchat.wx.secret} # 公众号的appsecret token: ${mallchat.wx.token} # 接口配置里的Token值 - aesKey: ${mallchat.wx.aesKey} # 接口配置里的EncodingAESKey值 \ No newline at end of file + aesKey: ${mallchat.wx.aesKey} # 接口配置里的EncodingAESKey值 +chatai: + chatgpt: + use: true + AIUserId: 10452 + key: xxxxx + chatglm2: + use: true + url: xxxxx + minute: 3 # 每个用户每3分钟可以请求一次 + AIUserId: 10451 diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chat/domain/vo/request/ChatMessageReq.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chat/domain/vo/request/ChatMessageReq.java index a6a6095..2fd2040 100644 --- a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chat/domain/vo/request/ChatMessageReq.java +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chat/domain/vo/request/ChatMessageReq.java @@ -11,9 +11,13 @@ import javax.validation.constraints.NotNull; /** + * 聊天信息点播 * Description: 消息发送请求体 * Author: abin * Date: 2023-03-23 + * + * @author zhaoyuhang + * @date 2023/06/30 */ @Data @Builder diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/enums/ChatGPTModelEnum.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/enums/ChatGPTModelEnum.java new file mode 100644 index 0000000..6b73b0f --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/enums/ChatGPTModelEnum.java @@ -0,0 +1,92 @@ +package com.abin.mallchat.custom.chatai.enums; + +import lombok.AllArgsConstructor; +import lombok.Getter; + +import java.util.Arrays; +import java.util.Map; +import java.util.function.Function; +import java.util.stream.Collectors; + +@AllArgsConstructor +@Getter +public enum ChatGPTModelEnum { + // chat + GPT_35_TURBO("gpt-3.5-turbo", 3, 40000), + GPT_35_TURBO_0301("gpt-3.5-turbo-0301", 3, 40000), + GPT_35_TURBO_0613("gpt-3.5-turbo-0613", 3, 40000), + GPT_35_TURBO_16K("gpt-3.5-turbo-16k", 3, 40000), + GPT_35_TURBO_16K_0613("gpt-3.5-turbo-16k-0613", 3, 40000), + // text + ADA("ada", 60, 150000), + ADA_CODE_SEARCH_CODE("ada-code-search-code", 60, 150000), + ADA_CODE_SEARCH_TEXT("ada-code-search-text", 60, 150000), + ADA_SEARCH_DOCUMENT("ada-search-document", 60, 150000), + ADA_SEARCH_QUERY("ada-search-query", 60, 150000), + ADA_SIMILARITY("ada-similarity", 60, 150000), + BABBAGE("babbage", 60, 150000), + BABBAGE_CODE_SEARCH_CODE("babbage-code-search-code", 60, 150000), + BABBAGE_CODE_SEARCH_TEXT("babbage-code-search-text", 60, 150000), + BABBAGE_SEARCH_DOCUMENT("babbage-search-document", 60, 150000), + BABBAGE_SEARCH_QUERY("babbage-search-query", 60, 150000), + BABBAGE_SIMILARITY("babbage-similarity", 60, 150000), + CODE_DAVINCI_EDIT_001("code-davinci-edit-001", 20, 150000), + CODE_SEARCH_ADA_CODE_001("code-search-ada-code-001", 60, 150000), + CODE_SEARCH_ADA_TEXT_001("code-search-ada-text-001", 60, 150000), + CODE_SEARCH_BABBAGE_CODE_001("code-search-babbage-code-001", 60, 150000), + CODE_SEARCH_BABBAGE_TEXT_001("code-search-babbage-text-001", 60, 150000), + CURIE("curie", 60, 150000), + CURIE_INSTRUCT_BETA("curie-instruct-beta", 60, 150000), + CURIE_SEARCH_DOCUMENT("curie-search-document", 60, 150000), + CURIE_SEARCH_QUERY("curie-search-query", 60, 150000), + CURIE_SIMILARITY("curie-similarity", 60, 150000), + DAVINCI("davinci", 60, 150000), + DAVINCI_INSTRUCT_BETA("davinci-instruct-beta", 60, 150000), + DAVINCI_SEARCH_DOCUMENT("davinci-search-document", 60, 150000), + DAVINCI_SEARCH_QUERY("davinci-search-query", 60, 150000), + DAVINCI_SIMILARITY("davinci-similarity", 60, 150000), + TEXT_ADA_001("text-ada-001", 60, 150000), + TEXT_BABBAGE_001("text-babbage-001", 60, 150000), + TEXT_CURIE_001("text-curie-001", 60, 150000), + TEXT_DAVINCI_001("text-davinci-001", 60, 150000), + TEXT_DAVINCI_002("text-davinci-002", 60, 150000), + TEXT_DAVINCI_003("text-davinci-003", 60, 150000), + TEXT_DAVINCI_EDIT_001("text-davinci-edit-001", 20, 150000), + TEXT_EMBEDDING_ADA_002("text-embedding-ada-002", 60, 150000), + TEXT_SEARCH_ADA_DOC_001("text-search-ada-doc-001", 60, 150000), + TEXT_SEARCH_ADA_QUERY_001("text-search-ada-query-001", 60, 150000), + TEXT_SEARCH_BABBAGE_DOC_001("text-search-babbage-doc-001", 60, 150000), + TEXT_SEARCH_BABBAGE_QUERY_001("text-search-babbage-query-001", 60, 150000), + TEXT_SEARCH_CURIE_DOC_001("text-search-curie-doc-001", 60, 150000), + TEXT_SEARCH_CURIE_QUERY_001("text-search-curie-query-001", 60, 150000), + TEXT_SEARCH_DAVINCI_DOC_001("text-search-davinci-doc-001", 60, 150000), + TEXT_SEARCH_DAVINCI_QUERY_001("text-search-davinci-query-001", 60, 150000), + TEXT_SIMILARITY_ADA_001("text-similarity-ada-001", 60, 150000), + TEXT_SIMILARITY_BABBAGE_001("text-similarity-babbage-001", 60, 150000), + TEXT_SIMILARITY_CURIE_001("text-similarity-curie-001", 60, 150000), + TEXT_SIMILARITY_DAVINCI_001("text-similarity-davinci-001", 60, 150000); + + /** + * 名字 + */ + private final String name; + /** + * 每分钟请求数 + */ + private final Integer RPM; + /** + * 每分钟令牌数 + */ + private final Integer TPM; + + private static final Map cache; + + static { + cache = Arrays.stream(ChatGPTModelEnum.values()).collect(Collectors.toMap(ChatGPTModelEnum::getName, Function.identity())); + } + + public static ChatGPTModelEnum of(String name) { + return cache.get(name); + } + +} diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/AbstractChatAIHandler.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/AbstractChatAIHandler.java new file mode 100644 index 0000000..04e5a6e --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/AbstractChatAIHandler.java @@ -0,0 +1,130 @@ +package com.abin.mallchat.custom.chatai.handler; + +import cn.hutool.core.thread.NamedThreadFactory; +import com.abin.mallchat.common.chat.domain.entity.Message; +import com.abin.mallchat.common.chat.domain.enums.MessageTypeEnum; +import com.abin.mallchat.common.common.exception.BusinessException; +import com.abin.mallchat.common.common.handler.GlobalUncaughtExceptionHandler; +import com.abin.mallchat.custom.chat.domain.vo.request.ChatMessageReq; +import com.abin.mallchat.custom.chat.domain.vo.request.msg.TextMsgReq; +import com.abin.mallchat.custom.chat.service.ChatService; +import com.abin.mallchat.custom.user.domain.vo.response.user.UserInfoResp; +import com.abin.mallchat.custom.user.service.UserService; +import lombok.extern.slf4j.Slf4j; +import org.apache.commons.lang3.StringUtils; +import org.springframework.beans.factory.DisposableBean; +import org.springframework.beans.factory.InitializingBean; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Component; + +import javax.annotation.PostConstruct; +import java.util.Collections; +import java.util.concurrent.ExecutorService; +import java.util.concurrent.LinkedBlockingQueue; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.concurrent.TimeUnit; + +@Slf4j +@Component +public abstract class AbstractChatAIHandler implements DisposableBean, InitializingBean { + public static ExecutorService EXECUTOR; + + @Autowired + protected ChatService chatService; + @Autowired + protected UserService userService; + + @PostConstruct + private void init() { + ChatAIHandlerFactory.register(getChatAIUserId(), getChatAIName(), this); + } + // 获取机器人id + public abstract Long getChatAIUserId(); + // 获取机器人名称 + public abstract String getChatAIName(); + + public void chat(Message message) { + if (!supports(message)) { + return; + } + EXECUTOR.execute(() -> { + String text = doChat(message); + if (StringUtils.isNotBlank(text)) { + answerMsg(text, message.getRoomId(), message.getFromUid()); + } + }); + } + + /** + * 支持 + * + * @param message 消息 + * @return boolean true 支持 false 不支持 + */ + protected abstract boolean supports(Message message); + + /** + * 执行聊天 + * + * @param message 消息 + * @return {@link String} AI回答的内容 + */ + protected abstract String doChat(Message message); + + + protected void answerMsg(String text, Long roomId, Long uid) { + UserInfoResp userInfo = userService.getUserInfo(uid); + text = "@" + userInfo.getName() + " " + text; + if (text.length() < 450) { + save(text, roomId, uid); + }else { + int maxLen = 450; + int len = text.length(); + int count = (len + maxLen - 1) / maxLen; + + for (int i = 0; i < count; i++) { + int start = i * maxLen; + int end = Math.min(start + maxLen, len); + save(text.substring(start, end), roomId, uid); + } + } + } + + private void save(String text, Long roomId, Long uid) { + ChatMessageReq answerReq = new ChatMessageReq(); + answerReq.setRoomId(roomId); + answerReq.setMsgType(MessageTypeEnum.TEXT.getType()); + TextMsgReq textMsgReq = new TextMsgReq(); + textMsgReq.setContent(text); + textMsgReq.setAtUidList(Collections.singletonList(uid)); + answerReq.setBody(textMsgReq); + chatService.sendMsg(answerReq, getChatAIUserId()); + } + + @Override + public void afterPropertiesSet() { + EXECUTOR = new ThreadPoolExecutor( + 10, + 10, + 0L, + TimeUnit.MILLISECONDS, + new LinkedBlockingQueue<>(15), + new NamedThreadFactory("openAI-chat-gpt", + null, + false, + new GlobalUncaughtExceptionHandler()), + (r, executor) -> { + throw new BusinessException("别问的太快了，我的脑子不够用了"); + }); + } + + @Override + public void destroy() throws Exception { + EXECUTOR.shutdown(); + if (!EXECUTOR.awaitTermination(30, TimeUnit.SECONDS)) { //最多等30秒，处理不完就拉倒 + if (log.isErrorEnabled()) { + log.error("Timed out while waiting for executor [{}] to terminate", EXECUTOR); + } + } + } +} diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/ChatAIHandlerFactory.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/ChatAIHandlerFactory.java new file mode 100644 index 0000000..8d6356d --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/ChatAIHandlerFactory.java @@ -0,0 +1,42 @@ +package com.abin.mallchat.custom.chatai.handler; + +import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; +import com.baomidou.mybatisplus.core.toolkit.StringUtils; + +import java.util.List; +import java.util.Map; +import java.util.concurrent.ConcurrentHashMap; + +public class ChatAIHandlerFactory { + private static final Map CHATAI_ID_MAP = new ConcurrentHashMap<>(); + private static final Map CHATAI_NAME_MAP = new ConcurrentHashMap<>(); + + public static void register(Long aIUserId, String name, AbstractChatAIHandler chatAIHandler) { + CHATAI_ID_MAP.put(aIUserId, chatAIHandler); + CHATAI_NAME_MAP.put(name, chatAIHandler); + } + + public static AbstractChatAIHandler getChatAIHandlerById(List userIds) { + if (CollectionUtils.isEmpty(userIds)) { + return null; + } + for (Long userId : userIds) { + AbstractChatAIHandler chatAIHandler = CHATAI_ID_MAP.get(userId); + if (chatAIHandler != null) { + return chatAIHandler; + } + } + return null; + } + public static AbstractChatAIHandler getChatAIHandlerByName(String text) { + if (StringUtils.isBlank(text)) { + return null; + } + for (Map.Entry entry : CHATAI_NAME_MAP.entrySet()) { + if (text.contains("@"+entry.getKey())) { + return entry.getValue(); + } + } + return null; + } +} diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/ChatGLM2Handler.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/ChatGLM2Handler.java new file mode 100644 index 0000000..11350ca --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/ChatGLM2Handler.java @@ -0,0 +1,138 @@ +package com.abin.mallchat.custom.chatai.handler; + +import cn.hutool.http.HttpResponse; +import com.abin.mallchat.common.chat.domain.entity.Message; +import com.abin.mallchat.common.chat.domain.entity.msg.MessageExtra; +import com.abin.mallchat.common.common.constant.RedisKey; +import com.abin.mallchat.common.common.utils.RedisUtils; +import com.abin.mallchat.custom.chatai.properties.ChatGLM2Properties; +import com.abin.mallchat.custom.chatai.utils.ChatGLM2Utils; +import lombok.extern.slf4j.Slf4j; +import org.apache.commons.lang3.StringUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Component; +import org.springframework.util.CollectionUtils; + +import java.util.Arrays; +import java.util.Date; +import java.util.List; +import java.util.Random; +import java.util.concurrent.TimeUnit; + +import static com.abin.mallchat.common.common.constant.RedisKey.USER_GLM2_TIME_LAST; + +@Slf4j +@Component +public class ChatGLM2Handler extends AbstractChatAIHandler { + + private static final List ERROR_MSG = Arrays.asList( + "还摸鱼呢？你不下班我还要下班呢。。。。", + "没给钱，矿工了。。。。", + "服务器被你们玩儿坏了。。。。", + "你们这群人，我都不想理你们了。。。。", + "艾特我那是另外的价钱。。。。", + "得加钱"); + + + private static final Random RANDOM = new Random(); + + @Autowired + private ChatGLM2Properties glm2Properties; + + @Override + public Long getChatAIUserId() { + return glm2Properties.getAIUserId(); + } + + @Override + public String getChatAIName() { + if (StringUtils.isNotBlank(glm2Properties.getAIUserName())) { + return glm2Properties.getAIUserName(); + } + String name = userService.getUserInfo(glm2Properties.getAIUserId()).getName(); + glm2Properties.setAIUserName(name); + return name; + } + + @Override + protected String doChat(Message message) { + String content = message.getContent().replace("@" +glm2Properties.getAIUserName(), "").trim(); + Long uid = message.getFromUid(); + Long minute; + String text; + if ((minute = userMinutesLater(uid)) > 0) { + text = "你太快了 " + minute + "分钟后重试"; + } else { + HttpResponse response = null; + try { + response = ChatGLM2Utils + .create() + .url(glm2Properties.getUrl()) + .prompt(content) + .timeout(glm2Properties.getTimeout()) + .send(); + text = ChatGLM2Utils.parseText(response); + } catch (Exception e) { + e.printStackTrace(); + return getErrorText(); + } + if (StringUtils.isNotBlank(text)) { + RedisUtils.set(RedisKey.getKey(USER_GLM2_TIME_LAST, uid), new Date(), glm2Properties.getMinute(), TimeUnit.MINUTES); + } + } + return text; + } + + private static String getErrorText() { + int index = RANDOM.nextInt(ERROR_MSG.size()); + return ERROR_MSG.get(index); + } + + /** + * 用户多少分钟后才能再次聊天 + * + * @param uid + * @return + */ + private Long userMinutesLater(Long uid) { + // 获取用户最后聊天时间 + Date lastChatTime = RedisUtils.get(RedisKey.getKey(USER_GLM2_TIME_LAST, uid), Date.class); + if (lastChatTime == null) { + // 如果没有聊天记录，则可以立即聊天 + return 0L; + } + // 计算当前时间和上次聊天时间之间的时间差 + long now = System.currentTimeMillis(); + long lastChatTimeMillis = lastChatTime.getTime(); + long durationMillis = now - lastChatTimeMillis; + long minutes = TimeUnit.MILLISECONDS.toMinutes(durationMillis); + // 计算剩余等待时间 + long remainingMinutes = glm2Properties.getMinute() - minutes; + return remainingMinutes > 0 ? remainingMinutes : 0L; + } + + + @Override + protected boolean supports(Message message) { + if (!glm2Properties.isUse()) { + return false; + } + /* 前端传@信息后取消注释 */ + MessageExtra extra = message.getExtra(); + if (extra == null) { + return false; + } + if (CollectionUtils.isEmpty(extra.getAtUidList())) { + return false; + } + if (!extra.getAtUidList().contains(glm2Properties.getAIUserId())) { + return false; + } + + if (StringUtils.isBlank(message.getContent())) { + return false; + } + return StringUtils.contains(message.getContent(), "@" + glm2Properties.getAIUserName()) + && StringUtils.isNotBlank(message.getContent().replace(glm2Properties.getAIUserName(), "").trim()); + } +} diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/GPTChatAIHandler.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/GPTChatAIHandler.java new file mode 100644 index 0000000..f814713 --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/handler/GPTChatAIHandler.java @@ -0,0 +1,98 @@ +package com.abin.mallchat.custom.chatai.handler; + +import cn.hutool.http.HttpResponse; +import com.abin.mallchat.common.chat.domain.entity.Message; +import com.abin.mallchat.common.common.constant.RedisKey; +import com.abin.mallchat.common.common.utils.DateUtils; +import com.abin.mallchat.common.common.utils.RedisUtils; +import com.abin.mallchat.custom.chatai.properties.ChatGPTProperties; +import com.abin.mallchat.custom.chatai.utils.ChatGPTUtils; +import org.apache.commons.lang3.StringUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Component; + +import java.util.concurrent.TimeUnit; + +@Component +public class GPTChatAIHandler extends AbstractChatAIHandler { + + @Autowired + private ChatGPTProperties chatGPTProperties; + + @Override + public Long getChatAIUserId() { + return chatGPTProperties.getAIUserId(); + } + + @Override + public String getChatAIName() { + if (StringUtils.isNotBlank(chatGPTProperties.getAIUserName())) { + return chatGPTProperties.getAIUserName(); + } + String name = userService.getUserInfo(chatGPTProperties.getAIUserId()).getName(); + chatGPTProperties.setAIUserName(name); + return name; + } + + @Override + protected String doChat(Message message) { + String content = message.getContent().replace("@" + chatGPTProperties.getAIUserName(), "").trim(); + Long uid = message.getFromUid(); + Long chatNum; + String text; + if ((chatNum = getUserChatNum(uid)) > chatGPTProperties.getLimit()) { + text = "你今天已经和我聊了" + chatNum + "次了，我累了，明天再聊吧"; + } else { + HttpResponse response = null; + try { + response = ChatGPTUtils.create(chatGPTProperties.getKey()) + .proxyUrl(chatGPTProperties.getProxyUrl()) + .model(chatGPTProperties.getModelName()) + .prompt(content) + .send(); + text = ChatGPTUtils.parseText(response); + userChatNumInrc(uid); + } catch (Exception e) { + e.printStackTrace(); + text = "我累了，明天再聊吧"; + } + } + return text; + } + + private Long userChatNumInrc(Long uid) { + return RedisUtils.inc(RedisKey.getKey(RedisKey.USER_CHAT_NUM, uid), DateUtils.getEndTimeByToday().intValue(), TimeUnit.MILLISECONDS); + } + + private Long getUserChatNum(Long uid) { + Long num = RedisUtils.get(RedisKey.getKey(RedisKey.USER_CHAT_NUM, uid), Long.class); + return num == null ? 0 : num; + + } + + + @Override + protected boolean supports(Message message) { + if (!chatGPTProperties.isUse()) { + return false; + } + /* 前端传@信息后取消注释 */ + +// MessageExtra extra = message.getExtra(); +// if (extra == null) { +// return false; +// } +// if (CollectionUtils.isEmpty(extra.getAtUidList())) { +// return false; +// } +// if (!extra.getAtUidList().contains(chatGPTProperties.getAIUserId())) { +// return false; +// } + + if (StringUtils.isBlank(message.getContent())) { + return false; + } + return StringUtils.contains(message.getContent(), "@" + chatGPTProperties.getAIUserName()) + && StringUtils.isNotBlank(message.getContent().replace(chatGPTProperties.getAIUserName(), "").trim()); + } +} diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/properties/ChatGLM2Properties.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/properties/ChatGLM2Properties.java new file mode 100644 index 0000000..2ccc99b --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/properties/ChatGLM2Properties.java @@ -0,0 +1,48 @@ +package com.abin.mallchat.custom.chatai.properties; + +import lombok.Data; +import org.springframework.boot.context.properties.ConfigurationProperties; +import org.springframework.stereotype.Component; + +/** + * ChatGLM2 配置文件 + * + * @author zhaoyuhang + * @date 2023/06/30 + */ +@Data +@Component +@ConfigurationProperties(prefix = "chatai.chatglm2") +public class ChatGLM2Properties { + + /** + * 使用 + */ + private boolean use; + + /** + * url + */ + private String url; + + /** + * 机器人 id + */ + private Long AIUserId; + + /** + * 机器人名称 + */ + private String AIUserName; + + /** + * 每个用户每?分钟可以请求一次 + */ + private Long minute = 3L; + + /** + * 超时 + */ + private Integer timeout = 60*1000; + +} diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/properties/ChatGPTProperties.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/properties/ChatGPTProperties.java new file mode 100644 index 0000000..b7a5c56 --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/properties/ChatGPTProperties.java @@ -0,0 +1,44 @@ +package com.abin.mallchat.custom.chatai.properties; + +import lombok.Data; +import org.springframework.boot.context.properties.ConfigurationProperties; +import org.springframework.stereotype.Component; + + +@Data +@Component +@ConfigurationProperties(prefix = "chatai.chatgpt") +public class ChatGPTProperties { + + /** + * 是否使用openAI + */ + private boolean use; + /** + * 机器人 id + */ + private Long AIUserId; + + /** + * 机器人名称 + */ + private String AIUserName; + /** + * 模型名称 + */ + private String modelName = "text-davinci-003"; + /** + * openAI key + */ + private String key; + /** + * 代理地址 + */ + private String proxyUrl; + + /** + * 用户每天条数限制 + */ + private Integer limit = 5; + +} diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/service/IChatAIService.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/service/IChatAIService.java new file mode 100644 index 0000000..8cded22 --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/service/IChatAIService.java @@ -0,0 +1,8 @@ +package com.abin.mallchat.custom.chatai.service; + +import com.abin.mallchat.common.chat.domain.entity.Message; + +public interface IChatAIService { + + void chat(Message message); +} diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/service/impl/ChatAIServiceImpl.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/service/impl/ChatAIServiceImpl.java new file mode 100644 index 0000000..885c227 --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/service/impl/ChatAIServiceImpl.java @@ -0,0 +1,26 @@ +package com.abin.mallchat.custom.chatai.service.impl; + +import com.abin.mallchat.common.chat.domain.entity.Message; +import com.abin.mallchat.common.chat.domain.entity.msg.MessageExtra; +import com.abin.mallchat.custom.chatai.handler.AbstractChatAIHandler; +import com.abin.mallchat.custom.chatai.handler.ChatAIHandlerFactory; +import com.abin.mallchat.custom.chatai.service.IChatAIService; +import lombok.extern.slf4j.Slf4j; +import org.springframework.stereotype.Service; + +@Slf4j +@Service +public class ChatAIServiceImpl implements IChatAIService { + @Override + public void chat(Message message) { + MessageExtra extra = message.getExtra(); + if (extra == null) { + return; + } +// AbstractChatAIHandler chatAI = ChatAIHandlerFactory.getChatAIHandlerByName(message.getContent()); + AbstractChatAIHandler chatAI = ChatAIHandlerFactory.getChatAIHandlerById(extra.getAtUidList()); + if (chatAI != null) { + chatAI.chat(message); + } + } +} \ No newline at end of file diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/utils/ChatGLM2Utils.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/utils/ChatGLM2Utils.java new file mode 100644 index 0000000..7ac9e1a --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/utils/ChatGLM2Utils.java @@ -0,0 +1,106 @@ +package com.abin.mallchat.custom.chatai.utils; + +import cn.hutool.http.HttpResponse; +import cn.hutool.http.HttpUtil; +import cn.hutool.json.JSONObject; +import lombok.extern.slf4j.Slf4j; + +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +@Slf4j +public class ChatGLM2Utils { + + + private final Map headers; + /** + * 超时30秒 + */ + private Integer timeout = 60 * 1000; + + private String url; + /** + * 提示词 + */ + private String prompt; + + /** + * 历史 + */ + private List history; + + + public ChatGLM2Utils() { + HashMap _headers_ = new HashMap<>(); + _headers_.put("Content-Type", "application/json"); + this.headers = _headers_; + } + + public static ChatGLM2Utils create() { + return new ChatGLM2Utils(); + } + + + public ChatGLM2Utils url(String url) { + this.url = url; + return this; + } + + + public ChatGLM2Utils timeout(int timeout) { + this.timeout = timeout; + return this; + } + + public ChatGLM2Utils prompt(String prompt) { + this.prompt = prompt; + return this; + } + + public HttpResponse send() { + JSONObject param = new JSONObject(); + param.set("prompt", prompt); + log.info("headers >>> " + headers); + log.info("param >>> " + param); + return HttpUtil.createPost(url) + .addHeaders(headers) + .body(param.toString()) + .timeout(timeout) + .execute(); + } + + public static String parseText(String body) { + log.info("body >>> " + body); + JSONObject jsonObj = new JSONObject(body); + if (200 != jsonObj.getInt("status")) { + log.error("status >>> " + jsonObj.getInt("status")); + return "闹脾气了，等会再试试吧~"; + } + return jsonObj.getStr("response"); + } + + public static String parseText(HttpResponse response) { + return parseText(response.body()); + } + + + public static void main(String[] args) { + HttpResponse send = null; + try { + send = ChatGLM2Utils + .create() + .url("http://vastmiao.natapp1.cc") + .timeout(60 * 1000) + .prompt("Spring的启动流程是什么") + .send(); + } catch (Exception e) { + throw new RuntimeException(e); + } + System.out.println("send = " + send); + + System.out.println("parseText(send) = " + parseText(send)); + } + + +} \ No newline at end of file diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/utils/ChatGPTUtils.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/utils/ChatGPTUtils.java new file mode 100644 index 0000000..094cd39 --- /dev/null +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/chatai/utils/ChatGPTUtils.java @@ -0,0 +1,168 @@ +package com.abin.mallchat.custom.chatai.utils; + +import cn.hutool.http.HttpResponse; +import cn.hutool.http.HttpUtil; +import cn.hutool.json.JSONArray; +import cn.hutool.json.JSONObject; +import com.abin.mallchat.common.common.exception.BusinessException; +import lombok.extern.slf4j.Slf4j; +import org.apache.commons.lang3.StringUtils; + +import java.util.HashMap; +import java.util.Map; + +@Slf4j +public class ChatGPTUtils { + + private static final String URL = "https://api.openai.com/v1/completions"; + + private String model = "text-davinci-003"; + + private final Map headers; + /** + * 超时30秒 + */ + private Integer timeout = 30 * 1000; + /** + * 参数用于指定生成文本的最大长度。 + * 它表示生成的文本中最多包含多少个 token。一个 token 可以是一个单词、一个标点符号或一个空格。 + */ + private int maxTokens = 2048; + /** + * 用于控制生成文本的多样性。 + * 较高的温度会导致更多的随机性和多样性，但可能会降低生成文本的质量。默认值为 1，建议在 0.7 到 1.3 之间调整。 + */ + private Object temperature = 1; + /** + * 用于控制生成文本的多样性。 + * 它会根据概率选择最高的几个单词，而不是选择概率最高的单词。默认值为 1，建议在 0.7 到 0.9 之间调整。 + */ + private Object topP = 0.9; + /** + * 用于控制生成文本中重复单词的数量。 + * 较高的惩罚值会导致更少的重复单词，但可能会降低生成文本的流畅性。默认值为 0，建议在 0 到 2 之间调整。 + */ + private Object frequencyPenalty = 0.0; + /** + * 用于控制生成文本中出现特定单词的数量。 + * 较高的惩罚值会导致更少的特定单词，但可能会降低生成文本的流畅性。默认值为 0，建议在 0 到 2 之间调整。 + */ + private Object presencePenalty = 0.6; + + /** + * 提示词 + */ + private String prompt; + + private String proxyUrl; + + public ChatGPTUtils(String key) { + HashMap _headers_ = new HashMap<>(); + _headers_.put("Content-Type", "application/json"); + if (StringUtils.isBlank(key)) { + throw new BusinessException("openAi key is blank"); + } + _headers_.put("Authorization", "Bearer " + key); + this.headers = _headers_; + } + + public static ChatGPTUtils create(String key) { + return new ChatGPTUtils(key); + } + + public static String parseText(HttpResponse response) { + return parseText(response.body()); + } + + public static String parseText(String body) { + log.info("body >>> " + body); + JSONObject jsonObj = new JSONObject(body); + JSONObject error = jsonObj.getJSONObject("error"); + if (error != null) { + log.error("error >>> " + error); + return "闹脾气了，等会再试试吧~"; + } + JSONArray choicesArr = jsonObj.getJSONArray("choices"); + JSONObject choiceObj = choicesArr.getJSONObject(0); + return choiceObj.getStr("text"); + } + + public ChatGPTUtils model(String model) { + this.model = model; + return this; + } + + public ChatGPTUtils timeout(int timeout) { + this.timeout = timeout; + return this; + } + + public ChatGPTUtils maxTokens(int maxTokens) { + this.maxTokens = maxTokens; + return this; + } + + public ChatGPTUtils temperature(int temperature) { + this.temperature = temperature; + return this; + } + + public ChatGPTUtils topP(int topP) { + this.topP = topP; + return this; + } + + public ChatGPTUtils frequencyPenalty(int frequencyPenalty) { + this.frequencyPenalty = frequencyPenalty; + return this; + } + + public ChatGPTUtils presencePenalty(int presencePenalty) { + this.presencePenalty = presencePenalty; + return this; + } + + public ChatGPTUtils prompt(String prompt) { + this.prompt = prompt; + return this; + } + + public ChatGPTUtils proxyUrl(String proxyUrl) { + this.proxyUrl = proxyUrl; + return this; + } + + public HttpResponse send() { + JSONObject param = new JSONObject(); + param.set("model", model); + param.set("prompt", prompt); + param.set("max_tokens", maxTokens); + param.set("temperature", temperature); + param.set("top_p", topP); + param.set("frequency_penalty", frequencyPenalty); + param.set("presence_penalty", presencePenalty); + log.info("headers >>> " + headers); + log.info("param >>> " + param); + return HttpUtil.createPost(StringUtils.isNotBlank(proxyUrl) ? proxyUrl : URL) + .addHeaders(headers) + .body(param.toString()) + .timeout(timeout) + .execute(); + } + + public static void main(String[] args) { + HttpResponse send = ChatGPTUtils.create("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") + .timeout(30 * 1000) + .prompt("Spring的启动流程是什么") + .send(); + System.out.println("send = " + send); + // JSON 数据 + // JSON 数据 + JSONObject jsonObj = new JSONObject(send.body()); + JSONArray choicesArr = jsonObj.getJSONArray("choices"); + JSONObject choiceObj = choicesArr.getJSONObject(0); + String text = choiceObj.getStr("text"); + System.out.println("text = " + text); + + } +} \ No newline at end of file diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/common/event/listener/MessageSendListener.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/common/event/listener/MessageSendListener.java index 15b7cab..ba8bcf0 100644 --- a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/common/event/listener/MessageSendListener.java +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/common/event/listener/MessageSendListener.java @@ -5,9 +5,11 @@ import com.abin.mallchat.common.chat.domain.entity.Message; import com.abin.mallchat.common.common.event.MessageSendEvent; import com.abin.mallchat.custom.chat.domain.vo.response.ChatMessageResp; import com.abin.mallchat.custom.chat.service.ChatService; +import com.abin.mallchat.custom.chatai.service.IChatAIService; import com.abin.mallchat.custom.user.service.WebSocketService; import com.abin.mallchat.custom.user.service.adapter.WSAdapter; import lombok.extern.slf4j.Slf4j; +import org.jetbrains.annotations.NotNull; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.scheduling.annotation.Async; import org.springframework.stereotype.Component; @@ -27,6 +29,8 @@ public class MessageSendListener { private ChatService chatService; @Autowired private MessageDao messageDao; + @Autowired + private IChatAIService openAIService; @Async @TransactionalEventListener(classes = MessageSendEvent.class, fallbackExecution = true) @@ -36,4 +40,10 @@ public class MessageSendListener { webSocketService.sendToAllOnline(WSAdapter.buildMsgSend(msgResp), message.getFromUid()); } + @TransactionalEventListener(classes = MessageSendEvent.class, fallbackExecution = true) + public void handlerMsg(@NotNull MessageSendEvent event) { + Message message = messageDao.getById(event.getMsgId()); + openAIService.chat(message); + } + }