Browse Source

add echart page

tycoding 1 year ago
parent
commit
a694468f11

+ 1 - 1
langchat-aigc/src/main/java/cn/tycoding/langchat/aigc/entity/AigcConversation.java

@@ -60,7 +60,7 @@ public class AigcConversation implements Serializable {
      * Token消耗量
      */
     @TableField(exist = false)
-    private String tokenUsed;
+    private Integer tokenUsed;
     /**
      * 最后一次对话时间
      */

+ 1 - 0
langchat-aigc/src/main/java/cn/tycoding/langchat/aigc/service/impl/AigcMessageServiceImpl.java

@@ -66,6 +66,7 @@ public class AigcMessageServiceImpl extends ServiceImpl<AigcMessageMapper, AigcM
                 if (!messageList.isEmpty()) {
                     i.setChatTotal(messageList.size());
                     i.setEndTime(messageList.get(0).getCreateTime());
+                    i.setTokenUsed(messageList.stream().filter(m -> m.getTokens() != null).mapToInt(AigcMessage::getTokens).sum());
                 }
             });
         }