fix error handling in LlamaEvent

also fixes #4
alexrozanski · May 3, 2023 · 74eb467 · 74eb467
1 parent 616a2b9
commit 74eb467
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Sources/llama/LlamaRunner.swift b/Sources/llama/LlamaRunner.swift
@@ -36,7 +36,7 @@ public class LlamaRunner {
     case initializing
     case generatingOutput
     case completed
-    case failed(error: Error)
+    case failed(error: Error?)
   }
 
   public let modelURL: URL

diff --git a/Sources/llamaObjCxx/bridge/LlamaEvent.mm b/Sources/llamaObjCxx/bridge/LlamaEvent.mm
@@ -78,7 +78,7 @@ + (instancetype)completed
 
 + (instancetype)failedWithError:(nonnull NSError *)error
 {
-  _LlamaEvent *event = [[_LlamaEvent alloc] initWithEventType:LlamaEventTypeFailed data:{}];
+  _LlamaEvent *event = [[_LlamaEvent alloc] initWithEventType:LlamaEventTypeFailed data:{ .failed_error = error }];
   return event;
 }