From 63b01db83d9aad8a15276b4236a9a22e4aeef065 Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期二, 05 五月 2026 12:30:59 +0800
Subject: [PATCH] # Agent数据分析V3.0.1.7
---
src/main/java/com/zy/ai/gateway/AiGatewayService.java | 11 +++++++++++
1 files changed, 11 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/zy/ai/gateway/AiGatewayService.java b/src/main/java/com/zy/ai/gateway/AiGatewayService.java
index 71a256b..122f16b 100644
--- a/src/main/java/com/zy/ai/gateway/AiGatewayService.java
+++ b/src/main/java/com/zy/ai/gateway/AiGatewayService.java
@@ -8,6 +8,7 @@
import com.zy.ai.gateway.adapter.AiProviderAdapterRegistry;
import com.zy.ai.gateway.model.AiRequest;
import com.zy.ai.gateway.model.AiResponse;
+import com.zy.ai.service.AiTokenUsageService;
import com.zy.ai.service.LlmCallLogService;
import com.zy.ai.service.LlmRoutingService;
import lombok.RequiredArgsConstructor;
@@ -33,6 +34,7 @@
private final LlmRoutingService llmRoutingService;
private final AiProviderAdapterRegistry adapterRegistry;
private final LlmCallLogService llmCallLogService;
+ private final AiTokenUsageService aiTokenUsageService;
@Value("${llm.base-url:}")
private String fallbackBaseUrl;
@@ -201,6 +203,15 @@
item.setExtra(cut(extraPayload(route, response), 512));
item.setCreateTime(new Date());
llmCallLogService.saveIgnoreError(item);
+
+ // 绱姞 token 鍒扮嫭绔嬪瓨鍌�
+ if (success && response != null && response.getUsage() != null) {
+ aiTokenUsageService.incrementTokens(
+ response.getUsage().getInputTokens() == null ? 0 : response.getUsage().getInputTokens(),
+ response.getUsage().getOutputTokens() == null ? 0 : response.getUsage().getOutputTokens(),
+ response.getUsage().getTotalTokens() == null ? 0 : response.getUsage().getTotalTokens(),
+ 1);
+ }
}
private String responseText(AiResponse response) {
--
Gitblit v1.9.1