Files
firebase-ios-sdk/GeminiLanguageModel/Sources/GeminiAPIDataModels/GenerateContent/HistoryPart.swift
T
Andrew Heard 3796a55fdf feat(ai): add GeminiLanguageModel (#16601)
Implemented `GeminiLanguageModel`, providing an adapter for Google's
Gemini models conforming to Apple's `FoundationModels.LanguageModel`
protocol. This PR includes text to text generation capabilities; other
modalities and Gemini features will be handled in follow-up PRs.

**Note:** The `GeminiLanguageModel` implementation is guarded by the
environment variable `GEMINI_LANGUAGE_MODEL`. By default none of the
`GeminiLanguageModel` implementation is included.

Key additions:
- `GeminiLanguageModel`: Implements `LanguageModel` protocol with
  reasoning capabilities, configuring execution via `ModelResource`
  and `EndpointConfiguration`.
- `GeminiLanguageModelExecutor`: Converts prompt instructions and
  history into Gemini streaming requests, yielding responses as
  `LanguageModelSession.ResponseStream`.
- `GeminiTranscriptTranslator`: Translates Apple `Transcript` entries
  and segments into Gemini `Content` structures while preserving thought
  signatures and validating supported transcript features.
- `GeminiLanguageModel.Error` and `GeminiErrorMapper`: Translates Gemini
  API, HTTP, and network errors into standard `LanguageModelError` cases
  (`rateLimited`, `timeout`, `guardrailViolation`, `refusal`, and
  `unsupportedLanguageOrLocale`) or `GeminiLanguageModel.Error` cases
  (`serviceUnavailable`, `networkFailure`, `modelNotFound`, `apiError`).
  The `APIError` payload is designed for seamless forward compatibility
  with both GenerateContent and the upcoming Interactions API.
- Testing infrastructure: Adds `ConditionTrait.requireFoundationModels`
  to conditionally run Foundation Models tests only on supported
  platforms (macOS 27+, iOS 27+, watchOS 27+, visionOS 27+), along with
  shared test constants, mock fixtures, and comprehensive unit and
  integration test suites.
2026-09-04 18:05:45 -04:00

108 lines
4.0 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 `HistoryPart`.
///
/// ### Gemini Developer API
///
/// Type: `HistoryPart`
///
/// A single message from the conversation history list. This can represent a
/// plain text message, a request to invoke a client-side function or the output
/// from invoking a client-side function.
///
/// ### Gemini Enterprise Agent Platform
///
/// Type: `HistoryPart`
///
/// A single message from the conversation history list. This can represent a
/// plain text message, a request to invoke a client-side function or the output
/// from invoking a client-side function.
package struct HistoryPart: Codable, Sendable, Equatable, Hashable {
/// Optional. A plain text message from one party in the conversation.
package let text: String?
/// Optional. A request to invoke a client-side function
package let functionCall: FunctionCall?
/// Optional. The output from a client-side function invocation
package let functionResponse: FunctionResponse?
/// Optional. Inline media bytes.
package let inlineData: Blob?
/// Optional. URI based data.
package let fileData: FileData?
/// Optional. Code generated by the model that is meant to be executed.
package let executableCode: ExecutableCode?
/// Optional. Result of executing the `ExecutableCode`.
package let codeExecutionResult: CodeExecutionResult?
/// Optional. Indicates if the part is thought from the model.
package let thought: Bool?
/// Optional. Represents an opaque signature for the thought so it can be reused in
/// subsequent requests.
package let thoughtSignature: String?
/// Creates a new `HistoryPart`.
///
/// - Parameters:
/// - text: Optional. A plain text message from one party in the conversation.
/// - functionCall: Optional. A request to invoke a client-side function
/// - functionResponse: Optional. The output from a client-side function invocation
/// - inlineData: Optional. Inline media bytes.
/// - fileData: Optional. URI based data.
/// - executableCode: Optional. Code generated by the model that is meant to be executed.
/// - codeExecutionResult: Optional. Result of executing the `ExecutableCode`.
/// - thought: Optional. Indicates if the part is thought from the model.
/// - thoughtSignature: Optional. Represents an opaque signature for the thought so it can be reused in
package init(
text: String? = nil,
functionCall: FunctionCall? = nil,
functionResponse: FunctionResponse? = nil,
inlineData: Blob? = nil,
fileData: FileData? = nil,
executableCode: ExecutableCode? = nil,
codeExecutionResult: CodeExecutionResult? = nil,
thought: Bool? = nil,
thoughtSignature: String? = nil
) {
self.text = text
self.functionCall = functionCall
self.functionResponse = functionResponse
self.inlineData = inlineData
self.fileData = fileData
self.executableCode = executableCode
self.codeExecutionResult = codeExecutionResult
self.thought = thought
self.thoughtSignature = thoughtSignature
}
enum CodingKeys: String, CodingKey {
case text = "text"
case functionCall = "functionCall"
case functionResponse = "functionResponse"
case inlineData = "inlineData"
case fileData = "fileData"
case executableCode = "executableCode"
case codeExecutionResult = "codeExecutionResult"
case thought = "thought"
case thoughtSignature = "thoughtSignature"
}
}