Browse Source

fix: workflow trace none type error (#5758)

Joe 10 months ago
parent
commit
ffb07eb24b

+ 3 - 1
api/core/ops/langfuse_trace/langfuse_trace.py

@@ -121,7 +121,9 @@ class LangFuseDataTrace(BaseTraceInstance):
             node_type = node_execution.node_type
             node_type = node_execution.node_type
             status = node_execution.status
             status = node_execution.status
             if node_type == "llm":
             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:
             else:
                 inputs = json.loads(node_execution.inputs) if node_execution.inputs else {}
                 inputs = json.loads(node_execution.inputs) if node_execution.inputs else {}
             outputs = (
             outputs = (

+ 3 - 1
api/core/ops/langsmith_trace/langsmith_trace.py

@@ -114,7 +114,9 @@ class LangSmithDataTrace(BaseTraceInstance):
             node_type = node_execution.node_type
             node_type = node_execution.node_type
             status = node_execution.status
             status = node_execution.status
             if node_type == "llm":
             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:
             else:
                 inputs = json.loads(node_execution.inputs) if node_execution.inputs else {}
                 inputs = json.loads(node_execution.inputs) if node_execution.inputs else {}
             outputs = (
             outputs = (