@@ -164,7 +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))
167
+ client.webSocketSession(getBidiEndpoint(location)) { applyCommonHeaders() }
168
168
169
169
fun generateContentStream (
170
170
request : GenerateContentRequest
@@ -191,12 +191,7 @@ internal constructor(
191
191
throw FirebaseCommonAIException .from(e)
192
192
}
193
193
194
- private fun HttpRequestBuilder.applyCommonConfiguration (request : Request ) {
195
- when (request) {
196
- is GenerateContentRequest -> setBody<GenerateContentRequest >(request)
197
- is CountTokensRequest -> setBody<CountTokensRequest >(request)
198
- is GenerateImageRequest -> setBody<GenerateImageRequest >(request)
199
- }
194
+ private fun HttpRequestBuilder.applyCommonHeaders () {
200
195
contentType(ContentType .Application .Json )
201
196
header(" x-goog-api-key" , key)
202
197
header(" x-goog-api-client" , apiClient)
@@ -205,6 +200,14 @@ internal constructor(
205
200
header(" X-Firebase-AppVersion" , appVersion)
206
201
}
207
202
}
203
+ private fun HttpRequestBuilder.applyCommonConfiguration (request : Request ) {
204
+ when (request) {
205
+ is GenerateContentRequest -> setBody<GenerateContentRequest >(request)
206
+ is CountTokensRequest -> setBody<CountTokensRequest >(request)
207
+ is GenerateImageRequest -> setBody<GenerateImageRequest >(request)
208
+ }
209
+ applyCommonHeaders()
210
+ }
208
211
209
212
private suspend fun HttpRequestBuilder.applyHeaderProvider () {
210
213
if (headerProvider != null ) {
0 commit comments