Spaces:
Paused
Paused
File size: 2,273 Bytes
8d1819a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
import re
import traceback
import asyncio
def handle_error(e: Exception):
# if asyncio.CancelledError, re-raise
if isinstance(e, asyncio.CancelledError):
raise e
def error_text(e: Exception):
return str(e)
def format_error(e: Exception, start_entries=6, end_entries=4):
# format traceback from the provided exception instead of the most recent one
traceback_text = ''.join(traceback.format_exception(type(e), e, e.__traceback__))
# Split the traceback into lines
lines = traceback_text.split("\n")
if not start_entries and not end_entries:
trimmed_lines = []
else:
# Find all "File" lines
file_indices = [
i for i, line in enumerate(lines) if line.strip().startswith("File ")
]
# If we found at least one "File" line, trim the middle if there are more than start_entries+end_entries lines
if len(file_indices) > start_entries + end_entries:
start_index = max(0, len(file_indices) - start_entries - end_entries)
trimmed_lines = (
lines[: file_indices[start_index]]
+ [
f"\n>>> {len(file_indices) - start_entries - end_entries} stack lines skipped <<<\n"
]
+ lines[file_indices[start_index + end_entries] :]
)
else:
# If no "File" lines found, or not enough to trim, just return the original traceback
trimmed_lines = lines
# Find the error message at the end
error_message = ""
for line in reversed(lines):
# match both simple errors and module.path.Error patterns
if re.match(r"[\w\.]+Error:\s*", line):
error_message = line
break
# Combine the trimmed traceback with the error message
if not trimmed_lines:
result = error_message
else:
result = "Traceback (most recent call last):\n" + "\n".join(trimmed_lines)
if error_message:
result += f"\n\n{error_message}"
# at least something
if not result:
result = str(e)
return result
class RepairableException(Exception):
"""An exception type indicating errors that can be surfaced to the LLM for potential self-repair."""
pass
|