From 81303ee089e50dc4521261c16029012146c99b5a Mon Sep 17 00:00:00 2001
From: yunqilee69 <yunqilee69@gmail.com>
Date: Thu, 4 Dec 2025 12:18:49 +0800
Subject: [PATCH] fix: Add Usage with the ZhiPu model

Signed-off-by: yunqilee69 <yunqilee69@gmail.com>
---
 .../java/org/springframework/ai/zhipuai/ZhiPuAiChatModel.java | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/spring-ai-zhipuai/src/main/java/org/springframework/ai/zhipuai/ZhiPuAiChatModel.java b/models/spring-ai-zhipuai/src/main/java/org/springframework/ai/zhipuai/ZhiPuAiChatModel.java
index 2c9ff3e54ff..db2f85c8e74 100644
--- a/models/spring-ai-zhipuai/src/main/java/org/springframework/ai/zhipuai/ZhiPuAiChatModel.java
+++ b/models/spring-ai-zhipuai/src/main/java/org/springframework/ai/zhipuai/ZhiPuAiChatModel.java
@@ -418,7 +418,7 @@ private DefaultUsage getDefaultUsage(ZhiPuAiApi.Usage usage) {
 	}
 
 	/**
-	 * Convert the ChatCompletionChunk into a ChatCompletion. The Usage is set to null.
+	 * Convert the ChatCompletionChunk into a ChatCompletion.
 	 * @param chunk the ChatCompletionChunk to convert
 	 * @return the ChatCompletion
 	 */
@@ -432,7 +432,7 @@ private ChatCompletion chunkToChatCompletion(ChatCompletionChunk chunk) {
 		}).toList();
 
 		return new ChatCompletion(chunk.id(), choices, chunk.created(), chunk.model(), chunk.systemFingerprint(),
-				"chat.completion", null);
+				"chat.completion", chunk.usage());
 	}
 
 	private List<ZhiPuAiApi.FunctionTool> getFunctionTools(List<ToolDefinition> toolDefinitions) {