diff --git a/guardrails/run/async_runner.py b/guardrails/run/async_runner.py index 762cfa0d7..2a06ce279 100644 --- a/guardrails/run/async_runner.py +++ b/guardrails/run/async_runner.py @@ -179,7 +179,7 @@ async def async_step( try: # Prepare: run pre-processing, and input validation. - if output: + if output is not None: instructions = None prompt = None msg_history = None diff --git a/guardrails/run/async_stream_runner.py b/guardrails/run/async_stream_runner.py index fc1fd4018..77e730780 100644 --- a/guardrails/run/async_stream_runner.py +++ b/guardrails/run/async_stream_runner.py @@ -161,7 +161,7 @@ async def async_step( outputs = Outputs() iteration = Iteration(inputs=inputs, outputs=outputs) call_log.iterations.push(iteration) - if output: + if output is not None: instructions = None prompt = None msg_history = None diff --git a/guardrails/run/runner.py b/guardrails/run/runner.py index eba815973..e9851f3f9 100644 --- a/guardrails/run/runner.py +++ b/guardrails/run/runner.py @@ -238,7 +238,7 @@ def step( try: # Prepare: run pre-processing, and input validation. - if output: + if output is not None: instructions = None prompt = None msg_history = None diff --git a/guardrails/run/stream_runner.py b/guardrails/run/stream_runner.py index 56e7873eb..6e2a84a11 100644 --- a/guardrails/run/stream_runner.py +++ b/guardrails/run/stream_runner.py @@ -116,7 +116,7 @@ def step( call_log.iterations.push(iteration) # Prepare: run pre-processing, and input validation. - if output: + if output is not None: instructions = None prompt = None msg_history = None