wehub-resource-sync
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
|
|
from fastapi import APIRouter, WebSocket
|
|
|
|
from vllm.logger import init_logger
|
|
|
|
from .connection import RealtimeConnection
|
|
|
|
logger = init_logger(__name__)
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.websocket("/v1/realtime")
|
|
async def realtime_endpoint(websocket: WebSocket):
|
|
"""WebSocket endpoint for realtime audio transcription.
|
|
|
|
Protocol:
|
|
1. Client connects to ws://host/v1/realtime
|
|
2. Server sends session.created event
|
|
3. Client optionally sends session.update with model/params
|
|
4. Client sends input_audio_buffer.commit when ready
|
|
5. Client sends input_audio_buffer.append events with base64 PCM16 chunks
|
|
6. Server processes and sends transcription.delta events
|
|
7. Server sends transcription.done with final text + usage
|
|
8. Repeat from step 5 for next utterance
|
|
9. Optionally, client sends input_audio_buffer.commit with final=True
|
|
to signal audio input is finished. Useful when streaming audio files
|
|
|
|
Audio format: PCM16, 16kHz, mono, base64-encoded
|
|
"""
|
|
app = websocket.app
|
|
serving = app.state.openai_serving_realtime
|
|
|
|
connection = RealtimeConnection(websocket, serving)
|
|
await connection.handle_connection()
|