fix: loadSession not taking paths with file://

mybigday · Jul 20, 2024 · 6d3797f · 6d3797f
1 parent 4c90a78
commit 6d3797f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/index.ts b/src/index.ts
@@ -75,7 +75,9 @@ export class LlamaContext {
    * Load cached prompt & completion state from a file.
    */
   async loadSession(filepath: string): Promise<NativeSessionLoadResult> {
-    return RNLlama.loadSession(this.id, filepath)
+    let path = filepath
+    if (path.startsWith('file://')) path = path.slice(7)
+    return RNLlama.loadSession(this.id, path)
   }
 
   /**