mirror of
https://github.com/mudler/LocalAI.git
synced 2026-07-01 03:46:41 -04:00
* feat(realtime): EOU-driven semantic_vad turn detection Add a `semantic_vad` turn-detection mode to the realtime API that feeds the transcription model live and decides "the user finished speaking" from the `<EOU>` end-of-utterance token rather than from silence alone. When EOU fires the turn commits immediately (~0.3s); otherwise it falls back to an eagerness-scaled silence threshold (low/med/high = 8/4/2s). Plumbing, bottom to top: - proto: `AudioTranscriptionLive` bidirectional RPC (config-first oneof, mono float PCM @16k, ready-ack / Unimplemented degrade signal) plus `TranscriptResult.eou` for the unary retranscribe gate. - pkg/grpc: client/server/base/embed scaffolding for the bidi stream, modeled on AudioTransformStream; release stream conns on terminal Recv. - parakeet-cpp: live transcription RPC with per-C-call engine locking (one live stream per turn, finalize+free at commit); bump parakeet.cpp to ABI v5 — incremental StreamingMel (no more quadratic per-feed mel recompute that delayed EOU on long turns) and the <EOU>/<EOB> split; strip the literal <EOU>/<EOB> from offline text and set Eou. - core/backend: LiveTranscriptionSession wrapper + pipeline `turn_detection:` config block (type/eagerness/retranscribe). - realtime: semantic_vad integration — live input captions streamed as transcription deltas while the user speaks, EOU-immediate commit with eagerness fallback, optional retranscribe gate (batch re-decode must also end in <EOU> to confirm), clause synthesis off the LLM token callback, and per-turn live-transcription / model_load telemetry. - UI: show the realtime pipeline components as a vertical list. Docs and tests included; opt-in via the pipeline YAML or per-session `session.update`. Non-streaming STT backends degrade to silence-only. Assisted-by: Claude Code:claude-opus-4-8 [Read] [Edit] [Write] [Bash] Assisted-by: Claude Code:claude-fable-5 [Read] [Edit] [Bash] Signed-off-by: Richard Palethorpe <io@richiejp.com> * feat(realtime): explicit formally-verified state machines + parakeet streaming driver The realtime API had several implicit state machines whose state was inferred from scattered booleans, channels, and five separate mutexes, leaving illegal/inconsistent states reachable. Make them explicit and keep the implementation in step with a formal design; rework the parakeet streaming backend along the same lines. Realtime state machines (M1-M5). Each is a sealed sum-type State/Event/Effect with a total, pure Next(state,event)->(state,[]effect) behind a single-writer Coordinator: M1 conncoord connection lifecycle: VAD toggle + once-only teardown (replaces vadServerStarted + a `done` channel closed from two sites). M2 turncoord turn detection: collapses speechStarted and the live-stream "turn open" flag into one state, so discardTurn can no longer desync them and suppress the next onset. M3 respcoord response coordination: serializes the dual-writer start/cancel so at most one response is live; one response.done per response.create. M4 compactcoord conversation compaction: single-flight (replaces the `compacting atomic.Bool` CAS). M5 ttscoord TTS pipeline: open->closing->closed, idempotent wait(), rejects enqueue-after-close (was a silent drop). The Coordinator/Sink/Next plumbing — only the sealed types and Next differed per machine — is extracted once into core/http/endpoints/openai/coordinator as a generic Coordinator[S,E,F]; each machine keeps its public API via type aliases, so no sink, call-site, or test moved. Hierarchy. session_lifecycle.fizz models M1 as the parent region with its children (M2/M3/M4) as one statechart and asserts ChildrenDieWithParent (conn torn => all children terminal, none start after teardown). respcoord and compactcoord gain an absorbing Terminated state + Shutdown event; conncoord's teardown drives the children terminal. This closes a compaction teardown gap: a fire-and-forget compaction could outlive a torn session — compactionSink now takes a session-scoped cancellable context + WaitGroup and joins the in-flight summarize+evict on shutdown. Formal verification. formal-verification/ holds one authoritative FizzBee spec per machine plus the composition spec, each with an always-assertion and a documented one-line edit that makes the checker fail (verified non-vacuous). scripts/realtime-conformance.sh is fail-closed: all Go conformance suites under -race AND a model-check of every .fizz spec; a missing FizzBee is a hard error (only the loud REALTIME_CONFORMANCE_SKIP_FIZZBEE=1 bypasses it, never in CI). FizzBee is pinned by sha256 and installed via scripts/install-fizzbee.sh into .tools/ (gitignored). Wired as make test-realtime-conformance, a CI workflow, and a pre-commit path filter. Go conformance tests are Ginkgo/Gomega (per the repo's forbidigo lint): transition tables + fixed-seed property walks + concurrent/-race specs, no rapid dependency. Design map: docs/design/realtime-state-machines.md. Parakeet streaming backend. The same treatment applied to the parakeet-cpp streaming paths: - AudioTranscriptionStream returns codes.Unimplemented for non-streaming models instead of decoding offline and emitting it as one delta + final. A client that asked for streaming learns the model cannot stream rather than receiving a batch result shaped like a stream. New grpcerrors.StreamTranscriptionUnsupported carries that signal; the HTTP /v1/audio/transcriptions stream path surfaces it as an SSE error event. Mirrors AudioTranscriptionLive, which already did this. - utteranceBoundary (boundary.go): a single definition of the end-of-utterance latch, replacing three open-coded finalEou toggles. Modelled as a two-valued type so illegal states are unrepresentable. - Shared decode driver (driver.go): streamFeedResult (one per-feed event) + feedChunk (hides the ABI v4 JSON vs text-only split) + feedSlices + flushTail. The feed loop is written once. - AudioTranscriptionLive becomes a bidi adapter: it streams the per-feed {delta,eou,eob,words} the realtime turn detector consumes and a terminal FinalResult carrying only Text. Segments/duration/eou are offline-only and no longer produced (nor read) on the live path; liveTraceState drops the terminal eou and keeps the per-feed eou_events count. - AudioTranscriptionStream + streamJSON merge into one driver-based function; streamSegmenter is generalized to the unified event with a text-only fallback that preserves the legacy (no-words) library's per-utterance segmentation. Verified: build/vet/gofumpt clean, golangci-lint 0 issues, all coordinator and parakeet packages under -race, the fail-closed conformance gate green, and make test-realtime (12 e2e WS+WebRTC). Assisted-by: Claude:claude-opus-4-8 [Claude Code] Signed-off-by: Richard Palethorpe <io@richiejp.com> --------- Signed-off-by: Richard Palethorpe <io@richiejp.com>
82 lines
3.5 KiB
Go
82 lines
3.5 KiB
Go
package main
|
|
|
|
// utteranceBoundary is the single definition of a small state machine that was
|
|
// previously open-coded three times — as a bare `finalEou` bool with an ad-hoc
|
|
// toggle — in the live feed (live.go), the file-stream text path, and the
|
|
// file-stream JSON path (goparakeetcpp.go).
|
|
//
|
|
// It answers one running question: does the decode currently rest on an
|
|
// end-of-utterance boundary? That is the value a closing FinalResult reports as
|
|
// .Eou and the realtime turn detector treats as a commit point.
|
|
//
|
|
// parakeet auto-resets its decoder after every <EOU>/<EOB>, so one streaming
|
|
// session is a sequence of utterances and this is a LATCH, not a monotonic
|
|
// flag: it closes on an <EOU> and reopens as soon as the next utterance starts.
|
|
// (Contrast the realtime API's per-turn `eouSeen`, which only ever goes
|
|
// false->true because each turn gets a fresh stream. Here the stream outlives
|
|
// the turn, so the boundary status must be able to reopen.)
|
|
//
|
|
// The only transitions, over the events one streamFeedResult carries — an
|
|
// <EOU>, an <EOB> (backchannel), or plain speech output (text and/or words):
|
|
//
|
|
// <EOU>
|
|
// open ───────────► closed
|
|
// ▲ ▲ │ │ │
|
|
// │ └─┘ <EOB>|speech │ │ <EOU>
|
|
// │ (stay open) │ └─┘ (stay closed)
|
|
// └──────────────────┘
|
|
// <EOB>|speech
|
|
//
|
|
// open = NOT on an utterance boundary: mid-utterance, the last boundary was
|
|
// a backchannel <EOB>, or the stream just began (the initial state).
|
|
// closed = the last meaningful event was an <EOU> with no later speech: a real
|
|
// turn boundary.
|
|
//
|
|
// A feed that carries nothing (no eou/eob/text/words — e.g. a finalize flush
|
|
// that produced no tail) is a no-op and leaves the state unchanged, matching
|
|
// the legacy "leave finalEou as it was" behaviour.
|
|
//
|
|
// The state carries no data, so it is modelled as a two-valued type (a named
|
|
// bool) rather than an int enum: every inhabitant is legal, so illegal states
|
|
// are unrepresentable — the payload-free analog of the sealed sum types the
|
|
// realtime machines use (those need interfaces because their states carry data,
|
|
// e.g. Active{ID}, where "Active with no ID" is the illegal combination a scalar
|
|
// cannot even express).
|
|
type utteranceBoundary bool
|
|
|
|
const (
|
|
// boundaryOpen is the zero value (false), so a fresh decode starts open —
|
|
// exactly the legacy `var finalEou bool` (false) initial condition.
|
|
boundaryOpen utteranceBoundary = false
|
|
boundaryClosed utteranceBoundary = true
|
|
)
|
|
|
|
// observe folds one decode increment into the latch and returns the new state.
|
|
//
|
|
// <EOU> takes priority when a single feed carries both an <EOU> and speech
|
|
// (e.g. {"text":"hello","eou":1}): the utterance both produced that text AND
|
|
// ended, so the decode rests on the boundary. This matches the legacy
|
|
// eou-checked-first ordering at every call site.
|
|
func (b utteranceBoundary) observe(r streamFeedResult) utteranceBoundary {
|
|
switch {
|
|
case r.Eou:
|
|
return boundaryClosed
|
|
case r.Eob || r.Delta != "" || len(r.Words) > 0:
|
|
return boundaryOpen
|
|
default:
|
|
return b
|
|
}
|
|
}
|
|
|
|
// ended reports whether the decode currently rests on an end-of-utterance
|
|
// boundary (a real <EOU>, not a backchannel <EOB>). This is what a closing
|
|
// FinalResult carries as .Eou.
|
|
func (b utteranceBoundary) ended() bool { return b == boundaryClosed }
|
|
|
|
func (b utteranceBoundary) String() string {
|
|
if b == boundaryClosed {
|
|
return "closed"
|
|
}
|
|
return "open"
|
|
}
|