update MultiResourceComponent to cache googleAI object

Author: David Motsonashvili

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

@@ -21,7 +21,6 @@ import com.google.firebase.FirebaseApp
 import com.google.firebase.app
 import com.google.firebase.vertexai.type.Content
 import com.google.firebase.vertexai.type.GenerationConfig
-import com.google.firebase.vertexai.type.GenerativeBackend
 import com.google.firebase.vertexai.type.ImagenGenerationConfig
 import com.google.firebase.vertexai.type.ImagenSafetySettings
 import com.google.firebase.vertexai.type.PublicPreviewAPI
@@ -94,7 +93,7 @@ public class FirebaseGoogleAI internal constructor(private val proxy: FirebaseVe
     @JvmStatic
     public fun getInstance(app: FirebaseApp = Firebase.app): FirebaseGoogleAI {
       val multiResourceComponent = app[FirebaseVertexAIMultiResourceComponent::class.java]
-      return FirebaseGoogleAI(multiResourceComponent.get(GenerativeBackend.GOOGLE_AI, "UNUSED"))
+      return multiResourceComponent.getGoogle()
     }
   }
 }

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

@@ -147,7 +147,7 @@ internal constructor(
     @JvmOverloads
     public fun getInstance(app: FirebaseApp = Firebase.app, location: String): FirebaseVertexAI {
       val multiResourceComponent = app[FirebaseVertexAIMultiResourceComponent::class.java]
-      return multiResourceComponent.get(GenerativeBackend.VERTEX_AI, location)
+      return multiResourceComponent.getVertex(location)
     }
   }
 }

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

@@ -31,21 +31,39 @@ import com.google.firebase.vertexai.type.GenerativeBackend
 internal class FirebaseVertexAIMultiResourceComponent(
   private val app: FirebaseApp,
   private val appCheckProvider: Provider<InteropAppCheckTokenProvider>,
-  private val internalAuthProvider: Provider<InternalAuthProvider>
+  private val internalAuthProvider: Provider<InternalAuthProvider>,
 ) {
 
-  @GuardedBy("this") private val instances: MutableMap<String, FirebaseVertexAI> = mutableMapOf()
+  @GuardedBy("this")
+  private val vertexInstances: MutableMap<String, FirebaseVertexAI> = mutableMapOf()
 
-  fun get(generativeBackend: GenerativeBackend, location: String): FirebaseVertexAI =
+  @GuardedBy("this") private val googleInstance: MutableList<FirebaseGoogleAI> = mutableListOf()
+
+  fun getVertex(location: String): FirebaseVertexAI =
     synchronized(this) {
-      instances[location]
+      vertexInstances[location]
         ?: FirebaseVertexAI(
             app,
-            generativeBackend,
+            GenerativeBackend.VERTEX_AI,
             location,
             appCheckProvider,
-            internalAuthProvider
+            internalAuthProvider,
+          )
+          .also { vertexInstances[location] = it }
+    }
+
+  fun getGoogle(): FirebaseGoogleAI =
+    synchronized(this) {
+      googleInstance.getOrNull(0)
+        ?: FirebaseGoogleAI(
+            FirebaseVertexAI(
+              app,
+              GenerativeBackend.GOOGLE_AI,
+              "UNUSED",
+              appCheckProvider,
+              internalAuthProvider,
+            )
           )
-          .also { instances[location] = it }
+          .also { googleInstance.add(it) }
     }
 }