@@ -26,10 +26,8 @@ import com.google.firebase.vertexai.common.shared.FunctionCall
26
26
import com.google.firebase.vertexai.common.shared.FunctionCallPart
27
27
import com.google.firebase.vertexai.common.shared.FunctionResponse
28
28
import com.google.firebase.vertexai.common.shared.FunctionResponsePart
29
- import com.google.firebase.vertexai.common.shared.HarmBlockThreshold
30
29
import com.google.firebase.vertexai.type.BlobPart
31
30
import com.google.firebase.vertexai.type.BlockReason
32
- import com.google.firebase.vertexai.type.BlockThreshold
33
31
import com.google.firebase.vertexai.type.Candidate
34
32
import com.google.firebase.vertexai.type.Citation
35
33
import com.google.firebase.vertexai.type.CitationMetadata
@@ -41,6 +39,7 @@ import com.google.firebase.vertexai.type.FunctionCallingConfig
41
39
import com.google.firebase.vertexai.type.FunctionDeclaration
42
40
import com.google.firebase.vertexai.type.GenerateContentResponse
43
41
import com.google.firebase.vertexai.type.GenerationConfig
42
+ import com.google.firebase.vertexai.type.HarmBlockThreshold
44
43
import com.google.firebase.vertexai.type.HarmCategory
45
44
import com.google.firebase.vertexai.type.HarmProbability
46
45
import com.google.firebase.vertexai.type.HarmSeverity
@@ -138,13 +137,16 @@ internal fun ToolConfig.toInternal() =
138
137
)
139
138
)
140
139
141
- internal fun BlockThreshold .toInternal () =
140
+ internal fun HarmBlockThreshold .toInternal () =
142
141
when (this ) {
143
- BlockThreshold .NONE -> HarmBlockThreshold .BLOCK_NONE
144
- BlockThreshold .ONLY_HIGH -> HarmBlockThreshold .BLOCK_ONLY_HIGH
145
- BlockThreshold .MEDIUM_AND_ABOVE -> HarmBlockThreshold .BLOCK_MEDIUM_AND_ABOVE
146
- BlockThreshold .LOW_AND_ABOVE -> HarmBlockThreshold .BLOCK_LOW_AND_ABOVE
147
- BlockThreshold .UNSPECIFIED -> HarmBlockThreshold .UNSPECIFIED
142
+ HarmBlockThreshold .NONE ->
143
+ com.google.firebase.vertexai.common.shared.HarmBlockThreshold .BLOCK_NONE
144
+ HarmBlockThreshold .ONLY_HIGH ->
145
+ com.google.firebase.vertexai.common.shared.HarmBlockThreshold .BLOCK_ONLY_HIGH
146
+ HarmBlockThreshold .MEDIUM_AND_ABOVE ->
147
+ com.google.firebase.vertexai.common.shared.HarmBlockThreshold .BLOCK_MEDIUM_AND_ABOVE
148
+ HarmBlockThreshold .LOW_AND_ABOVE ->
149
+ com.google.firebase.vertexai.common.shared.HarmBlockThreshold .BLOCK_LOW_AND_ABOVE
148
150
}
149
151
150
152
internal fun Tool.toInternal () =
@@ -258,8 +260,7 @@ internal fun com.google.firebase.vertexai.common.server.FinishReason?.toPublic()
258
260
com.google.firebase.vertexai.common.server.FinishReason .SAFETY -> FinishReason .SAFETY
259
261
com.google.firebase.vertexai.common.server.FinishReason .STOP -> FinishReason .STOP
260
262
com.google.firebase.vertexai.common.server.FinishReason .OTHER -> FinishReason .OTHER
261
- com.google.firebase.vertexai.common.server.FinishReason .UNSPECIFIED -> FinishReason .UNSPECIFIED
262
- com.google.firebase.vertexai.common.server.FinishReason .UNKNOWN -> FinishReason .UNKNOWN
263
+ else -> FinishReason .UNKNOWN
263
264
}
264
265
265
266
internal fun com.google.firebase.vertexai.common.shared.HarmCategory.toPublic () =
@@ -270,7 +271,7 @@ internal fun com.google.firebase.vertexai.common.shared.HarmCategory.toPublic()
270
271
HarmCategory .SEXUALLY_EXPLICIT
271
272
com.google.firebase.vertexai.common.shared.HarmCategory .DANGEROUS_CONTENT ->
272
273
HarmCategory .DANGEROUS_CONTENT
273
- com.google.firebase.vertexai.common.shared. HarmCategory . UNKNOWN -> HarmCategory .UNKNOWN
274
+ else -> HarmCategory .UNKNOWN
274
275
}
275
276
276
277
internal fun com.google.firebase.vertexai.common.server.HarmProbability.toPublic () =
@@ -280,9 +281,7 @@ internal fun com.google.firebase.vertexai.common.server.HarmProbability.toPublic
280
281
com.google.firebase.vertexai.common.server.HarmProbability .LOW -> HarmProbability .LOW
281
282
com.google.firebase.vertexai.common.server.HarmProbability .NEGLIGIBLE ->
282
283
HarmProbability .NEGLIGIBLE
283
- com.google.firebase.vertexai.common.server.HarmProbability .UNSPECIFIED ->
284
- HarmProbability .UNSPECIFIED
285
- com.google.firebase.vertexai.common.server.HarmProbability .UNKNOWN -> HarmProbability .UNKNOWN
284
+ else -> HarmProbability .UNKNOWN
286
285
}
287
286
288
287
internal fun com.google.firebase.vertexai.common.server.HarmSeverity.toPublic () =
@@ -291,16 +290,14 @@ internal fun com.google.firebase.vertexai.common.server.HarmSeverity.toPublic()
291
290
com.google.firebase.vertexai.common.server.HarmSeverity .MEDIUM -> HarmSeverity .MEDIUM
292
291
com.google.firebase.vertexai.common.server.HarmSeverity .LOW -> HarmSeverity .LOW
293
292
com.google.firebase.vertexai.common.server.HarmSeverity .NEGLIGIBLE -> HarmSeverity .NEGLIGIBLE
294
- com.google.firebase.vertexai.common.server.HarmSeverity .UNSPECIFIED -> HarmSeverity .UNSPECIFIED
295
- com.google.firebase.vertexai.common.server.HarmSeverity .UNKNOWN -> HarmSeverity .UNKNOWN
293
+ else -> HarmSeverity .UNKNOWN
296
294
}
297
295
298
296
internal fun com.google.firebase.vertexai.common.server.BlockReason.toPublic () =
299
297
when (this ) {
300
- com.google.firebase.vertexai.common.server.BlockReason .UNSPECIFIED -> BlockReason .UNSPECIFIED
301
298
com.google.firebase.vertexai.common.server.BlockReason .SAFETY -> BlockReason .SAFETY
302
299
com.google.firebase.vertexai.common.server.BlockReason .OTHER -> BlockReason .OTHER
303
- com.google.firebase.vertexai.common.server. BlockReason . UNKNOWN -> BlockReason .UNKNOWN
300
+ else -> BlockReason .UNKNOWN
304
301
}
305
302
306
303
internal fun com.google.firebase.vertexai.common.GenerateContentResponse.toPublic ():
0 commit comments