Add support for word-level audio transcription timestamp granularity (#733)

* Add support for audio transcription timestamp_granularities word

* Fixup multiple timestamp granularities
This commit is contained in:
Alireza Ghasemi
2024-05-07 16:10:07 +03:30
committed by GitHub
parent c9953a7b05
commit 3334a9c78a
3 changed files with 35 additions and 6 deletions

View File

@@ -24,6 +24,10 @@ func TestAudioWithFailingFormBuilder(t *testing.T) {
Temperature: 0.5,
Language: "en",
Format: AudioResponseFormatSRT,
TimestampGranularities: []TranscriptionTimestampGranularity{
TranscriptionTimestampGranularitySegment,
TranscriptionTimestampGranularityWord,
},
}
mockFailedErr := fmt.Errorf("mock form builder fail")
@@ -47,7 +51,7 @@ func TestAudioWithFailingFormBuilder(t *testing.T) {
return nil
}
failOn := []string{"model", "prompt", "temperature", "language", "response_format"}
failOn := []string{"model", "prompt", "temperature", "language", "response_format", "timestamp_granularities[]"}
for _, failingField := range failOn {
failForField = failingField
mockFailedErr = fmt.Errorf("mock form builder fail on field %s", failingField)