feat: add session resumption.

PiperOrigin-RevId: 745227055

Author: MarkDaoust

src/main/java/com/google/genai/types/LiveClientSessionResumptionConfig.java

@@ -0,0 +1,85 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Auto-generated code. Do not edit.
+
+package com.google.genai.types;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.google.auto.value.AutoValue;
+import com.google.genai.JsonSerializable;
+import java.util.Optional;
+
+/**
+ * Configuration of session resumption mechanism.
+ *
+ * <p>Included in `BidiGenerateContentSetup.session_resumption`. If included server will send
+ * SessionResumptionUpdate messages.
+ */
+@AutoValue
+@JsonDeserialize(builder = LiveClientSessionResumptionConfig.Builder.class)
+public abstract class LiveClientSessionResumptionConfig extends JsonSerializable {
+  /**
+   * Session resumption handle of previous session (session to restore).
+   *
+   * <p>If not present new session will be started.
+   */
+  @JsonProperty("handle")
+  public abstract Optional<String> handle();
+
+  /**
+   * If set the server will to send `last_consumed_client_message_index` in the
+   * `session_resumption_update` messages to allow for transparent reconnections.
+   */
+  @JsonProperty("transparent")
+  public abstract Optional<Boolean> transparent();
+
+  /** Instantiates a builder for LiveClientSessionResumptionConfig. */
+  public static Builder builder() {
+    return new AutoValue_LiveClientSessionResumptionConfig.Builder();
+  }
+
+  /** Creates a builder with the same values as this instance. */
+  public abstract Builder toBuilder();
+
+  /** Builder for LiveClientSessionResumptionConfig. */
+  @AutoValue.Builder
+  public abstract static class Builder {
+    /**
+     * For internal usage. Please use `LiveClientSessionResumptionConfig.builder()` for
+     * instantiation.
+     */
+    @JsonCreator
+    private static Builder create() {
+      return new AutoValue_LiveClientSessionResumptionConfig.Builder();
+    }
+
+    @JsonProperty("handle")
+    public abstract Builder handle(String handle);
+
+    @JsonProperty("transparent")
+    public abstract Builder transparent(boolean transparent);
+
+    public abstract LiveClientSessionResumptionConfig build();
+  }
+
+  /** Deserializes a JSON string to a LiveClientSessionResumptionConfig object. */
+  public static LiveClientSessionResumptionConfig fromJson(String jsonString) {
+    return JsonSerializable.fromJsonString(jsonString, LiveClientSessionResumptionConfig.class);
+  }
+}

src/main/java/com/google/genai/types/LiveServerGoAway.java

@@ -0,0 +1,66 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Auto-generated code. Do not edit.
+
+package com.google.genai.types;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.google.auto.value.AutoValue;
+import com.google.genai.JsonSerializable;
+import java.util.Optional;
+
+/** Server will not be able to service client soon. */
+@AutoValue
+@JsonDeserialize(builder = LiveServerGoAway.Builder.class)
+public abstract class LiveServerGoAway extends JsonSerializable {
+  /**
+   * The remaining time before the connection will be terminated as ABORTED. The minimal time
+   * returned here is specified differently together with the rate limits for a given model.
+   */
+  @JsonProperty("timeLeft")
+  public abstract Optional<String> timeLeft();
+
+  /** Instantiates a builder for LiveServerGoAway. */
+  public static Builder builder() {
+    return new AutoValue_LiveServerGoAway.Builder();
+  }
+
+  /** Creates a builder with the same values as this instance. */
+  public abstract Builder toBuilder();
+
+  /** Builder for LiveServerGoAway. */
+  @AutoValue.Builder
+  public abstract static class Builder {
+    /** For internal usage. Please use `LiveServerGoAway.builder()` for instantiation. */
+    @JsonCreator
+    private static Builder create() {
+      return new AutoValue_LiveServerGoAway.Builder();
+    }
+
+    @JsonProperty("timeLeft")
+    public abstract Builder timeLeft(String timeLeft);
+
+    public abstract LiveServerGoAway build();
+  }
+
+  /** Deserializes a JSON string to a LiveServerGoAway object. */
+  public static LiveServerGoAway fromJson(String jsonString) {
+    return JsonSerializable.fromJsonString(jsonString, LiveServerGoAway.class);
+  }
+}

src/main/java/com/google/genai/types/LiveServerSessionResumptionUpdate.java

@@ -0,0 +1,101 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Auto-generated code. Do not edit.
+
+package com.google.genai.types;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.google.auto.value.AutoValue;
+import com.google.genai.JsonSerializable;
+import java.util.Optional;
+
+/**
+ * Update of the session resumption state.
+ *
+ * <p>Only sent if `session_resumption` was set in the connection config.
+ */
+@AutoValue
+@JsonDeserialize(builder = LiveServerSessionResumptionUpdate.Builder.class)
+public abstract class LiveServerSessionResumptionUpdate extends JsonSerializable {
+  /** New handle that represents state that can be resumed. Empty if `resumable`=false. */
+  @JsonProperty("newHandle")
+  public abstract Optional<String> newHandle();
+
+  /**
+   * True if session can be resumed at this point. It might be not possible to resume session at
+   * some points. In that case we send update empty new_handle and resumable=false. Example of such
+   * case could be model executing function calls or just generating. Resuming session (using
+   * previous session token) in such state will result in some data loss.
+   */
+  @JsonProperty("resumable")
+  public abstract Optional<Boolean> resumable();
+
+  /**
+   * Index of last message sent by client that is included in state represented by this
+   * SessionResumptionToken. Only sent when `SessionResumptionConfig.transparent` is set.
+   *
+   * <p>Presence of this index allows users to transparently reconnect and avoid issue of losing
+   * some part of realtime audio input/video. If client wishes to temporarily disconnect (for
+   * example as result of receiving GoAway) they can do it without losing state by buffering
+   * messages sent since last `SessionResmumptionTokenUpdate`. This field will enable them to limit
+   * buffering (avoid keeping all requests in RAM).
+   *
+   * <p>Note: This should not be used for when resuming a session at some time later -- in those
+   * cases partial audio and video frames arelikely not needed.
+   */
+  @JsonProperty("lastConsumedClientMessageIndex")
+  public abstract Optional<Long> lastConsumedClientMessageIndex();
+
+  /** Instantiates a builder for LiveServerSessionResumptionUpdate. */
+  public static Builder builder() {
+    return new AutoValue_LiveServerSessionResumptionUpdate.Builder();
+  }
+
+  /** Creates a builder with the same values as this instance. */
+  public abstract Builder toBuilder();
+
+  /** Builder for LiveServerSessionResumptionUpdate. */
+  @AutoValue.Builder
+  public abstract static class Builder {
+    /**
+     * For internal usage. Please use `LiveServerSessionResumptionUpdate.builder()` for
+     * instantiation.
+     */
+    @JsonCreator
+    private static Builder create() {
+      return new AutoValue_LiveServerSessionResumptionUpdate.Builder();
+    }
+
+    @JsonProperty("newHandle")
+    public abstract Builder newHandle(String newHandle);
+
+    @JsonProperty("resumable")
+    public abstract Builder resumable(boolean resumable);
+
+    @JsonProperty("lastConsumedClientMessageIndex")
+    public abstract Builder lastConsumedClientMessageIndex(Long lastConsumedClientMessageIndex);
+
+    public abstract LiveServerSessionResumptionUpdate build();
+  }
+
+  /** Deserializes a JSON string to a LiveServerSessionResumptionUpdate object. */
+  public static LiveServerSessionResumptionUpdate fromJson(String jsonString) {
+    return JsonSerializable.fromJsonString(jsonString, LiveServerSessionResumptionUpdate.class);
+  }
+}

src/main/java/com/google/genai/types/SessionResumptionConfig.java

@@ -0,0 +1,82 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Auto-generated code. Do not edit.
+
+package com.google.genai.types;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.google.auto.value.AutoValue;
+import com.google.genai.JsonSerializable;
+import java.util.Optional;
+
+/**
+ * Configuration of session resumption mechanism.
+ *
+ * <p>Included in `LiveConnectConfig.session_resumption`. If included server will send
+ * `LiveServerSessionResumptionUpdate` messages.
+ */
+@AutoValue
+@JsonDeserialize(builder = SessionResumptionConfig.Builder.class)
+public abstract class SessionResumptionConfig extends JsonSerializable {
+  /**
+   * Session resumption handle of previous session (session to restore).
+   *
+   * <p>If not present new session will be started.
+   */
+  @JsonProperty("handle")
+  public abstract Optional<String> handle();
+
+  /**
+   * If set the server will send `last_consumed_client_message_index` in the
+   * `session_resumption_update` messages to allow for transparent reconnections.
+   */
+  @JsonProperty("transparent")
+  public abstract Optional<Boolean> transparent();
+
+  /** Instantiates a builder for SessionResumptionConfig. */
+  public static Builder builder() {
+    return new AutoValue_SessionResumptionConfig.Builder();
+  }
+
+  /** Creates a builder with the same values as this instance. */
+  public abstract Builder toBuilder();
+
+  /** Builder for SessionResumptionConfig. */
+  @AutoValue.Builder
+  public abstract static class Builder {
+    /** For internal usage. Please use `SessionResumptionConfig.builder()` for instantiation. */
+    @JsonCreator
+    private static Builder create() {
+      return new AutoValue_SessionResumptionConfig.Builder();
+    }
+
+    @JsonProperty("handle")
+    public abstract Builder handle(String handle);
+
+    @JsonProperty("transparent")
+    public abstract Builder transparent(boolean transparent);
+
+    public abstract SessionResumptionConfig build();
+  }
+
+  /** Deserializes a JSON string to a SessionResumptionConfig object. */
+  public static SessionResumptionConfig fromJson(String jsonString) {
+    return JsonSerializable.fromJsonString(jsonString, SessionResumptionConfig.class);
+  }
+}