Update core/localmodel.mjs

core/localmodel.mjs

@@ -11,7 +11,15 @@ export const APLocalModel = {
   wllama: null,
   label: 'MiMo-V2.6-Distill-Qwen-9B (local en este navegador)',
   async cleanupCache() {
-    try { const keys = await caches.keys(); for (const k of keys) await caches.delete(k); } catch (_) {}
+    // Solo la caché del propio modelo/wllama de ESTE origen; nada más.
+    try {
+      for (const name of await caches.keys()) {
+        const c = await caches.open(name);
+        for (const req of await c.keys()) {
+          if (/wllama|MiMo-V2\.6/i.test(req.url)) await c.delete(req);
+        }
+      }
+    } catch (_) {}
   },
   async load(onProgress) {
     if (this.state === 'ready' || this.state === 'downloading') return;
@@ -39,16 +47,25 @@ export const APLocalModel = {
       throw e;
     }
   },
+  stop() { this._abort = true; },
   async chat(messages, { maxTokens = 640, temperature = 0.2, onToken } = {}) {
     if (this.state !== 'ready') throw new Error('modelo local no cargado');
-    const stream = await this.wllama.createChatCompletion({ messages, max_tokens: maxTokens, temperature, stream: true });
-    let answer = '';
-    for await (const chunk of stream) {
-      const delta = (chunk.choices && chunk.choices[0] && chunk.choices[0].delta) || {};
-      const tk = delta.content || '';
-      if (tk) { answer += tk; if (onToken) onToken(answer); }
+    this.state = 'generating'; this._abort = false;
+    let stream = null;
+    try {
+      stream = await this.wllama.createChatCompletion({ messages, max_tokens: maxTokens, temperature, stream: true });
+      let answer = '';
+      for await (const chunk of stream) {
+        if (this._abort) break; // salir del bucle cierra el iterador y frena la generación
+        const delta = (chunk.choices && chunk.choices[0] && chunk.choices[0].delta) || {};
+        const tk = delta.content || '';
+        if (tk) { answer += tk; if (onToken) onToken(answer); }
+      }
+      return answer;
+    } finally {
+      this._abort = false;
+      if (this.state === 'generating') this.state = 'ready';
     }
-    return answer;
   },
 };
 globalThis.APLocalModel = APLocalModel;