GeminiService - support for explicit timeouts

Author: peterbanda

google-gemini-client/src/main/scala/io/cequence/openaiscala/gemini/service/GeminiServiceFactory.scala

@@ -10,6 +10,7 @@ import io.cequence.openaiscala.service.ChatProviderSettings
 import io.cequence.openaiscala.service.StreamedServiceTypes.OpenAIChatCompletionStreamedService
 
 import scala.concurrent.ExecutionContext
+import io.cequence.wsclient.service.ws.Timeouts
 
 /**
  * Factory for creating instances of the [[GeminiService]] and an OpenAI adapter for
@@ -20,11 +21,12 @@ object GeminiServiceFactory extends GeminiServiceConsts with EnvHelper {
   private val apiKeyEnv = ChatProviderSettings.gemini.apiKeyEnvVariable
 
   def apply(
-    apiKey: String = getEnvValue(apiKeyEnv)
+    apiKey: String = getEnvValue(apiKeyEnv),
+    timeouts: Option[Timeouts] = None
   )(
     implicit ec: ExecutionContext,
     materializer: Materializer
-  ): GeminiService = new GeminiServiceImpl(apiKey)
+  ): GeminiService = new GeminiServiceImpl(apiKey, timeouts)
 
   /**
    * Create a new instance of the [[OpenAIChatCompletionService]] wrapping the SonarService

google-gemini-client/src/main/scala/io/cequence/openaiscala/gemini/service/impl/GeminiServiceImpl.scala

@@ -22,9 +22,11 @@ import io.cequence.wsclient.service.{WSClientEngine, WSClientEngineStreamExtra}
 import play.api.libs.json._
 
 import scala.concurrent.{ExecutionContext, Future}
+import io.cequence.wsclient.service.ws.Timeouts
 
 private[service] class GeminiServiceImpl(
-  apiKey: String
+  apiKey: String,
+  timeouts: Option[Timeouts] = None
 )(
   override implicit val ec: ExecutionContext,
   implicit val materializer: Materializer
@@ -40,7 +42,8 @@ private[service] class GeminiServiceImpl(
       WsRequestContext(
         extraParams = Seq(
           Param.key.toString() -> apiKey
-        )
+        ),
+        explTimeouts = timeouts
       )
     )