✨ 新增记录群或者好友已使用的tokens信息

2024-07-05 09:44:07 +08:00
parent 9e8c3f5e6f
commit f747bf5ead
3 changed files with 18 additions and 0 deletions
--- a/plugin/plugins/ai.go
+++ b/plugin/plugins/ai.go
@@ -142,6 +142,9 @@ func AI(m *plugin.MessageContext) {
 		return
 	}

+	// 异步更新一下已使用的AI次数
+	go service.UpdateUsedAiTokens(m.FromUser, resp.Usage.TotalTokens)
+
 	// 保存一下AI 返回的消息，消息 Id 使用传入 Id 的负数
 	var replyMessage entity.Message
 	replyMessage.MsgId = -m.MsgId