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 index 01e9c94b..6887d0d0 100644 --- a/generativeai/src/main/java/com/google/ai/client/generativeai/Chat.kt +++ b/generativeai/src/main/java/com/google/ai/client/generativeai/Chat.kt @@ -51,7 +51,7 @@ class Chat(private val model: GenerativeModel, val history: MutableList */ suspend fun sendMessage(prompt: Content): GenerateContentResponse { prompt.assertComesFromUser() - assertNoOngoingCall() + attemptLock() val response = model.generateContent(*history.toTypedArray(), prompt) @@ -91,7 +91,7 @@ class Chat(private val model: GenerativeModel, val history: MutableList */ fun sendMessageStream(prompt: Content): Flow { prompt.assertComesFromUser() - assertNoOngoingCall() + attemptLock() val flow = model.generateContentStream(*history.toTypedArray(), prompt) val bitmaps = LinkedList() @@ -163,7 +163,7 @@ class Chat(private val model: GenerativeModel, val history: MutableList } } - private fun assertNoOngoingCall() { + private fun attemptLock() { if (!lock.tryAcquire()) { throw IllegalStateException( "This chat instance currently has an ongoing request, please wait for it to complete " +