@dataville/dataville-mcp 0.1.1 → 0.1.3

package/dist/client.js

@@ -1,6 +1,6 @@
 const DEFAULT_BASE_URL = "https://api.dataville.com";
 // Keep in sync with package.json version (enforced by a test).
-export const VERSION = "0.1.1";
+export const VERSION = "0.1.3";
 const USER_AGENT = `dataville-mcp/${VERSION}`;
 export class DatavilleApiError extends Error {
     status;
@@ -10,6 +10,21 @@ export class DatavilleApiError extends Error {
         this.name = "DatavilleApiError";
     }
 }
+/**
+ * Raised when the API accepted the request but did not recognise our API key.
+ *
+ * Dataville's auth middleware never rejects outright — an unrecognised key
+ * falls through to anonymous access, which still returns data but under the
+ * much lower anonymous rate limit and without attributing usage to the
+ * account. Left unchecked that failure is invisible inside an MCP client, so
+ * we surface it explicitly instead of returning results that look fine.
+ */
+export class DatavilleAuthError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = "DatavilleAuthError";
+    }
+}
 function getConfig() {
     const apiKey = process.env.DATAVILLE_API_KEY;
     if (!apiKey) {
@@ -34,9 +49,27 @@ export async function searchDataSource(source, keywords, params) {
     });
     const body = await response.json().catch(() => undefined);
     if (!response.ok) {
-        const message = (body && typeof body === "object" && "error" in body && String(body.error)) ||
-            `Dataville API request failed with status ${response.status}`;
+        // Errors come back as { status: "error", data: { error: "..." } }. Those
+        // messages are useful to the model — "no results for X", or the full list
+        // of valid sources — so prefer them over a bare status code.
+        const apiMessage = body && typeof body === "object"
+            ? body.data?.error ??
+                body.error
+            : undefined;
+        const message = typeof apiMessage === "string" && apiMessage.length > 0
+            ? apiMessage
+            : `Dataville API request failed with status ${response.status}`;
         throw new DatavilleApiError(response.status, message);
     }
+    // We always send an API key, so an "anonymous" account state means the key
+    // was not accepted. Fail loudly rather than silently serving anonymous-tier
+    // results that are neither attributed nor billed to the user's account.
+    if (body &&
+        typeof body === "object" &&
+        body.account_state === "anonymous") {
+        throw new DatavilleAuthError("Your DATAVILLE_API_KEY was not recognised, so this request fell back to anonymous access " +
+            "(much lower rate limits, and usage is not attributed to your account). " +
+            "Check the key in your MCP client config, or generate a new one at https://app.dataville.com/api-keys.");
+    }
     return body;
 }

package/dist/index.js

@@ -2,7 +2,7 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { z } from "zod";
-import { searchDataSource, DatavilleApiError, VERSION } from "./client.js";
+import { searchDataSource, VERSION } from "./client.js";
 import { DATAVILLE_SOURCES } from "./sources.js";
 const server = new McpServer({
     name: "dataville-mcp-server",
@@ -32,7 +32,7 @@ server.registerTool("search_dataville", {
         return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
     }
     catch (error) {
-        const message = error instanceof DatavilleApiError ? error.message : String(error);
+        const message = error instanceof Error ? error.message : String(error);
         return { content: [{ type: "text", text: message }], isError: true };
     }
 });

package/dist/sources.js

@@ -1,14 +1,17 @@
 // Hand-maintained list of data sources supported by the Dataville API.
-// Update when Dataville adds or removes a supported source.
+// The `name` values must match the source names the API accepts — a mismatch
+// makes the source unreachable. To check against the live API, request an
+// unknown source (e.g. /nosuchsource/foo); the error lists every valid name.
 export const DATAVILLE_SOURCES = [
     { name: "wikipedia", description: "Wikipedia articles" },
     { name: "arxiv", description: "arXiv preprints" },
     { name: "gutenberg", description: "Project Gutenberg public-domain books" },
     { name: "census", description: "US Census Bureau data" },
     { name: "fooddata", description: "USDA FoodData Central" },
-    { name: "pwc", description: "Papers with Code — ML papers, code, and benchmarks" },
+    { name: "paperswithcode", description: "Papers with Code — ML papers, code, and benchmarks" },
     { name: "edgar", description: "SEC EDGAR filings" },
     { name: "openalex", description: "OpenAlex scholarly works" },
     { name: "pypi", description: "PyPI package metadata" },
     { name: "stackexchange", description: "Stack Exchange Q&A" },
+    { name: "news", description: "News front-page headlines and summaries" },
 ];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dataville/dataville-mcp",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "MCP server exposing Dataville's data source API as tools for MCP clients",
   "private": false,
   "license": "MIT",
@@ -22,7 +22,8 @@
     "prepublishOnly": "npm run build",
     "dev": "tsx watch src/index.ts",
     "start": "node dist/index.js",
-    "test": "tsx --test src/__tests__/*.test.ts"
+    "test": "tsx --test src/__tests__/*.test.ts",
+    "test:integration": "tsx --test src/__tests__/*.itest.ts"
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.12.0",