// 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 PredictionService
  import Foundation
  #if canImport(FoundationNetworking)
    import FoundationNetworking
  #endif
  import GoogleApi
  import GoogleCloudLocation
  import GoogleCloudWkt
  import GoogleIAMV1
  import GoogleLongRunning
  import GoogleCloudGax
  import struct Logging.Logger

  extension Clients {
    final class PredictionServiceLogging: PredictionServiceStub {
      let inner: any PredictionServiceStub
      let logger: Logger

      public init(_ inner: any PredictionServiceStub, logger: Logger) {
        var logger = logger
        logger[metadataKey: "gcp.artifact.id"] = "google-cloud-aiplatform-v1"
        logger[metadataKey: "gcp.client.service"] = "aiplatform"
        logger[metadataKey: "gcp.experimental.swift.client"] = "PredictionService"
        self.inner = inner
        self.logger = logger
      }

      func _intercept<Input, Output>(
        request: Input,
        options: GoogleCloudGax.RequestOptions,
        name: Swift.String,
        action: (Input, GoogleCloudGax.RequestOptions) async throws -> Output,
      ) async throws -> Output {
        var logger = logger
        logger[metadataKey: "gcp.experimental.swift.request.id"] = "\(UUID())"
        logger[metadataKey: "gcp.experimental.swift.method"] = .string(name)
        logger.debug("enter  : \(request) \(options)")
        do {
          let output = try await action(request, options)
          logger.debug("success: \(request) \(options) \(output)")
          return output
        } catch let error {
          logger.debug("error  : \(request) \(options) \(error)")
          throw error
        }
      }

      public func predict(
        request: PredictRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudAIPlatformV1.PredictResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "predict",
          action: {
            (r: PredictRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleCloudAIPlatformV1.PredictResponse
            in
            return try await self.inner.predict(request: r, options: o)
          })
      }

      public func rawPredict(
        request: RawPredictRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleApi.HttpBody {
        try await self._intercept(
          request: request,
          options: options,
          name: "rawPredict",
          action: {
            (r: RawPredictRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleApi.HttpBody
            in
            return try await self.inner.rawPredict(request: r, options: o)
          })
      }

      public func streamRawPredict(
        request: StreamRawPredictRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleApi.HttpBody {
        try await self._intercept(
          request: request,
          options: options,
          name: "streamRawPredict",
          action: {
            (r: StreamRawPredictRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleApi.HttpBody
            in
            return try await self.inner.streamRawPredict(request: r, options: o)
          })
      }

      public func directPredict(
        request: DirectPredictRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudAIPlatformV1.DirectPredictResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "directPredict",
          action: {
            (r: DirectPredictRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleCloudAIPlatformV1.DirectPredictResponse
            in
            return try await self.inner.directPredict(request: r, options: o)
          })
      }

      public func directRawPredict(
        request: DirectRawPredictRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudAIPlatformV1.DirectRawPredictResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "directRawPredict",
          action: {
            (r: DirectRawPredictRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleCloudAIPlatformV1.DirectRawPredictResponse
            in
            return try await self.inner.directRawPredict(request: r, options: o)
          })
      }

      public func serverStreamingPredict(
        request: StreamingPredictRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudAIPlatformV1.StreamingPredictResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "serverStreamingPredict",
          action: {
            (r: StreamingPredictRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleCloudAIPlatformV1.StreamingPredictResponse
            in
            return try await self.inner.serverStreamingPredict(request: r, options: o)
          })
      }

      public func explain(
        request: ExplainRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudAIPlatformV1.ExplainResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "explain",
          action: {
            (r: ExplainRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleCloudAIPlatformV1.ExplainResponse
            in
            return try await self.inner.explain(request: r, options: o)
          })
      }

      public func generateContent(
        request: GenerateContentRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudAIPlatformV1.GenerateContentResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "generateContent",
          action: {
            (r: GenerateContentRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleCloudAIPlatformV1.GenerateContentResponse
            in
            return try await self.inner.generateContent(request: r, options: o)
          })
      }

      public func streamGenerateContent(
        request: GenerateContentRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudAIPlatformV1.GenerateContentResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "streamGenerateContent",
          action: {
            (r: GenerateContentRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleCloudAIPlatformV1.GenerateContentResponse
            in
            return try await self.inner.streamGenerateContent(request: r, options: o)
          })
      }

      public func embedContent(
        request: EmbedContentRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudAIPlatformV1.EmbedContentResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "embedContent",
          action: {
            (r: EmbedContentRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleCloudAIPlatformV1.EmbedContentResponse
            in
            return try await self.inner.embedContent(request: r, options: o)
          })
      }

      public func listLocations(
        request: GoogleCloudLocation.ListLocationsRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudLocation.ListLocationsResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "listLocations",
          action: {
            (r: GoogleCloudLocation.ListLocationsRequest, o: GoogleCloudGax.RequestOptions)
              async throws -> GoogleCloudLocation.ListLocationsResponse
            in
            return try await self.inner.listLocations(request: r, options: o)
          })
      }

      public func getLocation(
        request: GoogleCloudLocation.GetLocationRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleCloudLocation.Location {
        try await self._intercept(
          request: request,
          options: options,
          name: "getLocation",
          action: {
            (r: GoogleCloudLocation.GetLocationRequest, o: GoogleCloudGax.RequestOptions)
              async throws -> GoogleCloudLocation.Location
            in
            return try await self.inner.getLocation(request: r, options: o)
          })
      }

      public func setIamPolicy(
        request: GoogleIAMV1.SetIamPolicyRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleIAMV1.Policy {
        try await self._intercept(
          request: request,
          options: options,
          name: "setIamPolicy",
          action: {
            (r: GoogleIAMV1.SetIamPolicyRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleIAMV1.Policy
            in
            return try await self.inner.setIamPolicy(request: r, options: o)
          })
      }

      public func getIamPolicy(
        request: GoogleIAMV1.GetIamPolicyRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleIAMV1.Policy {
        try await self._intercept(
          request: request,
          options: options,
          name: "getIamPolicy",
          action: {
            (r: GoogleIAMV1.GetIamPolicyRequest, o: GoogleCloudGax.RequestOptions) async throws
              -> GoogleIAMV1.Policy
            in
            return try await self.inner.getIamPolicy(request: r, options: o)
          })
      }

      public func testIamPermissions(
        request: GoogleIAMV1.TestIamPermissionsRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleIAMV1.TestIamPermissionsResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "testIamPermissions",
          action: {
            (r: GoogleIAMV1.TestIamPermissionsRequest, o: GoogleCloudGax.RequestOptions)
              async throws -> GoogleIAMV1.TestIamPermissionsResponse
            in
            return try await self.inner.testIamPermissions(request: r, options: o)
          })
      }

      public func listOperations(
        request: GoogleLongRunning.ListOperationsRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleLongRunning.ListOperationsResponse {
        try await self._intercept(
          request: request,
          options: options,
          name: "listOperations",
          action: {
            (r: GoogleLongRunning.ListOperationsRequest, o: GoogleCloudGax.RequestOptions)
              async throws -> GoogleLongRunning.ListOperationsResponse
            in
            return try await self.inner.listOperations(request: r, options: o)
          })
      }

      public func getOperation(
        request: GoogleLongRunning.GetOperationRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleLongRunning.Operation {
        try await self._intercept(
          request: request,
          options: options,
          name: "getOperation",
          action: {
            (r: GoogleLongRunning.GetOperationRequest, o: GoogleCloudGax.RequestOptions)
              async throws -> GoogleLongRunning.Operation
            in
            return try await self.inner.getOperation(request: r, options: o)
          })
      }

      public func deleteOperation(
        request: GoogleLongRunning.DeleteOperationRequest, options: GoogleCloudGax.RequestOptions
      ) async throws {
        try await self._intercept(
          request: request,
          options: options,
          name: "deleteOperation",
          action: {
            (r: GoogleLongRunning.DeleteOperationRequest, o: GoogleCloudGax.RequestOptions)
              async throws -> Void in
            return try await self.inner.deleteOperation(request: r, options: o)
          })
      }

      public func cancelOperation(
        request: GoogleLongRunning.CancelOperationRequest, options: GoogleCloudGax.RequestOptions
      ) async throws {
        try await self._intercept(
          request: request,
          options: options,
          name: "cancelOperation",
          action: {
            (r: GoogleLongRunning.CancelOperationRequest, o: GoogleCloudGax.RequestOptions)
              async throws -> Void in
            return try await self.inner.cancelOperation(request: r, options: o)
          })
      }

      public func waitOperation(
        request: GoogleLongRunning.WaitOperationRequest, options: GoogleCloudGax.RequestOptions
      ) async throws -> GoogleLongRunning.Operation {
        try await self._intercept(
          request: request,
          options: options,
          name: "waitOperation",
          action: {
            (r: GoogleLongRunning.WaitOperationRequest, o: GoogleCloudGax.RequestOptions)
              async throws -> GoogleLongRunning.Operation
            in
            return try await self.inner.waitOperation(request: r, options: o)
          })
      }
    }
  }
#endif
