Skip to content

Commit 1564511

Browse files
committed
refactoring
1 parent 1920904 commit 1564511

3 files changed

Lines changed: 9 additions & 5 deletions

File tree

src/main/java/com/github/jlangch/venice/impl/docgen/cheatsheet/modules/ModuleOpenAiJavaSection.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,8 @@ public DocSection section() {
6565
final DocSection audio = new DocSection("Audio", id());
6666
all.addSection(audio);
6767
audio.addItem(diBuilder.getDocItem("openai-java/create-speech", false));
68+
audio.addItem(diBuilder.getDocItem("openai-java/transcribe", false));
69+
audio.addItem(diBuilder.getDocItem("openai-java/transcription-text", false));
6870

6971
final DocSection response = new DocSection("Response", id());
7072
all.addSection(response);

src/main/java/com/github/jlangch/venice/util/openai/TokenUsage.java

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -116,18 +116,19 @@ public static TokenUsage of(final ImagesResponse response) {
116116
public static TokenUsage of(final TranscriptionCreateResponse response) {
117117
Objects.requireNonNull(response);
118118

119-
final Transcription transcription = response.asTranscription();
119+
final Transcription transcription = response.transcription().orElse(null);
120120
if (transcription == null) {
121121
return new TokenUsage();
122122
}
123123
else {
124-
final Transcription.Usage usage = transcription.usage().orElseGet(null);
125-
if (usage == null) {
124+
final Transcription.Usage usage = transcription.usage().orElse(null);
125+
if (usage == null || !usage.isTokens()) {
126126
return new TokenUsage();
127127
}
128-
Transcription.Usage.Tokens tokens = usage.asTokens();
129128

130-
Transcription.Usage.Tokens.InputTokenDetails details = tokens.inputTokenDetails().orElse(null);
129+
final Transcription.Usage.Tokens tokens = usage.asTokens();
130+
131+
final Transcription.Usage.Tokens.InputTokenDetails details = tokens.inputTokenDetails().orElse(null);
131132

132133
final long audioTokens = details == null ? 0 : details.audioTokens().orElse(0L);
133134
final long textTokens = details == null ? 0 : details.textTokens().orElse(0L);

src/main/resources/com/github/jlangch/venice/openai-java.venice

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1626,6 +1626,7 @@
16261626
usage [response]
16271627

16281628
{ :pre [(or (instance-of? :ChatCompletionTraditionalResponse response)
1629+
(instance-of? :TranscriptionCreateResponse response)
16291630
(instance-of? :ImagesResponse response))] }
16301631

16311632
(-<> (. :TokenUsage :of response)

0 commit comments

Comments
 (0)