tipi/kyutai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Make the api key configurable.

Browse Source
This commit is contained in:
Laurent 2025-06-25 08:21:27 +02:00
parent 2198f1d660
commit b5a106357e
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/asr-streaming-query.py
View File

@ -21,7 +21,6 @@ import websockets
# Desired audio properties # Desired audio properties
TARGET_SAMPLE_RATE = 24000 TARGET_SAMPLE_RATE = 24000
TARGET_CHANNELS = 1 # Mono TARGET_CHANNELS = 1 # Mono
HEADERS = {"kyutai-api-key": "open_token"}
all_text = [] all_text = []
transcript = [] transcript = []
finished = False finished = False
@ -101,10 +100,11 @@ async def send_messages(websocket, rtf: float):
print("Connection closed while sending messages.") print("Connection closed while sending messages.")
async def stream_audio(url: str, rtf: float): async def stream_audio(url: str, rtf: float, api_key: str):
"""Stream audio data to a WebSocket server.""" """Stream audio data to a WebSocket server."""
async with websockets.connect(url, additional_headers=HEADERS) as websocket: headers = {"kyutai-api-key": api_key}
async with websockets.connect(url, additional_headers=headers) as websocket:
send_task = asyncio.create_task(send_messages(websocket, rtf)) send_task = asyncio.create_task(send_messages(websocket, rtf))
receive_task = asyncio.create_task(receive_messages(websocket)) receive_task = asyncio.create_task(receive_messages(websocket))
await asyncio.gather(send_task, receive_task) await asyncio.gather(send_task, receive_task)
@ -115,6 +115,7 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("in_file") parser.add_argument("in_file")
parser.add_argument("--transcript") parser.add_argument("--transcript")
parser.add_argument("--api-key", default="open_key")
parser.add_argument( parser.add_argument(
"--url", "--url",
help="The url of the server to which to send the audio", help="The url of the server to which to send the audio",
@ -124,7 +125,7 @@ if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
url = f"{args.url}/api/asr-streaming" url = f"{args.url}/api/asr-streaming"
asyncio.run(stream_audio(url, args.rtf)) asyncio.run(stream_audio(url, args.rtf, args.api_key))
print(" ".join(all_text)) print(" ".join(all_text))
if args.transcript is not None: if args.transcript is not None:
with open(args.transcript, "w") as fobj: with open(args.transcript, "w") as fobj: