@librechat/agents 3.0.40 → 3.0.42

package/src/graphs/Graph.ts

@@ -35,6 +35,7 @@ import {
   GraphEvents,
   Providers,
   StepTypes,
+  Constants,
 } from '@/common';
 import {
   formatAnthropicArtifactContent,
@@ -608,7 +609,7 @@ export class StandardGraph extends Graph<t.BaseGraphState, t.GraphNode> {
     client.abortHandler = undefined;
   }
 
-  createCallModel(agentId = 'default', currentModel?: t.ChatModel) {
+  createCallModel(agentId = 'default') {
     return async (
       state: t.BaseGraphState,
       config?: RunnableConfig
@@ -621,15 +622,23 @@ export class StandardGraph extends Graph<t.BaseGraphState, t.GraphNode> {
         throw new Error(`Agent context not found for agentId: ${agentId}`);
       }
 
-      const model = this.overrideModel ?? currentModel;
-      if (!model) {
-        throw new Error('No Graph model found');
-      }
       if (!config) {
         throw new Error('No config provided');
       }
 
-      // Ensure token calculations are complete before proceeding
+      const toolsForBinding = agentContext.getToolsForBinding();
+      let model =
+        this.overrideModel ??
+        this.initializeModel({
+          tools: toolsForBinding,
+          provider: agentContext.provider,
+          clientOptions: agentContext.clientOptions,
+        });
+
+      if (agentContext.systemRunnable) {
+        model = agentContext.systemRunnable.pipe(model as Runnable);
+      }
+
       if (agentContext.tokenCalculationPromise) {
         await agentContext.tokenCalculationPromise;
       }
@@ -703,6 +712,23 @@ export class StandardGraph extends Graph<t.BaseGraphState, t.GraphNode> {
 
       const isLatestToolMessage = lastMessageY instanceof ToolMessage;
 
+      if (
+        isLatestToolMessage &&
+        lastMessageY.name === Constants.TOOL_SEARCH_REGEX &&
+        typeof lastMessageY.artifact === 'object' &&
+        lastMessageY.artifact != null
+      ) {
+        const artifact = lastMessageY.artifact as {
+          tool_references?: Array<{ tool_name: string }>;
+        };
+        if (artifact.tool_references && artifact.tool_references.length > 0) {
+          const discoveredNames = artifact.tool_references.map(
+            (ref) => ref.tool_name
+          );
+          agentContext.markToolsAsDiscovered(discoveredNames);
+        }
+      }
+
       if (
         isLatestToolMessage &&
         agentContext.provider === Providers.ANTHROPIC
@@ -849,16 +875,6 @@ export class StandardGraph extends Graph<t.BaseGraphState, t.GraphNode> {
       throw new Error(`Agent context not found for agentId: ${agentId}`);
     }
 
-    let currentModel = this.initializeModel({
-      tools: agentContext.tools,
-      provider: agentContext.provider,
-      clientOptions: agentContext.clientOptions,
-    });
-
-    if (agentContext.systemRunnable) {
-      currentModel = agentContext.systemRunnable.pipe(currentModel);
-    }
-
     const agentNode = `${AGENT}${agentId}` as const;
     const toolNode = `${TOOLS}${agentId}` as const;
 
@@ -878,7 +894,7 @@ export class StandardGraph extends Graph<t.BaseGraphState, t.GraphNode> {
     });
 
     const workflow = new StateGraph(StateAnnotation)
-      .addNode(agentNode, this.createCallModel(agentId, currentModel))
+      .addNode(agentNode, this.createCallModel(agentId))
       .addNode(
         toolNode,
         this.initializeTools({