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.
75 lines
2.4 KiB
Swift
75 lines
2.4 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)
|
|
import Foundation
|
|
import Testing
|
|
|
|
@testable import FoundationModelsFake
|
|
|
|
// MARK: - Encoding
|
|
|
|
/// Helpers for comparing encoded schemas.
|
|
enum SchemaJSON {
|
|
/// Returns a schema encoded compactly with sorted keys.
|
|
///
|
|
/// - Parameter schema: The schema to encode.
|
|
/// - Returns: The JSON text.
|
|
static func compactSortedJSON(_ schema: GenerationSchema) throws -> String {
|
|
let encoder = JSONEncoder()
|
|
encoder.outputFormatting = [.sortedKeys, .withoutEscapingSlashes]
|
|
return String(decoding: try encoder.encode(schema), as: UTF8.self)
|
|
}
|
|
}
|
|
|
|
// MARK: - Test types
|
|
|
|
/// A `Generable` type used only for its schema; its content conversions are placeholders.
|
|
protocol SchemaOnlyGenerable: Generable {}
|
|
|
|
/// The error thrown when a schema-only type is created from content.
|
|
struct SchemaOnlyError: Error {}
|
|
|
|
extension SchemaOnlyGenerable {
|
|
/// Always throws; schema-only types aren't created from content.
|
|
///
|
|
/// - Parameter content: Ignored.
|
|
init(_ content: GeneratedContent) throws {
|
|
throw SchemaOnlyError()
|
|
}
|
|
|
|
/// Placeholder content.
|
|
var generatedContent: GeneratedContent {
|
|
"".generatedContent
|
|
}
|
|
}
|
|
|
|
/// Hand-written `Generable` types, built the way the `@Generable` macro expands them.
|
|
enum SchemaTestTypes {
|
|
/// A simple object with two string properties.
|
|
struct Address: SchemaOnlyGenerable {
|
|
/// The macro-equivalent schema.
|
|
static var generationSchema: GenerationSchema {
|
|
GenerationSchema(
|
|
type: Self.self,
|
|
properties: [
|
|
GenerationSchema.Property(name: "street", type: String.self),
|
|
GenerationSchema.Property(name: "city", type: String.self),
|
|
]
|
|
)
|
|
}
|
|
}
|
|
}
|
|
#endif // compiler(>=6.4)
|