Skip to content

Commit 97baa66

Browse files
committed
fix: raise exception anytime a run fails
Signed-off-by: Donnie Adams <[email protected]>
1 parent 4dece0e commit 97baa66

File tree

2 files changed

+13
-3
lines changed

2 files changed

+13
-3
lines changed

gptscript/run.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,9 @@ async def text(self) -> str:
6262
finally:
6363
self._task = None
6464

65-
return f"run encountered an error: {self._err} with error output: {self._errput}" if self._err != "" else self._output
65+
if self._err != "":
66+
raise Exception(f"run encountered an error: {self._err} with error output: {self._errput}")
67+
return self._output
6668

6769
def err(self):
6870
return self._err

tests/test_gptscript.py

+10-2
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,12 @@ async def test_restart_failed_run(gptscript):
181181
]
182182

183183
run = gptscript.evaluate(tools, Options(disableCache=True, env=["EXIT_CODE=1"]))
184-
await run.text()
184+
185+
try:
186+
await run.text()
187+
assert False, "Expected run to fail"
188+
except:
189+
pass
185190

186191
assert run.state() == RunState.Error, "Unexpected run state after exit 1"
187192

@@ -697,7 +702,10 @@ async def process_event(r: Run, frame: CallFrame | RunFrame | PromptFrame):
697702
event_handlers=[process_event],
698703
)
699704

700-
out = await run.text()
705+
try:
706+
out = await run.text()
707+
except Exception as e:
708+
out = str(e)
701709

702710
assert not prompt_event_found, "Prompt event occurred"
703711
assert "prompt event occurred" in out, "Unexpected output: " + out

0 commit comments

Comments
 (0)