From 18724d35479853b85a728626ed5c9da333ba68b5 Mon Sep 17 00:00:00 2001 From: David Motsonashvili Date: Mon, 18 Dec 2023 14:09:32 -0800 Subject: [PATCH] rename attempt lock --- .../src/main/java/com/google/ai/client/generativeai/Chat.kt | 6 +++--- 1 file changed, 3 insertions(+), 3 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 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 " +