refactor: Replace List of ImageGenerated by String in ImageGenerations

Author: renatoarg

ychat/src/commonMain/kotlin/co/yml/ychat/domain/mapper/ImageGenerationsMapper.kt

@@ -5,9 +5,9 @@ import co.yml.ychat.data.dto.ImageGenerationsParamsDto
 import co.yml.ychat.domain.model.ImageGenerated
 import co.yml.ychat.domain.model.ImageGenerationsParams
 
-internal fun ImageGenerationsDto.toImageGenerated(): List<ImageGenerated> {
+internal fun ImageGenerationsDto.toImageGenerated(): List<String> {
     return this.data.map {
-        ImageGenerated(it.url)
+        it.url
     }
 }
 

ychat/src/commonMain/kotlin/co/yml/ychat/domain/usecases/ImageGenerationsUseCase.kt

@@ -8,7 +8,7 @@ import co.yml.ychat.domain.model.ImageGenerationsParams
 
 internal data class ImageGenerationsUseCase(private val chatGptApi: ChatGptApi) {
 
-    suspend fun requestImageGenerations(params: ImageGenerationsParams): List<ImageGenerated> {
+    suspend fun requestImageGenerations(params: ImageGenerationsParams): List<String> {
         val requestDto = params.toImageGenerationsParamsDto()
         val response = chatGptApi.imageGenerations(requestDto)
         return response.getBodyOrThrow().toImageGenerated()

ychat/src/commonMain/kotlin/co/yml/ychat/entrypoint/features/ImageGenerations.kt

@@ -14,7 +14,7 @@ interface ImageGenerations {
     fun setResponseFormat(responseFormat: String): ImageGenerations
 
     @Throws(CancellationException::class, ChatGptException::class)
-    suspend fun execute(prompt: String): List<ImageGenerated>
+    suspend fun execute(prompt: String): List<String>
 
     fun execute(prompt: String, callback: YChat.Callback<List<String>>)
 }

ychat/src/commonMain/kotlin/co/yml/ychat/entrypoint/impl/ImageGenerationsImpl.kt

@@ -34,15 +34,15 @@ internal class ImageGenerationsImpl(
         return this
     }
 
-    override suspend fun execute(prompt: String): List<ImageGenerated> {
+    override suspend fun execute(prompt: String): List<String> {
         params.prompt = prompt
         return imageGenerationsUseCase.requestImageGenerations(params)
     }
 
     override fun execute(prompt: String, callback: YChat.Callback<List<String>>) {
         scope.launch {
             kotlin.runCatching { execute(prompt) }
-                .onSuccess { callback.onSuccess(it.map { it.url }) }
+                .onSuccess { callback.onSuccess(it) }
                 .onFailure { callback.onError(it) }
         }
     }