🐛 Fix generate token usage type (#100)

* 🐛 Fix generate token usage type * 🍻 Remove tutu
polyfact · Nov 15, 2023 · 86bb86a · 86bb86a
1 parent 75a8b55
commit 86bb86a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/generate.ts b/lib/generate.ts
@@ -220,7 +220,11 @@ export class Generation extends Readable implements Promise<string> {
         return new Promise<GenerationResult>((resolve, reject) => {
             this.on("error", reject);
             this.on("infos", (data) => {
-                resolve(data);
+                resolve({
+                    result: data.result,
+                    tokenUsage: data.token_usage,
+                    ressources: data.ressources,
+                });
             });
         });
     }

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
     "name": "polyfire-js",
-    "version": "0.2.12",
+    "version": "0.2.13",
     "main": "index.js",
     "types": "index.d.ts",
     "author": "Lancelot Owczarczak <[email protected]>",