mirror of
https://github.com/firebase/firebase-ios-sdk
synced 2026-11-09 22:36:01 +05:45
Andrew Heard
bc9f9ea1da
Add FoundationModelsFake, a nested Swift package that fakes Apple's Foundation Models framework, so that GeminiLanguageModel (GLM) can build and run its unit tests on Linux and on OS versions without the real framework. It's used only for development and testing. - GLM depends on the fake through a `FoundationModelsFake` package trait, which is on by default where Darwin isn't available (e.g., Linux). With the trait off, the fake isn't built. The root `Package.swift` and podspecs can't reach it. - The fake mirrors Apple's public API and availability in the 27.2 SDKs, including `@Generable` macro expansions. It omits PrivateCloudComputeLanguageModel, OCRTool, BarcodeReaderTool, the adapter and feedback APIs, and deprecated APIs. SystemLanguageModel is always unavailable. Below Swift 6.4 it compiles to an empty module. - Decoding rejects transcript images with `file:` URLs and schemas whose references never resolve, instead of reading local files or overflowing the stack. - Without ImageIO, GLM passes PNG, JPEG, WebP, HEIC, and HEIF images through to Gemini after checking their signatures, and rejects images that would need re-encoding. - Add CI jobs that run both packages' tests against the fake: one on Linux in a digest-pinned `swift:6.4.2` container, and one on macOS with Xcode 27.2. Skip `sdk.ai` for changes that only touch the fake.
106 lines
3.8 KiB
Swift
106 lines
3.8 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.
|
|
|
|
#if compiler(>=6.4) && (FoundationModelsFake || canImport(FoundationModels))
|
|
import Foundation
|
|
#if FoundationModelsFake
|
|
import FoundationModelsFake
|
|
#else
|
|
import FoundationModels
|
|
#endif
|
|
import GeminiAPIClient
|
|
import GeminiTestUtilities
|
|
import Testing
|
|
|
|
@testable import GeminiLanguageModel
|
|
|
|
/// Integration tests for content generation using `GeminiLanguageModel` with a basic feature set.
|
|
@Suite(
|
|
"Basic Content Generation Integration Tests",
|
|
.requireFoundationModels,
|
|
.tags(.integration)
|
|
)
|
|
struct BasicContentGenerationIntegrationTests {
|
|
@Test(
|
|
.requireIntegrationTestingBackend,
|
|
arguments: IntegrationTestingBackend.availableBackends
|
|
)
|
|
@available(macOS 27.0, iOS 27.0, watchOS 27.0, visionOS 27.0, *)
|
|
func sessionRespond(backend: IntegrationTestingBackend) async throws {
|
|
let model = try await backend.makeModel()
|
|
let session = LanguageModelSession(model: model)
|
|
|
|
let response = try await session.respond(to: "Reply with the single word 'HELLO'.")
|
|
|
|
#expect(!response.content.isEmpty)
|
|
#expect(response.content.localizedCaseInsensitiveContains("HELLO"))
|
|
#expect(response.usage.totalTokenCount > 0)
|
|
#expect(response.usage.input.totalTokenCount > 0)
|
|
#expect(response.usage.output.totalTokenCount > 0)
|
|
}
|
|
|
|
@Test(
|
|
.requireIntegrationTestingBackend,
|
|
arguments: IntegrationTestingBackend.availableBackends
|
|
)
|
|
@available(macOS 27.0, iOS 27.0, watchOS 27.0, visionOS 27.0, *)
|
|
func sessionRespondMultiTurn(backend: IntegrationTestingBackend) async throws {
|
|
let model = try await backend.makeModel()
|
|
let session = LanguageModelSession(model: model)
|
|
|
|
_ = try await session.respond(to: "My favorite color is teal.")
|
|
let response = try await session.respond(
|
|
to: "What is my favorite color? Answer in one word."
|
|
)
|
|
|
|
#expect(!response.content.isEmpty)
|
|
#expect(response.content.localizedCaseInsensitiveContains("teal"))
|
|
#expect(response.usage.totalTokenCount > 0)
|
|
#expect(response.usage.input.totalTokenCount > 0)
|
|
#expect(response.usage.output.totalTokenCount > 0)
|
|
}
|
|
|
|
@Test(
|
|
.requireIntegrationTestingBackend,
|
|
arguments: IntegrationTestingBackend.availableBackends
|
|
)
|
|
@available(macOS 27.0, iOS 27.0, watchOS 27.0, visionOS 27.0, *)
|
|
func sessionStreamResponse(backend: IntegrationTestingBackend) async throws {
|
|
let model = try await backend.makeModel()
|
|
let session = LanguageModelSession(model: model)
|
|
|
|
let stream = session.streamResponse(
|
|
to: "List the numbers 1 through 5, one per line."
|
|
)
|
|
var finalContent = ""
|
|
var snapshotCount = 0
|
|
var finalUsage: LanguageModelSession.Usage?
|
|
for try await snapshot in stream {
|
|
finalContent = snapshot.content
|
|
snapshotCount += 1
|
|
finalUsage = snapshot.usage
|
|
}
|
|
|
|
#expect(snapshotCount > 1)
|
|
#expect(!finalContent.isEmpty)
|
|
#expect(finalContent.contains("1"))
|
|
#expect(finalContent.contains("5"))
|
|
let usage = try #require(finalUsage)
|
|
#expect(usage.totalTokenCount > 0)
|
|
#expect(usage.input.totalTokenCount > 0)
|
|
#expect(usage.output.totalTokenCount > 0)
|
|
}
|
|
}
|
|
#endif // compiler(>=6.4) && (FoundationModelsFake || canImport(FoundationModels))
|