@skyramp/mcp 0.0.19 → 0.0.21

package/build/prompts/modularization-prompts.js

@@ -2,70 +2,7 @@ export function getModularizationPrompt(filePath, language) {
     return `
 **CRITICAL: Refactor code with modularization principles. Respond with complete, modularized code only WITHOUT ERRORS AND WRITTEN BACK TO THE FILE ${filePath}**
 
-**STEP 1: SEARCH & ANALYZE EXISTING CODE BASED ON TEST LANGUAGE AND NOT ALL LANGUAGE FILES - MANDATORY**
-You MUST search for existing test files by language and NOT ALL LANGUAGE FILES:
-${language === "python" ? `Python: .py files with "skyramp"` : ""}
-
-${language === "typescript" || language === "javascript"
-        ? `TypeScript/JavaScript: .ts/.js files with "@skyramp/skyramp"`
-        : ""}
-
-${language === "java" ? `Java: .java files with "SkyrampClient"` : ""}
-
-You MUST look for reusable patterns: authentication, HTTP helpers, data setup, validation, configuration, cleanup functions EXCLUDE current test file from the search.
-
-**STEP 2: IMPORT & REUSE (HIGHEST PRIORITY) - MANDATORY**
-You MUST import existing functions instead of duplicating. If you find reusable functions, you MUST import them:
-${language === "python"
-        ? `\`\`\`python
-from path.to.existing import helper_function
-from utils.auth_helpers import authenticate
-\`\`\``
-        : ""}
-
-${language === "typescript" || language === "javascript"
-        ? `\`\`\`typescript/javascript
-import { helperFunction } from './path/to/existing';
-import { authenticate } from './utils/authHelpers';
-\`\`\``
-        : ""}
-
-${language === "java"
-        ? `\`\`\`java
-import path.to.existing.helper_function;
-import utils.auth_helpers.authenticate;
-\`\`\``
-        : ""}
-
-if ${language === "python"
-        ? `IMPORTANT:
-FOR PYTHON:
-- If functions are in different folders, use relative path manipulation with os.path. NEVER hardcode absolute paths:
-  \`\`\`python
-  import sys
-  import os
-  # Add parent directory to path to access sibling folders
-  sys.path.append(os.path.dirname(os.path.dirname(__file__)))
-  from e2e.e2e_test import helper_function
-  
-  # OR if importing from same directory level
-  sys.path.append(os.path.dirname(__file__))
-  from e2e_test import helper_function
-  \`\`\`
-- Use aliases to avoid naming conflicts and prevent test execution:
-  \`\`\`python
-  # If importing a test file, use alias to prevent auto-execution
-  import e2e_test as e2e_helpers
-  # Then call: e2e_helpers.setup_auth_headers()
-  \`\`\`
-- CRITICAL: When importing test files, they may auto-execute. Use \`if __name__ == "__main__":\` guards in source files.
-  `
-        : ""}
-
-**STEP 3: REPLACE DUPLICATE CODE - MANDATORY**
-You MUST replace duplicate code with imported function calls. If you imported functions, you MUST use them.
-
-**STEP 4: CREATE MINIMAL NEW HELPERS (ONLY IF NEEDED)**
+**CRITICAL: DO NOT CHANGE THE FUNCTIONALITY OF THE TESTS. ONLY MODULARIZE THE CODE.**
 Only if NO existing function can be reused:
 1. Group related lines logically (e.g., login steps together) to improve readability and parameterize the test function(s) with meaningful parameters to make it more flexible.
 2. If functions are already modularized (like breakpoint_section_0, breakpoint_section_1), then rename them with meaningful names WITHOUT ANY OTHER CHANGES.

package/build/prompts/startTraceCollectionPrompts.js

@@ -25,7 +25,7 @@ export function registerStartTraceCollectionPrompt(mcpServer) {
 * To start a playwright trace collection session using agent, run the following command:
   Start playwright trace collection with default settings.
 
-**CRITICAL: FOR UI TESTS ALWAYS SHOW THE ABOVE PROMPT FOR PLAYWRIGHT TRACE COLLECTION. NEVER SHOW THE CLI COMMANDS.**
+**CRITICAL: NEVER SHOW THE CLI COMMANDS.**
 `,
                 },
             },

package/build/tools/executeSkyrampTestTool.js

@@ -5,6 +5,8 @@ import { Writable } from "stream";
 import { logger } from "../utils/logger.js";
 import { stripVTControlCharacters } from "util";
 import fs from "fs";
+const EXECUTOR_DOCKER_IMAGE = "skyramp/executor:v1.2.16";
+const DOCKER_PLATFORM = "linux/amd64";
 export function registerExecuteSkyrampTestTool(server) {
     server.registerTool("skyramp_execute_test", {
         description: `Execute a Skyramp-generated test in isolated containerized environments for reliable, deterministic testing.
@@ -54,7 +56,6 @@ For detailed documentation visit: https://www.skyramp.dev/docs/quickstart`,
         },
     }, async (params) => {
         var docker = new Docker();
-        var image = "public.ecr.aws/skyramp/rampup/runner:mcp-v0.0.9";
         const containerMountPath = "/home/user";
         // Use path.basename for safer filename extraction
         const filename = path.basename(params.testFile);
@@ -116,14 +117,17 @@ For detailed documentation visit: https://www.skyramp.dev/docs/quickstart`,
             var statusCode = 0;
             // check if image exists
             const images = await docker.listImages();
-            const imageExists = images.some((img) => (img.RepoTags && img.RepoTags.includes(image)) ||
-                (img.RepoDigests && img.RepoDigests.includes(image)));
+            const imageExists = images.some((img) => (img.RepoTags && img.RepoTags.includes(EXECUTOR_DOCKER_IMAGE)) ||
+                (img.RepoDigests &&
+                    img.RepoDigests.includes(EXECUTOR_DOCKER_IMAGE)));
             if (!imageExists) {
                 // Pull the image if it does not exist
                 await new Promise((resolve, reject) => {
-                    docker.pull(image, (err, stream) => {
+                    docker.pull(EXECUTOR_DOCKER_IMAGE, { platform: DOCKER_PLATFORM }, (err, stream) => {
                         if (err)
                             return reject(err);
+                        if (!stream)
+                            return reject(new Error("No stream received from docker pull"));
                         docker.modem.followProgress(stream, (err, res) => {
                             if (err)
                                 return reject(err);
@@ -133,7 +137,10 @@ For detailed documentation visit: https://www.skyramp.dev/docs/quickstart`,
                 });
             }
             await docker
-                .run(image, command, stream, { Env: env, HostConfig: hostConfig })
+                .run(EXECUTOR_DOCKER_IMAGE, command, stream, {
+                Env: env,
+                HostConfig: hostConfig,
+            })
                 .then(function (data) {
                 var result = data[0];
                 var container = data[1];

package/build/tools/stopTraceCollectionTool.js

@@ -4,7 +4,6 @@ import { z } from "zod";
 import { TELEMETRY_entrypoint_FIELD_NAME, } from "../utils/utils.js";
 import { logger } from "../utils/logger.js";
 import { baseSchema } from "../types/TestTypes.js";
-import path from "path";
 import { existsSync, mkdirSync } from "fs";
 export function registerTraceStopTool(server) {
     server.registerTool("skyramp_stop_trace_collection", {
@@ -59,11 +58,6 @@ For detailed documentation visit: https://www.skyramp.dev/docs/load-test/advance
             if (!existsSync(params.outputDir)) {
                 mkdirSync(params.outputDir, { recursive: true });
             }
-            // Temporary fix until library is updated
-            if (params.outputDir) {
-                params.traceOutputFile = path.join(params.outputDir, params.traceOutputFile);
-                params.playwrightOutput = path.join(params.outputDir, params.playwrightOutput);
-            }
             let errList = {
                 content: [],
                 isError: true,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skyramp/mcp",
-  "version": "0.0.19",
+  "version": "0.0.21",
   "main": "build/index.js",
   "type": "module",
   "bin": {
@@ -42,7 +42,7 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.11.4",
-    "@skyramp/skyramp": "^1.2.15",
+    "@skyramp/skyramp": "^1.2.16",
     "dockerode": "^4.0.6",
     "zod": "^3.25.3"
   },