Skip to content

Commit 023db26

Browse files
author
David Motsonashvili
committed
clean up config object types
1 parent 6dd08ee commit 023db26

File tree

2 files changed

+10
-19
lines changed

2 files changed

+10
-19
lines changed

firebase-ai/src/main/kotlin/com/google/firebase/ai/type/AudioTranscriptionConfig.kt

Lines changed: 6 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -16,20 +16,12 @@
1616

1717
package com.google.firebase.ai.type
1818

19-
import kotlinx.serialization.json.JsonObject
19+
import kotlinx.serialization.Serializable
2020

21-
/**
22-
* The audio transcription configuration.
23-
* @property enable If true, the server will use Gemini to transcribe the audio.
24-
* @property prefixPrompt Prefix prompt for the audio transcription op. This is useful to override
25-
* the default prefix prompt that only asks the model to transcribe the audio. Overriding can be
26-
* useful to provide additional context to the model such as what language is expected to be spoken
27-
* in the audio.
28-
*/
29-
public class AudioTranscriptionConfig(
30-
internal val enable: Boolean? = null,
31-
internal val prefixPrompt: String? = null
32-
) {
21+
/** The audio transcription configuration. Its presence enables audio transcription */
22+
public class AudioTranscriptionConfig() {
23+
24+
@Serializable internal object Internal
3325

34-
internal fun toInternal() = JsonObject(emptyMap())
26+
internal fun toInternal() = Internal
3527
}

firebase-ai/src/main/kotlin/com/google/firebase/ai/type/LiveClientSetupMessage.kt

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ package com.google.firebase.ai.type
1818

1919
import kotlinx.serialization.ExperimentalSerializationApi
2020
import kotlinx.serialization.Serializable
21-
import kotlinx.serialization.json.JsonObject
2221

2322
/**
2423
* First message in a live session.
@@ -34,8 +33,8 @@ internal class LiveClientSetupMessage(
3433
val generationConfig: LiveGenerationConfig.Internal?,
3534
val tools: List<Tool.Internal>?,
3635
val systemInstruction: Content.Internal?,
37-
val inputAudioTranscription: JsonObject?,
38-
val outputAudioTranscription: JsonObject?,
36+
val inputAudioTranscription: AudioTranscriptionConfig.Internal?,
37+
val outputAudioTranscription: AudioTranscriptionConfig.Internal?,
3938
) {
4039
@Serializable
4140
internal class Internal(val setup: LiveClientSetup) {
@@ -45,8 +44,8 @@ internal class LiveClientSetupMessage(
4544
val generationConfig: LiveGenerationConfig.Internal?,
4645
val tools: List<Tool.Internal>?,
4746
val systemInstruction: Content.Internal?,
48-
val inputAudioTranscription: JsonObject?,
49-
val outputAudioTranscription: JsonObject?,
47+
val inputAudioTranscription: AudioTranscriptionConfig.Internal?,
48+
val outputAudioTranscription: AudioTranscriptionConfig.Internal?,
5049
)
5150
}
5251

0 commit comments

Comments
 (0)