Skip to content

Commit 6d74628

Browse files
committed
await tripwire triggered instead of task
1 parent 260bdf0 commit 6d74628

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/agents/run.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1143,7 +1143,9 @@ async def _start_streaming(
11431143
finally:
11441144
if streamed_result._input_guardrails_task:
11451145
try:
1146-
await streamed_result._input_guardrails_task
1146+
await AgentRunner._input_guardrail_tripwire_triggered_for_stream(
1147+
streamed_result
1148+
)
11471149
except Exception as e:
11481150
logger.debug(
11491151
f"Error in streamed_result finalize for agent {current_agent.name} - {e}"

0 commit comments

Comments
 (0)