diff --git a/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py b/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py index be43639fc0..8db1e432fc 100644 --- a/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py +++ b/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py @@ -186,6 +186,8 @@ class ParameterExtractorNode(LLMNode): "usage": None, "function": {} if not prompt_message_tools else jsonable_encoder(prompt_message_tools[0]), "tool_call": None, + "model_provider": model_config.provider, + "model_name": model_config.model, } try: 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 0ec44eefac..b4f34a3bef 100644 --- a/api/core/workflow/nodes/question_classifier/question_classifier_node.py +++ b/api/core/workflow/nodes/question_classifier/question_classifier_node.py @@ -130,6 +130,8 @@ class QuestionClassifierNode(LLMNode): ), "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}