File tree 1 file changed +2
-4
lines changed
firebase-ai/src/main/kotlin/com/google/firebase/ai/common 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -164,9 +164,7 @@ internal constructor(
164
164
" wss://firebasevertexai.googleapis.com/ws/google.firebase.vertexai.v1beta.LlmBidiService/BidiGenerateContent/locations/$location ?key=$key "
165
165
166
166
suspend fun getWebSocketSession (location : String ): ClientWebSocketSession =
167
- client.webSocketSession(getBidiEndpoint(location)) {
168
- applyCommonHeaders()
169
- }
167
+ client.webSocketSession(getBidiEndpoint(location)) { applyCommonHeaders() }
170
168
171
169
fun generateContentStream (
172
170
request : GenerateContentRequest
@@ -208,7 +206,7 @@ internal constructor(
208
206
is CountTokensRequest -> setBody<CountTokensRequest >(request)
209
207
is GenerateImageRequest -> setBody<GenerateImageRequest >(request)
210
208
}
211
- applyCommonHeaders()
209
+ applyCommonHeaders()
212
210
}
213
211
214
212
private suspend fun HttpRequestBuilder.applyHeaderProvider () {
You can’t perform that action at this time.
0 commit comments