@animaapp/anima-sdk-react 0.2.0 → 0.2.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@animaapp/anima-sdk-react",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "type": "module",
   "description": "Anima's JavaScript utilities library",
   "author": "Anima App, Inc.",

package/src/useAnimaCodegen.ts

@@ -4,7 +4,6 @@ import type {
   GetCodeParams,
   StreamCodgenMessage,
 } from "@animaapp/anima-sdk";
-import { convertCodegenFilesToAnimaFiles } from "@animaapp/anima-sdk";
 import { EventSource } from "eventsource";
 import { useImmer } from "use-immer";
 
@@ -160,12 +159,7 @@ export const useAnimaCodegen = ({
           event.data
         ) as StreamMessageByType<"generating_code">;
         if (message.payload.status === "success") {
-          const codegenFiles = message.payload.files as Record<
-            string,
-            { code: string; type: "code" }
-          >;
-
-          result.files = convertCodegenFilesToAnimaFiles(codegenFiles);
+          result.files = message.payload.files;
         }
 
         updateStatus((draft) => {
@@ -225,7 +219,12 @@ export const useAnimaCodegen = ({
         }
       });
 
-      es.addEventListener("done", () => {
+      es.addEventListener("done", (event) => {
+        const message = JSON.parse(
+          event.data
+        ) as StreamMessageByType<"done">;
+        result.tokenUsage = message.payload.tokenUsage;
+
         updateStatus((draft) => {
           draft.status = "success";
           draft.result = result as AnimaSDKResult;

package/src/utils.ts

@@ -1,8 +1,8 @@
 export const arrayBufferToBase64 = (buffer: ArrayBuffer) => {
-  var binary = "";
-  var bytes = new Uint8Array(buffer);
-  var len = bytes.byteLength;
-  for (var i = 0; i < len; i++) {
+  let binary = "";
+  const bytes = new Uint8Array(buffer);
+  const len = bytes.byteLength;
+  for (let i = 0; i < len; i++) {
     binary += String.fromCharCode(bytes[i]);
   }