@@ -282,24 +282,24 @@ async def resolve(
282282 async for chunk in stream .aiter_lines ():
283283 iter_time = time .time ()
284284
285- if (
286- (iterations := response_handler .add_streaming_line (chunk ))
287- is None
288- or iterations < 0
289- or end_reached
290- ):
291- end_reached = end_reached or iterations is None
285+ if request_info .timings .first_request_iteration is None :
286+ request_info .timings .first_request_iteration = iter_time
287+ request_info .timings .last_request_iteration = iter_time
288+ request_info .timings .request_iterations += 1
289+
290+ iterations = response_handler .add_streaming_line (chunk )
291+ if iterations is None or end_reached :
292+ end_reached = True
293+ continue
294+ if iterations <= 0 :
292295 continue
293296
294- if (
295- request_info .timings .first_iteration is None
296- or request_info .timings .iterations is None
297- ):
298- request_info .timings .first_iteration = iter_time
299- request_info .timings .iterations = 0
297+ if request_info .timings .first_token_iteration is None :
298+ request_info .timings .first_token_iteration = iter_time
299+ request_info .timings .token_iterations = 0
300300
301- request_info .timings .last_iteration = iter_time
302- request_info .timings .iterations += iterations
301+ request_info .timings .last_token_iteration = iter_time
302+ request_info .timings .token_iterations += iterations
303303
304304 request_info .timings .request_end = time .time ()
305305 yield response_handler .compile_streaming (request ), request_info
0 commit comments