Skip to content

Commit 2a64479

Browse files
committed
Only initialize Gemini Nano if remote_config specifies using Nano
1 parent a8c6c33 commit 2a64479

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

data/src/main/java/com/android/developers/androidify/data/GeminiNanoGenerationDataSource.kt

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
*/
1616
package com.android.developers.androidify.data
1717

18+
import com.android.developers.androidify.RemoteConfigDataSource
1819
import timber.log.Timber
1920
import javax.inject.Inject
2021
import javax.inject.Singleton
@@ -25,11 +26,15 @@ interface GeminiNanoGenerationDataSource {
2526
}
2627

2728
@Singleton
28-
internal class GeminiNanoGenerationDataSourceImpl @Inject constructor(private val downloader: GeminiNanoDownloader) :
29+
internal class GeminiNanoGenerationDataSourceImpl @Inject constructor(
30+
private val remoteConfigDataSource: RemoteConfigDataSource,
31+
private val downloader: GeminiNanoDownloader) :
2932
GeminiNanoGenerationDataSource {
3033

3134
override suspend fun initialize() {
32-
downloader.downloadModel()
35+
if (remoteConfigDataSource.useGeminiNano()) {
36+
downloader.downloadModel()
37+
}
3338
}
3439

3540
/**

0 commit comments

Comments
 (0)