wire streaming adapters into transcriber factory
This commit is contained in:
@@ -95,12 +95,23 @@ func NewTranscriber(config Config) (Transcriber, error) {
|
||||
return nil, fmt.Errorf("model %s is not a transcription model", config.Model)
|
||||
}
|
||||
|
||||
// streaming models not supported yet
|
||||
// streaming models use StreamingTranscriber
|
||||
if model.Streaming {
|
||||
return nil, fmt.Errorf("streaming model %s not supported yet (coming soon)", config.Model)
|
||||
var streamingAdapter StreamingAdapter
|
||||
switch model.AdapterType {
|
||||
case "elevenlabs-streaming":
|
||||
streamingAdapter = NewElevenLabsStreamingAdapter(model.Endpoint, config.APIKey, model.ID, config.Language)
|
||||
case "deepgram":
|
||||
streamingAdapter = NewDeepgramAdapter(model.Endpoint, config.APIKey, model.ID, config.Language)
|
||||
case "openai-realtime":
|
||||
streamingAdapter = NewOpenAIRealtimeAdapter(model.Endpoint, config.APIKey, model.ID, config.Language)
|
||||
default:
|
||||
return nil, fmt.Errorf("unsupported streaming adapter type: %s", model.AdapterType)
|
||||
}
|
||||
return NewStreamingTranscriber(streamingAdapter, config.Language), nil
|
||||
}
|
||||
|
||||
// create adapter based on model.AdapterType
|
||||
// batch models use SimpleTranscriber
|
||||
var adapter BatchAdapter
|
||||
switch model.AdapterType {
|
||||
case "openai":
|
||||
|
||||
@@ -155,14 +155,34 @@ func TestNewTranscriber(t *testing.T) {
|
||||
wantErr: false, // uses default model when empty
|
||||
},
|
||||
{
|
||||
name: "streaming model returns error",
|
||||
name: "elevenlabs streaming model creates StreamingTranscriber",
|
||||
config: Config{
|
||||
Provider: "elevenlabs",
|
||||
APIKey: "test-key",
|
||||
Language: "en",
|
||||
Model: "scribe_v1-streaming",
|
||||
},
|
||||
wantErr: true, // streaming not yet supported
|
||||
wantErr: false, // streaming is now supported
|
||||
},
|
||||
{
|
||||
name: "deepgram streaming model creates StreamingTranscriber",
|
||||
config: Config{
|
||||
Provider: "deepgram",
|
||||
APIKey: "test-key",
|
||||
Language: "en",
|
||||
Model: "nova-3",
|
||||
},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "openai realtime streaming model creates StreamingTranscriber",
|
||||
config: Config{
|
||||
Provider: "openai",
|
||||
APIKey: "test-key",
|
||||
Language: "en",
|
||||
Model: "gpt-4o-realtime-preview",
|
||||
},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "unknown model returns error",
|
||||
|
||||
Reference in New Issue
Block a user