@voltagent/core 2.6.0 → 2.6.1

package/dist/index.d.mts

@@ -9410,6 +9410,9 @@ declare class Agent {
     private executeWithModelFallback;
     private probeStreamStart;
     private cloneResultWithFullStream;
+    private withProbedFullStream;
+    private usesGetterBasedTeeingFullStream;
+    private findPropertyDescriptor;
     private toAsyncIterableStream;
     private discardStream;
     /**

package/dist/index.d.ts

@@ -9410,6 +9410,9 @@ declare class Agent {
     private executeWithModelFallback;
     private probeStreamStart;
     private cloneResultWithFullStream;
+    private withProbedFullStream;
+    private usesGetterBasedTeeingFullStream;
+    private findPropertyDescriptor;
     private toAsyncIterableStream;
     private discardStream;
     /**

package/dist/index.js

@@ -31223,8 +31223,13 @@ var Agent = class {
                 );
               }, "onFinish")
             });
-            const probeResult = await this.probeStreamStart(streamResult.fullStream, attemptState);
-            const streamResultForConsumption = probeResult.stream === streamResult.fullStream ? streamResult : this.cloneResultWithFullStream(streamResult, probeResult.stream);
+            const originalFullStream = streamResult.fullStream;
+            const probeResult = await this.probeStreamStart(originalFullStream, attemptState);
+            const streamResultForConsumption = this.withProbedFullStream(
+              streamResult,
+              originalFullStream,
+              probeResult.stream
+            );
             if (probeResult.status === "error") {
               this.discardStream(streamResultForConsumption.fullStream);
               const fallbackEligible = this.shouldFallbackOnError(probeResult.error);
@@ -32095,8 +32100,13 @@ var Agent = class {
                 }
               }, "onFinish")
             });
-            const probeResult = await this.probeStreamStart(streamResult.fullStream, attemptState);
-            const streamResultForConsumption = probeResult.stream === streamResult.fullStream ? streamResult : this.cloneResultWithFullStream(streamResult, probeResult.stream);
+            const originalFullStream = streamResult.fullStream;
+            const probeResult = await this.probeStreamStart(originalFullStream, attemptState);
+            const streamResultForConsumption = this.withProbedFullStream(
+              streamResult,
+              originalFullStream,
+              probeResult.stream
+            );
             if (probeResult.status === "error") {
               this.discardStream(streamResultForConsumption.fullStream);
               const fallbackEligible = this.shouldFallbackOnError(probeResult.error);
@@ -33713,6 +33723,30 @@ ${retrieverContext}`;
     });
     return clone;
   }
+  withProbedFullStream(result, originalFullStream, probedFullStream) {
+    if (probedFullStream === originalFullStream) {
+      return result;
+    }
+    if (this.usesGetterBasedTeeingFullStream(result)) {
+      return result;
+    }
+    return this.cloneResultWithFullStream(result, probedFullStream);
+  }
+  usesGetterBasedTeeingFullStream(result) {
+    const descriptor = this.findPropertyDescriptor(result, "fullStream");
+    return typeof descriptor?.get === "function" && typeof result.teeStream === "function";
+  }
+  findPropertyDescriptor(target, propertyName) {
+    let current = target;
+    while (current) {
+      const descriptor = Object.getOwnPropertyDescriptor(current, propertyName);
+      if (descriptor) {
+        return descriptor;
+      }
+      current = Object.getPrototypeOf(current);
+    }
+    return void 0;
+  }
   toAsyncIterableStream(stream) {
     const asyncStream = stream;
     if (!asyncStream[Symbol.asyncIterator]) {