整合if

alibaba · Feb 2, 2025 · a98e80c · a98e80c
1 parent a1161a2
commit a98e80c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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{