def send_chunk_to_response()

in aidial_interceptors_sdk/utils/_dial_sdk.py [0:0]

• 10 lines of code
• 5 McCabe index (conditional complexity)

def send_chunk_to_response(response: Response, chunk: BaseChunk | dict):
    if isinstance(chunk, dict):
        for choice in chunk.get("choices") or []:
            if (index := choice.get("index")) is not None:
                response._last_choice_index = max(
                    response._last_choice_index, index + 1
                )
        response._queue.put_nowait(ArbitraryChunk(chunk=chunk))
    else:
        response._queue.put_nowait(chunk)