feat: add support for keep alive probe in agent engines

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/types/AgentEngineConfig.java

@@ -228,6 +228,14 @@ public abstract class AgentEngineConfig extends JsonSerializable {
   @JsonProperty("agentConfigSource")
   public abstract Optional<ReasoningEngineSpecSourceCodeSpecAgentConfigSource> agentConfigSource();
 
+  /**
+   * Optional. Specifies the configuration for keep-alive probe. Contains configuration on a
+   * specified endpoint that a deployment host should use to keep the container alive based on the
+   * probe settings.
+   */
+  @JsonProperty("keepAliveProbe")
+  public abstract Optional<KeepAliveProbe> keepAliveProbe();
+
   /** The container spec for the Agent Engine. */
   @JsonProperty("containerSpec")
   public abstract Optional<ReasoningEngineSpecContainerSpec> containerSpec();
@@ -1021,6 +1029,38 @@ public Builder clearAgentConfigSource() {
       return agentConfigSource(Optional.empty());
     }
 
+    /**
+     * Setter for keepAliveProbe.
+     *
+     * <p>keepAliveProbe: Optional. Specifies the configuration for keep-alive probe. Contains
+     * configuration on a specified endpoint that a deployment host should use to keep the container
+     * alive based on the probe settings.
+     */
+    @JsonProperty("keepAliveProbe")
+    public abstract Builder keepAliveProbe(KeepAliveProbe keepAliveProbe);
+
+    /**
+     * Setter for keepAliveProbe builder.
+     *
+     * <p>keepAliveProbe: Optional. Specifies the configuration for keep-alive probe. Contains
+     * configuration on a specified endpoint that a deployment host should use to keep the container
+     * alive based on the probe settings.
+     */
+    @CanIgnoreReturnValue
+    public Builder keepAliveProbe(KeepAliveProbe.Builder keepAliveProbeBuilder) {
+      return keepAliveProbe(keepAliveProbeBuilder.build());
+    }
+
+    @ExcludeFromGeneratedCoverageReport
+    abstract Builder keepAliveProbe(Optional<KeepAliveProbe> keepAliveProbe);
+
+    /** Clears the value of keepAliveProbe field. */
+    @ExcludeFromGeneratedCoverageReport
+    @CanIgnoreReturnValue
+    public Builder clearKeepAliveProbe() {
+      return keepAliveProbe(Optional.empty());
+    }
+
     /**
      * Setter for containerSpec.
      *

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/types/CreateAgentEngineConfig.java

@@ -173,6 +173,14 @@ public abstract class CreateAgentEngineConfig extends JsonSerializable {
   @JsonProperty("buildOptions")
   public abstract Optional<Map<String, List<String>>> buildOptions();
 
+  /**
+   * Optional. Specifies the configuration for keep-alive probe. Contains configuration on a
+   * specified endpoint that a deployment host should use to keep the container alive based on the
+   * probe settings.
+   */
+  @JsonProperty("keepAliveProbe")
+  public abstract Optional<KeepAliveProbe> keepAliveProbe();
+
   /** Instantiates a builder for CreateAgentEngineConfig. */
   @ExcludeFromGeneratedCoverageReport
   public static Builder builder() {
@@ -727,6 +735,38 @@ public Builder clearBuildOptions() {
       return buildOptions(Optional.empty());
     }
 
+    /**
+     * Setter for keepAliveProbe.
+     *
+     * <p>keepAliveProbe: Optional. Specifies the configuration for keep-alive probe. Contains
+     * configuration on a specified endpoint that a deployment host should use to keep the container
+     * alive based on the probe settings.
+     */
+    @JsonProperty("keepAliveProbe")
+    public abstract Builder keepAliveProbe(KeepAliveProbe keepAliveProbe);
+
+    /**
+     * Setter for keepAliveProbe builder.
+     *
+     * <p>keepAliveProbe: Optional. Specifies the configuration for keep-alive probe. Contains
+     * configuration on a specified endpoint that a deployment host should use to keep the container
+     * alive based on the probe settings.
+     */
+    @CanIgnoreReturnValue
+    public Builder keepAliveProbe(KeepAliveProbe.Builder keepAliveProbeBuilder) {
+      return keepAliveProbe(keepAliveProbeBuilder.build());
+    }
+
+    @ExcludeFromGeneratedCoverageReport
+    abstract Builder keepAliveProbe(Optional<KeepAliveProbe> keepAliveProbe);
+
+    /** Clears the value of keepAliveProbe field. */
+    @ExcludeFromGeneratedCoverageReport
+    @CanIgnoreReturnValue
+    public Builder clearKeepAliveProbe() {
+      return keepAliveProbe(Optional.empty());
+    }
+
     public abstract CreateAgentEngineConfig build();
   }
 

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/types/KeepAliveProbe.java

@@ -0,0 +1,120 @@
+/*
+ * 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.cloud.vertexai.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.errorprone.annotations.CanIgnoreReturnValue;
+import com.google.genai.JsonSerializable;
+import java.util.Optional;
+
+/**
+ * Represents the configuration for keep-alive probe. Contains configuration on a specified endpoint
+ * that a deployment host should use to keep the container alive based on the probe settings.
+ */
+@AutoValue
+@JsonDeserialize(builder = KeepAliveProbe.Builder.class)
+public abstract class KeepAliveProbe extends JsonSerializable {
+  /** Optional. Specifies the HTTP GET configuration for the probe. */
+  @JsonProperty("httpGet")
+  public abstract Optional<KeepAliveProbeHttpGet> httpGet();
+
+  /**
+   * Optional. Specifies the maximum duration (in seconds) to keep the instance alive via this
+   * probe. Can be a maximum of 3600 seconds (1 hour).
+   */
+  @JsonProperty("maxSeconds")
+  public abstract Optional<Integer> maxSeconds();
+
+  /** Instantiates a builder for KeepAliveProbe. */
+  @ExcludeFromGeneratedCoverageReport
+  public static Builder builder() {
+    return new AutoValue_KeepAliveProbe.Builder();
+  }
+
+  /** Creates a builder with the same values as this instance. */
+  public abstract Builder toBuilder();
+
+  /** Builder for KeepAliveProbe. */
+  @AutoValue.Builder
+  public abstract static class Builder {
+    /** For internal usage. Please use `KeepAliveProbe.builder()` for instantiation. */
+    @JsonCreator
+    private static Builder create() {
+      return new AutoValue_KeepAliveProbe.Builder();
+    }
+
+    /**
+     * Setter for httpGet.
+     *
+     * <p>httpGet: Optional. Specifies the HTTP GET configuration for the probe.
+     */
+    @JsonProperty("httpGet")
+    public abstract Builder httpGet(KeepAliveProbeHttpGet httpGet);
+
+    /**
+     * Setter for httpGet builder.
+     *
+     * <p>httpGet: Optional. Specifies the HTTP GET configuration for the probe.
+     */
+    @CanIgnoreReturnValue
+    public Builder httpGet(KeepAliveProbeHttpGet.Builder httpGetBuilder) {
+      return httpGet(httpGetBuilder.build());
+    }
+
+    @ExcludeFromGeneratedCoverageReport
+    abstract Builder httpGet(Optional<KeepAliveProbeHttpGet> httpGet);
+
+    /** Clears the value of httpGet field. */
+    @ExcludeFromGeneratedCoverageReport
+    @CanIgnoreReturnValue
+    public Builder clearHttpGet() {
+      return httpGet(Optional.empty());
+    }
+
+    /**
+     * Setter for maxSeconds.
+     *
+     * <p>maxSeconds: Optional. Specifies the maximum duration (in seconds) to keep the instance
+     * alive via this probe. Can be a maximum of 3600 seconds (1 hour).
+     */
+    @JsonProperty("maxSeconds")
+    public abstract Builder maxSeconds(Integer maxSeconds);
+
+    @ExcludeFromGeneratedCoverageReport
+    abstract Builder maxSeconds(Optional<Integer> maxSeconds);
+
+    /** Clears the value of maxSeconds field. */
+    @ExcludeFromGeneratedCoverageReport
+    @CanIgnoreReturnValue
+    public Builder clearMaxSeconds() {
+      return maxSeconds(Optional.empty());
+    }
+
+    public abstract KeepAliveProbe build();
+  }
+
+  /** Deserializes a JSON string to a KeepAliveProbe object. */
+  @ExcludeFromGeneratedCoverageReport
+  public static KeepAliveProbe fromJson(String jsonString) {
+    return JsonSerializable.fromJsonString(jsonString, KeepAliveProbe.class);
+  }
+}

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/types/KeepAliveProbeHttpGet.java

@@ -0,0 +1,103 @@
+/*
+ * 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.cloud.vertexai.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.errorprone.annotations.CanIgnoreReturnValue;
+import com.google.genai.JsonSerializable;
+import java.util.Optional;
+
+/** Specifies the HTTP GET configuration for the probe. */
+@AutoValue
+@JsonDeserialize(builder = KeepAliveProbeHttpGet.Builder.class)
+public abstract class KeepAliveProbeHttpGet extends JsonSerializable {
+  /** Required. Specifies the path of the HTTP GET request (e.g., "/is_busy"). */
+  @JsonProperty("path")
+  public abstract Optional<String> path();
+
+  /** Optional. Specifies the port number on the container to which the request is sent. */
+  @JsonProperty("port")
+  public abstract Optional<Integer> port();
+
+  /** Instantiates a builder for KeepAliveProbeHttpGet. */
+  @ExcludeFromGeneratedCoverageReport
+  public static Builder builder() {
+    return new AutoValue_KeepAliveProbeHttpGet.Builder();
+  }
+
+  /** Creates a builder with the same values as this instance. */
+  public abstract Builder toBuilder();
+
+  /** Builder for KeepAliveProbeHttpGet. */
+  @AutoValue.Builder
+  public abstract static class Builder {
+    /** For internal usage. Please use `KeepAliveProbeHttpGet.builder()` for instantiation. */
+    @JsonCreator
+    private static Builder create() {
+      return new AutoValue_KeepAliveProbeHttpGet.Builder();
+    }
+
+    /**
+     * Setter for path.
+     *
+     * <p>path: Required. Specifies the path of the HTTP GET request (e.g., "/is_busy").
+     */
+    @JsonProperty("path")
+    public abstract Builder path(String path);
+
+    @ExcludeFromGeneratedCoverageReport
+    abstract Builder path(Optional<String> path);
+
+    /** Clears the value of path field. */
+    @ExcludeFromGeneratedCoverageReport
+    @CanIgnoreReturnValue
+    public Builder clearPath() {
+      return path(Optional.empty());
+    }
+
+    /**
+     * Setter for port.
+     *
+     * <p>port: Optional. Specifies the port number on the container to which the request is sent.
+     */
+    @JsonProperty("port")
+    public abstract Builder port(Integer port);
+
+    @ExcludeFromGeneratedCoverageReport
+    abstract Builder port(Optional<Integer> port);
+
+    /** Clears the value of port field. */
+    @ExcludeFromGeneratedCoverageReport
+    @CanIgnoreReturnValue
+    public Builder clearPort() {
+      return port(Optional.empty());
+    }
+
+    public abstract KeepAliveProbeHttpGet build();
+  }
+
+  /** Deserializes a JSON string to a KeepAliveProbeHttpGet object. */
+  @ExcludeFromGeneratedCoverageReport
+  public static KeepAliveProbeHttpGet fromJson(String jsonString) {
+    return JsonSerializable.fromJsonString(jsonString, KeepAliveProbeHttpGet.class);
+  }
+}

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/types/ReasoningEngineSpecDeploymentSpec.java

@@ -91,6 +91,14 @@ public abstract class ReasoningEngineSpecDeploymentSpec extends JsonSerializable
   @JsonProperty("secretEnv")
   public abstract Optional<List<SecretEnvVar>> secretEnv();
 
+  /**
+   * Optional. Specifies the configuration for keep-alive probe. Contains configuration on a
+   * specified endpoint that a deployment host should use to keep the container alive based on the
+   * probe settings.
+   */
+  @JsonProperty("keepAliveProbe")
+  public abstract Optional<KeepAliveProbe> keepAliveProbe();
+
   /** Instantiates a builder for ReasoningEngineSpecDeploymentSpec. */
   @ExcludeFromGeneratedCoverageReport
   public static Builder builder() {
@@ -350,6 +358,38 @@ public Builder clearSecretEnv() {
       return secretEnv(Optional.empty());
     }
 
+    /**
+     * Setter for keepAliveProbe.
+     *
+     * <p>keepAliveProbe: Optional. Specifies the configuration for keep-alive probe. Contains
+     * configuration on a specified endpoint that a deployment host should use to keep the container
+     * alive based on the probe settings.
+     */
+    @JsonProperty("keepAliveProbe")
+    public abstract Builder keepAliveProbe(KeepAliveProbe keepAliveProbe);
+
+    /**
+     * Setter for keepAliveProbe builder.
+     *
+     * <p>keepAliveProbe: Optional. Specifies the configuration for keep-alive probe. Contains
+     * configuration on a specified endpoint that a deployment host should use to keep the container
+     * alive based on the probe settings.
+     */
+    @CanIgnoreReturnValue
+    public Builder keepAliveProbe(KeepAliveProbe.Builder keepAliveProbeBuilder) {
+      return keepAliveProbe(keepAliveProbeBuilder.build());
+    }
+
+    @ExcludeFromGeneratedCoverageReport
+    abstract Builder keepAliveProbe(Optional<KeepAliveProbe> keepAliveProbe);
+
+    /** Clears the value of keepAliveProbe field. */
+    @ExcludeFromGeneratedCoverageReport
+    @CanIgnoreReturnValue
+    public Builder clearKeepAliveProbe() {
+      return keepAliveProbe(Optional.empty());
+    }
+
     public abstract ReasoningEngineSpecDeploymentSpec build();
   }