Files
firebase-ios-sdk/GeminiLanguageModel/Sources/GeminiAPIDataModels/GenerateContent/ImageConfig.swift
T

138 lines
5.3 KiB
Swift

// 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
//
// 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.
import Foundation
/// An internal data model for `ImageConfig`.
///
/// ### Gemini Developer API
///
/// Type: `GoogleAiGenerativelanguageV1betaImageConfig`
///
/// Config for image generation features.
///
/// ### Gemini Enterprise API
///
/// Type: `GoogleCloudAiplatformV1beta1ImageConfig`
///
/// Configuration for image generation.
///
/// This message allows you to control various aspects of image generation, such
/// as the output format, aspect ratio, and whether the model can generate
/// images of people.
package struct ImageConfig: Codable, Sendable, Equatable, Hashable {
/// Optional. The aspect ratio of the image to generate. Supported aspect ratios: `1:1`,
///
/// ### Gemini Developer API
///
/// Optional. The aspect ratio of the image to generate. Supported aspect ratios: `1:1`,
/// `1:4`, `4:1`, `1:8`, `8:1`, `2:3`, `3:2`, `3:4`, `4:3`, `4:5`, `5:4`,
/// `9:16`, `16:9`, or `21:9`.
///
/// If not specified, the model will choose a default aspect ratio based on any
/// reference images provided.
///
/// ### Gemini Enterprise API
///
/// Optional. The desired aspect ratio for the generated images. The following
/// aspect ratios are supported:
///
/// "1:1"
/// "2:3", "3:2"
/// "3:4", "4:3"
/// "4:5", "5:4"
/// "9:16", "16:9"
/// "21:9"
package let aspectRatio: String?
/// Optional. Specifies the size of generated images. Supported values are `512`, `1K`,
///
/// ### Gemini Developer API
///
/// Optional. Specifies the size of generated images. Supported values are `512`, `1K`,
/// `2K`, `4K`. If not specified, the model will use default value `1K`.
///
/// ### Gemini Enterprise API
///
/// Optional. Specifies the size of generated images. Supported values are `1K`, `2K`,
/// `4K`. If not specified, the model will use default value `1K`.
package let imageSize: String?
/// Optional. The image output format for generated images.
///
/// ### Gemini Developer API
///
/// > Important: This property is not supported in the Gemini Developer API.
///
/// ### Gemini Enterprise API
///
/// Optional. The image output format for generated images.
package let imageOutputOptions: ImageConfigImageOutputOptions?
/// Optional. Controls whether the model can generate people.
///
/// ### Gemini Developer API
///
/// > Important: This property is not supported in the Gemini Developer API.
///
/// ### Gemini Enterprise API
///
/// Optional. Controls whether the model can generate people.
package let personGeneration: PersonGeneration?
/// Optional. Controls whether prominent people (celebrities) generation is allowed. If
///
/// ### Gemini Developer API
///
/// > Important: This property is not supported in the Gemini Developer API.
///
/// ### Gemini Enterprise API
///
/// Optional. Controls whether prominent people (celebrities) generation is allowed. If
/// used with personGeneration, personGeneration enum would take precedence.
/// For instance, if ALLOW_NONE is set, all person generation would be blocked.
/// If this field is unspecified, the default behavior is to allow prominent
/// people.
package let prominentPeople: ProminentPeople?
/// Creates a new `ImageConfig`.
///
/// - Parameters:
/// - aspectRatio: Optional. The aspect ratio of the image to generate. Supported aspect ratios: `1:1`, (behavior varies by backend). For more details, see ``aspectRatio``.
/// - imageSize: Optional. Specifies the size of generated images. Supported values are `512`, `1K`, (behavior varies by backend). For more details, see ``imageSize``.
/// - imageOutputOptions: Optional. The image output format for generated images. (Gemini Enterprise API only). For more details, see ``imageOutputOptions``.
/// - personGeneration: Optional. Controls whether the model can generate people. (Gemini Enterprise API only). For more details, see ``personGeneration``.
/// - prominentPeople: Optional. Controls whether prominent people (celebrities) generation is allowed. If (Gemini Enterprise API only). For more details, see ``prominentPeople``.
package init(
aspectRatio: String? = nil,
imageSize: String? = nil,
imageOutputOptions: ImageConfigImageOutputOptions? = nil,
personGeneration: PersonGeneration? = nil,
prominentPeople: ProminentPeople? = nil
) {
self.aspectRatio = aspectRatio
self.imageSize = imageSize
self.imageOutputOptions = imageOutputOptions
self.personGeneration = personGeneration
self.prominentPeople = prominentPeople
}
enum CodingKeys: String, CodingKey {
case aspectRatio = "aspectRatio"
case imageSize = "imageSize"
case imageOutputOptions = "imageOutputOptions"
case personGeneration = "personGeneration"
case prominentPeople = "prominentPeople"
}
}