def set_finish_reason()

in aidial_adapter_openai/utils/streaming.py [0:0]

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

    def set_finish_reason(chunk: dict | None, finish_reason: str) -> dict:
        chunk = chunk or empty_chunk
        chunk["choices"] = chunk.get("choices") or [{"index": 0, "delta": {}}]
        chunk["choices"][0]["finish_reason"] = finish_reason
        return chunk