feat(server,stage-ui): bidirectional streaming TTS + audio path refactor

Why:
- Add a real bidirectional streaming TTS path: raw LLM tokens are
  forwarded to the upstream model (Volcengine v3 via the unspeech ws
  bridge) without client-side segmentation, so the model owns sentence
  splitting and audio chunks play as they arrive.
- Move audio endpoints out of /api/v1/openai/. `/audio/voices`,
  `/audio/models`, `/audio/voices/streaming` are not real OpenAI public
  APIs, and the streaming TTS surface has nothing to do with OpenAI —
  keeping them under /openai/ mislabelled the contract.
- Introduce `capabilities.speech.transport` on ProviderDefinition so
  future streaming providers (ElevenLabs / Cartesia / OpenAI Realtime)
  opt in without touching Stage.vue or the session factory.
- Unify Stage.vue's TTS path through a single StageTtsSession so the
  chat-orchestrator hooks no longer branch on provider id.

What:
- apps/server: new ws proxy /api/v1/audio/speech/ws bridges client ↔
  unspeech with auth, pre-flight flux check, billing from upstream
  session.finished.usage, OTel spans.
- apps/server: audio routes moved from /api/v1/openai/audio/* to
  /api/v1/audio/* (hard cutover; 404 sentinel tests added).
- apps/server: new /api/v1/audio/voices/streaming proxy reads voices
  from unspeech /api/voices?provider=volcengine.
- apps/server: new STREAMING_TTS_UPSTREAM configKV entry +
  scripts/seed-streaming-tts.ts.
- stage-ui: new libs/speech/streaming-pipeline.ts opens one ws per LLM
  intent (appendText / finish / cancel + onSentence / onError / onDone).
- stage-ui: new libs/speech/tts-session.ts — StageTtsSession interface
  with segmenter and streaming adapters; factory dispatches by
  capabilities.speech.transport instead of hard-coded provider id.
- stage-ui: providerOfficialSpeechStreaming with capabilities.speech =
  { transport: 'bidirectional-ws' }; settings page with model/voice
  picker + ws-based preview.
- stage-ui: Stage.vue chat hooks collapsed to a single currentSession;
  hot-swap watcher cancels mid-session on provider/voice/model change;
  unmount cancels and drains playback.

Tests:
- 9 streaming-pipeline tests (happy path / buffered / error / cancel /
  truncation)
- 11 tts-session tests (factory branch coverage + adapter contracts)
- 4 audio-speech-ws route tests (forwarding / billing / pre-flight /
  config-missing)
- 3 legacy-path 404 sentinels in v1 route tests
- Verification doc updated to reflect automated coverage.
This commit is contained in:
RainbowBird
2026-05-18 23:34:35 +08:00
parent a78eaaa4a9
commit ba9247fb47
15 changed files with 2033 additions and 95 deletions
+172
View File
@@ -0,0 +1,172 @@
#!/usr/bin/env tsx
/* eslint-disable no-console */
/**
* Seed the `STREAMING_TTS_UPSTREAM` configKV entry that powers
* `/api/v1/audio/speech/ws`.
*
* Why a separate script from `seed-router-config.ts`:
* - STREAMING_TTS_UPSTREAM is a sibling top-level configKV entry, not part
* of `LLM_ROUTER_CONFIG.tts.models`. Mixing the two write paths would
* make the merge semantics in `seed-router-config.ts` harder to follow.
* - The streaming surface has only one upstream (one unspeech instance
* per deployment), so a focused script avoids the multi-provider
* merge logic entirely.
*
* What the script writes:
* - `STREAMING_TTS_UPSTREAM.baseURL` — the unspeech ws endpoint, e.g.
* `ws://airi-unspeech.railway.internal:5933/v1/audio/speech/stream`
* or `wss://unspeech.example.com/v1/audio/speech/stream`.
* - `STREAMING_TTS_UPSTREAM.keys[0]` — the **upstream provider** key
* (Volcengine `X-Api-Key`) wrapped in an envelope ciphertext, not the
* unspeech key — unspeech itself has no auth concept, it just
* forwards the `Authorization` header verbatim to the upstream.
*
* Envelope AAD:
* - `modelName: 'streaming-tts'` and `keyEntryId: 'volcengine-prod-1'`.
* These must match the values `audio-speech-ws/index.ts` uses when
* decrypting (`STREAM_MODEL_LABEL_FALLBACK = 'streaming-tts'`).
*
* Cross-instance invalidation:
* - Published on `configkv:invalidate`. The audio-speech-ws route reads
* STREAMING_TTS_UPSTREAM fresh on every connection (no in-memory
* cache), so the publish is currently informational — it stays here
* for forward compatibility if we add caching later.
*
* Usage:
*
* STREAMING_TTS_UPSTREAM_URL="ws://airi-unspeech.railway.internal:5933/v1/audio/speech/stream" \
* VOLCENGINE_TTS_API_KEY="sk-..." \
* pnpm exec dotenvx run --env-file=.env.local -- \
* tsx scripts/seed-streaming-tts.ts
*
* # preview the ciphertext shape without writing:
* pnpm exec dotenvx run --env-file=.env.local -- \
* tsx scripts/seed-streaming-tts.ts --dry-run
*
* # rotate the key id (default 'volcengine-prod-1'):
* tsx scripts/seed-streaming-tts.ts --key-id volcengine-prod-2
*/
import type { ConfigKVService } from '../src/services/config-kv'
import process, { env, exit } from 'node:process'
import Redis from 'ioredis'
import { parseEnv } from '../src/libs/env'
import { createConfigKVService } from '../src/services/config-kv'
import { createEnvelopeCrypto } from '../src/utils/envelope-crypto'
// Must match `STREAM_MODEL_LABEL_FALLBACK` in
// apps/server/src/routes/audio-speech-ws/index.ts — the route decrypts
// with this AAD, so seeding under a different label would surface as
// `DECRYPT_FAILED` at session start.
const STREAM_AAD_MODEL_NAME = 'streaming-tts'
interface Args {
dryRun: boolean
keyId: string
}
function parseArgs(argv: string[]): Args {
const flags = new Set<string>()
const values: Record<string, string> = {}
for (let i = 0; i < argv.length; i++) {
const arg = argv[i]
if (!arg.startsWith('--'))
continue
const key = arg.slice(2)
const next = argv[i + 1]
if (next == null || next.startsWith('--')) {
flags.add(key)
}
else {
values[key] = next
i++
}
}
return {
dryRun: flags.has('dry-run'),
keyId: values['key-id'] ?? 'volcengine-prod-1',
}
}
async function main() {
const parsedEnv = parseEnv(env)
if (!parsedEnv.LLM_ROUTER_MASTER_KEY) {
console.error('error: LLM_ROUTER_MASTER_KEY env var is required (32 random bytes, base64)')
exit(1)
}
const args = parseArgs(process.argv.slice(2))
const upstreamURL = process.env.STREAMING_TTS_UPSTREAM_URL
const volcKey = process.env.VOLCENGINE_TTS_API_KEY
if (!upstreamURL || !volcKey) {
console.error('error: STREAMING_TTS_UPSTREAM_URL and VOLCENGINE_TTS_API_KEY env vars are required.')
console.error(' STREAMING_TTS_UPSTREAM_URL=ws://<unspeech-host>:5933/v1/audio/speech/stream')
console.error(' VOLCENGINE_TTS_API_KEY=<volcengine X-Api-Key>')
console.error('keys are read from env so they never appear in shell history or `ps`.')
exit(1)
}
// Soft-validate the scheme. unspeech is reachable over either ws:// (Railway
// internal networking) or wss:// (public TLS-terminated). Anything else
// (http://, https://) is almost certainly a copy-paste error from the
// unspeech REST endpoint URL.
if (!upstreamURL.startsWith('ws://') && !upstreamURL.startsWith('wss://')) {
console.error(`error: STREAMING_TTS_UPSTREAM_URL must start with ws:// or wss://, got: ${upstreamURL}`)
console.error('hint: the apps/server proxy opens a WebSocket — http:// will fail at the new WebSocket() call.')
exit(1)
}
const envelope = createEnvelopeCrypto({
masterKey: parsedEnv.LLM_ROUTER_MASTER_KEY,
previousMasterKey: parsedEnv.LLM_ROUTER_MASTER_KEY_PREVIOUS,
})
const ciphertext = envelope.encryptKey(volcKey, {
modelName: STREAM_AAD_MODEL_NAME,
keyEntryId: args.keyId,
})
const value = {
baseURL: upstreamURL,
keys: [{ id: args.keyId, ciphertext }],
adapterParams: {},
}
console.log(`mode: seed${args.dryRun ? ' (dry-run)' : ''}`)
console.log(`baseURL: ${upstreamURL}`)
console.log(`keys[0].id: ${args.keyId}`)
console.log(`ciphertext: <${ciphertext.length} chars>`)
if (args.dryRun) {
console.log('dry-run: no writes, no publish.')
return
}
const redis = new Redis(parsedEnv.REDIS_URL)
const configKV: ConfigKVService = createConfigKVService(redis)
// configKV.set runs the valibot validator (ttsUpstreamSchema) before
// committing; a malformed shape fails here instead of at first request.
await configKV.set('STREAMING_TTS_UPSTREAM', value as never)
const payload = JSON.stringify({
key: 'STREAMING_TTS_UPSTREAM',
version: Date.now(),
publishedAt: Date.now(),
})
await redis.publish('configkv:invalidate', payload)
console.log('STREAMING_TTS_UPSTREAM written.')
console.log('Published configkv:invalidate.')
await redis.quit()
}
main().catch((err) => {
console.error('seed-streaming-tts failed:', err)
exit(1)
})