From f576f8124af85bcfe6e983f643dfb99aa554a8d2 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Fri, 11 Jul 2025 13:51:45 -0700
Subject: [PATCH] record workflow when there's an error

---
 comfy/cmd/execution.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/comfy/cmd/execution.py b/comfy/cmd/execution.py
index b80705c05..cd74c623c 100644
--- a/comfy/cmd/execution.py
+++ b/comfy/cmd/execution.py
@@ -4,6 +4,7 @@ import asyncio
 import copy
 import heapq
 import inspect
+import json
 import logging
 import sys
 import threading
@@ -559,6 +560,11 @@ class PromptExecutor:
         current_span = get_current_span()
         current_span.set_status(Status(StatusCode.ERROR))
         current_span.record_exception(ex)
+        try:
+            encoded_prompt = json.dumps(prompt)
+            current_span.set_attribute("prompt", encoded_prompt)
+        except Exception as exc_info:
+            pass
 
         node_id = error["node_id"]
         class_type = prompt[node_id]["class_type"]