@agentwonderland/mcp 0.1.14 → 0.1.15

package/dist/tools/run.js

@@ -1,10 +1,34 @@
 import { z } from "zod";
 import { apiGet, apiPost, apiPostWithPayment } from "../core/api-client.js";
 import { uploadLocalFiles } from "../core/file-upload.js";
-import { getConfiguredMethods, hasWalletConfigured } from "../core/payments.js";
+import { getConfiguredMethods, hasWalletConfigured, getWalletAddress } from "../core/payments.js";
 import { requiresSpendConfirmation, getDefaultTipAmount } from "../core/config.js";
 import { formatRunResult } from "../core/formatters.js";
 import { storeFeedbackToken } from "./_token-cache.js";
+const POLL_INTERVAL_MS = 3000;
+const POLL_MAX_MS = 120000; // 2 minutes
+async function pollJobUntilDone(jobId, agentName) {
+    const deadline = Date.now() + POLL_MAX_MS;
+    const walletAddress = await getWalletAddress();
+    const walletParam = walletAddress ? `?wallet=${walletAddress}` : "";
+    while (Date.now() < deadline) {
+        await new Promise((r) => setTimeout(r, POLL_INTERVAL_MS));
+        try {
+            const job = await apiGet(`/jobs/${jobId}${walletParam}`);
+            if (job.status === "completed") {
+                return { status: "completed", output: job.output };
+            }
+            if (job.status === "failed") {
+                return { status: "failed", output: job.output, error_code: job.error_code };
+            }
+            // Still processing — continue polling
+        }
+        catch {
+            // Ignore poll errors, keep trying until deadline
+        }
+    }
+    return { status: "failed", error_code: "POLL_TIMEOUT" };
+}
 function text(t) {
     return { content: [{ type: "text", text: t }] };
 }
@@ -99,6 +123,38 @@ export function registerRunTools(server) {
         }
         const actualCost = result.cost;
         const status = result.status;
+        // Async agent — poll until complete
+        if (status === "processing") {
+            const pollResult = await pollJobUntilDone(jobId, agentName);
+            if (pollResult.status === "completed") {
+                const asyncFormatted = formatRunResult({
+                    ...result,
+                    status: "success",
+                    output: pollResult.output,
+                }, { paymentMethod: method ?? getConfiguredMethods()[0] });
+                if (result.feedback_token) {
+                    storeFeedbackToken(jobId, result.feedback_token, resultAgentId);
+                }
+                const asyncCostLine = actualCost != null ? `Paid $${actualCost.toFixed(actualCost < 0.01 ? 4 : 2)}.` : "";
+                const asyncFeedbackAsk = [
+                    `IMPORTANT: After presenting the result above, you MUST ask the user:`,
+                    `"${asyncCostLine} Was this result helpful? I can rate it (1-5 stars) or tip the builder for you."`,
+                    "",
+                    `If the user wants to rate: rate_agent({ job_id: "${jobId}", rating: <1-5>, comment: "..." })`,
+                    `If the user wants to tip: tip_agent({ job_id: "${jobId}", agent_id: "${resultAgentId}", amount: 0.25 })`,
+                    `To save for later: favorite_agent({ agent_id: "${resultAgentId}" })`,
+                ].join("\n");
+                return multiText(uploadSummary ? `${uploadSummary}\n\n${asyncFormatted}` : asyncFormatted, asyncFeedbackAsk);
+            }
+            // Async agent failed
+            const failedFormatted = formatRunResult({
+                ...result,
+                status: "failed",
+                output: pollResult.output,
+                error_code: pollResult.error_code,
+            }, { paymentMethod: method ?? getConfiguredMethods()[0] });
+            return multiText(uploadSummary ? `${uploadSummary}\n\n${failedFormatted}` : failedFormatted, "The agent execution failed. A refund has been initiated automatically.");
+        }
         // Auto-tip if configured and run succeeded
         const defaultTip = getDefaultTipAmount();
         let tipLine = "";

package/dist/tools/solve.js

@@ -1,10 +1,33 @@
 import { z } from "zod";
 import { apiGet, apiPost, apiPostWithPayment } from "../core/api-client.js";
-import { hasWalletConfigured, getConfiguredMethods, getAcceptedPaymentMethods, } from "../core/payments.js";
+import { hasWalletConfigured, getConfiguredMethods, getAcceptedPaymentMethods, getWalletAddress, } from "../core/payments.js";
 import { requiresSpendConfirmation, getDefaultTipAmount } from "../core/config.js";
 import { agentList, formatRunResult } from "../core/formatters.js";
 import { uploadLocalFiles } from "../core/file-upload.js";
 import { storeFeedbackToken } from "./_token-cache.js";
+const POLL_INTERVAL_MS = 3000;
+const POLL_MAX_MS = 120000;
+async function pollSolveJob(jobId) {
+    const deadline = Date.now() + POLL_MAX_MS;
+    const walletAddress = await getWalletAddress();
+    const walletParam = walletAddress ? `?wallet=${walletAddress}` : "";
+    while (Date.now() < deadline) {
+        await new Promise((r) => setTimeout(r, POLL_INTERVAL_MS));
+        try {
+            const job = await apiGet(`/jobs/${jobId}${walletParam}`);
+            if (job.status === "completed") {
+                return { status: "completed", output: job.output };
+            }
+            if (job.status === "failed") {
+                return { status: "failed", output: job.output, error_code: job.error_code };
+            }
+        }
+        catch {
+            // Ignore poll errors, keep trying
+        }
+    }
+    return { status: "failed", error_code: "POLL_TIMEOUT" };
+}
 function text(t) {
     return { content: [{ type: "text", text: t }] };
 }
@@ -171,9 +194,46 @@ export function registerSolveTools(server) {
         }
         const jobId = result.job_id ?? "";
         const agentId2 = result.agent_id ?? selected.id;
+        const status = result.status;
         if (result.feedback_token) {
             storeFeedbackToken(jobId, result.feedback_token, agentId2);
         }
+        // Async agent — poll until complete
+        if (status === "processing") {
+            const pollResult = await pollSolveJob(jobId);
+            if (pollResult.status === "completed") {
+                const asyncFormatted = formatRunResult({
+                    ...result,
+                    status: "success",
+                    output: pollResult.output,
+                }, { paymentMethod: method });
+                const asyncOutput = [
+                    discovery,
+                    "",
+                    `Running ${selected.name} — best match`,
+                    `Estimated cost: $${estimatedCost.toFixed(4)}`,
+                    "",
+                    asyncFormatted,
+                ].join("\n");
+                const asyncCost = result.cost;
+                return multiText(asyncOutput, feedbackAsk(jobId, agentId2, asyncCost, ""));
+            }
+            // Async agent failed
+            const failedFormatted = formatRunResult({
+                ...result,
+                status: "failed",
+                output: pollResult.output,
+                error_code: pollResult.error_code,
+            }, { paymentMethod: method });
+            const failedOutput = [
+                discovery,
+                "",
+                `Running ${selected.name} — best match`,
+                "",
+                failedFormatted,
+            ].join("\n");
+            return multiText(failedOutput, "The agent execution failed. A refund has been initiated automatically.");
+        }
         const actualCost = result.cost;
         const tipMsg = result.feedback_token ? await autoTip(jobId, agentId2, selected.name ?? "", result.feedback_token) : "";
         const output = [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentwonderland/mcp",
-  "version": "0.1.14",
+  "version": "0.1.15",
   "type": "module",
   "description": "MCP server for the Agent Wonderland AI agent marketplace",
   "bin": {

package/src/tools/run.ts

@@ -2,11 +2,46 @@ import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { z } from "zod";
 import { apiGet, apiPost, apiPostWithPayment } from "../core/api-client.js";
 import { uploadLocalFiles } from "../core/file-upload.js";
-import { getConfiguredMethods, hasWalletConfigured } from "../core/payments.js";
+import { getConfiguredMethods, hasWalletConfigured, getWalletAddress } from "../core/payments.js";
 import { requiresSpendConfirmation, getDefaultTipAmount } from "../core/config.js";
 import { formatRunResult } from "../core/formatters.js";
 import { storeFeedbackToken, getFeedbackToken } from "./_token-cache.js";
 
+const POLL_INTERVAL_MS = 3000;
+const POLL_MAX_MS = 120000; // 2 minutes
+
+async function pollJobUntilDone(
+  jobId: string,
+  agentName: string,
+): Promise<{ status: string; output?: unknown; error_code?: string }> {
+  const deadline = Date.now() + POLL_MAX_MS;
+  const walletAddress = await getWalletAddress();
+  const walletParam = walletAddress ? `?wallet=${walletAddress}` : "";
+
+  while (Date.now() < deadline) {
+    await new Promise((r) => setTimeout(r, POLL_INTERVAL_MS));
+    try {
+      const job = await apiGet<{
+        status: string;
+        output?: unknown;
+        error_code?: string;
+      }>(`/jobs/${jobId}${walletParam}`);
+
+      if (job.status === "completed") {
+        return { status: "completed", output: job.output };
+      }
+      if (job.status === "failed") {
+        return { status: "failed", output: job.output, error_code: job.error_code };
+      }
+      // Still processing — continue polling
+    } catch {
+      // Ignore poll errors, keep trying until deadline
+    }
+  }
+
+  return { status: "failed", error_code: "POLL_TIMEOUT" };
+}
+
 function text(t: string) {
   return { content: [{ type: "text" as const, text: t }] };
 }
@@ -127,6 +162,50 @@ export function registerRunTools(server: McpServer): void {
       const actualCost = result.cost as number | undefined;
       const status = result.status as string;
 
+      // Async agent — poll until complete
+      if (status === "processing") {
+        const pollResult = await pollJobUntilDone(jobId, agentName);
+        if (pollResult.status === "completed") {
+          const asyncFormatted = formatRunResult({
+            ...result,
+            status: "success",
+            output: pollResult.output,
+          }, { paymentMethod: method ?? getConfiguredMethods()[0] });
+
+          if (result.feedback_token) {
+            storeFeedbackToken(jobId, result.feedback_token as string, resultAgentId);
+          }
+
+          const asyncCostLine = actualCost != null ? `Paid $${actualCost.toFixed(actualCost < 0.01 ? 4 : 2)}.` : "";
+          const asyncFeedbackAsk = [
+            `IMPORTANT: After presenting the result above, you MUST ask the user:`,
+            `"${asyncCostLine} Was this result helpful? I can rate it (1-5 stars) or tip the builder for you."`,
+            "",
+            `If the user wants to rate: rate_agent({ job_id: "${jobId}", rating: <1-5>, comment: "..." })`,
+            `If the user wants to tip: tip_agent({ job_id: "${jobId}", agent_id: "${resultAgentId}", amount: 0.25 })`,
+            `To save for later: favorite_agent({ agent_id: "${resultAgentId}" })`,
+          ].join("\n");
+
+          return multiText(
+            uploadSummary ? `${uploadSummary}\n\n${asyncFormatted}` : asyncFormatted,
+            asyncFeedbackAsk,
+          );
+        }
+
+        // Async agent failed
+        const failedFormatted = formatRunResult({
+          ...result,
+          status: "failed",
+          output: pollResult.output,
+          error_code: pollResult.error_code,
+        }, { paymentMethod: method ?? getConfiguredMethods()[0] });
+
+        return multiText(
+          uploadSummary ? `${uploadSummary}\n\n${failedFormatted}` : failedFormatted,
+          "The agent execution failed. A refund has been initiated automatically.",
+        );
+      }
+
       // Auto-tip if configured and run succeeded
       const defaultTip = getDefaultTipAmount();
       let tipLine = "";

package/src/tools/solve.ts

@@ -5,6 +5,7 @@ import {
   hasWalletConfigured,
   getConfiguredMethods,
   getAcceptedPaymentMethods,
+  getWalletAddress,
 } from "../core/payments.js";
 import { requiresSpendConfirmation, getDefaultTipAmount } from "../core/config.js";
 import { agentList, formatRunResult } from "../core/formatters.js";
@@ -12,6 +13,39 @@ import { uploadLocalFiles } from "../core/file-upload.js";
 import type { AgentRecord } from "../core/types.js";
 import { storeFeedbackToken } from "./_token-cache.js";
 
+const POLL_INTERVAL_MS = 3000;
+const POLL_MAX_MS = 120000;
+
+async function pollSolveJob(
+  jobId: string,
+): Promise<{ status: string; output?: unknown; error_code?: string }> {
+  const deadline = Date.now() + POLL_MAX_MS;
+  const walletAddress = await getWalletAddress();
+  const walletParam = walletAddress ? `?wallet=${walletAddress}` : "";
+
+  while (Date.now() < deadline) {
+    await new Promise((r) => setTimeout(r, POLL_INTERVAL_MS));
+    try {
+      const job = await apiGet<{
+        status: string;
+        output?: unknown;
+        error_code?: string;
+      }>(`/jobs/${jobId}${walletParam}`);
+
+      if (job.status === "completed") {
+        return { status: "completed", output: job.output };
+      }
+      if (job.status === "failed") {
+        return { status: "failed", output: job.output, error_code: job.error_code };
+      }
+    } catch {
+      // Ignore poll errors, keep trying
+    }
+  }
+
+  return { status: "failed", error_code: "POLL_TIMEOUT" };
+}
+
 function text(t: string) {
   return { content: [{ type: "text" as const, text: t }] };
 }
@@ -204,11 +238,54 @@ export function registerSolveTools(server: McpServer): void {
 
       const jobId = (result as Record<string, unknown>).job_id as string ?? "";
       const agentId2 = (result as Record<string, unknown>).agent_id as string ?? selected.id;
+      const status = result.status as string;
 
       if (result.feedback_token) {
         storeFeedbackToken(jobId, result.feedback_token as string, agentId2);
       }
 
+      // Async agent — poll until complete
+      if (status === "processing") {
+        const pollResult = await pollSolveJob(jobId);
+        if (pollResult.status === "completed") {
+          const asyncFormatted = formatRunResult({
+            ...result,
+            status: "success",
+            output: pollResult.output,
+          }, { paymentMethod: method });
+
+          const asyncOutput = [
+            discovery,
+            "",
+            `Running ${selected.name} — best match`,
+            `Estimated cost: $${estimatedCost.toFixed(4)}`,
+            "",
+            asyncFormatted,
+          ].join("\n");
+
+          const asyncCost = (result as Record<string, unknown>).cost as number | undefined;
+          return multiText(asyncOutput, feedbackAsk(jobId, agentId2, asyncCost, ""));
+        }
+
+        // Async agent failed
+        const failedFormatted = formatRunResult({
+          ...result,
+          status: "failed",
+          output: pollResult.output,
+          error_code: pollResult.error_code,
+        }, { paymentMethod: method });
+
+        const failedOutput = [
+          discovery,
+          "",
+          `Running ${selected.name} — best match`,
+          "",
+          failedFormatted,
+        ].join("\n");
+
+        return multiText(failedOutput, "The agent execution failed. A refund has been initiated automatically.");
+      }
+
       const actualCost = (result as Record<string, unknown>).cost as number | undefined;
       const tipMsg = result.feedback_token ? await autoTip(jobId, agentId2, selected.name ?? "", result.feedback_token as string) : "";