def create_stage_chunk()

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

• 26 lines of code
• 1 McCabe index (conditional complexity)

def create_stage_chunk(name: str, content: str, stream: bool) -> dict:
    id = generate_id()
    created = generate_created()

    stage = {
        "index": 0,
        "name": name,
        "content": content,
        "status": "completed",
    }

    custom_content = {"stages": [stage]}

    return build_chunk(
        id,
        "stop",
        {
            "role": "assistant",
            "content": "",
            "custom_content": custom_content,
        },
        created,
        stream,
        usage={
            "completion_tokens": 0,
            "prompt_tokens": 0,
            "total_tokens": 0,
        },
    )