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