| @@ -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: | |||
| @@ -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} | |||