Refresh health insights when AI keys change

Author: Tinnci

app/src/main/java/com/driezy/medlog/ui/screen/health/HealthViewModel.kt

@@ -4,6 +4,7 @@ import android.util.Log
 import androidx.lifecycle.ViewModel
 import com.driezy.medlog.ui.BaseViewModel
 import androidx.lifecycle.viewModelScope
+import com.driezy.medlog.ai.AiApiKeyStore
 import com.driezy.medlog.data.model.HealthRecord
 import com.driezy.medlog.data.model.HealthRecordSource
 import com.driezy.medlog.data.model.HealthType
@@ -96,6 +97,7 @@ class HealthViewModel @Inject constructor(
     private val repository: HealthRepository,
     private val prefsRepository: UserPreferencesRepository,
     private val insightGeneration: HealthInsightGenerationUseCase,
+    private val apiKeyStore: AiApiKeyStore,
     private val voiceInputController: VoiceInputController,
 ) : BaseViewModel() {
 
@@ -223,7 +225,8 @@ class HealthViewModel @Inject constructor(
             combine(
                 repository.getAllRecords(),
                 prefsRepository.settingsFlow,
-            ) { records, settings ->
+                apiKeyStore.availableProviders,
+            ) { records, settings, _ ->
                 records to settings.userHeightCm
             }
                 .catch { e -> Log.e("HealthVM", "Failed to build health insights", e) }

app/src/test/java/com/driezy/medlog/ui/screen/health/HealthInsightsAiConnectionCoverageTest.kt

@@ -0,0 +1,33 @@
+package com.driezy.medlog.ui.screen.health
+
+import java.io.File
+import org.junit.Assert.assertTrue
+import org.junit.Test
+
+class HealthInsightsAiConnectionCoverageTest {
+    private val projectRoot = generateSequence(File("").absoluteFile) { it.parentFile }
+        .first { File(it, "settings.gradle.kts").exists() }
+
+    private fun source(path: String) = File(projectRoot, path).readText()
+
+    @Test
+    fun `health insights refresh when cloud ai key availability changes`() {
+        val viewModel = source("app/src/main/java/com/driezy/medlog/ui/screen/health/HealthViewModel.kt")
+
+        assertTrue(viewModel.contains("private val apiKeyStore: AiApiKeyStore"))
+        assertTrue(viewModel.contains("apiKeyStore.availableProviders"))
+        assertTrue(viewModel.contains("generateWithStatus("))
+    }
+
+    @Test
+    fun `health insight use case reports cloud success and records health insight usage`() {
+        val useCase = source("app/src/main/java/com/driezy/medlog/domain/health/HealthInsightGenerationUseCase.kt")
+        val cache = source("app/src/main/java/com/driezy/medlog/domain/health/HealthAiPipelineCache.kt")
+
+        assertTrue(useCase.contains("HealthCloudInsightGenerationGate.evaluate"))
+        assertTrue(useCase.contains("AiExecutionStatus.CloudSuccess"))
+        assertTrue(useCase.contains("AiExecutionStatus.providerError"))
+        assertTrue(cache.contains("AiUsageFeature.HEALTH_INSIGHT"))
+        assertTrue(cache.contains("AiAnalysisKind.HEALTH_INSIGHT"))
+    }
+}