Skip to content

Commit 6746bcc

Browse files
committed
chore: simplify code more
1 parent 6ea3cc7 commit 6746bcc

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

src/strands/event_loop/event_loop.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -154,18 +154,12 @@ async def event_loop_cycle(
154154
model_events = _handle_model_execution(
155155
agent, cycle_span, cycle_trace, invocation_state, tracer, structured_output_context
156156
)
157-
try:
158-
async for model_event in model_events:
159-
if not isinstance(model_event, ModelStopReason):
160-
yield model_event
161-
finally:
162-
await model_events.aclose()
157+
async for model_event in model_events:
158+
if not isinstance(model_event, ModelStopReason):
159+
yield model_event
163160

164161
stop_reason, message, *_ = model_event["stop"]
165162
yield ModelMessageEvent(message=message)
166-
except Exception as e:
167-
tracer.end_span_with_error(cycle_span, str(e), e)
168-
raise
169163
except BaseException as e:
170164
tracer.end_span_with_error(cycle_span, str(e), e)
171165
raise

0 commit comments

Comments
 (0)