Junjie
2 天以前 63b01db83d9aad8a15276b4236a9a22e4aeef065
src/main/java/com/zy/ai/service/LlmChatService.java
@@ -41,6 +41,7 @@
    private final LlmSpringAiClientService llmSpringAiClientService;
    private final AiGatewayService aiGatewayService;
    private final OpenAiChatCompletionsMapper openAiChatCompletionsMapper;
    private final AiTokenUsageService aiTokenUsageService;
    @Value("${llm.base-url:}")
    private String fallbackBaseUrl;
@@ -469,6 +470,16 @@
        item.setExtra(cut(buildExtraPayload(responseObj == null ? null : responseObj.getUsage(), extra), 512));
        item.setCreateTime(new Date());
        llmCallLogService.saveIgnoreError(item);
        // 累加 token 到独立存储
        if (success && responseObj != null && responseObj.getUsage() != null) {
            ChatCompletionResponse.Usage usage = responseObj.getUsage();
            aiTokenUsageService.incrementTokens(
                    usage.getPromptTokens() == null ? 0 : usage.getPromptTokens(),
                    usage.getCompletionTokens() == null ? 0 : usage.getCompletionTokens(),
                    usage.getTotalTokens() == null ? 0 : usage.getTotalTokens(),
                    1);
        }
    }
    private ChatCompletionResponse usageResponse(ChatCompletionResponse.Usage usage) {