@@ -28,7 +28,6 @@ import com.google.firebase.vertexai.common.shared.FunctionResponse
28
28
import com.google.firebase.vertexai.common.shared.FunctionResponsePart
29
29
import com.google.firebase.vertexai.type.BlobPart
30
30
import com.google.firebase.vertexai.type.BlockReason
31
- import com.google.firebase.vertexai.type.HarmBlockThreshold
32
31
import com.google.firebase.vertexai.type.Candidate
33
32
import com.google.firebase.vertexai.type.Citation
34
33
import com.google.firebase.vertexai.type.CitationMetadata
@@ -40,6 +39,7 @@ import com.google.firebase.vertexai.type.FunctionCallingConfig
40
39
import com.google.firebase.vertexai.type.FunctionDeclaration
41
40
import com.google.firebase.vertexai.type.GenerateContentResponse
42
41
import com.google.firebase.vertexai.type.GenerationConfig
42
+ import com.google.firebase.vertexai.type.HarmBlockThreshold
43
43
import com.google.firebase.vertexai.type.HarmCategory
44
44
import com.google.firebase.vertexai.type.HarmProbability
45
45
import com.google.firebase.vertexai.type.HarmSeverity
@@ -139,11 +139,14 @@ internal fun ToolConfig.toInternal() =
139
139
140
140
internal fun HarmBlockThreshold.toInternal () =
141
141
when (this ) {
142
- HarmBlockThreshold .NONE -> com.google.firebase.vertexai.common.shared.HarmBlockThreshold .BLOCK_NONE
143
- HarmBlockThreshold .ONLY_HIGH -> com.google.firebase.vertexai.common.shared.HarmBlockThreshold .BLOCK_ONLY_HIGH
144
- HarmBlockThreshold .MEDIUM_AND_ABOVE -> com.google.firebase.vertexai.common.shared.HarmBlockThreshold .BLOCK_MEDIUM_AND_ABOVE
145
- HarmBlockThreshold .LOW_AND_ABOVE -> com.google.firebase.vertexai.common.shared.HarmBlockThreshold .BLOCK_LOW_AND_ABOVE
146
- HarmBlockThreshold .UNKNOWN -> com.google.firebase.vertexai.common.shared.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
147
150
}
148
151
149
152
internal fun Tool.toInternal () =
0 commit comments