@probelabs/probe 0.6.0-rc137 → 0.6.0-rc138

package/build/agent/ProbeAgent.js

@@ -1270,6 +1270,9 @@ When troubleshooting:
               temperature: 0.3,
             });
 
+            // Get the promise reference BEFORE consuming stream (doesn't lock it)
+            const usagePromise = result.usage;
+
             // Collect the streamed response - stream all content for now
             for await (const delta of result.textStream) {
               assistantResponseContent += delta;
@@ -1279,8 +1282,8 @@ When troubleshooting:
               }
             }
 
-            // Record token usage
-            const usage = await result.usage;
+            // Record token usage - await the promise AFTER stream is consumed
+            const usage = await usagePromise;
             if (usage) {
               this.tokenCounter.recordUsage(usage, result.experimental_providerMetadata);
             }

package/build/agent/index.js

@@ -49394,13 +49394,14 @@ You are working with a repository located at: ${searchDirectory}
                   maxTokens: maxResponseTokens,
                   temperature: 0.3
                 });
+                const usagePromise = result.usage;
                 for await (const delta of result.textStream) {
                   assistantResponseContent += delta;
                   if (options.onStream) {
                     options.onStream(delta);
                   }
                 }
-                const usage = await result.usage;
+                const usage = await usagePromise;
                 if (usage) {
                   this.tokenCounter.recordUsage(usage, result.experimental_providerMetadata);
                 }

package/cjs/agent/ProbeAgent.cjs

@@ -74587,13 +74587,14 @@ You are working with a repository located at: ${searchDirectory}
                   maxTokens: maxResponseTokens,
                   temperature: 0.3
                 });
+                const usagePromise = result.usage;
                 for await (const delta of result.textStream) {
                   assistantResponseContent += delta;
                   if (options.onStream) {
                     options.onStream(delta);
                   }
                 }
-                const usage = await result.usage;
+                const usage = await usagePromise;
                 if (usage) {
                   this.tokenCounter.recordUsage(usage, result.experimental_providerMetadata);
                 }

package/cjs/index.cjs

@@ -74729,13 +74729,14 @@ You are working with a repository located at: ${searchDirectory}
                   maxTokens: maxResponseTokens,
                   temperature: 0.3
                 });
+                const usagePromise = result.usage;
                 for await (const delta of result.textStream) {
                   assistantResponseContent += delta;
                   if (options.onStream) {
                     options.onStream(delta);
                   }
                 }
-                const usage = await result.usage;
+                const usage = await usagePromise;
                 if (usage) {
                   this.tokenCounter.recordUsage(usage, result.experimental_providerMetadata);
                 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc137",
+  "version": "0.6.0-rc138",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",

package/src/agent/ProbeAgent.js

@@ -1270,6 +1270,9 @@ When troubleshooting:
               temperature: 0.3,
             });
 
+            // Get the promise reference BEFORE consuming stream (doesn't lock it)
+            const usagePromise = result.usage;
+
             // Collect the streamed response - stream all content for now
             for await (const delta of result.textStream) {
               assistantResponseContent += delta;
@@ -1279,8 +1282,8 @@ When troubleshooting:
               }
             }
 
-            // Record token usage
-            const usage = await result.usage;
+            // Record token usage - await the promise AFTER stream is consumed
+            const usage = await usagePromise;
             if (usage) {
               this.tokenCounter.recordUsage(usage, result.experimental_providerMetadata);
             }