From 2a64479e648d41d3f1a893274ab266d834833a40 Mon Sep 17 00:00:00 2001
From: Caren <calren24@gmail.com>
Date: Thu, 16 Oct 2025 21:26:24 -0700
Subject: [PATCH] Only initialize Gemini Nano if remote_config specifies using
 Nano

---
 .../androidify/data/GeminiNanoGenerationDataSource.kt    | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/data/src/main/java/com/android/developers/androidify/data/GeminiNanoGenerationDataSource.kt b/data/src/main/java/com/android/developers/androidify/data/GeminiNanoGenerationDataSource.kt
index ad05d44e..cc5e880c 100644
--- a/data/src/main/java/com/android/developers/androidify/data/GeminiNanoGenerationDataSource.kt
+++ b/data/src/main/java/com/android/developers/androidify/data/GeminiNanoGenerationDataSource.kt
@@ -15,6 +15,7 @@
  */
 package com.android.developers.androidify.data
 
+import com.android.developers.androidify.RemoteConfigDataSource
 import timber.log.Timber
 import javax.inject.Inject
 import javax.inject.Singleton
@@ -25,11 +26,15 @@ interface GeminiNanoGenerationDataSource {
 }
 
 @Singleton
-internal class GeminiNanoGenerationDataSourceImpl @Inject constructor(private val downloader: GeminiNanoDownloader) :
+internal class GeminiNanoGenerationDataSourceImpl @Inject constructor(
+    private val remoteConfigDataSource: RemoteConfigDataSource,
+    private val downloader: GeminiNanoDownloader) :
     GeminiNanoGenerationDataSource {
 
     override suspend fun initialize() {
-        downloader.downloadModel()
+        if (remoteConfigDataSource.useGeminiNano()) {
+            downloader.downloadModel()
+        }
     }
 
     /**