From efc8888372e606e615866e9d8017e9b870128473 Mon Sep 17 00:00:00 2001
From: Davide Delbianco <davide.delbianco@outlook.com>
Date: Tue, 1 Jul 2025 11:54:39 +0200
Subject: [PATCH] chore: standardize output of `LLMNode` based nodes to return
 `usage` and `finish_reason` in `output` instead of `process_data`

---
 .../nodes/question_classifier/question_classifier_node.py     | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/api/core/workflow/nodes/question_classifier/question_classifier_node.py b/api/core/workflow/nodes/question_classifier/question_classifier_node.py
index a518167cc6..c1f21942ef 100644
--- a/api/core/workflow/nodes/question_classifier/question_classifier_node.py
+++ b/api/core/workflow/nodes/question_classifier/question_classifier_node.py
@@ -140,12 +140,10 @@ class QuestionClassifierNode(LLMNode):
                 "prompts": PromptMessageUtil.prompt_messages_to_prompt_for_saving(
                     model_mode=model_config.mode, prompt_messages=prompt_messages
                 ),
-                "usage": jsonable_encoder(usage),
-                "finish_reason": finish_reason,
                 "model_provider": model_config.provider,
                 "model_name": model_config.model,
             }
-            outputs = {"class_name": category_name, "class_id": category_id}
+            outputs = {"class_name": category_name, "class_id": category_id, "usage": jsonable_encoder(usage), "finish_reason": finish_reason}
 
             return NodeRunResult(
                 status=WorkflowNodeExecutionStatus.SUCCEEDED,