update how we set error message

This commit is contained in:
Richard Yu 2025-12-03 15:02:11 -08:00
parent 380b6aec8e
commit 048c413adc
2 changed files with 9 additions and 3 deletions

View File

@ -86,8 +86,12 @@ def normalize_history_item(prompt_id, history_item, include_outputs=False):
error_message = None error_message = None
if status == JobStatus.ERROR and status_info: if status == JobStatus.ERROR and status_info:
messages = status_info.get('messages', []) messages = status_info.get('messages', [])
if messages: for entry in messages:
error_message = messages[0] if isinstance(messages[0], str) else str(messages[0]) if isinstance(entry, (list, tuple)) and len(entry) >= 2 and entry[0] == 'execution_error':
detail = entry[1]
if isinstance(detail, dict):
error_message = str(detail.get('exception_message', ''))
break
execution_time = history_item.get('execution_time') execution_time = history_item.get('execution_time')

View File

@ -294,7 +294,9 @@ class TestNormalizeHistoryItem:
'status': { 'status': {
'status_str': 'error', 'status_str': 'error',
'completed': False, 'completed': False,
'messages': ['Node failed: OutOfMemory'] 'messages': [
('execution_error', {'exception_message': 'Node failed: OutOfMemory', 'node_id': '5'})
]
}, },
'outputs': {}, 'outputs': {},
'execution_time': 1.0, 'execution_time': 1.0,