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

pull/17997/head
Novice 1 year 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, "usage": None,
"function": {} if not prompt_message_tools else jsonable_encoder(prompt_message_tools[0]), "function": {} if not prompt_message_tools else jsonable_encoder(prompt_message_tools[0]),
"tool_call": None, "tool_call": None,
"model_provider": model_config.provider,
"model_name": model_config.model,
} }
try: try:

@ -130,6 +130,8 @@ class QuestionClassifierNode(LLMNode):
), ),
"usage": jsonable_encoder(usage), "usage": jsonable_encoder(usage),
"finish_reason": finish_reason, "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}

Loading…
Cancel
Save