diff options
| -rw-r--r-- | continuedev/src/continuedev/libs/util/traceback_parsers.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/continuedev/src/continuedev/libs/util/traceback_parsers.py b/continuedev/src/continuedev/libs/util/traceback_parsers.py index 4e7a24e0..1885cc79 100644 --- a/continuedev/src/continuedev/libs/util/traceback_parsers.py +++ b/continuedev/src/continuedev/libs/util/traceback_parsers.py @@ -1,6 +1,11 @@ +PYTHON_TRACEBACK_PREFIX = "Traceback (most recent call last):" + + def get_python_traceback(output: str) -> str: - if "Traceback (most recent call last):" in output or "SyntaxError" in output: - return output + if PYTHON_TRACEBACK_PREFIX in output: + return PYTHON_TRACEBACK_PREFIX + output.split(PYTHON_TRACEBACK_PREFIX)[-1] + elif "SyntaxError" in output: + return "SyntaxError" + output.split("SyntaxError")[-1] else: return None |