fix(ios): removeLoraAdapters

mybigday · Nov 21, 2024 · 1569797 · 1569797
1 parent 8560f03
commit 1569797
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/ios/RNLlama.mm b/ios/RNLlama.mm
@@ -277,6 +277,10 @@ - (NSArray *)supportedEvents {
                  withRejecter:(RCTPromiseRejectBlock)reject)
 {
     RNLlamaContext *context = llamaContexts[[NSNumber numberWithDouble:contextId]];
+    if (context == nil) {
+        reject(@"llama_error", @"Context not found", nil);
+        return;
+    }
     [context applyLoraAdapters:loraAdapters];
     resolve(nil);
 }
@@ -286,6 +290,10 @@ - (NSArray *)supportedEvents {
                  withRejecter:(RCTPromiseRejectBlock)reject)
 {
     RNLlamaContext *context = llamaContexts[[NSNumber numberWithDouble:contextId]];
+    if (context == nil) {
+        reject(@"llama_error", @"Context not found", nil);
+        return;
+    }
     [context removeLoraAdapters];
     resolve(nil);
 }
@@ -295,6 +303,10 @@ - (NSArray *)supportedEvents {
                  withRejecter:(RCTPromiseRejectBlock)reject)
 {
     RNLlamaContext *context = llamaContexts[[NSNumber numberWithDouble:contextId]];
+    if (context == nil) {
+        reject(@"llama_error", @"Context not found", nil);
+        return;
+    }
     resolve([context getLoadedLoraAdapters]);
 }
 

diff --git a/ios/RNLlamaContext.mm b/ios/RNLlamaContext.mm
@@ -576,7 +576,7 @@ - (void)applyLoraAdapters:(NSArray *)loraAdapters {
 }
 
 - (void)removeLoraAdapters {
-    [llama removeLoraAdapters];
+    llama->removeLoraAdapters();
 }
 
 - (NSArray *)getLoadedLoraAdapters {