Skip to content

Commit 91f932e

Browse files
committed
GeminiServiceFactory - support for explicit timeouts
1 parent f3ac0b8 commit 91f932e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,12 +41,13 @@ object GeminiServiceFactory extends GeminiServiceConsts with EnvHelper {
4141
* @return
4242
*/
4343
def asOpenAI(
44-
apiKey: String = getEnvValue(apiKeyEnv)
44+
apiKey: String = getEnvValue(apiKeyEnv),
45+
timeouts: Option[Timeouts] = None
4546
)(
4647
implicit ec: ExecutionContext,
4748
materializer: Materializer
4849
): OpenAIChatCompletionStreamedService =
4950
new OpenAIGeminiChatCompletionService(
50-
new GeminiServiceImpl(apiKey)
51+
new GeminiServiceImpl(apiKey, timeouts)
5152
)
5253
}

0 commit comments

Comments
 (0)