// 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
//
//     http://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.

syntax = "proto3";

package google.cloud.aiplatform.v1;

import "google/api/field_behavior.proto";
import "google/api/resource.proto";
import "google/cloud/aiplatform/v1/encryption_spec.proto";
import "google/cloud/aiplatform/v1/env_var.proto";
import "google/cloud/aiplatform/v1/service_networking.proto";
import "google/protobuf/struct.proto";
import "google/protobuf/timestamp.proto";

option csharp_namespace = "Google.Cloud.AIPlatform.V1";
option go_package = "cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb";
option java_multiple_files = true;
option java_outer_classname = "ReasoningEngineProto";
option java_package = "com.google.cloud.aiplatform.v1";
option php_namespace = "Google\\Cloud\\AIPlatform\\V1";
option ruby_package = "Google::Cloud::AIPlatform::V1";

// ReasoningEngine configurations
message ReasoningEngineSpec {
  // User provided package spec like pickled object and package requirements.
  message PackageSpec {
    // Optional. The Cloud Storage URI of the pickled python object.
    string pickle_object_gcs_uri = 1 [(google.api.field_behavior) = OPTIONAL];

    // Optional. The Cloud Storage URI of the dependency files in tar.gz format.
    string dependency_files_gcs_uri = 2
        [(google.api.field_behavior) = OPTIONAL];

    // Optional. The Cloud Storage URI of the `requirements.txt` file
    string requirements_gcs_uri = 3 [(google.api.field_behavior) = OPTIONAL];

    // Optional. The Python version. Currently support 3.8, 3.9, 3.10, 3.11.
    // If not specified, default value is 3.10.
    string python_version = 4 [(google.api.field_behavior) = OPTIONAL];
  }

  // The specification of a Reasoning Engine deployment.
  message DeploymentSpec {
    // Optional. Environment variables to be set with the Reasoning Engine
    // deployment. The environment variables can be updated through the
    // UpdateReasoningEngine API.
    repeated EnvVar env = 1 [(google.api.field_behavior) = OPTIONAL];

    // Optional. Environment variables where the value is a secret in Cloud
    // Secret Manager.
    // To use this feature, add 'Secret Manager Secret Accessor' role
    // (roles/secretmanager.secretAccessor) to AI Platform Reasoning Engine
    // Service Agent.
    repeated SecretEnvVar secret_env = 2
        [(google.api.field_behavior) = OPTIONAL];

    // Optional. Configuration for PSC-I.
    PscInterfaceConfig psc_interface_config = 4
        [(google.api.field_behavior) = OPTIONAL];

    // Optional. The minimum number of application instances that will be kept
    // running at all times. Defaults to 1. Range: [0, 10].
    optional int32 min_instances = 5 [(google.api.field_behavior) = OPTIONAL];

    // Optional. The maximum number of application instances that can be
    // launched to handle increased traffic. Defaults to 100. Range: [1, 1000].
    //
    // If VPC-SC or PSC-I is enabled, the acceptable range is [1, 100].
    optional int32 max_instances = 6 [(google.api.field_behavior) = OPTIONAL];

    // Optional. Resource limits for each container. Only 'cpu' and 'memory'
    // keys are supported. Defaults to {"cpu": "4", "memory": "4Gi"}.
    //
    //   * The only supported values for CPU are '1', '2', '4', '6' and '8'. For
    //   more information, go to
    //   https://cloud.google.com/run/docs/configuring/cpu.
    //   * The only supported values for memory are '1Gi', '2Gi', ... '32 Gi'.
    //   * For required cpu on different memory values, go to
    //   https://cloud.google.com/run/docs/configuring/memory-limits
    map<string, string> resource_limits = 7
        [(google.api.field_behavior) = OPTIONAL];

    // Optional. Concurrency for each container and agent server. Recommended
    // value: 2 * cpu + 1. Defaults to 9.
    optional int32 container_concurrency = 8
        [(google.api.field_behavior) = OPTIONAL];
  }

  // Optional. The service account that the Reasoning Engine artifact runs as.
  // It should have "roles/storage.objectViewer" for reading the user project's
  // Cloud Storage and "roles/aiplatform.user" for using Vertex extensions. If
  // not specified, the Vertex AI Reasoning Engine Service Agent in the project
  // will be used.
  optional string service_account = 1 [(google.api.field_behavior) = OPTIONAL];

  // Optional. User provided package spec of the ReasoningEngine.
  // Ignored when users directly specify a deployment image through
  // `deployment_spec.first_party_image_override`, but keeping the
  // field_behavior to avoid introducing breaking changes.
  PackageSpec package_spec = 2 [(google.api.field_behavior) = OPTIONAL];

  // Optional. The specification of a Reasoning Engine deployment.
  DeploymentSpec deployment_spec = 4 [(google.api.field_behavior) = OPTIONAL];

  // Optional. Declarations for object class methods in OpenAPI specification
  // format.
  repeated google.protobuf.Struct class_methods = 3
      [(google.api.field_behavior) = OPTIONAL];

  // Optional. The OSS agent framework used to develop the agent.
  // Currently supported values: "google-adk", "langchain", "langgraph", "ag2",
  // "llama-index", "custom".
  string agent_framework = 5 [(google.api.field_behavior) = OPTIONAL];
}

// ReasoningEngine provides a customizable runtime for models to determine
// which actions to take and in which order.
message ReasoningEngine {
  option (google.api.resource) = {
    type: "aiplatform.googleapis.com/ReasoningEngine"
    pattern: "projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}"
    plural: "reasoningEngines"
    singular: "reasoningEngine"
  };

  // Identifier. The resource name of the ReasoningEngine.
  // Format:
  // `projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}`
  string name = 1 [(google.api.field_behavior) = IDENTIFIER];

  // Required. The display name of the ReasoningEngine.
  string display_name = 2 [(google.api.field_behavior) = REQUIRED];

  // Optional. The description of the ReasoningEngine.
  string description = 7 [(google.api.field_behavior) = OPTIONAL];

  // Optional. Configurations of the ReasoningEngine
  ReasoningEngineSpec spec = 3 [(google.api.field_behavior) = OPTIONAL];

  // Output only. Timestamp when this ReasoningEngine was created.
  google.protobuf.Timestamp create_time = 4
      [(google.api.field_behavior) = OUTPUT_ONLY];

  // Output only. Timestamp when this ReasoningEngine was most recently updated.
  google.protobuf.Timestamp update_time = 5
      [(google.api.field_behavior) = OUTPUT_ONLY];

  // Optional. Used to perform consistent read-modify-write updates. If not set,
  // a blind "overwrite" update happens.
  string etag = 6 [(google.api.field_behavior) = OPTIONAL];

  // Customer-managed encryption key spec for a ReasoningEngine. If set, this
  // ReasoningEngine and all sub-resources of this ReasoningEngine will be
  // secured by this key.
  EncryptionSpec encryption_spec = 11;
}