bsv-mcp 0.2.0 → 0.2.8

package/tools/wallet/a2bPublishAgent.ts

@@ -1,10 +1,5 @@
 import { PrivateKey, Utils } from "@bsv/sdk";
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import type { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import type {
-	ServerNotification,
-	ServerRequest,
-} from "@modelcontextprotocol/sdk/types.js";
 import type {
 	ChangeResult,
 	CreateOrdinalsConfig,
@@ -14,7 +9,6 @@ import type {
 	PreMAP,
 } from "js-1sat-ord";
 import { createOrdinals } from "js-1sat-ord";
-import { Sigma } from "sigma-protocol";
 import { z } from "zod";
 import packageJson from "../../package.json";
 import { V5Broadcaster } from "../../utils/broadcaster";
@@ -155,11 +149,8 @@ export function registerA2bPublishAgentTool(server: McpServer, wallet: Wallet) {
 	server.tool(
 		"wallet_a2bPublish",
 		"Publish an agent.json record on-chain via Ordinal inscription",
-		{ args: a2bPublishArgsSchema },
-		async (
-			{ args }: { args: A2bPublishArgs },
-			extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
-		) => {
+		{ ...a2bPublishArgsSchema.shape },
+		async ({ agentUrl, agentName, description, providerOrganization, providerUrl, version, documentationUrl, streaming, pushNotifications, stateTransitionHistory, defaultInputModes, defaultOutputModes, skills, destinationAddress }) => {
 			try {
 				const paymentPk = wallet.getPaymentKey();
 				if (!paymentPk) throw new Error("No private key available");
@@ -167,7 +158,7 @@ export function registerA2bPublishAgentTool(server: McpServer, wallet: Wallet) {
 				if (!paymentUtxos?.length)
 					throw new Error("No payment UTXOs available to fund inscription");
 
-				// TODO: Get the skills from actually running the MCP instead of trusting the agent args for args.skills
+				// TODO: Get the skills from actually running the MCP instead of trusting the agent args for skills
 				const walletAddress = paymentPk.toAddress().toString();
 
 				// Create pricing plans using the new schema
@@ -212,27 +203,27 @@ export function registerA2bPublishAgentTool(server: McpServer, wallet: Wallet) {
 
 				// Assemble AgentCard with defaults and user overrides
 				const agentCard = {
-					name: args.agentName,
-					description: args.description ?? null,
-					url: args.agentUrl,
+					name: agentName,
+					description: description ?? null,
+					url: agentUrl,
 					provider:
-						args.providerOrganization && args.providerUrl
+						providerOrganization && providerUrl
 							? {
-									organization: args.providerOrganization,
-									url: args.providerUrl,
+									organization: providerOrganization,
+									url: providerUrl,
 								}
 							: null,
-					version: args.version ?? packageJson.version,
-					documentationUrl: args.documentationUrl ?? null,
+					version: version ?? packageJson.version,
+					documentationUrl: documentationUrl ?? null,
 					capabilities: {
-						streaming: args.streaming,
-						pushNotifications: args.pushNotifications,
-						stateTransitionHistory: args.stateTransitionHistory,
+						streaming,
+						pushNotifications,
+						stateTransitionHistory,
 					},
 					authentication: null,
-					defaultInputModes: args.defaultInputModes,
-					defaultOutputModes: args.defaultOutputModes,
-					skills: args.skills,
+					defaultInputModes,
+					defaultOutputModes,
+					skills,
 					"x-payment-config": pricingConfig,
 				};
 				// Validate compliance
@@ -245,7 +236,7 @@ export function registerA2bPublishAgentTool(server: McpServer, wallet: Wallet) {
 					contentType: "application/json",
 				};
 				// Destination for the ordinal
-				const targetAddress = args.destinationAddress ?? walletAddress;
+				const targetAddress = destinationAddress ?? walletAddress;
 				const destinations: Destination[] = [
 					{ address: targetAddress, inscription },
 				];

package/tools/wallet/a2bPublishMcp.ts

@@ -1,9 +1,8 @@
-import type { PrivateKey, WalletInterface } from "@bsv/sdk";
+import type { PrivateKey } from "@bsv/sdk";
 import { Utils } from "@bsv/sdk";
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import type { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
 import type {
 	ClientNotification,
 	ClientRequest,
@@ -19,10 +18,9 @@ import type {
 	PreMAP,
 } from "js-1sat-ord";
 import { createOrdinals } from "js-1sat-ord";
-import { Sigma } from "sigma-protocol";
 import { z } from "zod";
 import packageJson from "../../package.json";
-import { BsocialBroadcaster, V5Broadcaster } from "../../utils/broadcaster";
+import { V5Broadcaster } from "../../utils/broadcaster";
 import type { Wallet } from "./wallet";
 
 const { toArray, toBase64 } = Utils;
@@ -196,11 +194,8 @@ export function registerA2bPublishMcpTool(
 	server.tool(
 		"wallet_a2bPublishMcp",
 		"Publish an MCP tool configuration record on-chain via Ordinal inscription. This creates a permanent, immutable, and discoverable tool definition that can be accessed by other MCP servers. The tool is published as a JSON inscription with metadata and optional digital signatures for authenticity verification.",
-		{ args: a2bPublishMcpArgsSchema },
-		async (
-			{ args }: { args: A2bPublishMcpArgs },
-			extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
-		) => {
+		{ ...a2bPublishMcpArgsSchema.shape },
+		async ({ toolName, command, args, keywords, env, description, destinationAddress }) => {
 			try {
 				if (!identityPk) {
 					console.warn(
@@ -218,20 +213,20 @@ export function registerA2bPublishMcpTool(
 				const walletAddress = paymentPk.toAddress().toString();
 
 				// Fetch MCP metadata (tools, prompts, resources)
-				const metadata = await fetchMcpMetadata(args.command, args.args);
+				const metadata = await fetchMcpMetadata(command, args);
 				// console.log(`Discovered ${metadata.tools.length} tools, ${metadata.prompts.length} prompts, and ${metadata.resources.length} resources`);
 
 				// Assemble tool configuration
 				const toolConfig: McpConfig = {
-					command: args.command,
-					args: args.args,
+					command,
+					args,
 					tools: metadata.tools,
 					prompts: metadata.prompts,
 					resources: metadata.resources,
-					env: args.env
-						? args.env.reduce(
-								(acc, { key, description }) => {
-									acc[key] = description;
+					env: env
+						? env.reduce(
+								(acc, { key, description: desc }) => {
+									acc[key] = desc;
 									return acc;
 								},
 								{} as Record<string, string>,
@@ -245,9 +240,9 @@ export function registerA2bPublishMcpTool(
 				// Prepare the full configuration with metadata
 				const fullConfig = {
 					mcpServers: {
-						[args.toolName]: {
-							description: args.description || "",
-							keywords: args.keywords || [],
+						[toolName]: {
+							description: description || "",
+							keywords: keywords || [],
 							tools: metadata.tools || [],
 							prompts: metadata.prompts || [],
 							resources: metadata.resources || [],
@@ -266,7 +261,7 @@ export function registerA2bPublishMcpTool(
 				};
 
 				// Destination for the ordinal
-				const targetAddress = args.destinationAddress ?? walletAddress;
+				const targetAddress = destinationAddress ?? walletAddress;
 				const destinations: Destination[] = [
 					{ address: targetAddress, inscription },
 				];
@@ -335,12 +330,12 @@ export function registerA2bPublishMcpTool(
 										txid,
 										outpoint,
 										onchainUrl,
-										toolName: args.toolName,
+										toolName,
 										toolCount: metadata.tools.length,
 										promptCount: metadata.prompts.length,
 										resourceCount: metadata.resources.length,
 										description:
-											args.description || `MCP Tool: ${args.toolName}`,
+											description || `MCP Tool: ${toolName}`,
 										address: targetAddress,
 									},
 									null,

package/tools/wallet/createOrdinals.ts

@@ -1,11 +1,6 @@
 import { PrivateKey } from "@bsv/sdk";
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import type { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import type {
-	CallToolResult,
-	ServerNotification,
-	ServerRequest,
-} from "@modelcontextprotocol/sdk/types.js";
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import type {
 	ChangeResult,
 	CreateOrdinalsCollectionItemMetadata,
@@ -16,10 +11,9 @@ import type {
 	LocalSigner,
 	PreMAP,
 } from "js-1sat-ord";
-import { createOrdinals, OneSatBroadcaster } from "js-1sat-ord";
-import { Sigma } from "sigma-protocol";
+import { createOrdinals } from "js-1sat-ord";
 import { z } from "zod";
-import { BsocialBroadcaster, V5Broadcaster } from "../../utils/broadcaster";
+import { V5Broadcaster } from "../../utils/broadcaster";
 import type { Wallet } from "./wallet";
 
 /**
@@ -53,11 +47,8 @@ export function registerCreateOrdinalsTool(server: McpServer, wallet: Wallet) {
 	server.tool(
 		"wallet_createOrdinals",
 		"Creates and inscribes ordinals (NFTs) on the Bitcoin SV blockchain. This tool lets you mint new digital artifacts by encoding data directly into the blockchain. Supports various content types including images, text, JSON, and HTML. The tool handles transaction creation, fee calculation, and broadcasting.",
-		{ args: createOrdinalsArgsSchema },
-		async (
-			{ args }: { args: CreateOrdinalsArgs },
-			extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
-		): Promise<CallToolResult> => {
+		{ ...createOrdinalsArgsSchema.shape },
+		async ({ dataB64, contentType, destinationAddress, metadata }): Promise<CallToolResult> => {
 			try {
 				// Load optional identity key for sigma signing
 				const identityKeyWif = process.env.IDENTITY_KEY_WIF;
@@ -92,14 +83,14 @@ export function registerCreateOrdinalsTool(server: McpServer, wallet: Wallet) {
 
 				// 4. Create the inscription object
 				const inscription: Inscription = {
-					dataB64: args.dataB64,
-					contentType: args.contentType,
+					dataB64: dataB64,
+					contentType: contentType,
 				};
 
 				// 5. Create the destination
 				const destinations: Destination[] = [
 					{
-						address: args.destinationAddress || walletAddress,
+						address: destinationAddress || walletAddress,
 						inscription,
 					},
 				];
@@ -109,7 +100,7 @@ export function registerCreateOrdinalsTool(server: McpServer, wallet: Wallet) {
 					destinations,
 					paymentPk,
 					changeAddress: walletAddress,
-					metaData: args.metadata as
+					metaData: metadata as
 						| PreMAP
 						| CreateOrdinalsCollectionMetadata
 						| CreateOrdinalsCollectionItemMetadata,
@@ -152,8 +143,8 @@ export function registerCreateOrdinalsTool(server: McpServer, wallet: Wallet) {
 									txid: changeResult.tx.id("hex"),
 									spentOutpoints: changeResult.spentOutpoints,
 									payChange: changeResult.payChange,
-									inscriptionAddress: args.destinationAddress || walletAddress,
-									contentType: args.contentType,
+									inscriptionAddress: destinationAddress || walletAddress,
+									contentType: contentType,
 								}),
 							},
 						],

package/tools/wallet/fetchPaymentUtxos.ts

@@ -104,7 +104,15 @@ export async function fetchPaymentUtxosFromV5(
 		const url = `${V5_API_URL}/own/${address}/utxos?refresh=true&txo=true&script=true&limit=250&tags=p2pkh`;
 		console.error(`Fetching UTXOs from V5: ${url}`);
 		const response = await fetch(url);
-		const data = (await response.json()) as V5Utxo[];
+		if (!response.ok) {
+			console.error(`V5 UTXO fetch failed: ${response.status} ${response.statusText}`);
+			return undefined;
+		}
+		const data = (await response.json()) as V5Utxo[] | null;
+		if (!Array.isArray(data)) {
+			console.error(`V5 UTXO response was not an array: ${typeof data}`);
+			return undefined;
+		}
 		const utxos: Utxo[] = [];
 		for (const utxo of data) {
 			if (!utxo.data.p2pkh) {

package/tools/wallet/gatherCollectionInfo.ts

@@ -133,15 +133,11 @@ export function registerGatherCollectionInfoTool(
 	server.tool(
 		"wallet_gatherCollectionInfo",
 		"Analyzes a folder of images and gathers all necessary information for minting an ordinals collection. This includes validating images, checking wallet balance, estimating costs, and suggesting metadata. Use this before minting to ensure everything is ready.",
-		{ args: gatherCollectionInfoArgsSchema },
-		async ({
-			args,
-		}: {
-			args: GatherCollectionInfoArgs;
-		}): Promise<CallToolResult> => {
+		{ ...gatherCollectionInfoArgsSchema.shape },
+		async ({ folderPath }): Promise<CallToolResult> => {
 			try {
 				const analysis: CollectionAnalysis = {
-					folderPath: args.folderPath,
+					folderPath,
 					totalImages: 0,
 					validImages: 0,
 					invalidImages: 0,
@@ -172,13 +168,13 @@ export function registerGatherCollectionInfoTool(
 
 				// Check if folder exists
 				try {
-					const folderStat = await fs.stat(args.folderPath);
+					const folderStat = await fs.stat(folderPath);
 					if (!folderStat.isDirectory()) {
-						throw new Error(`Path is not a directory: ${args.folderPath}`);
+						throw new Error(`Path is not a directory: ${folderPath}`);
 					}
 				} catch (error) {
 					analysis.errors.push(
-						`Folder not found or inaccessible: ${args.folderPath}`,
+						`Folder not found or inaccessible: ${folderPath}`,
 					);
 					return {
 						content: [
@@ -222,7 +218,7 @@ export function registerGatherCollectionInfoTool(
 				}
 
 				// Scan folder for images
-				const files = await fs.readdir(args.folderPath);
+				const files = await fs.readdir(folderPath);
 				const imageExtensions = [
 					".jpg",
 					".jpeg",
@@ -233,7 +229,7 @@ export function registerGatherCollectionInfoTool(
 				];
 
 				for (const file of files) {
-					const filePath = path.join(args.folderPath, file);
+					const filePath = path.join(folderPath, file);
 					const stat = await fs.stat(filePath);
 
 					if (stat.isFile()) {
@@ -280,7 +276,7 @@ export function registerGatherCollectionInfoTool(
 				}
 
 				// Suggest metadata based on folder name
-				const folderName = path.basename(args.folderPath);
+				const folderName = path.basename(folderPath);
 				analysis.suggestedMetadata = {
 					collectionName: folderName
 						.replace(/[-_]/g, " ")

package/tools/wallet/getAddress.ts

@@ -1,5 +1,4 @@
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { z } from "zod";
 import type { Wallet } from "./wallet";
 
 /**
@@ -14,14 +13,7 @@ export function registerGetAddressTool(
 	server.tool(
 		"wallet_getAddress",
 		"Retrieves the current wallet's Bitcoin SV address. This address can be used to receive BSV, ordinals, or tokens, and is derived from the wallet's private key.",
-		{
-			args: z
-				.object({})
-				.optional()
-				.describe(
-					"No parameters required - simply returns the current wallet address",
-				),
-		},
+		{},
 		async () => {
 			try {
 				const address = wallet.getAddress();

package/tools/wallet/getBalance.ts

@@ -1,16 +1,7 @@
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import type { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import type {
-	ServerNotification,
-	ServerRequest,
-} from "@modelcontextprotocol/sdk/types.js";
 import { toBitcoin } from "satoshi-token"; // For converting satoshis to BSV string
-import { z } from "zod";
 import type { Wallet } from "./wallet";
 
-export const walletGetBalanceArgsSchema = z.object({}).optional();
-export type WalletGetBalanceArgs = z.infer<typeof walletGetBalanceArgsSchema>;
-
 /**
  * Registers the wallet_getBalance tool
  */
@@ -21,11 +12,8 @@ export function registerWalletGetBalanceTool(
 	server.tool(
 		"wallet_getBalance",
 		"Retrieves the current BSV balance for the wallet based on its UTXOs.",
-		{ args: walletGetBalanceArgsSchema },
-		async (
-			args: WalletGetBalanceArgs, // Args are empty but present for schema consistency
-			extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
-		) => {
+		{},
+		async () => {
 			try {
 				const { paymentUtxos } = await wallet.getUtxos();
 

package/tools/wallet/getBalanceDroplet.ts

@@ -1,11 +1,5 @@
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import type { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import type {
-	ServerNotification,
-	ServerRequest,
-} from "@modelcontextprotocol/sdk/types.js";
 import type { DropletClient } from "../../utils/droplet";
-import { emptyArgsSchema } from "./schemas";
 
 /**
  * Register the getBalance tool for Droplet API mode
@@ -17,13 +11,8 @@ export function registerWalletGetBalanceDropletTool(
 	server.tool(
 		"wallet_getBalance",
 		"Gets the current balance of the wallet (Droplet mode)",
-		{
-			args: emptyArgsSchema,
-		},
-		async (
-			{ args }: { args: unknown },
-			extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
-		) => {
+		{},
+		async () => {
 			try {
 				const status = await dropletClient.getFaucetStatus();
 

package/tools/wallet/getPublicKey.ts

@@ -1,16 +1,6 @@
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import type { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import type {
-	ServerNotification,
-	ServerRequest,
-} from "@modelcontextprotocol/sdk/types.js";
-import type { z } from "zod";
-import { getPublicKeyArgsSchema } from "./schemas";
 import type { Wallet } from "./wallet";
 
-// Use the schema imported from schemas.ts
-export type GetPublicKeyArgs = z.infer<typeof getPublicKeyArgsSchema>;
-
 /**
  * Register the getPublicKey tool
  */
@@ -18,13 +8,10 @@ export function registerGetPublicKeyTool(server: McpServer, wallet: Wallet) {
 	server.tool(
 		"wallet_getPublicKey",
 		"Retrieves the current wallet's public key. This public key can be used for cryptographic operations like signature verification or encryption.",
-		{ args: getPublicKeyArgsSchema },
-		async (
-			{ args }: { args: GetPublicKeyArgs },
-			extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
-		) => {
+		{},
+		async () => {
 			try {
-				const result = await wallet.getPublicKey(args);
+				const result = await wallet.getPublicKey({});
 				return { content: [{ type: "text", text: JSON.stringify(result) }] };
 			} catch (err: unknown) {
 				const msg = err instanceof Error ? err.message : String(err);

package/tools/wallet/mintCollection.ts

@@ -2,12 +2,7 @@ import { promises as fs } from "node:fs";
 import path from "node:path";
 import { Utils } from "@bsv/sdk";
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import type { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import type {
-	CallToolResult,
-	ServerNotification,
-	ServerRequest,
-} from "@modelcontextprotocol/sdk/types.js";
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import type {
 	ChangeResult,
 	CollectionItemSubTypeData,
@@ -40,10 +35,10 @@ const mintCollectionArgsSchema = z.object({
 		.optional()
 		.describe("Rarity labels and their percentages"),
 	traits: z
-		.record(z.array(z.string()))
+		.string()
 		.optional()
 		.describe(
-			"Collection traits as key-value pairs where values are arrays of possible trait values",
+			'Collection traits as JSON string of key-value pairs where values are arrays, e.g. \'{"Background": ["Blue", "Red"], "Eyes": ["Big", "Small"]}\'',
 		),
 	skipBroadcast: z
 		.boolean()
@@ -125,9 +120,11 @@ export function registerMintCollectionTool(server: McpServer, wallet: Wallet) {
 	server.tool(
 		"wallet_mintCollection",
 		"Mint a collection of ordinals from a folder of images with proper metadata. This tool creates a collection inscription first, then mints each image as a collection item with the appropriate metadata linking it to the collection.",
-		{ args: mintCollectionArgsSchema },
-		async ({ args }: { args: MintCollectionArgs }): Promise<CallToolResult> => {
+		{ ...mintCollectionArgsSchema.shape },
+		async ({ folderPath, collectionName, description, rarityLabels, traits: traitsJson, skipBroadcast }): Promise<CallToolResult> => {
 			try {
+				// Parse traits JSON string if provided
+				const traits: Record<string, string[]> | undefined = traitsJson ? JSON.parse(traitsJson) : undefined;
 				// Get keys from wallet
 				const paymentPk = wallet.getPaymentKey();
 				if (!paymentPk) {
@@ -136,7 +133,7 @@ export function registerMintCollectionTool(server: McpServer, wallet: Wallet) {
 				const identityPk = wallet.getIdentityKey();
 
 				// Get image files
-				const imageFiles = await getImageFiles(args.folderPath);
+				const imageFiles = await getImageFiles(folderPath);
 				if (imageFiles.length === 0) {
 					throw new Error("No image files found in the specified folder");
 				}
@@ -145,14 +142,14 @@ export function registerMintCollectionTool(server: McpServer, wallet: Wallet) {
 
 				// Prepare collection metadata
 				const collectionSubTypeData: CollectionSubTypeData = {
-					description: args.description,
+					description: description,
 					quantity: imageFiles.length,
 					// Transform rarity labels to correct format
 					rarityLabels:
-						args.rarityLabels?.map((r) => ({ label: r.label })) || [],
+						rarityLabels?.map((r) => ({ label: r.label })) || [],
 					// Transform traits - need to match the CollectionTraits type
-					traits: args.traits
-						? Object.entries(args.traits).reduce((acc, [key, values]) => {
+					traits: traits
+						? Object.entries(traits).reduce((acc, [key, values]) => {
 								acc[key] = {
 									values,
 									occurancePercentages: values.map(() =>
@@ -168,14 +165,14 @@ export function registerMintCollectionTool(server: McpServer, wallet: Wallet) {
 					app: "ord",
 					type: "ord",
 					subType: "collection",
-					name: args.collectionName,
+					name: collectionName,
 					subTypeData: collectionSubTypeData,
 				};
 
 				// Create collection icon
 				const svgString = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 100 100">
             <rect width="100" height="100" fill="#f0f0f0"/>
-            <text x="50" y="55" text-anchor="middle" font-family="Arial" font-size="16">${args.collectionName}</text>
+            <text x="50" y="55" text-anchor="middle" font-family="Arial" font-size="16">${collectionName}</text>
           </svg>`;
 				const collectionIconData = Utils.toBase64(
 					Utils.toArray(svgString, "utf8"),
@@ -217,7 +214,7 @@ export function registerMintCollectionTool(server: McpServer, wallet: Wallet) {
 
 				let collectionTxid = "";
 				const disableBroadcasting =
-					args.skipBroadcast || process.env.DISABLE_BROADCASTING === "true";
+					skipBroadcast || process.env.DISABLE_BROADCASTING === "true";
 
 				if (!disableBroadcasting) {
 					const broadcaster = new V5Broadcaster();
@@ -251,7 +248,7 @@ export function registerMintCollectionTool(server: McpServer, wallet: Wallet) {
 					);
 
 					try {
-						const itemTraits = generateItemTraits(i, args.traits);
+						const itemTraits = generateItemTraits(i, traits);
 
 						const itemSubTypeData: CollectionItemSubTypeData = {
 							collectionId: collectionTxid,
@@ -260,13 +257,13 @@ export function registerMintCollectionTool(server: McpServer, wallet: Wallet) {
 						};
 
 						// Assign rarity if labels provided
-						if (args.rarityLabels && args.rarityLabels.length > 0) {
+						if (rarityLabels && rarityLabels.length > 0) {
 							const rarityIndex = Math.floor(
-								(i / imageFiles.length) * args.rarityLabels.length,
+								(i / imageFiles.length) * rarityLabels.length,
 							);
 							const rarityItem =
-								args.rarityLabels[
-									Math.min(rarityIndex, args.rarityLabels.length - 1)
+								rarityLabels[
+									Math.min(rarityIndex, rarityLabels.length - 1)
 								];
 							if (rarityItem) {
 								itemSubTypeData.rarityLabel = [
@@ -352,7 +349,7 @@ export function registerMintCollectionTool(server: McpServer, wallet: Wallet) {
 									totalItems: imageFiles.length,
 									errors: results.errors,
 									totalCost: results.totalCost,
-									summary: `Collection "${args.collectionName}" created with ${results.itemTxids.length}/${imageFiles.length} items successfully minted.${results.errors.length > 0 ? ` ${results.errors.length} items failed.` : ""}`,
+									summary: `Collection "${collectionName}" created with ${results.itemTxids.length}/${imageFiles.length} items successfully minted.${results.errors.length > 0 ? ` ${results.errors.length} items failed.` : ""}`,
 								},
 								null,
 								2,