// Code generated by sidekick. DO NOT EDIT.
//
// Copyright 2026 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.

#if VertexRagDataService
  import Foundation
  import GoogleCloudWkt

  /// Config for the embedding model to use for RAG.
  public struct RagEmbeddingModelConfig: Codable, Equatable, GoogleCloudWkt._AnyPackable,
    Sendable
  {
    /// The model config to use.
    public var modelConfig: OneOf_ModelConfig? = nil

    /// Initialize a new instance of `RagEmbeddingModelConfig`.
    public init() {}

    /// Use `config` to return a new instance of this object, with some fields updated.
    ///
    /// Commonly used to initialize the value, for example:
    ///
    /// ```
    /// let value = RagEmbeddingModelConfig().with { $0.vertexPredictionEndpoint = ... }
    /// ```
    public func with(_ config: (inout Self) throws -> Swift.Void) rethrows -> Self {
      var copy = self
      try config(&copy)
      return copy
    }

    private enum CodingKeys: Swift.String, CodingKey {
      case vertexPredictionEndpoint = "vertexPredictionEndpoint"
    }

    public init(from decoder: Decoder) throws {
      let container = try decoder.container(keyedBy: CodingKeys.self)

      var modelConfig: OneOf_ModelConfig? = nil
      let modelConfigCheckAndSet = {
        if modelConfig != nil {
          throw DecodingError.dataCorrupted(
            DecodingError.Context(
              codingPath: decoder.codingPath,
              debugDescription: "Multiple values set for oneof 'modelConfig'"))
        }
        modelConfig = $0
      }
      if let vertexPredictionEndpoint = try container.decodeIfPresent(
        RagEmbeddingModelConfig.VertexPredictionEndpoint?.self, forKey: .vertexPredictionEndpoint)
      {
        try modelConfigCheckAndSet(.vertexPredictionEndpoint(vertexPredictionEndpoint))
      }
      self.modelConfig = modelConfig
    }

    public func encode(to encoder: Encoder) throws {
      var container = encoder.container(keyedBy: CodingKeys.self)

      if let choice = self.modelConfig {
        switch choice {
        case .vertexPredictionEndpoint(let value):
          try container.encode(value, forKey: .vertexPredictionEndpoint)
        }
      }
    }

    /// Config representing a model hosted on Vertex Prediction Endpoint.
    public struct VertexPredictionEndpoint: Codable, Equatable, GoogleCloudWkt._AnyPackable,
      Sendable
    {
      /// Required. The endpoint resource name.
      /// Format:
      /// `projects/{project}/locations/{location}/publishers/{publisher}/models/{model}`
      /// or
      /// `projects/{project}/locations/{location}/endpoints/{endpoint}`
      public var endpoint: Swift.String = Swift.String()

      /// Output only. The resource name of the model that is deployed on the
      /// endpoint. Present only when the endpoint is not a publisher model.
      /// Pattern:
      /// `projects/{project}/locations/{location}/models/{model}`
      public var model: Swift.String = Swift.String()

      /// Output only. Version ID of the model that is deployed on the endpoint.
      /// Present only when the endpoint is not a publisher model.
      public var modelVersionId: Swift.String = Swift.String()

      /// Initialize a new instance of `VertexPredictionEndpoint`.
      public init() {}

      /// Use `config` to return a new instance of this object, with some fields updated.
      ///
      /// Commonly used to initialize the value, for example:
      ///
      /// ```
      /// let value = VertexPredictionEndpoint().with { $0.endpoint = ... }
      /// ```
      public func with(_ config: (inout Self) throws -> Swift.Void) rethrows -> Self {
        var copy = self
        try config(&copy)
        return copy
      }

      public static var _anyTypeUrl: Swift.String {
        return
          "type.googleapis.com/google.cloud.aiplatform.v1.RagEmbeddingModelConfig.VertexPredictionEndpoint"
      }
      public init(fromAny any: GoogleCloudWkt.`Any`) throws {
        self = try GoogleCloudWkt._slowAnyDeserialize(Self.self, from: any)
      }
      public func _pack() throws -> GoogleCloudWkt.Struct {
        return try GoogleCloudWkt._slowAnySerialize(message: self)
      }
    }

    /// The model config to use.
    public enum OneOf_ModelConfig: Codable, Equatable, Sendable {
      /// The Vertex AI Prediction Endpoint that either refers to a publisher model
      /// or an endpoint that is hosting a 1P fine-tuned text embedding model.
      /// Endpoints hosting non-1P fine-tuned text embedding models are
      /// currently not supported.
      /// This is used for dense vector search.
      indirect case vertexPredictionEndpoint(RagEmbeddingModelConfig.VertexPredictionEndpoint?)
    }

    public static var _anyTypeUrl: Swift.String {
      return "type.googleapis.com/google.cloud.aiplatform.v1.RagEmbeddingModelConfig"
    }
    public init(fromAny any: GoogleCloudWkt.`Any`) throws {
      self = try GoogleCloudWkt._slowAnyDeserialize(Self.self, from: any)
    }
    public func _pack() throws -> GoogleCloudWkt.Struct {
      return try GoogleCloudWkt._slowAnySerialize(message: self)
    }
  }
#endif
