diff --git a/common/src/test/java/com/google/ai/client/generativeai/common/StreamingSnapshotTests.kt b/common/src/test/java/com/google/ai/client/generativeai/common/StreamingSnapshotTests.kt index 333b9c2a..5acb58a3 100644 --- a/common/src/test/java/com/google/ai/client/generativeai/common/StreamingSnapshotTests.kt +++ b/common/src/test/java/com/google/ai/client/generativeai/common/StreamingSnapshotTests.kt @@ -25,12 +25,12 @@ import io.kotest.assertions.throwables.shouldThrow import io.kotest.matchers.nulls.shouldNotBeNull import io.kotest.matchers.shouldBe import io.ktor.http.HttpStatusCode +import kotlin.time.Duration.Companion.seconds import kotlinx.coroutines.flow.collect import kotlinx.coroutines.flow.first import kotlinx.coroutines.flow.toList import kotlinx.coroutines.withTimeout import org.junit.Test -import kotlin.time.Duration.Companion.seconds internal class StreamingSnapshotTests { private val testTimeout = 5.seconds @@ -91,7 +91,6 @@ internal class StreamingSnapshotTests { val part = responseList.first().candidates?.first()?.content?.parts?.first() as? TextPart part.shouldNotBeNull() part.text shouldsContain "\"" - } }