@elixium.ai/mcp-server 0.2.0 → 0.2.2

package/dist/index.js

@@ -10,6 +10,14 @@ const API_URL = process.env.ELIXIUM_API_URL || "https://elixium.ai/api";
 const BOARD_SLUG = process.env.ELIXIUM_BOARD_SLUG;
 const LANE_STYLE_ENV = process.env.ELIXIUM_LANE_STYLE;
 const USER_EMAIL = process.env.ELIXIUM_USER_EMAIL; // Optional: Override requester email for stories
+// UUID v4 format validation — prevents 500s from partial/truncated IDs
+const UUID_RE = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
+function assertUUID(value, label) {
+    if (!UUID_RE.test(value)) {
+        throw new Error(`Invalid ${label}: "${value}" is not a valid UUID. ` +
+            `Expected format: xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx (full UUID from list_stories/list_epics).`);
+    }
+}
 const CLI_ARGS = process.argv.slice(2);
 const hasArg = (flag) => CLI_ARGS.includes(flag);
 const getArgValue = (flag) => {
@@ -283,7 +291,7 @@ const normalizeLane = async (lane) => {
     const map = style === "upper" ? LANE_UPPER : LANE_TITLE;
     return map[key] || (style === "upper" ? lane.trim().toUpperCase() : lane.trim());
 };
-const buildIterationContext = (stories, user = null) => {
+const buildIterationContext = (stories, user = null, infraProfile) => {
     const currentIteration = stories.filter((story) => normalizeLaneForComparison(story?.lane) === "current");
     const backlog = stories.filter((story) => normalizeLaneForComparison(story?.lane) === "backlog");
     // Cost summary across current iteration
@@ -294,7 +302,13 @@ const buildIterationContext = (stories, user = null) => {
         estimatedCount: withEstimates.length,
         unestimatedCount: iterationStories.length - withEstimates.length,
     };
-    return { currentIteration, backlog, user, costSummary };
+    return {
+        currentIteration,
+        backlog,
+        user,
+        costSummary,
+        ...(infraProfile ? { infrastructureProfile: infraProfile } : {}),
+    };
 };
 const createServer = () => {
     const server = new Server({
@@ -520,7 +534,15 @@ const createServer = () => {
                             description: "Optional constraints (e.g., budget limit, region, instance types)",
                         },
                     },
-                    required: ["storyId", "provider"],
+                    required: ["storyId"],
+                },
+            },
+            {
+                name: "get_infrastructure_profile",
+                description: "Get the workspace infrastructure profile. Returns cloud provider, regions, compliance frameworks, existing services, and constraints. Use this to understand the team's environment before making infrastructure recommendations.",
+                inputSchema: {
+                    type: "object",
+                    properties: {},
                 },
             },
             {
@@ -590,6 +612,20 @@ const createServer = () => {
                     required: ["storyId", "testPlan"],
                 },
             },
+            {
+                name: "approve_tests",
+                description: "Approve a proposed test plan so implementation can proceed. Transitions workflow from tests_proposed to tests_approved. This is the human approval gate in the TDD workflow.",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        storyId: {
+                            type: "string",
+                            description: "ID of the story whose test plan to approve",
+                        },
+                    },
+                    required: ["storyId"],
+                },
+            },
             {
                 name: "submit_for_review",
                 description: "Submit implementation for human review. Only works if tests are approved. Sets state to finished.",
@@ -727,6 +763,13 @@ ${config.branchingDefaults ? `- Trunk-Based: ${config.branchingDefaults.trunkBas
 - Auto-Merge: ${config.branchingDefaults.autoMerge ? "Yes" : "No"}
 - Source: ${config.branchingDefaults.source}` : "Not configured (branch-based default)"}
 
+## Infrastructure Profile
+${config.infrastructureProfile?.provider ? `- Provider: ${config.infrastructureProfile.provider.toUpperCase()}
+- Regions: ${config.infrastructureProfile.regions?.join(", ") || "Not set"}
+- Compliance: ${config.infrastructureProfile.complianceFrameworks?.join(", ").toUpperCase() || "None"}
+- Existing Services: ${config.infrastructureProfile.existingServices?.length || 0}
+> Use \`get_infrastructure_profile\` for full details.` : "Not configured. Set up in Settings → Infrastructure."}
+
 > **Tip:** Features can be configured at workspace or board level. Board settings override workspace defaults.
 `;
                     return {
@@ -782,9 +825,10 @@ ${config.branchingDefaults ? `- Trunk-Based: ${config.branchingDefaults.trunkBas
                     const hasContextData = Array.isArray(currentIteration) &&
                         Array.isArray(backlog) &&
                         (currentIteration.length > 0 || backlog.length > 0);
+                    const iterConfig = await fetchFeatureConfig();
                     const context = hasContextData
-                        ? contextData
-                        : buildIterationContext(await fetchStories(), contextData?.user ?? null);
+                        ? { ...contextData, ...(iterConfig.infrastructureProfile ? { infrastructureProfile: iterConfig.infrastructureProfile } : {}) }
+                        : buildIterationContext(await fetchStories(), contextData?.user ?? null, iterConfig.infrastructureProfile);
                     return {
                         content: [
                             { type: "text", text: JSON.stringify(context, null, 2) },
@@ -814,6 +858,7 @@ ${config.branchingDefaults ? `- Trunk-Based: ${config.branchingDefaults.trunkBas
                 case "record_learning": {
                     const args = request.params.arguments;
                     const { storyId, outcome_summary } = args;
+                    assertUUID(storyId, "storyId");
                     const response = await client.patch(`/stories/${storyId}`, {
                         outcome_summary,
                     });
@@ -829,6 +874,7 @@ ${config.branchingDefaults ? `- Trunk-Based: ${config.branchingDefaults.trunkBas
                     if (!storyId) {
                         throw new Error("storyId is required");
                     }
+                    assertUUID(storyId, "storyId");
                     // Guardrail: Block AI from setting accepted/rejected states (human-in-the-loop)
                     const blockedStates = ["accepted", "rejected"];
                     if (state && blockedStates.includes(state.toLowerCase())) {
@@ -882,6 +928,7 @@ ${config.branchingDefaults ? `- Trunk-Based: ${config.branchingDefaults.trunkBas
                     if (!epicId) {
                         throw new Error("epicId is required");
                     }
+                    assertUUID(epicId, "epicId");
                     const payload = Object.fromEntries(Object.entries(rest).filter(([, value]) => value !== undefined));
                     const response = await client.patch(`/epics/${epicId}`, payload);
                     return {
@@ -893,6 +940,7 @@ ${config.branchingDefaults ? `- Trunk-Based: ${config.branchingDefaults.trunkBas
                 case "prepare_implementation": {
                     const args = request.params.arguments;
                     const { storyId } = args;
+                    assertUUID(storyId, "storyId");
                     const storyResponse = await client.get(`/stories/${storyId}`);
                     const story = storyResponse.data;
                     // Validation & Guardrails
@@ -937,6 +985,7 @@ Here’s the smallest change that will validate it:
                     if (!storyId) {
                         throw new Error("storyId is required");
                     }
+                    assertUUID(storyId, "storyId");
                     const response = await client.post(`/stories/${storyId}/start`, {
                         branchPrefix: branchPrefix || "feat",
                         trunkBased,
@@ -1019,6 +1068,7 @@ ${result.workflow_reminder}
                     if (!storyId) {
                         throw new Error("storyId is required");
                     }
+                    assertUUID(storyId, "storyId");
                     if (!testPlan) {
                         throw new Error("testPlan is required");
                     }
@@ -1040,6 +1090,30 @@ ${(result.test_file_paths || []).map((p) => `- \`${p}\``).join("\n") || "No test
 ${result.message}
 
 > 🛑 **BLOCKED:** Implementation cannot proceed until a human approves this test plan.
+`;
+                    return {
+                        content: [{ type: "text", text: formattedResult.trim() }],
+                    };
+                }
+                case "approve_tests": {
+                    const args = request.params.arguments;
+                    const { storyId } = args;
+                    if (!storyId) {
+                        throw new Error("storyId is required");
+                    }
+                    assertUUID(storyId, "storyId");
+                    const response = await client.post(`/stories/${storyId}/approve-tests`);
+                    const result = response.data;
+                    const formattedResult = `
+# Test Plan Approved
+
+**Story ID:** ${storyId}
+**Workflow Stage:** ${result.workflow_stage}
+
+## Status
+${result.message}
+
+> ✅ **Approved:** Agent may now proceed with implementation.
 `;
                     return {
                         content: [{ type: "text", text: formattedResult.trim() }],
@@ -1051,6 +1125,7 @@ ${result.message}
                     if (!storyId) {
                         throw new Error("storyId is required");
                     }
+                    assertUUID(storyId, "storyId");
                     const response = await client.post(`/stories/${storyId}/submit-review`, {
                         commitHash,
                         testResults,
@@ -1105,11 +1180,23 @@ ${result.next_step}
                 }
                 case "estimate_cost": {
                     const args = request.params.arguments;
-                    const { storyId, provider, constraints } = args;
+                    const { storyId, constraints } = args;
+                    let { provider } = args;
                     if (!storyId)
                         throw new Error("storyId is required");
+                    assertUUID(storyId, "storyId");
+                    // Default provider from infrastructure profile if not explicitly provided
+                    if (!provider) {
+                        const costConfig = await fetchFeatureConfig();
+                        const profileProvider = costConfig.infrastructureProfile?.provider;
+                        if (profileProvider) {
+                            // Map gov variants to commercial for cost estimation API
+                            const govMap = { "aws-gov": "aws", "azure-gov": "azure" };
+                            provider = govMap[profileProvider] || profileProvider;
+                        }
+                    }
                     if (!provider)
-                        throw new Error("provider is required");
+                        throw new Error("provider is required (set it explicitly or configure an infrastructure profile)");
                     if (!["gcp", "aws", "azure", "self-hosted"].includes(provider)) {
                         throw new Error("provider must be one of: gcp, aws, azure, self-hosted");
                     }
@@ -1119,12 +1206,14 @@ ${result.next_step}
                     if (!storyData.description) {
                         throw new Error("Story needs a description before cost estimation");
                     }
-                    // Call the AI estimate-cost endpoint
+                    // Call the AI estimate-cost endpoint (with epicId/boardId for infra context)
                     const estimateRes = await client.post("/ai/estimate-cost", {
                         storyTitle: storyData.title,
                         description: storyData.description,
                         provider,
                         ...(constraints ? { constraints } : {}),
+                        ...(storyData.epicId ? { epicId: storyData.epicId } : {}),
+                        ...(storyData.boardId ? { boardId: storyData.boardId } : {}),
                     });
                     const estimate = estimateRes.data;
                     // Save the estimate to the story
@@ -1165,6 +1254,7 @@ ${(estimate.assumptions || []).map((a) => `- ${a}`).join("\n")}
                     const { epicId } = args;
                     if (!epicId)
                         throw new Error("epicId is required");
+                    assertUUID(epicId, "epicId");
                     // Fetch all stories and filter by epicId
                     const allStories = await fetchStories();
                     const epicStories = allStories.filter((s) => s.epicId === epicId);
@@ -1223,6 +1313,41 @@ ${unestimated.length === 0
                         content: [{ type: "text", text: formattedResult.trim() }],
                     };
                 }
+                case "get_infrastructure_profile": {
+                    const profileConfig = await fetchFeatureConfig();
+                    const profile = profileConfig.infrastructureProfile;
+                    if (!profile || !profile.provider) {
+                        return {
+                            content: [{
+                                    type: "text",
+                                    text: "# Infrastructure Profile\n\nNo infrastructure profile configured. Ask the workspace admin to set one up in **Settings → Infrastructure**.\n\nThis profile tells the AI about your cloud environment — provider, regions, compliance frameworks, and existing services — so cost estimates and recommendations are accurate.",
+                                }],
+                        };
+                    }
+                    const sections = ["# Workspace Infrastructure Profile\n"];
+                    sections.push(`**Cloud Provider:** ${profile.provider.toUpperCase()}`);
+                    if (profile.regions?.length) {
+                        sections.push(`**Regions:** ${profile.regions.join(", ")}`);
+                    }
+                    if (profile.complianceFrameworks?.length) {
+                        sections.push(`\n## Compliance Frameworks\n${profile.complianceFrameworks.map((f) => `- ${f.toUpperCase()}`).join("\n")}`);
+                    }
+                    if (profile.existingServices?.length) {
+                        sections.push(`\n## Existing Services\n| Service | Category | Specs |\n|---------|----------|-------|\n${profile.existingServices.map((s) => `| ${s.name} | ${s.category} | ${s.specs || "-"} |`).join("\n")}`);
+                    }
+                    if (profile.networkTopology) {
+                        sections.push(`\n## Network Topology\n${profile.networkTopology}`);
+                    }
+                    if (profile.additionalConstraints) {
+                        sections.push(`\n## Additional Constraints\n${profile.additionalConstraints}`);
+                    }
+                    if (profile.updatedAt) {
+                        sections.push(`\n---\n*Last updated: ${new Date(profile.updatedAt).toLocaleDateString()}*`);
+                    }
+                    return {
+                        content: [{ type: "text", text: sections.join("\n") }],
+                    };
+                }
                 default:
                     throw new Error("Unknown tool");
             }
@@ -1232,11 +1357,20 @@ ${unestimated.length === 0
             if (error.response) {
                 console.error("Response data:", error.response.data);
             }
+            let errorText = `Error: ${error.message}`;
+            if (error.response?.status) {
+                errorText += ` (HTTP ${error.response.status})`;
+            }
+            if (error.response?.data) {
+                const data = error.response.data;
+                const detail = typeof data === "string" ? data : JSON.stringify(data);
+                errorText += `\nDetails: ${detail}`;
+            }
             return {
                 content: [
                     {
                         type: "text",
-                        text: `Error: ${error.message}`,
+                        text: errorText,
                     },
                 ],
                 isError: true,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@elixium.ai/mcp-server",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "type": "module",
   "description": "MCP Server for Elixium.ai",
   "publishConfig": {