fix: workflow trace none type error (#5758)

2 years ago · ffb07eb24b
parent f101fcd0e7
commit ffb07eb24b
2 changed files with 6 additions and 2 deletions
--- a/api/core/ops/langfuse_trace/langfuse_trace.py
+++ b/api/core/ops/langfuse_trace/langfuse_trace.py
@ -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 = (
--- a/api/core/ops/langsmith_trace/langsmith_trace.py
+++ b/api/core/ops/langsmith_trace/langsmith_trace.py
@ -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 = (