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: vertex-sdk-bot

src/genai/types/common.ts

@@ -276,7 +276,7 @@ export declare interface ReasoningEngineSpecDeploymentSpecAgentGatewayConfig {
 
 /** Specifies the HTTP GET configuration for the probe. */
 export declare interface KeepAliveProbeHttpGet {
-  /** 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"). */
   path?: string;
   /** Optional. Specifies the port number on the container to which the request is sent. */
   port?: number;
@@ -715,6 +715,10 @@ export declare interface CreateAgentEngineConfig {
   buildOptions?: Record<string, string[]>;
   /** Agent Gateway configuration for a Reasoning Engine deployment. */
   agentGatewayConfig?: ReasoningEngineSpecDeploymentSpecAgentGatewayConfig;
+  /** 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. */
+  keepAliveProbe?: KeepAliveProbe;
 }
 
 /** Parameters for creating agent engines. */
@@ -1043,6 +1047,10 @@ export declare interface UpdateAgentEngineConfig {
   buildOptions?: Record<string, string[]>;
   /** Agent Gateway configuration for a Reasoning Engine deployment. */
   agentGatewayConfig?: ReasoningEngineSpecDeploymentSpecAgentGatewayConfig;
+  /** 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. */
+  keepAliveProbe?: KeepAliveProbe;
   /** The update mask to apply. For the `FieldMask` definition, see
       https://protobuf.dev/reference/protobuf/google.protobuf/#field-mask. */
   updateMask?: string;
@@ -2610,6 +2618,10 @@ export declare interface AgentEngineConfig {
   containerSpec?: ReasoningEngineSpecContainerSpec;
   /** Agent Gateway configuration for a Reasoning Engine deployment. */
   agentGatewayConfig?: ReasoningEngineSpecDeploymentSpecAgentGatewayConfig;
+  /** 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. */
+  keepAliveProbe?: KeepAliveProbe;
 }
 
 /** Config for checking a query job on an agent engine. */