fix(android): loadSession not taking paths with file:// (#66)

mybigday · Jul 22, 2024 · 39d1e05 · 39d1e05
1 parent 4c90a78
commit 39d1e05
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)
   }
 
   /**