From 1d2ec809c117a407eeef21d0d5b091dbae7ad031 Mon Sep 17 00:00:00 2001 From: David Motsonashvili Date: Tue, 28 Jan 2025 10:00:14 -0800 Subject: [PATCH] Add apiVersion parameter to RequestOptions --- .../com/google/firebase/vertexai/type/RequestOptions.kt | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/RequestOptions.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/RequestOptions.kt index 9aa648b6d07..ca87712ec22 100644 --- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/RequestOptions.kt +++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/RequestOptions.kt @@ -33,10 +33,12 @@ internal constructor( * Constructor for RequestOptions. * * @param timeoutInMillis the maximum amount of time, in milliseconds, for a request to take, from - * the first request to first response. + * the first request to first response. + * @param apiVersion changes the vertex API version the request is made to */ @JvmOverloads public constructor( - timeoutInMillis: Long = 180.seconds.inWholeMilliseconds - ) : this(timeout = timeoutInMillis.toDuration(DurationUnit.MILLISECONDS)) + timeoutInMillis: Long = 180.seconds.inWholeMilliseconds, + apiVersion: String = "v1beta", + ) : this(timeout = timeoutInMillis.toDuration(DurationUnit.MILLISECONDS), apiVersion = apiVersion) }