From a98e80c11165b788915cf1f3033fc552ae227ca6 Mon Sep 17 00:00:00 2001
From: kai2321 <a2269713951@163.com>
Date: Sun, 2 Feb 2025 22:16:23 +0800
Subject: [PATCH] =?UTF-8?q?=E6=95=B4=E5=90=88if?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 plugins/wasm-go/extensions/ai-proxy/provider/dify.go | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

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{