실시간 음성 전사

# 依赖: pip install "openai[realtime]"
import asyncio
import base64
from openai import AsyncOpenAI

# 复用官方 SDK 只需三处适配:
#   1) websocket_base_url 指向 AiHubMix 网关(而非 OpenAI 默认地址)
#   2) connect() 显式传 model —— 见上方 Note,网关握手必需
#   3) extra_query 带上 intent=transcription
client = AsyncOpenAI(
    api_key="sk-***",  # 替换为你的 AiHubMix API 密钥
    websocket_base_url="wss://aihubmix.com/v1",
)

async def main():
    async with client.realtime.connect(
        model="gpt-live-transcribe",           # 必传: 落进握手 URL,网关据此选路+计费
        extra_query={"intent": "transcription"},
    ) as conn:
        # 1) 配置转录会话
        await conn.session.update(session={
            "type": "transcription",
            "audio": {"input": {
                "format": {"type": "audio/pcm", "rate": 24000},
                "transcription": {
                    "model": "gpt-live-transcribe",
                    "languages": ["en", "zh"],
                    "prompt": "会议录音,含产品名与英文缩写",
                    "keywords": ["AiHubMix", "gpt-live-transcribe"],
                },
                "turn_detection": None,
                "noise_reduction": {"type": "near_field"},
            }},
        })

        # 2) 读取本地 PCM16 / 24kHz / 单声道 裸音频,分块发送
        async def send_audio():
            with open("audio_pcm16_24k.raw", "rb") as f:
                pcm = f.read()
            chunk = 24000 * 2 // 10  # 100ms = 采样率 × 2字节 ÷ 10
            for i in range(0, len(pcm), chunk):
                await conn.input_audio_buffer.append(
                    audio=base64.b64encode(pcm[i:i + chunk]).decode()
                )
                await asyncio.sleep(0.1)  # 模拟实时节奏
            # 无 VAD,音频发完后手动 commit 触发转录收尾
            await conn.input_audio_buffer.commit()

        asyncio.create_task(send_audio())

        # 3) 接收转录结果
        async for evt in conn:
            etype = getattr(evt, "type", "")
            if etype.endswith("transcription.delta"):
                print(getattr(evt, "delta", ""), end="", flush=True)
            elif etype.endswith("transcription.completed"):
                print("\n[完成]", getattr(evt, "transcript", ""))
                break  # 拿到完整结果即可退出
            elif etype == "error":
                print("\n[错误]", evt.to_dict())
                break

asyncio.run(main())

import asyncio
import base64
import json
import websockets

API_KEY = "sk-***"  # 替换为你的 AiHubMix API 密钥
URL = (
    "wss://aihubmix.com/v1/realtime"
    "?intent=transcription&model=gpt-live-transcribe"
)

async def main():
    # websockets >= 13 用 additional_headers；旧版本用 extra_headers
    async with websockets.connect(
        URL, additional_headers={"Authorization": f"Bearer {API_KEY}"}
    ) as ws:
        # 1) 配置转录会话
        await ws.send(json.dumps({
            "type": "session.update",
            "session": {
                "type": "transcription",
                "audio": {"input": {
                    "format": {"type": "audio/pcm", "rate": 24000},
                    "transcription": {"model": "gpt-live-transcribe", "language": "en"},
                    "turn_detection": None,
                    "noise_reduction": {"type": "near_field"},
                }},
            },
        }))

        # 2) 读取本地 PCM16 / 24kHz / 单声道 裸音频，分块发送
        async def send_audio():
            with open("audio_pcm16_24k.raw", "rb") as f:
                pcm = f.read()
            chunk = 24000 * 2 // 10  # 100ms = 采样率 × 2字节 ÷ 10
            for i in range(0, len(pcm), chunk):
                await ws.send(json.dumps({
                    "type": "input_audio_buffer.append",
                    "audio": base64.b64encode(pcm[i:i + chunk]).decode(),
                }))
                await asyncio.sleep(0.1)  # 模拟实时节奏
            # 无 VAD，音频发完后手动 commit 触发转录收尾
            await ws.send(json.dumps({"type": "input_audio_buffer.commit"}))

        asyncio.create_task(send_audio())

        # 3) 接收转录结果
        async for msg in ws:
            evt = json.loads(msg)
            etype = evt.get("type", "")
            if etype.endswith("transcription.delta"):
                print(evt.get("delta", ""), end="", flush=True)
            elif etype.endswith("transcription.completed"):
                print("\n[完成]", evt.get("transcript", ""))
                break  # 拿到完整结果即可退出
            elif etype == "error":
                print("\n[错误]", evt.get("error"))
                break

asyncio.run(main())

# 用 wscat 快速验证连通性与鉴权（需先 npm i -g wscat）
wscat -c "wss://aihubmix.com/v1/realtime?intent=transcription&model=gpt-live-transcribe" \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY"

# 连接成功后，粘贴一帧 session.update 即可开始（音频需自行 base64 后用 input_audio_buffer.append 发送）