prevent locked state for chat if an exception occurs

google-gemini · Dec 18, 2023 · f3cf350 · f3cf350
1 parent 18724d3
commit f3cf350
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/generativeai/src/main/java/com/google/ai/client/generativeai/Chat.kt b/generativeai/src/main/java/com/google/ai/client/generativeai/Chat.kt
@@ -52,14 +52,14 @@ class Chat(private val model: GenerativeModel, val history: MutableList<Content>
   suspend fun sendMessage(prompt: Content): GenerateContentResponse {
     prompt.assertComesFromUser()
     attemptLock()
-
-    val response = model.generateContent(*history.toTypedArray(), prompt)
-
-    lock.release()
-    history.add(prompt)
-    history.add(response.candidates.first().content)
-
-    return response
+    try {
+      val response = model.generateContent(*history.toTypedArray(), prompt)
+      history.add(prompt)
+      history.add(response.candidates.first().content)
+      return response
+    } finally {
+      lock.release()
+    }
   }
 
   /**