fix(langfuse): qusetion classify node can't see cost in langfuse (#17982)

pull/17997/head
Novice 10 months ago committed by GitHub
parent d80f4c7d3b
commit cd7fd100a7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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

Loading…
Cancel
Save