@voltagent/core 2.6.0 → 2.6.2

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]) {
@@ -33768,18 +33802,18 @@ ${retrieverContext}`;
     const toolRouting = this.resolveToolRouting(options);
     oc.systemContext.set(TOOL_ROUTING_CONTEXT_KEY, toolRouting);
     if (toolRouting === false) {
-      const supportNames = this.getToolRoutingSupportToolNames();
-      const allNames = /* @__PURE__ */ new Set([
-        ...Object.keys(preparedStaticTools),
-        ...Object.keys(preparedDynamicTools)
-      ]);
-      const conflicts = [...allNames].filter((name) => supportNames.has(name));
-      if (conflicts.length > 0) {
+      const conflicts = /* @__PURE__ */ new Set();
+      for (const tool2 of [...this.toolManager.getAllTools(), ...tempManager.getAllTools()]) {
+        if (this.isToolRoutingSupportTool(tool2)) {
+          conflicts.add(tool2.name);
+        }
+      }
+      if (conflicts.size > 0) {
         throw new Error(
           [
-            "toolRouting is disabled but reserved routing tool names are in use:",
-            conflicts.join(", "),
-            "Rename these tools or enable toolRouting to use the built-in names."
+            "toolRouting is disabled but internal routing support tools are in use:",
+            Array.from(conflicts).join(", "),
+            "Enable toolRouting or remove internal routing tools before disabling it for this request."
           ].join(" ")
         );
       }
@@ -34052,10 +34086,7 @@ ${retrieverContext}`;
     if (!tool2 || typeof tool2 !== "object") {
       return false;
     }
-    if (TOOL_ROUTING_INTERNAL_TOOL_SYMBOL in tool2) {
-      return true;
-    }
-    return tool2.name === TOOL_ROUTING_SEARCH_TOOL_NAME || tool2.name === TOOL_ROUTING_CALL_TOOL_NAME;
+    return Object.prototype.hasOwnProperty.call(tool2, TOOL_ROUTING_INTERNAL_TOOL_SYMBOL);
   }
   isToolExecutableForRouting(tool2) {
     if (isProviderTool(tool2)) {
@@ -35131,12 +35162,12 @@ ${retrieverContext}`;
     const memoryInstance = activeMemory || void 0;
     const toolRoutingConfig = this.toolRouting && typeof this.toolRouting === "object" ? this.toolRouting : void 0;
     const toolRoutingState = toolRoutingConfig ? (() => {
-      const supportNames = this.getToolRoutingSupportToolNames();
       const searchTool = this.toolManager.getToolByName(TOOL_ROUTING_SEARCH_TOOL_NAME);
       const callTool = this.toolManager.getToolByName(TOOL_ROUTING_CALL_TOOL_NAME);
       const searchApiTool = searchTool && this.isToolRoutingSupportTool(searchTool) ? new ToolManager([searchTool], this.logger).getToolsForApi()[0] : void 0;
       const callApiTool = callTool && this.isToolRoutingSupportTool(callTool) ? new ToolManager([callTool], this.logger).getToolsForApi()[0] : void 0;
-      const poolApiTools = this.toolPoolManager.getToolsForApi().filter((tool2) => !supportNames.has(tool2.name));
+      const poolTools = this.toolPoolManager.getAllTools().filter((tool2) => !this.isToolRoutingSupportTool(tool2));
+      const poolApiTools = poolTools.length > 0 ? new ToolManager(poolTools, this.logger).getToolsForApi() : [];
       const exposeApiTools = toolRoutingConfig.expose && toolRoutingConfig.expose.length > 0 ? new ToolManager(toolRoutingConfig.expose, this.logger).getToolsForApi() : [];
       return {
         search: searchApiTool,
@@ -35442,9 +35473,10 @@ ${retrieverContext}`;
   upsertToolRoutingSupportTool(tool2) {
     const existing = this.toolManager.getToolByName(tool2.name);
     if (existing && !this.isToolRoutingSupportTool(existing)) {
-      throw new Error(
-        `Tool routing requires reserved tool name "${tool2.name}". Rename the conflicting tool to enable tool routing.`
+      this.logger.debug(
+        `Tool routing support tool "${tool2.name}" not added because a user-defined tool with the same name exists.`
       );
+      return;
     }
     if (existing && this.isToolRoutingSupportTool(existing)) {
       this.toolManager.removeTool(tool2.name);