Add Agent Engine module. (#3429)

modules/agent-engine/variables.tf

@@ -0,0 +1,124 @@
+/**
+ * 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.
+ */
+
+variable "agent_engine_config" {
+  description = "The agent configuration."
+  type = object({
+    # Add validation once API stabilizes
+    agent_framework       = string
+    class_methods         = optional(list(any), [])
+    environment_variables = optional(map(string), {})
+    python_version        = optional(string, "3.12")
+    secret_environment_variables = optional(map(object({
+      secret_id = string
+      version   = optional(string, "latest")
+    })), {})
+  })
+}
+
+variable "bucket_config" {
+  description = "The GCS bucket configuration."
+  type = object({
+    create                      = optional(bool, true)
+    deletion_protection         = optional(bool, true)
+    name                        = optional(string)
+    uniform_bucket_level_access = optional(bool, true)
+  })
+  nullable = false
+  default  = {}
+}
+
+variable "context" {
+  description = "Context-specific interpolations."
+  type = object({
+    custom_roles   = optional(map(string), {})
+    iam_principals = optional(map(string), {})
+    locations      = optional(map(string), {})
+    kms_keys       = optional(map(string), {})
+    project_ids    = optional(map(string), {})
+  })
+  nullable = false
+  default  = {}
+}
+
+variable "description" {
+  description = "The Agent Engine description."
+  type        = string
+  nullable    = false
+  default     = "Terraform managed."
+}
+
+variable "encryption_key" {
+  description = "The full resource name of the Cloud KMS CryptoKey."
+  type        = string
+  default     = null
+}
+
+variable "generate_pickle" {
+  description = "Generate the pickle file from a source file."
+  type        = bool
+  nullable    = false
+  default     = true
+}
+
+variable "name" {
+  description = "The name of the agent."
+  type        = string
+  nullable    = false
+}
+
+variable "project_id" {
+  description = "The id of the project where to deploy the agent."
+  type        = string
+  nullable    = false
+}
+
+variable "region" {
+  description = "The region where to deploy the agent."
+  type        = string
+  nullable    = false
+}
+
+variable "service_account_config" {
+  description = "Service account configurations."
+  type = object({
+    create = optional(bool, true)
+    email  = optional(string)
+    name   = optional(string)
+    roles = optional(list(string), [
+      "roles/aiplatform.user",
+      "roles/storage.objectViewer",
+      # TODO: remove when b/441480710 is solved
+      "roles/viewer"
+    ])
+  })
+  nullable = false
+  default  = {}
+}
+
+variable "source_files" {
+  description = "The to source files path and names."
+  type = object({
+    dependencies        = optional(string, "dependencies.tar.gz")
+    path                = optional(string, "./src")
+    pickle_out          = optional(string, "pickle.pkl")
+    pickle_src          = optional(string, "agent.py")
+    pickle_src_var_name = optional(string, "local_agent")
+    requirements        = optional(string, "requirements.txt")
+  })
+  nullable = false
+  default  = {}
+}