fix: workflow trace none type error (#5758)

This commit is contained in:
Joe 2024-06-29 23:32:52 +08:00 committed by GitHub
parent f101fcd0e7
commit ffb07eb24b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 2 deletions

View File

@ -121,7 +121,9 @@ class LangFuseDataTrace(BaseTraceInstance):
node_type = node_execution.node_type
status = node_execution.status
if node_type == "llm":
inputs = json.loads(node_execution.process_data).get("prompts", {})
inputs = json.loads(node_execution.process_data).get(
"prompts", {}
) if node_execution.process_data else {}
else:
inputs = json.loads(node_execution.inputs) if node_execution.inputs else {}
outputs = (

View File

@ -114,7 +114,9 @@ class LangSmithDataTrace(BaseTraceInstance):
node_type = node_execution.node_type
status = node_execution.status
if node_type == "llm":
inputs = json.loads(node_execution.process_data).get("prompts", {})
inputs = json.loads(node_execution.process_data).get(
"prompts", {}
) if node_execution.process_data else {}
else:
inputs = json.loads(node_execution.inputs) if node_execution.inputs else {}
outputs = (