react-native-nitro-mlx 0.3.0 → 0.4.0

package/nitrogen/generated/ios/swift/Func_void_std__shared_ptr_ArrayBuffer_.swift

@@ -0,0 +1,47 @@
+///
+/// Func_void_std__shared_ptr_ArrayBuffer_.swift
+/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
+/// https://github.com/mrousavy/nitro
+/// Copyright © 2026 Marc Rousavy @ Margelo
+///
+
+import Foundation
+import NitroModules
+
+/**
+ * Wraps a Swift `(_ value: ArrayBuffer) -> Void` as a class.
+ * This class can be used from C++, e.g. to wrap the Swift closure as a `std::function`.
+ */
+public final class Func_void_std__shared_ptr_ArrayBuffer_ {
+  public typealias bridge = margelo.nitro.mlxreactnative.bridge.swift
+
+  private let closure: (_ value: ArrayBuffer) -> Void
+
+  public init(_ closure: @escaping (_ value: ArrayBuffer) -> Void) {
+    self.closure = closure
+  }
+
+  @inline(__always)
+  public func call(value: ArrayBuffer) -> Void {
+    self.closure(value)
+  }
+
+  /**
+   * Casts this instance to a retained unsafe raw pointer.
+   * This acquires one additional strong reference on the object!
+   */
+  @inline(__always)
+  public func toUnsafe() -> UnsafeMutableRawPointer {
+    return Unmanaged.passRetained(self).toOpaque()
+  }
+
+  /**
+   * Casts an unsafe pointer to a `Func_void_std__shared_ptr_ArrayBuffer_`.
+   * The pointer has to be a retained opaque `Unmanaged<Func_void_std__shared_ptr_ArrayBuffer_>`.
+   * This removes one strong reference from the object!
+   */
+  @inline(__always)
+  public static func fromUnsafe(_ pointer: UnsafeMutableRawPointer) -> Func_void_std__shared_ptr_ArrayBuffer_ {
+    return Unmanaged<Func_void_std__shared_ptr_ArrayBuffer_>.fromOpaque(pointer).takeRetainedValue()
+  }
+}

package/nitrogen/generated/ios/swift/GenerationStats.swift

@@ -19,8 +19,8 @@ public extension GenerationStats {
   /**
    * Create a new instance of `GenerationStats`.
    */
-  init(tokenCount: Double, tokensPerSecond: Double, timeToFirstToken: Double, totalTime: Double) {
-    self.init(tokenCount, tokensPerSecond, timeToFirstToken, totalTime)
+  init(tokenCount: Double, tokensPerSecond: Double, timeToFirstToken: Double, totalTime: Double, toolExecutionTime: Double) {
+    self.init(tokenCount, tokensPerSecond, timeToFirstToken, totalTime, toolExecutionTime)
   }
 
   var tokenCount: Double {
@@ -66,4 +66,15 @@ public extension GenerationStats {
       self.__totalTime = newValue
     }
   }
+  
+  var toolExecutionTime: Double {
+    @inline(__always)
+    get {
+      return self.__toolExecutionTime
+    }
+    @inline(__always)
+    set {
+      self.__toolExecutionTime = newValue
+    }
+  }
 }

package/nitrogen/generated/ios/swift/HybridLLMSpec.swift

@@ -21,6 +21,7 @@ public protocol HybridLLMSpec_protocol: HybridObject {
   func load(modelId: String, options: LLMLoadOptions?) throws -> Promise<Void>
   func generate(prompt: String) throws -> Promise<String>
   func stream(prompt: String, onToken: @escaping (_ token: String) -> Void, onToolCall: ((_ toolName: String, _ args: String) -> Void)?) throws -> Promise<String>
+  func streamWithEvents(prompt: String, onEvent: @escaping (_ eventJson: String) -> Void) throws -> Promise<String>
   func stop() throws -> Void
   func unload() throws -> Void
   func getLastGenerationStats() throws -> GenerationStats

package/nitrogen/generated/ios/swift/HybridLLMSpec_cxx.swift

@@ -232,6 +232,30 @@ open class HybridLLMSpec_cxx {
     }
   }
   
+  @inline(__always)
+  public final func streamWithEvents(prompt: std.string, onEvent: bridge.Func_void_std__string) -> bridge.Result_std__shared_ptr_Promise_std__string___ {
+    do {
+      let __result = try self.__implementation.streamWithEvents(prompt: String(prompt), onEvent: { () -> (String) -> Void in
+        let __wrappedFunction = bridge.wrap_Func_void_std__string(onEvent)
+        return { (__eventJson: String) -> Void in
+          __wrappedFunction.call(std.string(__eventJson))
+        }
+      }())
+      let __resultCpp = { () -> bridge.std__shared_ptr_Promise_std__string__ in
+        let __promise = bridge.create_std__shared_ptr_Promise_std__string__()
+        let __promiseHolder = bridge.wrap_std__shared_ptr_Promise_std__string__(__promise)
+        __result
+          .then({ __result in __promiseHolder.resolve(std.string(__result)) })
+          .catch({ __error in __promiseHolder.reject(__error.toCpp()) })
+        return __promise
+      }()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__resultCpp)
+    } catch (let __error) {
+      let __exceptionPtr = __error.toCpp()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__exceptionPtr)
+    }
+  }
+  
   @inline(__always)
   public final func stop() -> bridge.Result_void_ {
     do {

package/nitrogen/generated/ios/swift/HybridSTTSpec.swift

@@ -0,0 +1,66 @@
+///
+/// HybridSTTSpec.swift
+/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
+/// https://github.com/mrousavy/nitro
+/// Copyright © 2026 Marc Rousavy @ Margelo
+///
+
+import Foundation
+import NitroModules
+
+/// See ``HybridSTTSpec``
+public protocol HybridSTTSpec_protocol: HybridObject {
+  // Properties
+  var isLoaded: Bool { get }
+  var isTranscribing: Bool { get }
+  var isListening: Bool { get }
+  var modelId: String { get }
+
+  // Methods
+  func load(modelId: String, options: STTLoadOptions?) throws -> Promise<Void>
+  func transcribe(audio: ArrayBuffer) throws -> Promise<String>
+  func transcribeStream(audio: ArrayBuffer, onToken: @escaping (_ token: String) -> Void) throws -> Promise<String>
+  func startListening() throws -> Promise<Void>
+  func transcribeBuffer() throws -> Promise<String>
+  func stopListening() throws -> Promise<String>
+  func stop() throws -> Void
+  func unload() throws -> Void
+}
+
+public extension HybridSTTSpec_protocol {
+  /// Default implementation of ``HybridObject.toString``
+  func toString() -> String {
+    return "[HybridObject STT]"
+  }
+}
+
+/// See ``HybridSTTSpec``
+open class HybridSTTSpec_base {
+  private weak var cxxWrapper: HybridSTTSpec_cxx? = nil
+  public init() { }
+  public func getCxxWrapper() -> HybridSTTSpec_cxx {
+  #if DEBUG
+    guard self is HybridSTTSpec else {
+      fatalError("`self` is not a `HybridSTTSpec`! Did you accidentally inherit from `HybridSTTSpec_base` instead of `HybridSTTSpec`?")
+    }
+  #endif
+    if let cxxWrapper = self.cxxWrapper {
+      return cxxWrapper
+    } else {
+      let cxxWrapper = HybridSTTSpec_cxx(self as! HybridSTTSpec)
+      self.cxxWrapper = cxxWrapper
+      return cxxWrapper
+    }
+  }
+}
+
+/**
+ * A Swift base-protocol representing the STT HybridObject.
+ * Implement this protocol to create Swift-based instances of STT.
+ * ```swift
+ * class HybridSTT : HybridSTTSpec {
+ *   // ...
+ * }
+ * ```
+ */
+public typealias HybridSTTSpec = HybridSTTSpec_protocol & HybridSTTSpec_base

package/nitrogen/generated/ios/swift/HybridSTTSpec_cxx.swift

@@ -0,0 +1,286 @@
+///
+/// HybridSTTSpec_cxx.swift
+/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
+/// https://github.com/mrousavy/nitro
+/// Copyright © 2026 Marc Rousavy @ Margelo
+///
+
+import Foundation
+import NitroModules
+
+/**
+ * A class implementation that bridges HybridSTTSpec over to C++.
+ * In C++, we cannot use Swift protocols - so we need to wrap it in a class to make it strongly defined.
+ *
+ * Also, some Swift types need to be bridged with special handling:
+ * - Enums need to be wrapped in Structs, otherwise they cannot be accessed bi-directionally (Swift bug: https://github.com/swiftlang/swift/issues/75330)
+ * - Other HybridObjects need to be wrapped/unwrapped from the Swift TCxx wrapper
+ * - Throwing methods need to be wrapped with a Result<T, Error> type, as exceptions cannot be propagated to C++
+ */
+open class HybridSTTSpec_cxx {
+  /**
+   * The Swift <> C++ bridge's namespace (`margelo::nitro::mlxreactnative::bridge::swift`)
+   * from `MLXReactNative-Swift-Cxx-Bridge.hpp`.
+   * This contains specialized C++ templates, and C++ helper functions that can be accessed from Swift.
+   */
+  public typealias bridge = margelo.nitro.mlxreactnative.bridge.swift
+
+  /**
+   * Holds an instance of the `HybridSTTSpec` Swift protocol.
+   */
+  private var __implementation: any HybridSTTSpec
+
+  /**
+   * Holds a weak pointer to the C++ class that wraps the Swift class.
+   */
+  private var __cxxPart: bridge.std__weak_ptr_HybridSTTSpec_
+
+  /**
+   * Create a new `HybridSTTSpec_cxx` that wraps the given `HybridSTTSpec`.
+   * All properties and methods bridge to C++ types.
+   */
+  public init(_ implementation: any HybridSTTSpec) {
+    self.__implementation = implementation
+    self.__cxxPart = .init()
+    /* no base class */
+  }
+
+  /**
+   * Get the actual `HybridSTTSpec` instance this class wraps.
+   */
+  @inline(__always)
+  public func getHybridSTTSpec() -> any HybridSTTSpec {
+    return __implementation
+  }
+
+  /**
+   * Casts this instance to a retained unsafe raw pointer.
+   * This acquires one additional strong reference on the object!
+   */
+  public func toUnsafe() -> UnsafeMutableRawPointer {
+    return Unmanaged.passRetained(self).toOpaque()
+  }
+
+  /**
+   * Casts an unsafe pointer to a `HybridSTTSpec_cxx`.
+   * The pointer has to be a retained opaque `Unmanaged<HybridSTTSpec_cxx>`.
+   * This removes one strong reference from the object!
+   */
+  public class func fromUnsafe(_ pointer: UnsafeMutableRawPointer) -> HybridSTTSpec_cxx {
+    return Unmanaged<HybridSTTSpec_cxx>.fromOpaque(pointer).takeRetainedValue()
+  }
+
+  /**
+   * Gets (or creates) the C++ part of this Hybrid Object.
+   * The C++ part is a `std::shared_ptr<HybridSTTSpec>`.
+   */
+  public func getCxxPart() -> bridge.std__shared_ptr_HybridSTTSpec_ {
+    let cachedCxxPart = self.__cxxPart.lock()
+    if Bool(fromCxx: cachedCxxPart) {
+      return cachedCxxPart
+    } else {
+      let newCxxPart = bridge.create_std__shared_ptr_HybridSTTSpec_(self.toUnsafe())
+      __cxxPart = bridge.weakify_std__shared_ptr_HybridSTTSpec_(newCxxPart)
+      return newCxxPart
+    }
+  }
+
+  
+
+  /**
+   * Get the memory size of the Swift class (plus size of any other allocations)
+   * so the JS VM can properly track it and garbage-collect the JS object if needed.
+   */
+  @inline(__always)
+  public var memorySize: Int {
+    return MemoryHelper.getSizeOf(self.__implementation) + self.__implementation.memorySize
+  }
+
+  /**
+   * Call dispose() on the Swift class.
+   * This _may_ be called manually from JS.
+   */
+  @inline(__always)
+  public func dispose() {
+    self.__implementation.dispose()
+  }
+
+  /**
+   * Call toString() on the Swift class.
+   */
+  @inline(__always)
+  public func toString() -> String {
+    return self.__implementation.toString()
+  }
+
+  // Properties
+  public final var isLoaded: Bool {
+    @inline(__always)
+    get {
+      return self.__implementation.isLoaded
+    }
+  }
+  
+  public final var isTranscribing: Bool {
+    @inline(__always)
+    get {
+      return self.__implementation.isTranscribing
+    }
+  }
+  
+  public final var isListening: Bool {
+    @inline(__always)
+    get {
+      return self.__implementation.isListening
+    }
+  }
+  
+  public final var modelId: std.string {
+    @inline(__always)
+    get {
+      return std.string(self.__implementation.modelId)
+    }
+  }
+
+  // Methods
+  @inline(__always)
+  public final func load(modelId: std.string, options: bridge.std__optional_STTLoadOptions_) -> bridge.Result_std__shared_ptr_Promise_void___ {
+    do {
+      let __result = try self.__implementation.load(modelId: String(modelId), options: options.value)
+      let __resultCpp = { () -> bridge.std__shared_ptr_Promise_void__ in
+        let __promise = bridge.create_std__shared_ptr_Promise_void__()
+        let __promiseHolder = bridge.wrap_std__shared_ptr_Promise_void__(__promise)
+        __result
+          .then({ __result in __promiseHolder.resolve() })
+          .catch({ __error in __promiseHolder.reject(__error.toCpp()) })
+        return __promise
+      }()
+      return bridge.create_Result_std__shared_ptr_Promise_void___(__resultCpp)
+    } catch (let __error) {
+      let __exceptionPtr = __error.toCpp()
+      return bridge.create_Result_std__shared_ptr_Promise_void___(__exceptionPtr)
+    }
+  }
+  
+  @inline(__always)
+  public final func transcribe(audio: ArrayBuffer) -> bridge.Result_std__shared_ptr_Promise_std__string___ {
+    do {
+      let __result = try self.__implementation.transcribe(audio: audio)
+      let __resultCpp = { () -> bridge.std__shared_ptr_Promise_std__string__ in
+        let __promise = bridge.create_std__shared_ptr_Promise_std__string__()
+        let __promiseHolder = bridge.wrap_std__shared_ptr_Promise_std__string__(__promise)
+        __result
+          .then({ __result in __promiseHolder.resolve(std.string(__result)) })
+          .catch({ __error in __promiseHolder.reject(__error.toCpp()) })
+        return __promise
+      }()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__resultCpp)
+    } catch (let __error) {
+      let __exceptionPtr = __error.toCpp()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__exceptionPtr)
+    }
+  }
+  
+  @inline(__always)
+  public final func transcribeStream(audio: ArrayBuffer, onToken: bridge.Func_void_std__string) -> bridge.Result_std__shared_ptr_Promise_std__string___ {
+    do {
+      let __result = try self.__implementation.transcribeStream(audio: audio, onToken: { () -> (String) -> Void in
+        let __wrappedFunction = bridge.wrap_Func_void_std__string(onToken)
+        return { (__token: String) -> Void in
+          __wrappedFunction.call(std.string(__token))
+        }
+      }())
+      let __resultCpp = { () -> bridge.std__shared_ptr_Promise_std__string__ in
+        let __promise = bridge.create_std__shared_ptr_Promise_std__string__()
+        let __promiseHolder = bridge.wrap_std__shared_ptr_Promise_std__string__(__promise)
+        __result
+          .then({ __result in __promiseHolder.resolve(std.string(__result)) })
+          .catch({ __error in __promiseHolder.reject(__error.toCpp()) })
+        return __promise
+      }()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__resultCpp)
+    } catch (let __error) {
+      let __exceptionPtr = __error.toCpp()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__exceptionPtr)
+    }
+  }
+  
+  @inline(__always)
+  public final func startListening() -> bridge.Result_std__shared_ptr_Promise_void___ {
+    do {
+      let __result = try self.__implementation.startListening()
+      let __resultCpp = { () -> bridge.std__shared_ptr_Promise_void__ in
+        let __promise = bridge.create_std__shared_ptr_Promise_void__()
+        let __promiseHolder = bridge.wrap_std__shared_ptr_Promise_void__(__promise)
+        __result
+          .then({ __result in __promiseHolder.resolve() })
+          .catch({ __error in __promiseHolder.reject(__error.toCpp()) })
+        return __promise
+      }()
+      return bridge.create_Result_std__shared_ptr_Promise_void___(__resultCpp)
+    } catch (let __error) {
+      let __exceptionPtr = __error.toCpp()
+      return bridge.create_Result_std__shared_ptr_Promise_void___(__exceptionPtr)
+    }
+  }
+  
+  @inline(__always)
+  public final func transcribeBuffer() -> bridge.Result_std__shared_ptr_Promise_std__string___ {
+    do {
+      let __result = try self.__implementation.transcribeBuffer()
+      let __resultCpp = { () -> bridge.std__shared_ptr_Promise_std__string__ in
+        let __promise = bridge.create_std__shared_ptr_Promise_std__string__()
+        let __promiseHolder = bridge.wrap_std__shared_ptr_Promise_std__string__(__promise)
+        __result
+          .then({ __result in __promiseHolder.resolve(std.string(__result)) })
+          .catch({ __error in __promiseHolder.reject(__error.toCpp()) })
+        return __promise
+      }()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__resultCpp)
+    } catch (let __error) {
+      let __exceptionPtr = __error.toCpp()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__exceptionPtr)
+    }
+  }
+  
+  @inline(__always)
+  public final func stopListening() -> bridge.Result_std__shared_ptr_Promise_std__string___ {
+    do {
+      let __result = try self.__implementation.stopListening()
+      let __resultCpp = { () -> bridge.std__shared_ptr_Promise_std__string__ in
+        let __promise = bridge.create_std__shared_ptr_Promise_std__string__()
+        let __promiseHolder = bridge.wrap_std__shared_ptr_Promise_std__string__(__promise)
+        __result
+          .then({ __result in __promiseHolder.resolve(std.string(__result)) })
+          .catch({ __error in __promiseHolder.reject(__error.toCpp()) })
+        return __promise
+      }()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__resultCpp)
+    } catch (let __error) {
+      let __exceptionPtr = __error.toCpp()
+      return bridge.create_Result_std__shared_ptr_Promise_std__string___(__exceptionPtr)
+    }
+  }
+  
+  @inline(__always)
+  public final func stop() -> bridge.Result_void_ {
+    do {
+      try self.__implementation.stop()
+      return bridge.create_Result_void_()
+    } catch (let __error) {
+      let __exceptionPtr = __error.toCpp()
+      return bridge.create_Result_void_(__exceptionPtr)
+    }
+  }
+  
+  @inline(__always)
+  public final func unload() -> bridge.Result_void_ {
+    do {
+      try self.__implementation.unload()
+      return bridge.create_Result_void_()
+    } catch (let __error) {
+      let __exceptionPtr = __error.toCpp()
+      return bridge.create_Result_void_(__exceptionPtr)
+    }
+  }
+}

package/nitrogen/generated/ios/swift/HybridTTSSpec.swift

@@ -0,0 +1,63 @@
+///
+/// HybridTTSSpec.swift
+/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
+/// https://github.com/mrousavy/nitro
+/// Copyright © 2026 Marc Rousavy @ Margelo
+///
+
+import Foundation
+import NitroModules
+
+/// See ``HybridTTSSpec``
+public protocol HybridTTSSpec_protocol: HybridObject {
+  // Properties
+  var isLoaded: Bool { get }
+  var isGenerating: Bool { get }
+  var modelId: String { get }
+  var sampleRate: Double { get }
+
+  // Methods
+  func load(modelId: String, options: TTSLoadOptions?) throws -> Promise<Void>
+  func generate(text: String, options: TTSGenerateOptions?) throws -> Promise<ArrayBuffer>
+  func stream(text: String, onAudioChunk: @escaping (_ audio: ArrayBuffer) -> Void, options: TTSGenerateOptions?) throws -> Promise<Void>
+  func stop() throws -> Void
+  func unload() throws -> Void
+}
+
+public extension HybridTTSSpec_protocol {
+  /// Default implementation of ``HybridObject.toString``
+  func toString() -> String {
+    return "[HybridObject TTS]"
+  }
+}
+
+/// See ``HybridTTSSpec``
+open class HybridTTSSpec_base {
+  private weak var cxxWrapper: HybridTTSSpec_cxx? = nil
+  public init() { }
+  public func getCxxWrapper() -> HybridTTSSpec_cxx {
+  #if DEBUG
+    guard self is HybridTTSSpec else {
+      fatalError("`self` is not a `HybridTTSSpec`! Did you accidentally inherit from `HybridTTSSpec_base` instead of `HybridTTSSpec`?")
+    }
+  #endif
+    if let cxxWrapper = self.cxxWrapper {
+      return cxxWrapper
+    } else {
+      let cxxWrapper = HybridTTSSpec_cxx(self as! HybridTTSSpec)
+      self.cxxWrapper = cxxWrapper
+      return cxxWrapper
+    }
+  }
+}
+
+/**
+ * A Swift base-protocol representing the TTS HybridObject.
+ * Implement this protocol to create Swift-based instances of TTS.
+ * ```swift
+ * class HybridTTS : HybridTTSSpec {
+ *   // ...
+ * }
+ * ```
+ */
+public typealias HybridTTSSpec = HybridTTSSpec_protocol & HybridTTSSpec_base