Merge pull request #177 from WolframResearch/feature/deploy-all

Support DeployAgentTools[All] for deploying to every supported client

Author: rhennigan

Kernel/DeployAgentTools.wl

@@ -261,9 +261,15 @@ $$deployAgentToolsOptions = OptionsPattern @ { DeployAgentTools, InstallMCPServe
 (* ::**************************************************************************************************************:: *)
 (* ::Subsection::Closed:: *)
 (*Main Definition*)
+(* Default server is `Automatic` *)
 DeployAgentTools[ target_, opts: $$deployAgentToolsOptions ] :=
     catchMine @ DeployAgentTools[ target, Automatic, opts ];
 
+(* Deploy for all clients *)
+DeployAgentTools[ All, server_, opts: $$deployAgentToolsOptions ] :=
+    catchMine @ deployAllAgentTools[ server, opts ];
+
+(* Resolve automatic server *)
 DeployAgentTools[ target_, Automatic, opts: $$deployAgentToolsOptions ] :=
     catchMine @ DeployAgentTools[
         target,
@@ -278,11 +284,83 @@ DeployAgentTools[ target_, Automatic, opts: $$deployAgentToolsOptions ] :=
         opts
     ];
 
+(* Proceed with deployment *)
 DeployAgentTools[ target_, server_, opts: $$deployAgentToolsOptions ] :=
     catchMine @ deployAgentTools[ target, ensureMCPServerExists @ MCPServerObject @ server, opts ];
 
 DeployAgentTools // endExportedDefinition;
 
+(* ::**************************************************************************************************************:: *)
+(* ::Subsubsection::Closed:: *)
+(*deployAllAgentTools*)
+deployAllAgentTools // beginDefinition;
+
+deployAllAgentTools[ server0_, opts: $$deployAgentToolsOptions ] := Enclose[
+    Module[ { clients, server, results },
+        clients = ConfirmMatch[ Keys @ $SupportedMCPClients, { __String }, "Clients" ];
+
+        server = If[ server0 === Automatic,
+                     Automatic,
+                     ConfirmBy[ ensureMCPServerExists @ MCPServerObject @ server0, MCPServerObjectQ, "Server" ]
+                 ];
+
+        results = ConfirmMatch[
+            Map[ deployAgentToolsQuietly[ #, resolveServerForClient[ #, server ], opts ] &, clients ],
+            { (_AgentToolsDeployment | Missing[ "DeploymentExists", _ ] | Missing[ "Unsupported", _ ]).. },
+            "Results"
+        ];
+
+        If[ MemberQ[ results, Missing[ "DeploymentExists", _ ] ],
+            messagePrint[ "DeploymentsExistWarning" ]
+        ];
+
+        results
+    ],
+    throwInternalFailure
+];
+
+deployAllAgentTools // endDefinition;
+
+(* ::**************************************************************************************************************:: *)
+(* ::Subsubsection::Closed:: *)
+(*resolveServerForClient*)
+(* When `server === Automatic`, resolve each client's `DefaultToolset` directly
+   from the client name.  Going through the 2-arg `defaultToolsetForTarget`
+   inside `DeployAgentTools[target, Automatic, opts]` would let an explicit
+   `"ApplicationName" -> name` option override the per-client default, which
+   defeats the point of `All`. *)
+resolveServerForClient // beginDefinition;
+resolveServerForClient[ client_String, Automatic ] := defaultToolsetForTarget @ client;
+resolveServerForClient[ _, server_ ] := server;
+resolveServerForClient // endDefinition;
+
+(* ::**************************************************************************************************************:: *)
+(* ::Subsubsection::Closed:: *)
+(*deployAgentToolsQuietly*)
+deployAgentToolsQuietly // beginDefinition;
+
+deployAgentToolsQuietly[ target_, server_, opts: $$deployAgentToolsOptions ] :=
+    Quiet[
+        Replace[
+            catchAlways @ DeployAgentTools[ target, server, opts ],
+            {
+                (* Used to issue a single warning about OverwriteTarget: *)
+                Failure[ "DeployAgentTools::DeploymentExists", _ ] :>
+                    Missing[ "DeploymentExists", target ],
+
+                (* Some clients aren't supported on all operating systems: *)
+                Failure[ "DeployAgentTools::UnknownInstallLocation", _ ] :>
+                    Missing[ "Unsupported", { target, $OperatingSystem } ],
+
+                (* Other failures are propagated to the top level: *)
+                other_Failure :> throwTop @ other
+            }
+        ],
+        { DeployAgentTools::DeploymentExists, DeployAgentTools::UnknownInstallLocation }
+    ];
+
+deployAgentToolsQuietly // endDefinition;
+
 (* ::**************************************************************************************************************:: *)
 (* ::Subsection::Closed:: *)
 (*deployAgentTools*)

Kernel/Messages.wl

@@ -69,9 +69,10 @@ AgentTools::CodeInspectorInvalidConfidence  = "Confidence level must be between
 
 (* DeployAgentTools messages *)
 AgentTools::DeploymentExists                = "A deployment already exists for target `1`. Use OverwriteTarget -> True to replace it.";
+AgentTools::DeploymentsExistWarning         = "Warning: Some deployments already exist. Use OverwriteTarget -> True to replace them.";
 AgentTools::DeploymentNotFound              = "No deployment found with UUID \"`1`\".";
 AgentTools::InvalidDeploymentData           = "Invalid deployment data: `1`.";
-AgentTools::InvalidDeployTarget             = "Invalid deployment target: `1`. Expected a client name string, {name, directory}, or File[\[Ellipsis]].";
+AgentTools::InvalidDeployTarget             = "Invalid deployment target: `1`. Expected a client name string, {name, directory}, File[\[Ellipsis]], or All.";
 
 (* PacletTools messages *)
 AgentTools::PacletToolsInvalidPath          = "The path \"`1`\" does not exist. Provide an absolute path to either the paclet root directory or the definition notebook (.nb) file.";

README.md

@@ -273,6 +273,7 @@ CreateMCPServer["My MCP Server", <|
 | `UninstallMCPServer[client]` | Remove all servers from a client |
 | `UninstallMCPServer[client, name]` | Remove a specific server from a client |
 | `DeployAgentTools[target]` | Deploy tools to a client with tracked deployment management |
+| `DeployAgentTools[All]` | Deploy tools to every client in `$SupportedMCPClients` (see [docs/deploy-agent-tools.md](docs/deploy-agent-tools.md#deploying-to-all-clients)) |
 | `ValidateAgentToolsPacletExtension[paclet]` | Validate an `"AgentTools"` paclet extension |
 | `DeployedAgentTools[]` | List all tracked deployments |
 | `CreatePreferencesContent[]` | Build the preferences panel UI for managing deployed Wolfram toolsets (see [docs/preferences-content.md](docs/preferences-content.md)) |