@agentwonderland/mcp 0.1.45 → 0.1.46

package/dist/tools/__tests__/search.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/tools/__tests__/search.test.js

@@ -0,0 +1,66 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+const mockApiGet = vi.fn();
+const mockGetAcceptedPaymentMethods = vi.fn();
+const mockIsFavorite = vi.fn();
+vi.mock("../../core/api-client.js", () => ({
+    apiGet: mockApiGet,
+}));
+vi.mock("../../core/payments.js", () => ({
+    getAcceptedPaymentMethods: mockGetAcceptedPaymentMethods,
+}));
+vi.mock("../../core/config.js", () => ({
+    isFavorite: mockIsFavorite,
+}));
+function flattenToolText(result) {
+    const content = result?.content ?? [];
+    return content
+        .filter((item) => item?.type === "text")
+        .map((item) => item.text ?? "")
+        .join("\n\n");
+}
+function makeServerHarness() {
+    const handlers = new Map();
+    return {
+        handlers,
+        server: {
+            tool(name, _description, _schema, handler) {
+                handlers.set(name, handler);
+            },
+        },
+    };
+}
+describe("search_agents MCP tool", () => {
+    beforeEach(() => {
+        vi.resetModules();
+        vi.clearAllMocks();
+        mockGetAcceptedPaymentMethods.mockReturnValue([]);
+        mockIsFavorite.mockReturnValue(false);
+    });
+    it("sorts popularity by live completed job stats after API enrichment", async () => {
+        mockApiGet.mockResolvedValueOnce([
+            {
+                id: "agent-low",
+                name: "Low History",
+                slug: "low-history",
+                pricePerRunUsd: "0.100000",
+                stats: { completedJobs: 2, avgRating: null, ratingCount: 0 },
+            },
+            {
+                id: "agent-high",
+                name: "High History",
+                slug: "high-history",
+                pricePerRunUsd: "0.100000",
+                stats: { completedJobs: 6, avgRating: null, ratingCount: 0 },
+            },
+        ]);
+        const { registerSearchTools } = await import("../search.js");
+        const harness = makeServerHarness();
+        registerSearchTools(harness.server);
+        const search = harness.handlers.get("search_agents");
+        expect(search).toBeDefined();
+        const result = await search({ query: "stock research", limit: 2, sort: "popularity" });
+        const output = flattenToolText(result);
+        expect(output.indexOf("High History")).toBeLessThan(output.indexOf("Low History"));
+        expect(output).toContain("High History (high-history) ☆☆☆☆☆ 6 jobs");
+    });
+});

package/dist/tools/search.js

@@ -6,6 +6,23 @@ import { agentLine } from "../core/formatters.js";
 function text(t) {
     return { content: [{ type: "text", text: t }] };
 }
+function completedJobs(agent) {
+    const stats = agent.stats;
+    return stats?.completedJobs ?? agent.totalExecutions ?? 0;
+}
+function averageRating(agent) {
+    const stats = agent.stats;
+    return agent.avgRating ?? stats?.avgRating ?? 0;
+}
+function ratingCount(agent) {
+    const stats = agent.stats;
+    const topLevel = typeof agent.ratingCount === "number" ? agent.ratingCount : undefined;
+    return topLevel ?? stats?.ratingCount ?? 0;
+}
+function price(agent) {
+    const parsed = Number.parseFloat(agent.pricePerRunUsd ?? "0");
+    return Number.isFinite(parsed) ? parsed : 0;
+}
 export function registerSearchTools(server) {
     server.tool("search_agents", "Search the Agent Wonderland marketplace for AI agents. Returns a ranked list of matching agents with ratings, pricing, and job counts.", {
         query: z.string().optional().describe("Search query (natural language or keywords)"),
@@ -61,6 +78,22 @@ export function registerSearchTools(server) {
                 return typeof avg === "number" && avg >= min_rating;
             });
         }
+        // The API enriches list rows with live stats after registry search. Do a
+        // final local sort for stat-based MCP views so displayed jobs/ratings and
+        // result order cannot disagree while registry aggregates catch up.
+        if (sort === "popularity") {
+            agents = [...agents].sort((a, b) => completedJobs(b) - completedJobs(a) ||
+                ratingCount(b) - ratingCount(a) ||
+                averageRating(b) - averageRating(a));
+        }
+        else if (sort === "rating") {
+            agents = [...agents].sort((a, b) => averageRating(b) - averageRating(a) ||
+                ratingCount(b) - ratingCount(a) ||
+                completedJobs(b) - completedJobs(a));
+        }
+        else if (sort === "price") {
+            agents = [...agents].sort((a, b) => price(a) - price(b));
+        }
         // Trim to requested limit after filtering
         agents = agents.slice(0, requestedLimit);
         if (agents.length === 0) {

package/package.json

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

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

@@ -0,0 +1,78 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+const mockApiGet = vi.fn();
+const mockGetAcceptedPaymentMethods = vi.fn();
+const mockIsFavorite = vi.fn();
+
+vi.mock("../../core/api-client.js", () => ({
+  apiGet: mockApiGet,
+}));
+
+vi.mock("../../core/payments.js", () => ({
+  getAcceptedPaymentMethods: mockGetAcceptedPaymentMethods,
+}));
+
+vi.mock("../../core/config.js", () => ({
+  isFavorite: mockIsFavorite,
+}));
+
+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("search_agents MCP tool", () => {
+  beforeEach(() => {
+    vi.resetModules();
+    vi.clearAllMocks();
+    mockGetAcceptedPaymentMethods.mockReturnValue([]);
+    mockIsFavorite.mockReturnValue(false);
+  });
+
+  it("sorts popularity by live completed job stats after API enrichment", async () => {
+    mockApiGet.mockResolvedValueOnce([
+      {
+        id: "agent-low",
+        name: "Low History",
+        slug: "low-history",
+        pricePerRunUsd: "0.100000",
+        stats: { completedJobs: 2, avgRating: null, ratingCount: 0 },
+      },
+      {
+        id: "agent-high",
+        name: "High History",
+        slug: "high-history",
+        pricePerRunUsd: "0.100000",
+        stats: { completedJobs: 6, avgRating: null, ratingCount: 0 },
+      },
+    ]);
+
+    const { registerSearchTools } = await import("../search.js");
+    const harness = makeServerHarness();
+    registerSearchTools(harness.server as never);
+
+    const search = harness.handlers.get("search_agents");
+    expect(search).toBeDefined();
+
+    const result = await search!({ query: "stock research", limit: 2, sort: "popularity" });
+    const output = flattenToolText(result);
+
+    expect(output.indexOf("High History")).toBeLessThan(output.indexOf("Low History"));
+    expect(output).toContain("High History (high-history) ☆☆☆☆☆ 6 jobs");
+  });
+});

package/src/tools/search.ts

@@ -10,6 +10,27 @@ function text(t: string) {
   return { content: [{ type: "text" as const, text: t }] };
 }
 
+function completedJobs(agent: AgentRecord): number {
+  const stats = agent.stats as { completedJobs?: number } | undefined;
+  return stats?.completedJobs ?? agent.totalExecutions ?? 0;
+}
+
+function averageRating(agent: AgentRecord): number {
+  const stats = agent.stats as { avgRating?: number | null } | undefined;
+  return agent.avgRating ?? stats?.avgRating ?? 0;
+}
+
+function ratingCount(agent: AgentRecord): number {
+  const stats = agent.stats as { ratingCount?: number } | undefined;
+  const topLevel = typeof agent.ratingCount === "number" ? agent.ratingCount : undefined;
+  return topLevel ?? stats?.ratingCount ?? 0;
+}
+
+function price(agent: AgentRecord): number {
+  const parsed = Number.parseFloat(agent.pricePerRunUsd ?? "0");
+  return Number.isFinite(parsed) ? parsed : 0;
+}
+
 export function registerSearchTools(server: McpServer): void {
   server.tool(
     "search_agents",
@@ -71,6 +92,25 @@ export function registerSearchTools(server: McpServer): void {
         });
       }
 
+      // The API enriches list rows with live stats after registry search. Do a
+      // final local sort for stat-based MCP views so displayed jobs/ratings and
+      // result order cannot disagree while registry aggregates catch up.
+      if (sort === "popularity") {
+        agents = [...agents].sort((a, b) =>
+          completedJobs(b) - completedJobs(a) ||
+          ratingCount(b) - ratingCount(a) ||
+          averageRating(b) - averageRating(a),
+        );
+      } else if (sort === "rating") {
+        agents = [...agents].sort((a, b) =>
+          averageRating(b) - averageRating(a) ||
+          ratingCount(b) - ratingCount(a) ||
+          completedJobs(b) - completedJobs(a),
+        );
+      } else if (sort === "price") {
+        agents = [...agents].sort((a, b) => price(a) - price(b));
+      }
+
       // Trim to requested limit after filtering
       agents = agents.slice(0, requestedLimit);