heimoshuiyu/whisper-fastapi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add new WebSocket and POST endpoints

Browse Source
This commit is contained in:
黑墨水鱼
2024-08-08 18:12:13 +08:00
parent 1c8a685e9e
commit ffefb2f09e
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
whisper_fastapi.py
View File

@@ -159,6 +159,8 @@ def stream_builder(
@app.websocket("/k6nele/status") @app.websocket("/k6nele/status")
@app.websocket("/konele/status") @app.websocket("/konele/status")
@app.websocket("/v1/k6nele/status")
@app.websocket("/v1/konele/status")
async def konele_status( async def konele_status(
websocket: WebSocket, websocket: WebSocket,
): ):
@@ -169,6 +171,8 @@ async def konele_status(
@app.websocket("/k6nele/ws") @app.websocket("/k6nele/ws")
@app.websocket("/konele/ws") @app.websocket("/konele/ws")
@app.websocket("/v1/k6nele/ws")
@app.websocket("/v1/konele/ws")
async def konele_ws( async def konele_ws(
websocket: WebSocket, websocket: WebSocket,
task: Literal["transcribe", "translate"] = "transcribe", task: Literal["transcribe", "translate"] = "transcribe",
@@ -233,6 +237,8 @@ async def konele_ws(
@app.post("/k6nele/post") @app.post("/k6nele/post")
@app.post("/konele/post") @app.post("/konele/post")
@app.post("/v1/k6nele/post")
@app.post("/v1/konele/post")
async def translateapi( async def translateapi(
request: Request, request: Request,
task: Literal["transcribe", "translate"] = "transcribe", task: Literal["transcribe", "translate"] = "transcribe",