INNER CODE UNIT · Python

echo

awslabs/awsome-distributed-ai · architectures/sagemaker-hyperpod-eks/nvidia-cloud-functions/nvcf/sample-function/server.py:51

async def echo(request: EchoRequest):
    """
    Echo the message back. Supports optional delay, repetition, and streaming.
    """
    if request.stream:

        def stream_text():
            for _ in range(request.repeats):
                time.sleep(request.delay)
                yield f"data: {request.message}\n\n"

        return StreamingResponse(stream_text(), media_type="text/event-stream")
    else:
        time.sleep(request.delay)
        return {"echo": request.message * request.repeats}


# ---------------------------------------------------------------------------

View source record →