办学质量监测教学评价系统
ageerle
2025-04-22 1b793e822aaa55d55b4dc4bbb79a4682947b2846
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
@@ -54,6 +54,12 @@
        ChatMessageBo chatMessageBo = new ChatMessageBo();
        Object userId = LocalCache.CACHE.get("userId");
        if(userId!=null){
            chatMessageBo.setUserId((Long) userId);
        }else {
            chatMessageBo.setUserId(getUserId());
        }
        // 计算总token数
        ChatToken chatToken = chatTokenService.queryByUserId(getUserId(), modelName);
        if (chatToken == null) {
@@ -96,12 +102,6 @@
            chatToken.setModelName(chatMessageBo.getModelName());
            chatToken.setUserId(chatMessageBo.getUserId());
            chatTokenService.editToken(chatToken);
        }
        Object userId = LocalCache.CACHE.get("userId");
        if(userId!=null){
            chatMessageBo.setUserId((Long) userId);
        }else {
            chatMessageBo.setUserId(getUserId());
        }
        // 保存消息记录
        chatMessageService.insertByBo(chatMessageBo);