🐛 models: Fix crash on api error (#137)

* 🐛 models: Fix crash on api error * 🐛 tests: Increase test timeout
polyfact · Apr 30, 2024 · d5019a5 · d5019a5
1 parent 4c5b07e
commit d5019a5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/lib/generate.ts b/lib/generate.ts
@@ -366,10 +366,11 @@ function stream(
         ws.onopen = () => ws.send(JSON.stringify(requestBody));
         ws.onmessage = (data: unknown) => onMessage(data, resultStream);
         ws.onclose = () => resultStream.push(null);
+        ws.onerror = () => resultStream.push(null);
         resultStream.stop = () => {
             ws.send("STOP");
         };
-    })();
+    })().catch(console.error);
     return resultStream;
 }
 

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
     "name": "polyfire-js",
-    "version": "0.2.58",
+    "version": "0.2.59",
     "main": "index.js",
     "types": "index.d.ts",
     "author": "Lancelot Owczarczak <[email protected]>",

diff --git a/tests/generation.test.js b/tests/generation.test.js
@@ -96,4 +96,5 @@ test(
 
         expect(message).toBe("banana42");
     }),
+    10000,
 );