@skyramp/mcp 0.0.33 → 0.0.34

package/build/services/ModularizationService.js

@@ -29,7 +29,8 @@ export class ModularizationService {
             content: [
                 {
                     type: "text",
-                    text: prompt,
+                    text: prompt +
+                        `\n\n***CRITICAL: After modularization is complete, proceed to skyramp_reuse_code tool to fix any errors.`,
                 },
             ],
         };

package/build/services/TestGenerationService.js

@@ -1,5 +1,6 @@
 import { SkyrampClient } from "@skyramp/skyramp";
 import { analyzeOpenAPIWithGivenEndpoint } from "../utils/analyze-openapi.js";
+import * as fs from "fs";
 import { getPathParameterValidationError, OUTPUT_DIR_FIELD_NAME, PATH_PARAMS_FIELD_NAME, QUERY_PARAMS_FIELD_NAME, FORM_PARAMS_FIELD_NAME, validateParams, validatePath, validateRequestData, TELEMETRY_entrypoint_FIELD_NAME, } from "../utils/utils.js";
 import { getLanguageSteps } from "../utils/language-helper.js";
 import { logger } from "../utils/logger.js";
@@ -26,6 +27,11 @@ export class TestGenerationService {
             if (apiAnalysisResult) {
                 return apiAnalysisResult;
             }
+            if (params.scenarioFile) {
+                //read file and convert and pass to generateOptions as `rawTrace`
+                const scenarioFile = await fs.promises.readFile(params.scenarioFile, "utf8");
+                generateOptions.rawTrace = scenarioFile;
+            }
             const result = await this.executeGeneration(generateOptions);
             const testType = this.getTestType();
             const languageSteps = getLanguageSteps({

package/build/tools/generateIntegrationRestTool.js

@@ -9,9 +9,10 @@ const integrationTestSchema = z
     exclude: baseTraceSchema.shape.exclude.optional(),
     endpointURL: baseTestSchema.endpointURL.default(""),
     ...codeRefactoringSchema.shape,
-    rawTrace: z
+    scenarioFile: z
         .string()
-        .describe("Raw trace data  read from the scenario file to be used for test generation, this is not generated by skramp_start_trace_collection tool"),
+        .describe("Path to the scenario file to be used for test generation. This file is generated by the skyramp_scenario_test_generation tool.")
+        .optional(),
 })
     .omit({ method: true }).shape;
 export class IntegrationTestService extends TestGenerationService {
@@ -23,7 +24,7 @@ export class IntegrationTestService extends TestGenerationService {
             ...super.buildBaseGenerationOptions(params),
             responseData: params.responseData,
             playwrightInput: params.playwrightInput,
-            rawTrace: params.rawTrace,
+            scenarioFile: params.scenarioFile,
         };
     }
     async handleApiAnalysis(params, generateOptions) {

package/package.json

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