feat: add support for keep alive probe in agent engines

Keep alive probe allows reasoning engine users to configure a probe that a deployment host can use to keep the container alive, based on the probe settings. If the keep alive endpoint returns a 2xx status, the deployment host will make a best effort (up to 1 hour) to keep the container alive.

Reasoning engine users with custom container specs (BYOC) have the option to configure a custom keep alive probe while the users without custom container specs (BYOC) have the option to configure an empty keep alive probe {} and the reasoning engine platform will handle the configuration and logic for keep alive probe.

To opt in, users should set the keep alive probe field when creating or updating reasoning engines.

PiperOrigin-RevId: 899766876

Author: google-genai-bot

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

@@ -237,6 +237,14 @@ public abstract class AgentEngineConfig extends JsonSerializable {
   public abstract Optional<ReasoningEngineSpecDeploymentSpecAgentGatewayConfig>
       agentGatewayConfig();
 
+  /**
+   * 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 AgentEngineConfig. */
   @ExcludeFromGeneratedCoverageReport
   public static Builder builder() {
@@ -1085,6 +1093,38 @@ public Builder clearAgentGatewayConfig() {
       return agentGatewayConfig(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 AgentEngineConfig build();
   }
 

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

@@ -178,6 +178,14 @@ public abstract class CreateAgentEngineConfig extends JsonSerializable {
   public abstract Optional<ReasoningEngineSpecDeploymentSpecAgentGatewayConfig>
       agentGatewayConfig();
 
+  /**
+   * 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() {
@@ -763,6 +771,38 @@ public Builder clearAgentGatewayConfig() {
       return agentGatewayConfig(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/KeepAliveProbeHttpGet.java

@@ -30,7 +30,7 @@
 @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"`). */
+  /** Required. Specifies the path of the HTTP GET request (e.g., "/is_busy"). */
   @JsonProperty("path")
   public abstract Optional<String> path();
 
@@ -59,7 +59,7 @@ private static Builder create() {
     /**
      * Setter for path.
      *
-     * <p>path: Required. Specifies the path of the HTTP GET request (e.g., `"/is_busy"`).
+     * <p>path: Required. Specifies the path of the HTTP GET request (e.g., "/is_busy").
      */
     @JsonProperty("path")
     public abstract Builder path(String path);

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

@@ -178,6 +178,14 @@ public abstract class UpdateAgentEngineConfig extends JsonSerializable {
   public abstract Optional<ReasoningEngineSpecDeploymentSpecAgentGatewayConfig>
       agentGatewayConfig();
 
+  /**
+   * 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 update mask to apply. For the `FieldMask` definition, see
    * https://protobuf.dev/reference/protobuf/google.protobuf/#field-mask.
@@ -770,6 +778,38 @@ public Builder clearAgentGatewayConfig() {
       return agentGatewayConfig(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 updateMask.
      *