Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Stabilize OpenAiAudioApi test fixtures

Browse Source
This commit is contained in:
Christian Tzolov
2024-03-02 23:33:38 +01:00
parent 8703582160
commit ccd5767ead
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
models/spring-ai-openai/src/test/java/org/springframework/ai/openai/audio/api/OpenAiAudioApiIT.java
View File

@@ -59,13 +59,14 @@ public class OpenAiAudioApiIT {
TranslationRequest.builder().withModel(WhisperModel.WHISPER_1.getValue()).withFile(speech).build(),
StructuredResponse.class);
assertThat(translation.text()).isEqualTo("Hello, my name is Chris, and I love Spring AI.");
assertThat(translation.text().replaceAll(",", "")).isEqualTo("Hello my name is Chris and I love Spring AI.");
StructuredResponse transcriptionEnglish = audioApi.createTranscription(
TranscriptionRequest.builder().withModel(WhisperModel.WHISPER_1.getValue()).withFile(speech).build(),
StructuredResponse.class);
assertThat(transcriptionEnglish.text()).isEqualTo("Hello, my name is Chris, and I love Spring AI.");
assertThat(transcriptionEnglish.text().replaceAll(",", ""))
.isEqualTo("Hello my name is Chris and I love Spring AI.");
StructuredResponse transcriptionDutch = audioApi.createTranscription(
TranscriptionRequest.builder().withFile(speech).withLanguage("nl").build(), StructuredResponse.class);