diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/dify.go b/plugins/wasm-go/extensions/ai-proxy/provider/dify.go
index a89cfbe33a..5c21cc3b63 100644
--- a/plugins/wasm-go/extensions/ai-proxy/provider/dify.go
+++ b/plugins/wasm-go/extensions/ai-proxy/provider/dify.go
@@ -204,12 +204,12 @@ func (d *difyProvider) streamResponseDify2OpenAI(ctx wrapper.HttpContext, respon
 	}
 	if response.Event == "message_end" || response.Event == "workflow_finished" {
 		choice.FinishReason = finishReasonStop
-	}
-	if response.Event == "message_end" {
-		responseUsage = usage{
-			PromptTokens:     response.MetaData.Usage.PromptTokens,
-			CompletionTokens: response.MetaData.Usage.CompletionTokens,
-			TotalTokens:      response.MetaData.Usage.TotalTokens,
+		if response.Event == "message_end" {
+			responseUsage = usage{
+				PromptTokens:     response.MetaData.Usage.PromptTokens,
+				CompletionTokens: response.MetaData.Usage.CompletionTokens,
+				TotalTokens:      response.MetaData.Usage.TotalTokens,
+			}
 		}
 	}
 	return &chatCompletionResponse{