File tree Expand file tree Collapse file tree
mcp/mcp-cli/src/main/java/software/amazon/smithy/java/mcp/cli/commands Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -78,17 +78,16 @@ public void execute(ExecutionContext context) throws IOException {
7878 var config = context .config ();
7979 // By default, load all available tools
8080 if (toolBundles == null || toolBundles .isEmpty ()) {
81- toolBundles = new ArrayList <>( config .getToolBundles ()
81+ toolBundles = config .getToolBundles ()
8282 .entrySet ()
8383 .stream ()
8484 .filter (entry -> {
8585 // By default, only include smithy bundles if no bundles are specified.
8686 // We can undo this once we have fanout support for generic bundles.
87- var bundle = entry .getValue ();
88- return bundle .getValue () instanceof McpBundleConfig .SmithyModeledMember ;
87+ return entry .getValue ().type () == McpBundleConfig .Type .smithyModeled ;
8988 })
9089 .map (Map .Entry ::getKey )
91- .toList ()) ;
90+ .toList ();
9291 }
9392
9493 if (toolBundles .isEmpty () && !registryServer ) {
You can’t perform that action at this time.
0 commit comments