fix: usage clear issue and response status issue for done

aws-samples · Nov 20, 2024 · c525c46 · c525c46
1 parent 50a9125
commit c525c46
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/react-native/src/api/bedrock-api.ts b/react-native/src/api/bedrock-api.ts
@@ -73,9 +73,6 @@ export const invokeBedrockWithCallBack = async (
         const decoder = new TextDecoder();
         while (true) {
           const { done, value } = await reader.read();
-          if (done) {
-            break;
-          }
           const chunk = decoder.decode(value, { stream: true });
           if (
             chunk[chunk.length - 1] === '}' &&
@@ -95,7 +92,10 @@ export const invokeBedrockWithCallBack = async (
             callback(completeMessage, true, false, usage);
           } else {
             completeMessage += chunk;
-            callback(completeMessage, false, false);
+            callback(completeMessage, done, false);
+          }
+          if (done) {
+            break;
           }
         }
       })

diff --git a/react-native/src/chat/ChatScreen.tsx b/react-native/src/chat/ChatScreen.tsx
@@ -177,6 +177,7 @@ function ChatScreen(): React.JSX.Element {
       modeRef.current = mode;
       setChatStatus(ChatStatus.Init);
       sendEventRef.current('');
+      setUsage(undefined);
       if (initialSessionId === 0 || initialSessionId === -1) {
         startNewChat.current();
         return;