@pantheon.ai/mcp 0.0.4 → 0.0.6

package/dist/cli.js

@@ -854,7 +854,7 @@ function getServerErrorMessage(error) {
 
 //#endregion
 //#region package.json
-var version = "0.0.3";
+var version = "0.0.5";
 
 //#endregion
 //#region src/mcp/mcp.ts
@@ -873,14 +873,13 @@ async function mcpServer({ authorizationHeader }) {
 		name: "Pantheon MCP",
 		version,
 		websiteUrl: "https://pantheon-ai.tidb.ai/"
-	});
+	}, { instructions: "Use tools to accept pantheon data." });
 	function registerApiTool(name, api, { paramsSchema, bodySchema, ...options }) {
+		const inputSchemaShape = { params: paramsSchema };
+		if (bodySchema) inputSchemaShape.body = bodySchema;
 		mcpServer.registerTool(name, {
 			...options,
-			inputSchema: z.object({
-				params: paramsSchema,
-				body: bodySchema
-			})
+			inputSchema: z.object(inputSchemaShape)
 		}, async ({ params, body }) => {
 			try {
 				const result = await executor.execute(api, params, body);
@@ -902,7 +901,7 @@ async function mcpServer({ authorizationHeader }) {
 	registerApiTool("listProjects", listProjects, {
 		title: "List pantheon projects",
 		paramsSchema: pageParams,
-		bodySchema: z.null(),
+		bodySchema: null,
 		outputSchema: paged(projectSchema)
 	});
 	registerApiTool("createProject", createProject, {
@@ -920,13 +919,13 @@ async function mcpServer({ authorizationHeader }) {
 	registerApiTool("listProjectBranches", listProjectBranches, {
 		title: "List project branches",
 		paramsSchema: pageParams.extend({ projectId: z.string() }),
-		bodySchema: z.null(),
+		bodySchema: null,
 		outputSchema: paged(branchSchema)
 	});
 	registerApiTool("getProject", getProject, {
 		title: "Get pantheon project",
 		paramsSchema: z.object({ projectId: z.string() }),
-		bodySchema: z.null(),
+		bodySchema: null,
 		outputSchema: projectSchema
 	});
 	registerApiTool("getProjectBranch", getProjectBranch, {
@@ -935,7 +934,7 @@ async function mcpServer({ authorizationHeader }) {
 			projectId: z.string(),
 			branchId: z.string()
 		}),
-		bodySchema: z.null(),
+		bodySchema: null,
 		outputSchema: branchSchema
 	});
 	registerApiTool("getProjectBranchOutput", getProjectBranchOutput, {
@@ -945,7 +944,7 @@ async function mcpServer({ authorizationHeader }) {
 			branchId: z.string(),
 			fullOutput: z.boolean().optional().default(false)
 		}),
-		bodySchema: z.null(),
+		bodySchema: null,
 		outputSchema: z.string()
 	});
 	registerApiTool("createProjectExploration", createProjectExploration, {
@@ -962,7 +961,7 @@ async function mcpServer({ authorizationHeader }) {
 	registerApiTool("listProjectExplorations", listProjectExplorations, {
 		title: "List project explorations",
 		paramsSchema: z.object({ projectId: z.string() }),
-		bodySchema: z.null(),
+		bodySchema: null,
 		outputSchema: explorationSchema.array()
 	});
 	registerApiTool("getProjectExplorationResult", getProjectExplorationResult, {
@@ -971,7 +970,7 @@ async function mcpServer({ authorizationHeader }) {
 			projectId: z.string(),
 			explorationId: z.string()
 		}),
-		bodySchema: z.null(),
+		bodySchema: null,
 		outputSchema: explorationResultSchema
 	});
 	return mcpServer;
@@ -992,7 +991,7 @@ async function startStdio({ apiKey }) {
 //#region src/streamableHttp.ts
 const port = process.env.PORT ? parseInt(process.env.PORT) : 9986;
 async function startStreamableHttpServer({ apiKey }) {
-	const app = createMcpExpressApp();
+	const app = createMcpExpressApp({ allowedHosts: ["pantheon-ai.tidb.ai", "localhost"] });
 	app.all("/mcp", async (req, res) => {
 		const authorizationHeader = req.get("authorization") ?? (apiKey ? `Bearer ${apiKey}` : void 0);
 		let server;
@@ -1031,7 +1030,7 @@ async function startStreamableHttpServer({ apiKey }) {
 		}
 	});
 	app.listen(port);
-	console.log("Listening on port 3001");
+	console.log(`Listening on port ${port}`);
 }
 
 //#endregion

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pantheon.ai/mcp",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "type": "module",
   "description": "",
   "bin": "dist/cli.js",