Fixed catch in abort examples (#128)

ollama · Aug 6, 2024 · 8607c28 · 8607c28
1 parent f73dae5
commit 8607c28
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 18 deletions.
diff --git a/examples/abort/any-request.ts b/examples/abort/any-request.ts
@@ -6,8 +6,7 @@ setTimeout(() => {
   ollama.abort()
 }, 1000) // 1000 milliseconds = 1 second
 
-try {
-  ollama.generate({
+ollama.generate({
     model: 'llama3.1',
     prompt: 'Write a long story',
     stream: true,
@@ -17,11 +16,12 @@ try {
         process.stdout.write(chunk.response)
       }
     }
+  ).catch(
+    (error) => {
+      if (error.name === 'AbortError') {
+        console.log('The request has been aborted')
+      } else {
+        console.error('An error occurred:', error)
+      }
+    }
   )
-} catch (error) {
-  if (error.name === 'AbortError') {
-    console.log('The request has been aborted')
-  } else {
-    console.error('An error occurred:', error)
-  }
-}
diff --git a/examples/abort/specific-request.ts b/examples/abort/specific-request.ts
@@ -9,8 +9,7 @@ setTimeout(() => {
   stream.abort()
 }, 1000) // 1000 milliseconds = 1 second
 
-try {
-  ollama.generate({
+ollama.generate({
     model: 'llama3.1',
     prompt: 'Write a long story',
     stream: true,
@@ -21,11 +20,12 @@ try {
         process.stdout.write(chunk.response)
       }
     }
+  ).catch(
+    (error) => {
+      if (error.name === 'AbortError') {
+        console.log('The request has been aborted')
+      } else {
+        console.error('An error occurred:', error)
+      }
+    }
   )
-} catch (error) {
-  if (error.name === 'AbortError') {
-    console.log('The request has been aborted')
-  } else {
-    console.error('An error occurred:', error)
-  }
-}