Skip to content

Commit 445075b

Browse files
committed
nit
1 parent eba78c8 commit 445075b

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/FirebaseVertexAI.kt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,7 @@ internal constructor(
111111
return LiveGenerativeModel(
112112
"projects/${firebaseApp.options.projectId}/locations/${location}/publishers/google/models/${modelName}",
113113
firebaseApp.options.apiKey,
114+
firebaseApp,
114115
generationConfig,
115116
tools,
116117
toolConfig,

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/LiveGenerativeModel.kt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
package com.google.firebase.vertexai
1818

19+
import com.google.firebase.FirebaseApp
1920
import com.google.firebase.appcheck.interop.InteropAppCheckTokenProvider
2021
import com.google.firebase.auth.internal.InternalAuthProvider
2122
import com.google.firebase.vertexai.common.APIController
@@ -55,6 +56,7 @@ internal constructor(
5556
internal constructor(
5657
modelName: String,
5758
apiKey: String,
59+
firebaseApp: FirebaseApp,
5860
config: LiveGenerationConfig? = null,
5961
tools: List<Tool>? = null,
6062
toolConfig: ToolConfig? = null,
@@ -75,6 +77,7 @@ internal constructor(
7577
modelName,
7678
requestOptions,
7779
"gl-kotlin/${KotlinVersion.CURRENT} fire/${BuildConfig.VERSION_NAME}",
80+
firebaseApp,
7881
AppCheckHeaderProvider(TAG, appCheckTokenProvider, internalAuthProvider),
7982
),
8083
)

0 commit comments

Comments
 (0)