From 06910ad84e8da33fa6d5fe66cd69293ce8986d84 Mon Sep 17 00:00:00 2001 From: Nate Sesti Date: Fri, 16 Jun 2023 09:22:08 -0700 Subject: improved traceback parser --- continuedev/src/continuedev/libs/util/traceback_parsers.py | 9 +++++++-- 1 file 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 -- cgit v1.2.3-70-g09d2