69 lines
2.1 KiB
Swift
69 lines
2.1 KiB
Swift
|
// Generated by alef. Do not edit by hand.
|
||
|
|
// swift-format-ignore-file
|
||
|
|
// This file contains generated FFI glue for trait bridge registration.
|
||
|
|
|
||
|
|
import Foundation
|
||
|
|
import RustBridge
|
||
|
|
|
||
|
|
/// Protocol for outbound `EmbeddingBackend` implementations.
|
||
|
|
/// Conform your Swift class or struct to this protocol to implement
|
||
|
|
/// a Rust trait from the host side.
|
||
|
|
public protocol SwiftEmbeddingBackendBridge: AnyObject {
|
||
|
|
func dimensions() -> Int
|
||
|
|
func embed(texts: [String]) throws -> [[Float]]
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Internal adapter wrapping a `SwiftEmbeddingBackendBridge` conformer.
|
||
|
|
/// Marshals Swift types and trait calls to/from the C boundary.
|
||
|
|
/// Excluded/internal types are serialised to/from JSON strings.
|
||
|
|
final class SwiftEmbeddingBackendAdapter {
|
||
|
|
private let bridge: any SwiftEmbeddingBackendBridge
|
||
|
|
|
||
|
|
init(bridge: any SwiftEmbeddingBackendBridge) {
|
||
|
|
self.bridge = bridge
|
||
|
|
}
|
||
|
|
|
||
|
|
func dimensionsCall() -> Int {
|
||
|
|
let result = self.bridge.dimensions()
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func embedCall(texts: [String]) throws -> String {
|
||
|
|
do {
|
||
|
|
let result = try self.bridge.embed(texts: texts)
|
||
|
|
return marshal_ok_result(try JSONEncoder().encode(result))
|
||
|
|
} catch {
|
||
|
|
return marshal_error_result(error)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// MARK: - Marshalling helpers
|
||
|
|
|
||
|
|
private struct Empty: Codable {}
|
||
|
|
|
||
|
|
private func marshal_ok_result<T: Encodable>(_ value: T) -> String {
|
||
|
|
let encoder = JSONEncoder()
|
||
|
|
if let data = try? encoder.encode(value),
|
||
|
|
let jsonString = String(data: data, encoding: .utf8) {
|
||
|
|
return "{\"ok\": \(jsonString)}"
|
||
|
|
}
|
||
|
|
return "{\"ok\": null}"
|
||
|
|
}
|
||
|
|
|
||
|
|
private func marshal_encode_excluded<T: Encodable>(_ value: T) throws -> Data {
|
||
|
|
let encoder = JSONEncoder()
|
||
|
|
return try encoder.encode(value)
|
||
|
|
}
|
||
|
|
|
||
|
|
private func marshal_error_result(_ error: any Error) -> String {
|
||
|
|
let errorString = String(describing: error)
|
||
|
|
let encoder = JSONEncoder()
|
||
|
|
if let data = try? encoder.encode(errorString),
|
||
|
|
let jsonString = String(data: data, encoding: .utf8) {
|
||
|
|
return "{\"err\": \(jsonString)}"
|
||
|
|
}
|
||
|
|
return "{\"err\": \"unknown error\"}"
|
||
|
|
}
|