This commit is contained in:
68
packages/swift/Sources/Kreuzberg/SwiftEmbeddingBackendBridge.swift
generated
Normal file
68
packages/swift/Sources/Kreuzberg/SwiftEmbeddingBackendBridge.swift
generated
Normal file
@@ -0,0 +1,68 @@
|
||||
// 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\"}"
|
||||
}
|
||||
Reference in New Issue
Block a user