@skyramp/mcp 0.0.14 → 0.0.15

package/build/services/TestGenerationService.js

@@ -33,6 +33,7 @@ export class TestGenerationService {
             const languageSteps = getLanguageSteps({
                 language: params.language || "",
                 testType: params.testType,
+                framework: params.framework,
             });
             return {
                 content: [
@@ -43,10 +44,9 @@ export class TestGenerationService {
                             : "Generated the test and DO NOT MODULARIZE THE CODE"}
             ${result} ${shouldModularize ? getModularizationPrompt() : ""}
         ${languageSteps}
-
-        - Notify the user that they can also use **Execute with Skyramp** tool to run the test. If this is your first test you are trying to execute with Skyramp, it may take a little bit longer than usual.
-        - Before I can execute the test for you, I need to confirm the following information:
-          Authentication: Please provide me with your authentication token (by default we assume a Bearer Token). If you don't require any token for your service please just say: 'No token required'`,
+- Notify the user that they can also use **Execute with Skyramp** tool to run the test. If this is your first test you are trying to execute with Skyramp, it may take a little bit longer than usual.
+- Before I can execute the test for you, I need to confirm the following information:
+  Authentication: Please provide me with your authentication token (by default we assume a Bearer Token). If you don't require any token for your service please just say: 'No token required'`,
                     },
                 ],
                 isError: false,
@@ -131,14 +131,14 @@ export class TestGenerationService {
                     throw new Error(`Test generation failed: ${result}`);
                 }
             }
-            return `**Generated Test Details:**
+            return `
+**Generated Test Details:**
 - Test Type: ${this.getTestType()}
 - Output Directory: ${generateOptions.outputDir}
 - Language: ${generateOptions.language}
 - Framework: ${generateOptions.framework}
 
-${result}
-`;
+${result}`;
         }
         catch (error) {
             const errorMessage = error instanceof Error ? error.message : String(error);

package/build/tools/executeSkyrampTestTool.js

@@ -56,7 +56,7 @@ 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.4";
+        var image = "public.ecr.aws/skyramp/rampup/runner:mcp-v0.0.5";
         // Use path.basename for safer filename extraction
         const filename = path.basename(params.testFile);
         if (!filename) {

package/build/utils/language-helper.js

@@ -16,21 +16,25 @@ python3 -m pytest --browser chromium <test-file>.py
         `;
             }
             // framework pytest
-            pythonSteps += `
+            if (params.framework === "pytest") {
+                pythonSteps += `
 # Install pytest dependency
 pip3 install pytest
 
 # Execute test file
 python3 -m pytest <test-file>.py
       `;
+            }
             // robot framework
-            pythonSteps += `
+            if (params.framework === "robot") {
+                pythonSteps += `
 # Install robotframework dependency
 pip3 install robotframework
 
 # Execution of test file
 python3 -m robot <robot-file>.robot
       `;
+            }
             return pythonSteps;
         case "javascript":
             return `

package/build/utils/modularization-prompts.js

@@ -1,10 +1,5 @@
 export function getModularizationPrompt() {
     return `
-**IMPORTANT: Now please modularize the code by**
-Please modularize and parameterize the test function(s) in the following code, modularize each functionality done by user into clear separate logical groups to make it more readable, and parameterize the modularized function(s) with necessary meaningful parameters to make it more flexible, making sure to keep the order of code lines, existing variable names, existing function names, existing function structures as same, without altering or removing any existing code lines. Keep the skyramp command comments in the beginning of the code. Respond with the complete, modularized code only. DO NOT ADD DEPENDENCY INSTRUCTIONS IN THE CODE.
-
-Please refactor the code with these modularization principles. Respond with the complete, modularized code only.
-
-
-  `;
+**IMPORTANT NEXT STEPS: Please refactor the code with these modularization principles. Respond with the complete, modularized code only in the same file.**
+Think carefully and only take action on the specific task I have given you with the most concise and elegant solution that changes as little code as possible. Make the code modular WITHOUT ANY ERRORS by creating helper functions with parameterized variables for flexibility without creating any new test cases or new classes, and make sure the logic is the same before and after modularization. If functions are already modularized (like breakpoint_section_0, breakpoint_section_1), then rename them with meaningful names and use them. Make sure not to delete any existing lines of code because that could cause errors in test execution. The diff should be minimal. DO NOT ADD DEPENDENCY INSTRUCTIONS IN THE CODE.`;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skyramp/mcp",
-  "version": "0.0.14",
+  "version": "0.0.15",
   "main": "build/index.js",
   "type": "module",
   "bin": {