fix: fix stream text error

trevorwang · Apr 2, 2023 · c17a711 · c17a711
1 parent 3875353
commit c17a711
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/example/main.dart b/example/main.dart
@@ -67,11 +67,12 @@ void chatCompletionStsream(OpenaiClient client) {
       model: Model.gpt3_5Turbo,
       stream: true,
       messages: [
-        ChatMessage(content: "hello", role: ChatMessageRole.user),
+        ChatMessage(
+            content: "Act as a golang runtime", role: ChatMessageRole.user),
       ],
     ),
     onSuccess: (p0) {
-      // print(p0);
+      print(p0);
     },
   );
 }
diff --git a/lib/src/client.dart b/lib/src/client.dart
@@ -83,7 +83,9 @@ class OpenaiClient {
         if (e.startsWith(dataPrefix)) {
           final result = e.split(dataPrefix);
           if (result.length > 1) {
-            onSuccess?.call(jsonDecode(result[1]));
+            if (!result[1].contains('[DONE]')) {
+              onSuccess?.call(jsonDecode(result[1]));
+            }
           }
         } else {
           final res = jsonDecode(e);