fix: truncated JSON formate issue (#57)

Author: zhu-xiaowei

.github/workflows/release-template.yml

@@ -1,10 +1,12 @@
 name: Release Template
 
 on:
-  push:
-    branches: [ "main" ]
-  pull_request:
-    branches: [ "main" ]
+  workflow_dispatch:
+    inputs:
+      reason:
+        description: 'the reason for triggering this workflow'
+        required: false
+        default: 'manually publish the ecr images and templates'
 jobs:
   release_template:
     runs-on: ubuntu-latest

react-native/src/api/ollama-api.ts

@@ -100,6 +100,14 @@ export const invokeOllamaWithCallBack = async (
             callback(completeMessage, done, false);
           }
           if (done) {
+            if (lastChunk.length > 0) {
+              callback(
+                completeMessage + '\n\n' + '**Parse error**:\n' + lastChunk,
+                true,
+                true,
+                undefined
+              );
+            }
             return;
           }
         } catch (readError) {
@@ -155,11 +163,10 @@ const parseStreamData = (chunk: string, lastChunk: string = '') => {
         };
       }
     } catch (error) {
-      if (lastChunk.length > 0) {
-        return { error: error + chunk };
-      }
-      if (content.length > 0) {
+      if (dataChunk === dataChunks[dataChunks.length - 1]) {
         return { content, dataChunk, usage };
+      } else {
+        return { error: chunk };
       }
     }
   }

react-native/src/api/open-api.ts

@@ -134,6 +134,15 @@ export const invokeOpenAIWithCallBack = async (
             );
           }
           if (done) {
+            if (lastChunk.length > 0) {
+              callback(
+                completeMessage + '\n\n' + '**Parse error**:\n' + lastChunk,
+                true,
+                true,
+                undefined,
+                completeReasoning
+              );
+            }
             return;
           }
         } catch (readError) {
@@ -226,17 +235,11 @@ const parseStreamData = (chunk: string, lastChunk: string = '') => {
         };
       }
     } catch (error) {
-      if (lastChunk.length > 0) {
+      if (dataChunk === dataChunks[dataChunks.length - 1]) {
         return { reason, content, dataChunk, usage };
-      } else if (reason === '' && content === '') {
-        if (dataChunk === 'data: ') {
-          return { reason, content, dataChunk, usage };
-        }
+      } else {
         return { error: chunk };
       }
-      if (reason || content) {
-        return { reason, content, dataChunk, usage };
-      }
     }
   }
   return { reason, content, usage };