From ffefb2f09ee40b1401cf01696bf94e5883dde9e3 Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Thu, 8 Aug 2024 18:12:13 +0800
Subject: [PATCH] Add new WebSocket and POST endpoints

---
 whisper_fastapi.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/whisper_fastapi.py b/whisper_fastapi.py
index 153f42e..15fbf27 100644
--- a/whisper_fastapi.py
+++ b/whisper_fastapi.py
@@ -159,6 +159,8 @@ def stream_builder(
 
 @app.websocket("/k6nele/status")
 @app.websocket("/konele/status")
+@app.websocket("/v1/k6nele/status")
+@app.websocket("/v1/konele/status")
 async def konele_status(
     websocket: WebSocket,
 ):
@@ -169,6 +171,8 @@ async def konele_status(
 
 @app.websocket("/k6nele/ws")
 @app.websocket("/konele/ws")
+@app.websocket("/v1/k6nele/ws")
+@app.websocket("/v1/konele/ws")
 async def konele_ws(
     websocket: WebSocket,
     task: Literal["transcribe", "translate"] = "transcribe",
@@ -233,6 +237,8 @@ async def konele_ws(
 
 @app.post("/k6nele/post")
 @app.post("/konele/post")
+@app.post("/v1/k6nele/post")
+@app.post("/v1/konele/post")
 async def translateapi(
     request: Request,
     task: Literal["transcribe", "translate"] = "transcribe",