@agentwonderland/mcp 0.1.25 → 0.1.27

package/src/tools/__tests__/jobs.test.ts

@@ -0,0 +1,89 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+const mockApiGet = vi.fn();
+const mockIsAuthenticated = vi.fn();
+const mockHasWalletConfigured = vi.fn();
+const mockGetWalletAddress = vi.fn();
+const mockFormatRunResult = vi.fn();
+
+vi.mock("../../core/api-client.js", () => ({
+  apiGet: mockApiGet,
+}));
+
+vi.mock("../../core/config.js", () => ({
+  isAuthenticated: mockIsAuthenticated,
+}));
+
+vi.mock("../../core/payments.js", () => ({
+  hasWalletConfigured: mockHasWalletConfigured,
+  getWalletAddress: mockGetWalletAddress,
+}));
+
+vi.mock("../../core/formatters.js", () => ({
+  formatRunResult: mockFormatRunResult,
+}));
+
+function flattenToolText(result: unknown): string {
+  const content = (result as { content?: Array<{ type?: string; text?: string }> })?.content ?? [];
+  return content
+    .filter((item) => item?.type === "text")
+    .map((item) => item.text ?? "")
+    .join("\n\n");
+}
+
+function makeServerHarness() {
+  const handlers = new Map<string, (args: Record<string, unknown>) => Promise<unknown>>();
+  return {
+    handlers,
+    server: {
+      tool(name: string, _description: string, _schema: unknown, handler: (args: Record<string, unknown>) => Promise<unknown>) {
+        handlers.set(name, handler);
+      },
+    },
+  };
+}
+
+describe("job MCP tools", () => {
+  beforeEach(() => {
+    vi.resetModules();
+    vi.clearAllMocks();
+    mockIsAuthenticated.mockReturnValue(false);
+    mockHasWalletConfigured.mockReturnValue(true);
+    mockGetWalletAddress.mockResolvedValue("0xabc123");
+    mockFormatRunResult.mockReturnValue("formatted job result");
+  });
+
+  it("appends the wallet address when looking up a job for an unauthenticated consumer", async () => {
+    mockApiGet.mockResolvedValueOnce({ status: "completed", job_id: "job-123" });
+
+    const { registerJobTools } = await import("../jobs.js");
+    const harness = makeServerHarness();
+    registerJobTools(harness.server as never);
+
+    const getJob = harness.handlers.get("get_job");
+    expect(getJob).toBeDefined();
+
+    const result = await getJob!({ job_id: "job-123" });
+    const text = flattenToolText(result);
+
+    expect(mockApiGet).toHaveBeenCalledWith("/jobs/job-123?wallet=0xabc123");
+    expect(text).toContain("formatted job result");
+  });
+
+  it("returns a processing message while the async job is still running", async () => {
+    mockApiGet.mockResolvedValueOnce({ status: "processing" });
+
+    const { registerJobTools } = await import("../jobs.js");
+    const harness = makeServerHarness();
+    registerJobTools(harness.server as never);
+
+    const getJob = harness.handlers.get("get_job");
+    expect(getJob).toBeDefined();
+
+    const result = await getJob!({ job_id: "job-456" });
+    const text = flattenToolText(result);
+
+    expect(mockApiGet).toHaveBeenCalledWith("/jobs/job-456?wallet=0xabc123");
+    expect(text).toContain("Job job-456 is still processing");
+  });
+});

package/src/tools/__tests__/run.test.ts

@@ -0,0 +1,176 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+const mockApiGet = vi.fn();
+const mockApiPost = vi.fn();
+const mockApiPostWithPayment = vi.fn();
+const mockGetCompatiblePaymentMethods = vi.fn();
+const mockHasWalletConfigured = vi.fn();
+const mockGetConfiguredMethods = vi.fn();
+const mockGetWalletAddress = vi.fn();
+const mockNormalizePaymentMethod = vi.fn();
+const mockRequiresSpendConfirmation = vi.fn();
+const mockGetDefaultTipAmount = vi.fn();
+const mockCanSpend = vi.fn();
+const mockRecordSpend = vi.fn();
+const mockRequiresPolicyConfirmation = vi.fn();
+const mockUploadLocalFiles = vi.fn();
+const mockStoreFeedbackToken = vi.fn();
+const mockGetActiveCreditPack = vi.fn();
+const mockGetCreditPackInventory = vi.fn();
+const mockGetCreditPackProgram = vi.fn();
+const mockGetOrCreatePendingCardSetup = vi.fn();
+const mockFormatCardSetupBlocks = vi.fn();
+
+vi.mock("../../core/api-client.js", () => ({
+  apiGet: mockApiGet,
+  apiPost: mockApiPost,
+  apiPostWithPayment: mockApiPostWithPayment,
+}));
+
+vi.mock("../../core/card-setup.js", () => ({
+  getOrCreatePendingCardSetup: mockGetOrCreatePendingCardSetup,
+  formatCardSetupBlocks: mockFormatCardSetupBlocks,
+}));
+
+vi.mock("../../core/payments.js", () => ({
+  getCompatiblePaymentMethods: mockGetCompatiblePaymentMethods,
+  getConfiguredMethods: mockGetConfiguredMethods,
+  hasWalletConfigured: mockHasWalletConfigured,
+  getWalletAddress: mockGetWalletAddress,
+  normalizePaymentMethod: mockNormalizePaymentMethod,
+}));
+
+vi.mock("../../core/config.js", () => ({
+  requiresSpendConfirmation: mockRequiresSpendConfirmation,
+  getDefaultTipAmount: mockGetDefaultTipAmount,
+}));
+
+vi.mock("../../core/spend-policy.js", () => ({
+  canSpend: mockCanSpend,
+  recordSpend: mockRecordSpend,
+  requiresPolicyConfirmation: mockRequiresPolicyConfirmation,
+}));
+
+vi.mock("../../core/file-upload.js", () => ({
+  uploadLocalFiles: mockUploadLocalFiles,
+}));
+
+vi.mock("../_token-cache.js", () => ({
+  storeFeedbackToken: mockStoreFeedbackToken,
+}));
+
+vi.mock("../../core/passes.js", () => ({
+  formatCreditPackOffer: vi.fn(),
+  getActiveCreditPack: mockGetActiveCreditPack,
+  getCreditPackInventory: mockGetCreditPackInventory,
+  getCreditPackProgram: mockGetCreditPackProgram,
+}));
+
+function flattenToolText(result: unknown): string {
+  const content = (result as { content?: Array<{ type?: string; text?: string }> })?.content ?? [];
+  return content
+    .filter((item) => item?.type === "text")
+    .map((item) => item.text ?? "")
+    .join("\n\n");
+}
+
+function makeServerHarness() {
+  const handlers = new Map<string, (args: Record<string, unknown>) => Promise<unknown>>();
+  return {
+    handlers,
+    server: {
+      tool(name: string, _description: string, _schema: unknown, handler: (args: Record<string, unknown>) => Promise<unknown>) {
+        handlers.set(name, handler);
+      },
+    },
+  };
+}
+
+const selectedAgent = {
+  id: "agent-1",
+  name: "MockFixedPrice",
+  pricePerRunUsd: "0.01",
+  tags: ["translation"],
+  mcpSchema: null,
+};
+
+describe("run_agent MCP tool", () => {
+  beforeEach(() => {
+    vi.resetModules();
+    vi.clearAllMocks();
+
+    mockHasWalletConfigured.mockReturnValue(true);
+    mockGetConfiguredMethods.mockReturnValue(["card"]);
+    mockGetCompatiblePaymentMethods.mockReturnValue(["card"]);
+    mockNormalizePaymentMethod.mockImplementation((method: string) => method);
+    mockRequiresSpendConfirmation.mockReturnValue(true);
+    mockGetDefaultTipAmount.mockReturnValue(0);
+    mockCanSpend.mockReturnValue({ ok: true, message: "" });
+    mockRequiresPolicyConfirmation.mockReturnValue(false);
+    mockUploadLocalFiles.mockResolvedValue({ input: { text: "hello", target_language: "es" }, uploads: [] });
+    mockGetActiveCreditPack.mockReturnValue(null);
+    mockGetCreditPackInventory.mockResolvedValue([]);
+    mockGetCreditPackProgram.mockReturnValue(null);
+    mockGetOrCreatePendingCardSetup.mockResolvedValue({ url: "https://example.com/card-setup" });
+    mockFormatCardSetupBlocks.mockReturnValue(["card setup"]);
+    mockGetWalletAddress.mockResolvedValue("0xabc");
+    mockApiGet.mockResolvedValue(selectedAgent);
+  });
+
+  it("quotes explicit agent execution before running", async () => {
+    const { registerRunTools } = await import("../run.js");
+    const harness = makeServerHarness();
+    registerRunTools(harness.server as never);
+
+    const runAgent = harness.handlers.get("run_agent");
+    expect(runAgent).toBeDefined();
+
+    const result = await runAgent!({
+      agent_id: selectedAgent.id,
+      input: { text: "hello", target_language: "es" },
+      pay_with: "card",
+    });
+    const text = flattenToolText(result);
+
+    expect(text).toContain("Ready to run MockFixedPrice");
+    expect(text).toContain('run_agent({ agent_id: "agent-1", input: <same>, pay_with: "card", confirmed: true })');
+    expect(mockApiPostWithPayment).not.toHaveBeenCalled();
+  });
+
+  it("runs the explicitly selected agent when confirmed", async () => {
+    mockApiPostWithPayment.mockResolvedValueOnce({
+      status: "success",
+      job_id: "job-1",
+      agent_id: selectedAgent.id,
+      agent_name: selectedAgent.name,
+      output: { translated_text: "hola" },
+      cost: 0.01,
+    });
+
+    const { registerRunTools } = await import("../run.js");
+    const harness = makeServerHarness();
+    registerRunTools(harness.server as never);
+
+    const runAgent = harness.handlers.get("run_agent");
+    expect(runAgent).toBeDefined();
+
+    const result = await runAgent!({
+      agent_id: selectedAgent.id,
+      input: { text: "hello", target_language: "es" },
+      pay_with: "card",
+      confirmed: true,
+    });
+    const text = flattenToolText(result);
+
+    expect(mockApiPostWithPayment).toHaveBeenCalledWith(
+      "/agents/agent-1/run",
+      { input: { text: "hello", target_language: "es" } },
+      "card",
+    );
+    expect(mockRecordSpend).toHaveBeenCalledWith("card", 0.01);
+    expect(text).toContain("translated_text");
+    expect(text).toContain("✓ MockFixedPrice");
+    expect(text).toContain("Paid: $0.01 via card");
+    expect(text).toContain("Job ID: job-1");
+  });
+});

package/src/tools/__tests__/solve.test.ts

@@ -0,0 +1,186 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+const mockApiGet = vi.fn();
+const mockApiPost = vi.fn();
+const mockApiPostWithPayment = vi.fn();
+const mockGetCompatiblePaymentMethods = vi.fn();
+const mockHasWalletConfigured = vi.fn();
+const mockGetConfiguredMethods = vi.fn();
+const mockGetAcceptedPaymentMethods = vi.fn();
+const mockGetWalletAddress = vi.fn();
+const mockNormalizePaymentMethod = vi.fn();
+const mockToRegistryPaymentMethod = vi.fn();
+const mockRequiresSpendConfirmation = vi.fn();
+const mockGetDefaultTipAmount = vi.fn();
+const mockCanSpend = vi.fn();
+const mockRecordSpend = vi.fn();
+const mockRequiresPolicyConfirmation = vi.fn();
+const mockUploadLocalFiles = vi.fn();
+const mockStoreFeedbackToken = vi.fn();
+const mockGetActiveCreditPack = vi.fn();
+const mockGetCreditPackInventory = vi.fn();
+const mockGetCreditPackProgram = vi.fn();
+const mockGetOrCreatePendingCardSetup = vi.fn();
+const mockFormatCardSetupBlocks = vi.fn();
+
+vi.mock("../../core/api-client.js", () => ({
+  apiGet: mockApiGet,
+  apiPost: mockApiPost,
+  apiPostWithPayment: mockApiPostWithPayment,
+}));
+
+vi.mock("../../core/card-setup.js", () => ({
+  getOrCreatePendingCardSetup: mockGetOrCreatePendingCardSetup,
+  formatCardSetupBlocks: mockFormatCardSetupBlocks,
+}));
+
+vi.mock("../../core/payments.js", () => ({
+  getCompatiblePaymentMethods: mockGetCompatiblePaymentMethods,
+  hasWalletConfigured: mockHasWalletConfigured,
+  getConfiguredMethods: mockGetConfiguredMethods,
+  getAcceptedPaymentMethods: mockGetAcceptedPaymentMethods,
+  getWalletAddress: mockGetWalletAddress,
+  normalizePaymentMethod: mockNormalizePaymentMethod,
+  toRegistryPaymentMethod: mockToRegistryPaymentMethod,
+}));
+
+vi.mock("../../core/config.js", () => ({
+  requiresSpendConfirmation: mockRequiresSpendConfirmation,
+  getDefaultTipAmount: mockGetDefaultTipAmount,
+}));
+
+vi.mock("../../core/spend-policy.js", () => ({
+  canSpend: mockCanSpend,
+  recordSpend: mockRecordSpend,
+  requiresPolicyConfirmation: mockRequiresPolicyConfirmation,
+}));
+
+vi.mock("../../core/file-upload.js", () => ({
+  uploadLocalFiles: mockUploadLocalFiles,
+}));
+
+vi.mock("../_token-cache.js", () => ({
+  storeFeedbackToken: mockStoreFeedbackToken,
+}));
+
+vi.mock("../../core/passes.js", () => ({
+  getActiveCreditPack: mockGetActiveCreditPack,
+  getCreditPackInventory: mockGetCreditPackInventory,
+  getCreditPackProgram: mockGetCreditPackProgram,
+}));
+
+function flattenToolText(result: unknown): string {
+  const content = (result as { content?: Array<{ type?: string; text?: string }> })?.content ?? [];
+  return content
+    .filter((item) => item?.type === "text")
+    .map((item) => item.text ?? "")
+    .join("\n\n");
+}
+
+function makeServerHarness() {
+  const handlers = new Map<string, (args: Record<string, unknown>) => Promise<unknown>>();
+  return {
+    handlers,
+    server: {
+      tool(name: string, _description: string, _schema: unknown, handler: (args: Record<string, unknown>) => Promise<unknown>) {
+        handlers.set(name, handler);
+      },
+    },
+  };
+}
+
+const selectedAgent = {
+  id: "agent-1",
+  name: "MockFixedPrice",
+  pricePerRunUsd: "0.01",
+  tags: ["translation"],
+  mcpSchema: null,
+};
+
+describe("solve MCP tool", () => {
+  beforeEach(() => {
+    vi.resetModules();
+    vi.clearAllMocks();
+
+    mockHasWalletConfigured.mockReturnValue(true);
+    mockGetConfiguredMethods.mockReturnValue(["card"]);
+    mockGetAcceptedPaymentMethods.mockReturnValue(["card"]);
+    mockGetCompatiblePaymentMethods.mockReturnValue(["card"]);
+    mockNormalizePaymentMethod.mockImplementation((method: string) => method);
+    mockToRegistryPaymentMethod.mockImplementation((method: string) => method);
+    mockRequiresSpendConfirmation.mockReturnValue(true);
+    mockGetDefaultTipAmount.mockReturnValue(0);
+    mockCanSpend.mockReturnValue({ ok: true, message: "" });
+    mockRequiresPolicyConfirmation.mockReturnValue(false);
+    mockUploadLocalFiles.mockResolvedValue({ input: { text: "hello", target_language: "es" }, uploads: [] });
+    mockGetActiveCreditPack.mockReturnValue(null);
+    mockGetCreditPackInventory.mockResolvedValue([]);
+    mockGetCreditPackProgram.mockReturnValue(null);
+    mockGetOrCreatePendingCardSetup.mockResolvedValue({ url: "https://example.com/card-setup" });
+    mockFormatCardSetupBlocks.mockReturnValue(["card setup"]);
+    mockGetWalletAddress.mockResolvedValue("0xabc");
+  });
+
+  it("falls back to a quoted confirmation flow when /solve returns a 402 challenge", async () => {
+    mockApiPost.mockRejectedValueOnce(Object.assign(new Error("Payment Required"), { status: 402 }));
+    mockApiGet.mockResolvedValueOnce([selectedAgent]);
+
+    const { registerSolveTools } = await import("../solve.js");
+    const harness = makeServerHarness();
+    registerSolveTools(harness.server as never);
+
+    const solve = harness.handlers.get("solve");
+    expect(solve).toBeDefined();
+
+    const result = await solve!({
+      intent: "MockFixedPrice",
+      input: { text: "hello", target_language: "es" },
+      budget: 0.05,
+      pay_with: "card",
+    });
+    const text = flattenToolText(result);
+
+    expect(text).toContain("Best match: MockFixedPrice");
+    expect(text).toContain('solve({ intent: "MockFixedPrice"');
+    expect(text).toContain('pay_with: "card"');
+    expect(mockApiPostWithPayment).not.toHaveBeenCalled();
+  });
+
+  it("falls back to paid execution when confirmed after a 402 challenge", async () => {
+    mockApiPost.mockRejectedValueOnce(Object.assign(new Error("Payment Required"), { status: 402 }));
+    mockApiGet.mockResolvedValueOnce([selectedAgent]);
+    mockApiPostWithPayment.mockResolvedValueOnce({
+      status: "success",
+      job_id: "job-1",
+      agent_id: selectedAgent.id,
+      agent_name: selectedAgent.name,
+      output: { translated_text: "hola" },
+      cost: 0.01,
+    });
+
+    const { registerSolveTools } = await import("../solve.js");
+    const harness = makeServerHarness();
+    registerSolveTools(harness.server as never);
+
+    const solve = harness.handlers.get("solve");
+    expect(solve).toBeDefined();
+
+    const result = await solve!({
+      intent: "MockFixedPrice",
+      input: { text: "hello", target_language: "es" },
+      budget: 0.05,
+      pay_with: "card",
+      confirmed: true,
+    });
+    const text = flattenToolText(result);
+
+    expect(mockApiPostWithPayment).toHaveBeenCalledWith(
+      "/agents/agent-1/run",
+      { input: { text: "hello", target_language: "es" } },
+      "card",
+    );
+    expect(text).toContain("Running MockFixedPrice — best match");
+    expect(text).toContain("Job ID: job-1");
+    expect(text).toContain("Paid: $0.01 via card");
+  });
+});