Fix ViewModel scope break and enable Mistral key rotation in service

Author: Android-PowerUser

app/src/main/kotlin/com/google/ai/sample/ScreenCaptureApiClients.kt

@@ -71,7 +71,13 @@ data class ServiceMistralResponseMessage(
     val content: String
 )
 
-internal suspend fun callMistralApi(modelName: String, apiKey: String, chatHistory: List<Content>, inputContent: Content): Pair<String?, String?> {
+internal suspend fun callMistralApi(
+    modelName: String,
+    apiKey: String,
+    chatHistory: List<Content>,
+    inputContent: Content,
+    availableApiKeys: List<String> = listOf(apiKey)
+): Pair<String?, String?> {
     var responseText: String? = null
     var errorMessage: String? = null
 
@@ -130,7 +136,8 @@ internal suspend fun callMistralApi(modelName: String, apiKey: String, chatHisto
             .addHeader("Authorization", "Bearer $apiKey")
             .build()
 
-        val coordinated = MistralRequestCoordinator.execute(apiKeys = listOf(apiKey), maxAttempts = 4) { key ->
+        val keysForCoordinator = availableApiKeys.filter { it.isNotBlank() }.distinct().ifEmpty { listOf(apiKey) }
+        val coordinated = MistralRequestCoordinator.execute(apiKeys = keysForCoordinator, maxAttempts = maxOf(4, keysForCoordinator.size * 3)) { key ->
             client.newCall(
                 request.newBuilder()
                     .header("Authorization", "Bearer $key")

app/src/main/kotlin/com/google/ai/sample/ScreenCaptureService.kt

@@ -297,7 +297,16 @@ class ScreenCaptureService : Service() {
                             if (apiProvider == ApiProvider.VERCEL) {
                                 responseText = callVercelApi(applicationContext, modelName, apiKey, chatHistoryDtos, inputContentDto)
                             } else if (apiProvider == ApiProvider.MISTRAL) {
-                                val result = callMistralApi(modelName, apiKey, chatHistory, inputContent)
+                                val availableMistralKeys = ApiKeyManager.getInstance(applicationContext)
+                                    .getApiKeys(ApiProvider.MISTRAL)
+                                    .filter { it.isNotBlank() }
+                                val result = callMistralApi(
+                                    modelName = modelName,
+                                    apiKey = apiKey,
+                                    chatHistory = chatHistory,
+                                    inputContent = inputContent,
+                                    availableApiKeys = availableMistralKeys
+                                )
                                 responseText = result.first
                                 errorMessage = result.second
                             } else if (apiProvider == ApiProvider.PUTER) {

app/src/main/kotlin/com/google/ai/sample/feature/multimodal/PhotoReasoningViewModel.kt

@@ -2076,8 +2076,6 @@ private fun processCommands(text: String) {
             refreshStopButtonState()
         }
     }
-}
-
     private fun executeAccessibilityCommand(command: Command, shouldTrackCommand: Boolean) {
         ScreenOperatorAccessibilityService.executeCommand(command)
         if (shouldTrackCommand) {