Skip to content

Commit 7eb304e

Browse files
committed
Intermediate working state for refactoring progress and output
1 parent 24a3472 commit 7eb304e

File tree

10 files changed

+3881
-2512
lines changed

10 files changed

+3881
-2512
lines changed

src/guidellm/backends/openai.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)