@autobe/agent 0.21.0 → 0.22.0

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteCodeTemplate.d.ts

@@ -0,0 +1,31 @@
+import { AutoBeOpenApi, AutoBeRealizeAuthorization } from "@autobe/interface";
+import { IAutoBeRealizeScenarioResult } from "../structures/IAutoBeRealizeScenarioResult";
+/**
+ * Generate a TypeScript function template for the Realize Agent to complete
+ *
+ * This utility creates a function skeleton with proper imports, parameters, and
+ * return type that the AI needs to implement. The template includes:
+ *
+ * - Import statements for required dependencies
+ * - Function signature with all parameters
+ * - Return type specification
+ *
+ * @example
+ *   // Returns a template like:
+ *   ```typescript
+ *   import { MyGlobal } from "../MyGlobal";
+ *   // ... other imports
+ *
+ *   async function post__users_create(
+ *     body: IUserCreateRequest
+ *   ): Promise<IUserResponse> {
+ *     ...
+ *   }
+ *   ```;
+ *
+ * @param scenario - The realize scenario containing function metadata
+ * @param operation - OpenAPI operation definition
+ * @param authorization - Authorization context if endpoint is authenticated
+ * @returns Complete TypeScript code template as a formatted string
+ */
+export declare function getRealizeWriteCodeTemplate(scenario: IAutoBeRealizeScenarioResult, operation: AutoBeOpenApi.IOperation, authorization: AutoBeRealizeAuthorization | null): string;

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteCodeTemplate.js

@@ -0,0 +1,86 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getRealizeWriteCodeTemplate = getRealizeWriteCodeTemplate;
+const utils_1 = require("@autobe/utils");
+const getRealizeWriteImportStatements_1 = require("./getRealizeWriteImportStatements");
+/**
+ * Generate a TypeScript function template for the Realize Agent to complete
+ *
+ * This utility creates a function skeleton with proper imports, parameters, and
+ * return type that the AI needs to implement. The template includes:
+ *
+ * - Import statements for required dependencies
+ * - Function signature with all parameters
+ * - Return type specification
+ *
+ * @example
+ *   // Returns a template like:
+ *   ```typescript
+ *   import { MyGlobal } from "../MyGlobal";
+ *   // ... other imports
+ *
+ *   async function post__users_create(
+ *     body: IUserCreateRequest
+ *   ): Promise<IUserResponse> {
+ *     ...
+ *   }
+ *   ```;
+ *
+ * @param scenario - The realize scenario containing function metadata
+ * @param operation - OpenAPI operation definition
+ * @param authorization - Authorization context if endpoint is authenticated
+ * @returns Complete TypeScript code template as a formatted string
+ */
+function getRealizeWriteCodeTemplate(scenario, operation, authorization) {
+    var _a, _b, _c;
+    // Collect all function parameters in order
+    const functionParameters = [];
+    // Add authentication parameter if needed (e.g., user: IUser, admin: IAdmin)
+    if (authorization && authorization.role.name) {
+        // Debug: Log the values to check what's being used
+        const authParameter = `${authorization.role.name}: ${authorization.payload.name}`;
+        functionParameters.push(authParameter);
+    }
+    // Add path parameters (e.g., id, postId, etc.)
+    const pathParameters = operation.parameters.map((param) => {
+        const paramType = param.schema.type;
+        const paramFormat = "format" in param.schema
+            ? ` & tags.Format<'${param.schema.format}'>`
+            : "";
+        return `${param.name}: ${paramType}${paramFormat}`;
+    });
+    functionParameters.push(...pathParameters);
+    // Add request body parameter if present
+    if ((_a = operation.requestBody) === null || _a === void 0 ? void 0 : _a.typeName) {
+        const bodyParameter = `body: ${operation.requestBody.typeName}`;
+        functionParameters.push(bodyParameter);
+    }
+    // Format parameters for props object
+    const hasParameters = functionParameters.length > 0;
+    let formattedSignature;
+    if (hasParameters) {
+        const propsFields = functionParameters
+            .map((param) => `  ${param}`)
+            .join(";\n");
+        formattedSignature = `props: {\n${propsFields};\n}`;
+    }
+    else {
+        formattedSignature = "";
+    }
+    // Determine return type
+    const returnType = (_c = (_b = operation.responseBody) === null || _b === void 0 ? void 0 : _b.typeName) !== null && _c !== void 0 ? _c : "void";
+    // Generate the complete template
+    return utils_1.StringUtil.trim `
+    Complete the code below, disregard the import part and return only the function part.
+
+    \`\`\`typescript
+    ${(0, getRealizeWriteImportStatements_1.getRealizeWriteImportStatements)(operation).join("\n")} 
+
+    // ONLY YOU HAVE TO WRITE THIS, AND USE IMPORTED.
+    export async function ${scenario.functionName}(${formattedSignature}): Promise<${returnType}> {
+      ...
+    }
+    \`\`\`
+  `;
+}
+//# sourceMappingURL=getRealizeWriteCodeTemplate.js.map

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteCodeTemplate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getRealizeWriteCodeTemplate.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/realize/utils/getRealizeWriteCodeTemplate.ts"],"names":[],"mappings":";;AAkCA,kEA8DC;AA/FD,yCAA2C;AAG3C,uFAAoF;AAEpF;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AACH,SAAgB,2BAA2B,CACzC,QAAsC,EACtC,SAAmC,EACnC,aAAgD;;IAEhD,2CAA2C;IAC3C,MAAM,kBAAkB,GAAa,EAAE,CAAC;IAExC,4EAA4E;IAC5E,IAAI,aAAa,IAAI,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAC7C,mDAAmD;QACnD,MAAM,aAAa,GAAG,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QAClF,kBAAkB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACzC,CAAC;IAED,+CAA+C;IAC/C,MAAM,cAAc,GAAG,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QACxD,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC;QACpC,MAAM,WAAW,GACf,QAAQ,IAAI,KAAK,CAAC,MAAM;YACtB,CAAC,CAAC,mBAAmB,KAAK,CAAC,MAAM,CAAC,MAAM,IAAI;YAC5C,CAAC,CAAC,EAAE,CAAC;QACT,OAAO,GAAG,KAAK,CAAC,IAAI,KAAK,SAAS,GAAG,WAAW,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IACH,kBAAkB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;IAE3C,wCAAwC;IACxC,IAAI,MAAA,SAAS,CAAC,WAAW,0CAAE,QAAQ,EAAE,CAAC;QACpC,MAAM,aAAa,GAAG,SAAS,SAAS,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QAChE,kBAAkB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACzC,CAAC;IAED,qCAAqC;IACrC,MAAM,aAAa,GAAG,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC;IAEpD,IAAI,kBAA0B,CAAC;IAC/B,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,WAAW,GAAG,kBAAkB;aACnC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,EAAE,CAAC;aAC5B,IAAI,CAAC,KAAK,CAAC,CAAC;QAEf,kBAAkB,GAAG,aAAa,WAAW,MAAM,CAAC;IACtD,CAAC;SAAM,CAAC;QACN,kBAAkB,GAAG,EAAE,CAAC;IAC1B,CAAC;IAED,wBAAwB;IACxB,MAAM,UAAU,GAAG,MAAA,MAAA,SAAS,CAAC,YAAY,0CAAE,QAAQ,mCAAI,MAAM,CAAC;IAE9D,iCAAiC;IACjC,OAAO,kBAAU,CAAC,IAAI,CAAA;;;;MAIlB,IAAA,iEAA+B,EAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;4BAG/B,QAAQ,CAAC,YAAY,IAAI,kBAAkB,cAAc,UAAU;;;;GAI5F,CAAC;AACJ,CAAC"}

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteDto.d.ts

@@ -0,0 +1,4 @@
+import { AutoBeOpenApi } from "@autobe/interface";
+import { ILlmSchema } from "@samchon/openapi";
+import { AutoBeContext } from "../../../context/AutoBeContext";
+export declare function getRealizeWriteDto<Model extends ILlmSchema.Model>(ctx: AutoBeContext<Model>, operation: AutoBeOpenApi.IOperation): Promise<Record<string, string>>;

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteDto.js

@@ -0,0 +1,55 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getRealizeWriteDto = getRealizeWriteDto;
+const openapi_1 = require("@samchon/openapi");
+function getRealizeWriteDto(ctx, operation) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const document = filterDocument(operation, ctx.state().interface.document);
+        const compiler = yield ctx.compiler();
+        const entries = Object.entries(yield compiler.interface.write(document));
+        const filter = (prefix, exclude) => {
+            const result = entries.filter(([key]) => key.startsWith(prefix) === true);
+            return Object.fromEntries(exclude
+                ? result.filter(([key]) => key.startsWith(exclude) === false)
+                : result);
+        };
+        const dto = filter("src/api/structures");
+        return dto;
+    });
+}
+function filterDocument(operation, document) {
+    const components = {
+        authorization: document.components.authorization,
+        schemas: {},
+    };
+    const visit = (typeName) => {
+        openapi_1.OpenApiTypeChecker.visit({
+            components: document.components,
+            schema: { $ref: `#/components/schemas/${typeName}` },
+            closure: (s) => {
+                if (openapi_1.OpenApiTypeChecker.isReference(s)) {
+                    const key = s.$ref.split("/").pop();
+                    components.schemas[key] = document.components.schemas[key];
+                }
+            },
+        });
+    };
+    if (operation.requestBody)
+        visit(operation.requestBody.typeName);
+    if (operation.responseBody)
+        visit(operation.responseBody.typeName);
+    return {
+        operations: [operation],
+        components,
+    };
+}
+//# sourceMappingURL=getRealizeWriteDto.js.map

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteDto.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getRealizeWriteDto.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/realize/utils/getRealizeWriteDto.ts"],"names":[],"mappings":";;;;;;;;;;;AAKA,gDA2BC;AA/BD,8CAAkE;AAIlE,SAAsB,kBAAkB,CACtC,GAAyB,EACzB,SAAmC;;QAEnC,MAAM,QAAQ,GAA4B,cAAc,CACtD,SAAS,EACT,GAAG,CAAC,KAAK,EAAE,CAAC,SAAU,CAAC,QAAQ,CAChC,CAAC;QAEF,MAAM,QAAQ,GAAoB,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;QACvD,MAAM,OAAO,GAAuB,MAAM,CAAC,OAAO,CAChD,MAAM,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CACzC,CAAC;QAEF,MAAM,MAAM,GAAG,CAAC,MAAc,EAAE,OAAgB,EAAE,EAAE;YAClD,MAAM,MAAM,GAAuB,OAAO,CAAC,MAAM,CAC/C,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,IAAI,CAC3C,CAAC;YACF,OAAO,MAAM,CAAC,WAAW,CACvB,OAAO;gBACL,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,KAAK,CAAC;gBAC7D,CAAC,CAAC,MAAM,CACX,CAAC;QACJ,CAAC,CAAC;QAEF,MAAM,GAAG,GAA2B,MAAM,CAAC,oBAAoB,CAAC,CAAC;QACjE,OAAO,GAAG,CAAC;IACb,CAAC;CAAA;AAED,SAAS,cAAc,CACrB,SAAmC,EACnC,QAAiC;IAEjC,MAAM,UAAU,GAA8B;QAC5C,aAAa,EAAE,QAAQ,CAAC,UAAU,CAAC,aAAa;QAChD,OAAO,EAAE,EAAE;KACZ,CAAC;IAEF,MAAM,KAAK,GAAG,CAAC,QAAgB,EAAE,EAAE;QACjC,4BAAkB,CAAC,KAAK,CAAC;YACvB,UAAU,EAAE,QAAQ,CAAC,UAAU;YAC/B,MAAM,EAAE,EAAE,IAAI,EAAE,wBAAwB,QAAQ,EAAE,EAAE;YACpD,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;gBACb,IAAI,4BAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;oBACtC,MAAM,GAAG,GAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAG,CAAC;oBAC7C,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;gBAC7D,CAAC;YACH,CAAC;SACF,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,IAAI,SAAS,CAAC,WAAW;QAAE,KAAK,CAAC,SAAS,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACjE,IAAI,SAAS,CAAC,YAAY;QAAE,KAAK,CAAC,SAAS,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEnE,OAAO;QACL,UAAU,EAAE,CAAC,SAAS,CAAC;QACvB,UAAU;KACX,CAAC;AACJ,CAAC"}

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteImportStatements.d.ts

@@ -0,0 +1,2 @@
+import { AutoBeOpenApi } from "@autobe/interface";
+export declare function getRealizeWriteImportStatements(operation: AutoBeOpenApi.IOperation): string[];

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteImportStatements.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getRealizeWriteImportStatements = getRealizeWriteImportStatements;
+function getRealizeWriteImportStatements(operation) {
+    const typeReferences = Array.from(new Set([operation.requestBody, operation.responseBody]
+        .filter((el) => el !== null)
+        .map((el) => el.typeName.split(".")[0])));
+    // Build the standard imports
+    const imports = [
+        'import jwt from "jsonwebtoken";',
+        'import { MyGlobal } from "../MyGlobal";',
+        'import typia, { tags } from "typia";',
+        'import { Prisma } from "@prisma/client";',
+        'import { v4 } from "uuid";',
+        'import { toISOStringSafe } from "../util/toISOStringSafe"',
+        ...typeReferences.map((ref) => `import { ${ref} } from "@ORGANIZATION/PROJECT-api/lib/structures/${ref}";`),
+    ];
+    return imports;
+}
+//# sourceMappingURL=getRealizeWriteImportStatements.js.map

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteImportStatements.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getRealizeWriteImportStatements.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/realize/utils/getRealizeWriteImportStatements.ts"],"names":[],"mappings":";;AAEA,0EA0BC;AA1BD,SAAgB,+BAA+B,CAC7C,SAAmC;IAEnC,MAAM,cAAc,GAAa,KAAK,CAAC,IAAI,CACzC,IAAI,GAAG,CACL,CAAC,SAAS,CAAC,WAAW,EAAE,SAAS,CAAC,YAAY,CAAC;SAC5C,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,IAAI,CAAC;SAC3B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAE,CAAC,CAC3C,CACF,CAAC;IAEF,6BAA6B;IAC7B,MAAM,OAAO,GAAG;QACd,iCAAiC;QACjC,yCAAyC;QACzC,sCAAsC;QACtC,0CAA0C;QAC1C,4BAA4B;QAC5B,2DAA2D;QAC3D,GAAG,cAAc,CAAC,GAAG,CACnB,CAAC,GAAG,EAAE,EAAE,CACN,YAAY,GAAG,qDAAqD,GAAG,IAAI,CAC9E;KACF,CAAC;IAEF,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteInputType.d.ts

@@ -0,0 +1,30 @@
+import { AutoBeOpenApi, AutoBeRealizeAuthorization } from "@autobe/interface";
+/**
+ * Generate TypeScript type definition string for provider function props
+ *
+ * This utility creates the exact props interface that the Realize Agent's
+ * generated provider functions must accept. It combines:
+ *
+ * - Authentication context (user/admin/member from authorization)
+ * - URL path parameters (e.g., id, boardId, postId)
+ * - Request body type (if applicable)
+ *
+ * The output is injected into REALIZE_WRITE_ARTIFACT.md template to tell the AI
+ * exactly what parameter type the function should accept.
+ *
+ * @example
+ *   // For DELETE /posts/:id (authenticated)
+ *   // Returns: "props: { user: IUser; id: string & tags.Format<'uuid'>; }"
+ *
+ * @example
+ *   // For GET /public/info (no auth, no params)
+ *   // Returns: "// No props parameter needed - function should have no parameters"
+ *
+ * @param operation - OpenAPI operation definition containing parameters and
+ *   auth requirements
+ * @param authorization - Authorization context with user/role information (if
+ *   authenticated endpoint)
+ * @returns TypeScript interface string like "props: { user: IUser; id: string;
+ *   body: IRequest; }"
+ */
+export declare function getRealizeWriteInputType(operation: AutoBeOpenApi.IOperation, authorization: AutoBeRealizeAuthorization | null): string;

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteInputType.js

@@ -0,0 +1,73 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getRealizeWriteInputType = getRealizeWriteInputType;
+const utils_1 = require("@autobe/utils");
+/**
+ * Generate TypeScript type definition string for provider function props
+ *
+ * This utility creates the exact props interface that the Realize Agent's
+ * generated provider functions must accept. It combines:
+ *
+ * - Authentication context (user/admin/member from authorization)
+ * - URL path parameters (e.g., id, boardId, postId)
+ * - Request body type (if applicable)
+ *
+ * The output is injected into REALIZE_WRITE_ARTIFACT.md template to tell the AI
+ * exactly what parameter type the function should accept.
+ *
+ * @example
+ *   // For DELETE /posts/:id (authenticated)
+ *   // Returns: "props: { user: IUser; id: string & tags.Format<'uuid'>; }"
+ *
+ * @example
+ *   // For GET /public/info (no auth, no params)
+ *   // Returns: "// No props parameter needed - function should have no parameters"
+ *
+ * @param operation - OpenAPI operation definition containing parameters and
+ *   auth requirements
+ * @param authorization - Authorization context with user/role information (if
+ *   authenticated endpoint)
+ * @returns TypeScript interface string like "props: { user: IUser; id: string;
+ *   body: IRequest; }"
+ */
+function getRealizeWriteInputType(operation, authorization) {
+    var _a, _b;
+    const functionParameterFields = [];
+    // Add authentication field (user/admin/member) if endpoint requires auth
+    const hasAuthentication = authorization && operation.authorizationRole;
+    if (hasAuthentication) {
+        const authFieldName = operation.authorizationRole;
+        const authFieldType = authorization.payload.name;
+        functionParameterFields.push(`${authFieldName}: ${authFieldType};`);
+    }
+    // Add URL path parameters (e.g., /posts/:id → id parameter)
+    const pathParameters = operation.parameters;
+    pathParameters.forEach((pathParam) => {
+        const paramName = pathParam.name;
+        const paramType = pathParam.schema.type;
+        const paramFormat = "format" in pathParam.schema
+            ? ` & tags.Format<'${pathParam.schema.format}'>`
+            : "";
+        const parameterField = `${paramName}: ${paramType}${paramFormat};`;
+        functionParameterFields.push(parameterField);
+    });
+    // Add request body if present (POST/PUT/PATCH operations)
+    const hasRequestBody = (_a = operation.requestBody) === null || _a === void 0 ? void 0 : _a.typeName;
+    if (hasRequestBody) {
+        const bodyTypeName = (_b = operation.requestBody) === null || _b === void 0 ? void 0 : _b.typeName;
+        functionParameterFields.push(`body: ${bodyTypeName};`);
+    }
+    // Format as TypeScript props interface or indicate no parameters needed
+    const hasParameters = functionParameterFields.length > 0;
+    if (!hasParameters) {
+        return `// No props parameter needed - function should have no parameters`;
+    }
+    const formattedFields = functionParameterFields
+        .map((field) => `  ${field}`)
+        .join("\n");
+    return utils_1.StringUtil.trim `
+    props: {
+    ${formattedFields}
+    }`;
+}
+//# sourceMappingURL=getRealizeWriteInputType.js.map

package/lib/agent/src/orchestrate/realize/utils/getRealizeWriteInputType.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getRealizeWriteInputType.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/realize/utils/getRealizeWriteInputType.ts"],"names":[],"mappings":";;AA+BA,4DAkDC;AAhFD,yCAA2C;AAE3C;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AACH,SAAgB,wBAAwB,CACtC,SAAmC,EACnC,aAAgD;;IAEhD,MAAM,uBAAuB,GAAa,EAAE,CAAC;IAE7C,yEAAyE;IACzE,MAAM,iBAAiB,GAAG,aAAa,IAAI,SAAS,CAAC,iBAAiB,CAAC;IACvE,IAAI,iBAAiB,EAAE,CAAC;QACtB,MAAM,aAAa,GAAG,SAAS,CAAC,iBAAiB,CAAC;QAClD,MAAM,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC;QACjD,uBAAuB,CAAC,IAAI,CAAC,GAAG,aAAa,KAAK,aAAa,GAAG,CAAC,CAAC;IACtE,CAAC;IAED,4DAA4D;IAC5D,MAAM,cAAc,GAAG,SAAS,CAAC,UAAU,CAAC;IAC5C,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;QACnC,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC;QACjC,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC;QACxC,MAAM,WAAW,GACf,QAAQ,IAAI,SAAS,CAAC,MAAM;YAC1B,CAAC,CAAC,mBAAmB,SAAS,CAAC,MAAM,CAAC,MAAM,IAAI;YAChD,CAAC,CAAC,EAAE,CAAC;QAET,MAAM,cAAc,GAAG,GAAG,SAAS,KAAK,SAAS,GAAG,WAAW,GAAG,CAAC;QACnE,uBAAuB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,0DAA0D;IAC1D,MAAM,cAAc,GAAG,MAAA,SAAS,CAAC,WAAW,0CAAE,QAAQ,CAAC;IACvD,IAAI,cAAc,EAAE,CAAC;QACnB,MAAM,YAAY,GAAG,MAAA,SAAS,CAAC,WAAW,0CAAE,QAAQ,CAAC;QACrD,uBAAuB,CAAC,IAAI,CAAC,SAAS,YAAY,GAAG,CAAC,CAAC;IACzD,CAAC;IAED,wEAAwE;IACxE,MAAM,aAAa,GAAG,uBAAuB,CAAC,MAAM,GAAG,CAAC,CAAC;IAEzD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,mEAAmE,CAAC;IAC7E,CAAC;IAED,MAAM,eAAe,GAAG,uBAAuB;SAC5C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,EAAE,CAAC;SAC5B,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,OAAO,kBAAU,CAAC,IAAI,CAAA;;MAElB,eAAe;MACf,CAAC;AACP,CAAC"}

package/lib/agent/src/orchestrate/realize/utils/replaceImportStatements.d.ts

@@ -1,4 +1,8 @@
+import { AutoBeOpenApi } from "@autobe/interface";
 import { ILlmSchema } from "@samchon/openapi";
 import { AutoBeContext } from "../../../context/AutoBeContext";
-import { IAutoBeTestScenarioArtifacts } from "../../test/structures/IAutoBeTestScenarioArtifacts";
-export declare function replaceImportStatements<Model extends ILlmSchema.Model>(ctx: AutoBeContext<Model>): (artifacts: IAutoBeTestScenarioArtifacts, code: string, decoratorType?: string) => Promise<string>;
+export declare function replaceImportStatements<Model extends ILlmSchema.Model>(ctx: AutoBeContext<Model>, props: {
+    operation: AutoBeOpenApi.IOperation;
+    code: string;
+    decoratorType?: string;
+}): Promise<string>;

package/lib/agent/src/orchestrate/realize/utils/replaceImportStatements.js

@@ -10,74 +10,68 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.replaceImportStatements = replaceImportStatements;
-function replaceImportStatements(ctx) {
-    return function (artifacts, code, decoratorType) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const typeReferences = Array.from(new Set(Object.keys(artifacts.document.components.schemas).map((key) => key.split(".")[0])));
-            const compiler = yield ctx.compiler();
-            code = yield compiler.typescript.beautify(code);
-            // Remove existing import statements using flexible regex patterns
-            code = code
-                .replace(/import\s*{\s*MyGlobal\s*}\s*from\s*["']\.\.\/MyGlobal["']\s*;?\s*/gm, "")
-                .replace(/import\s+typia\s*,\s*{\s*tags\s*}\s*from\s*["']typia["']\s*;?\s*/gm, "")
-                .replace(/import\s*{\s*tags\s*}\s*from\s*["']typia["']\s*;?\s*/gm, "")
-                .replace(/import\s*{\s*tags\s*,\s*typia\s*}\s*from\s*["']typia["']\s*;?\s*/gm, "")
-                .replace(/import\s+typia\s*from\s*["']typia["']\s*;?\s*/gm, "")
-                .replace(/import\s*{\s*Prisma\s*}\s*from\s*["']@prisma\/client["']\s*;?\s*/gm, "")
-                .replace(/import\s*{\s*v4\s*}\s*from\s*["']uuid["']\s*;?\s*/gm, "")
-                .replace(/import\s*{\s*toISOStringSafe\s*}\s*from\s*["']\.\.\/util\/toISOStringSafe["']\s*;?\s*/gm, "")
-                // Remove JWT import if it exists (to prevent duplication)
-                .replace(/import\s+jwt\s+from\s*["']jsonwebtoken["']\s*;?\s*/gm, "")
-                .replace(/import\s*\*\s*as\s+jwt\s+from\s*["']jsonwebtoken["']\s*;?\s*/gm, "");
-            // Remove any existing API structure imports
-            // Pattern 1: ../api/structures path
-            code = code.replace(/import\s*(?:type\s*)?{\s*[^}]+\s*}\s*from\s*["']\.\.\/api\/structures\/[^"']+["']\s*;?\s*/gm, "");
-            // Pattern 2: @ORGANIZATION/PROJECT-api/lib/structures path
-            code = code.replace(/import\s*(?:type\s*)?{\s*[^}]+\s*}\s*from\s*["']@ORGANIZATION\/PROJECT-api\/lib\/structures\/[^"']+["']\s*;?\s*/gm, "");
-            // Remove specific type imports that match our typeReferences
-            for (const ref of typeReferences) {
-                // Remove any import of this specific type from any path
-                const typeImportRegex = new RegExp(`import\\s*(?:type\\s*)?{\\s*${ref}\\s*}\\s*from\\s*["'][^"']+["']\\s*;?\\s*`, "gm");
-                code = code.replace(typeImportRegex, "");
-            }
-            // Remove any existing decoratorType imports if LLM mistakenly added them
-            if (decoratorType) {
-                const decoratorTypeRegex = new RegExp(`import\\s*(?:type\\s*)?{\\s*${decoratorType}\\s*}\\s*from\\s*["']\\.\\.\/decorators\/payload\/${decoratorType}["']\\s*;?\\s*`, "gm");
-                code = code.replace(decoratorTypeRegex, "");
-            }
-            // Build the standard imports
-            const imports = [
-                'import jwt from "jsonwebtoken";',
-                'import { MyGlobal } from "../MyGlobal";',
-                'import typia, { tags } from "typia";',
-                'import { Prisma } from "@prisma/client";',
-                'import { v4 } from "uuid";',
-                'import { toISOStringSafe } from "../util/toISOStringSafe"',
-                ...typeReferences.map((ref) => `import { ${ref} } from "@ORGANIZATION/PROJECT-api/lib/structures/${ref}";`),
-            ];
-            // Only add decoratorType import if it exists
-            if (decoratorType) {
-                imports.push(`import { ${decoratorType} } from "../decorators/payload/${decoratorType}"`);
-            }
-            code = [...imports, "", code].join("\n");
-            // Clean up formatting issues
-            code =
-                code
-                    // Remove lines with only whitespace
-                    .replace(/^\s+$/gm, "")
-                    // Replace 3+ consecutive newlines with exactly 2 newlines
-                    .replace(/\n{3,}/g, "\n\n")
-                    // Ensure proper spacing after import section
-                    .replace(/(import.*?;)(\s*)(\n(?!import|\s*$))/g, "$1\n\n$3")
-                    // Trim and ensure single trailing newline
-                    .trim() + "\n";
-            // fix escaped codes
-            code = code.replace(/\\n/g, "\n").replace(/\\"/g, '"').replace(/\\'/g, "'");
-            // Apply final beautification
-            code = yield compiler.typescript.beautify(code);
-            code = code.replaceAll("typia.tags.assert", "typia.assert");
-            return code;
-        });
-    };
+const getRealizeWriteImportStatements_1 = require("./getRealizeWriteImportStatements");
+function replaceImportStatements(ctx, props) {
+    return __awaiter(this, void 0, void 0, function* () {
+        let { operation, code, decoratorType } = props;
+        const typeReferences = Array.from(new Set([operation.requestBody, operation.responseBody]
+            .filter((el) => el !== null)
+            .map((el) => el.typeName.split(".")[0])));
+        const compiler = yield ctx.compiler();
+        code = yield compiler.typescript.beautify(code);
+        // Remove existing import statements using flexible regex patterns
+        code = code
+            .replace(/import\s*{\s*MyGlobal\s*}\s*from\s*["']\.\.\/MyGlobal["']\s*;?\s*/gm, "")
+            .replace(/import\s+typia\s*,\s*{\s*tags\s*}\s*from\s*["']typia["']\s*;?\s*/gm, "")
+            .replace(/import\s*{\s*tags\s*}\s*from\s*["']typia["']\s*;?\s*/gm, "")
+            .replace(/import\s*{\s*tags\s*,\s*typia\s*}\s*from\s*["']typia["']\s*;?\s*/gm, "")
+            .replace(/import\s+typia\s*from\s*["']typia["']\s*;?\s*/gm, "")
+            .replace(/import\s*{\s*Prisma\s*}\s*from\s*["']@prisma\/client["']\s*;?\s*/gm, "")
+            .replace(/import\s*{\s*v4\s*}\s*from\s*["']uuid["']\s*;?\s*/gm, "")
+            .replace(/import\s*{\s*toISOStringSafe\s*}\s*from\s*["']\.\.\/util\/toISOStringSafe["']\s*;?\s*/gm, "")
+            // Remove JWT import if it exists (to prevent duplication)
+            .replace(/import\s+jwt\s+from\s*["']jsonwebtoken["']\s*;?\s*/gm, "")
+            .replace(/import\s*\*\s*as\s+jwt\s+from\s*["']jsonwebtoken["']\s*;?\s*/gm, "");
+        // Remove any existing API structure imports
+        // Pattern 1: ../api/structures path
+        code = code.replace(/import\s*(?:type\s*)?{\s*[^}]+\s*}\s*from\s*["']\.\.\/api\/structures\/[^"']+["']\s*;?\s*/gm, "");
+        // Pattern 2: @ORGANIZATION/PROJECT-api/lib/structures path
+        code = code.replace(/import\s*(?:type\s*)?{\s*[^}]+\s*}\s*from\s*["']@ORGANIZATION\/PROJECT-api\/lib\/structures\/[^"']+["']\s*;?\s*/gm, "");
+        // Remove specific type imports that match our typeReferences
+        for (const ref of typeReferences) {
+            // Remove any import of this specific type from any path
+            const typeImportRegex = new RegExp(`import\\s*(?:type\\s*)?{\\s*${ref}\\s*}\\s*from\\s*["'][^"']+["']\\s*;?\\s*`, "gm");
+            code = code.replace(typeImportRegex, "");
+        }
+        // Remove any existing decoratorType imports if LLM mistakenly added them
+        if (decoratorType) {
+            const decoratorTypeRegex = new RegExp(`import\\s*(?:type\\s*)?{\\s*${decoratorType}\\s*}\\s*from\\s*["']\\.\\.\/decorators\/payload\/${decoratorType}["']\\s*;?\\s*`, "gm");
+            code = code.replace(decoratorTypeRegex, "");
+        }
+        // Build the standard imports
+        const imports = (0, getRealizeWriteImportStatements_1.getRealizeWriteImportStatements)(operation);
+        // Only add decoratorType import if it exists
+        if (decoratorType) {
+            imports.push(`import { ${decoratorType} } from "../decorators/payload/${decoratorType}"`);
+        }
+        code = [...imports, "", code].join("\n");
+        // Clean up formatting issues
+        code =
+            code
+                // Remove lines with only whitespace
+                .replace(/^\s+$/gm, "")
+                // Replace 3+ consecutive newlines with exactly 2 newlines
+                .replace(/\n{3,}/g, "\n\n")
+                // Ensure proper spacing after import section
+                .replace(/(import.*?;)(\s*)(\n(?!import|\s*$))/g, "$1\n\n$3")
+                // Trim and ensure single trailing newline
+                .trim() + "\n";
+        // fix escaped codes
+        code = code.replace(/\\n/g, "\n").replace(/\\"/g, '"').replace(/\\'/g, "'");
+        // Apply final beautification
+        code = yield compiler.typescript.beautify(code);
+        code = code.replaceAll("typia.tags.assert", "typia.assert");
+        return code;
+    });
 }
 //# sourceMappingURL=replaceImportStatements.js.map

package/lib/agent/src/orchestrate/realize/utils/replaceImportStatements.js.map

@@ -1 +1 @@
-{"version":3,"file":"replaceImportStatements.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/realize/utils/replaceImportStatements.ts"],"names":[],"mappings":";;;;;;;;;;;AAMA,0DA8HC;AA9HD,SAAgB,uBAAuB,CACrC,GAAyB;IAEzB,OAAO,UACL,SAAuC,EACvC,IAAY,EACZ,aAAsB;;YAEtB,MAAM,cAAc,GAAa,KAAK,CAAC,IAAI,CACzC,IAAI,GAAG,CACL,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,GAAG,CACpD,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAE,CAC5B,CACF,CACF,CAAC;YAEF,MAAM,QAAQ,GAAoB,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;YACvD,IAAI,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAChD,kEAAkE;YAClE,IAAI,GAAG,IAAI;iBACR,OAAO,CACN,qEAAqE,EACrE,EAAE,CACH;iBACA,OAAO,CACN,oEAAoE,EACpE,EAAE,CACH;iBACA,OAAO,CAAC,wDAAwD,EAAE,EAAE,CAAC;iBACrE,OAAO,CACN,oEAAoE,EACpE,EAAE,CACH;iBACA,OAAO,CAAC,iDAAiD,EAAE,EAAE,CAAC;iBAC9D,OAAO,CACN,oEAAoE,EACpE,EAAE,CACH;iBACA,OAAO,CAAC,qDAAqD,EAAE,EAAE,CAAC;iBAClE,OAAO,CACN,yFAAyF,EACzF,EAAE,CACH;gBACD,0DAA0D;iBACzD,OAAO,CAAC,sDAAsD,EAAE,EAAE,CAAC;iBACnE,OAAO,CACN,gEAAgE,EAChE,EAAE,CACH,CAAC;YAEJ,4CAA4C;YAC5C,oCAAoC;YACpC,IAAI,GAAG,IAAI,CAAC,OAAO,CACjB,6FAA6F,EAC7F,EAAE,CACH,CAAC;YACF,2DAA2D;YAC3D,IAAI,GAAG,IAAI,CAAC,OAAO,CACjB,mHAAmH,EACnH,EAAE,CACH,CAAC;YAEF,6DAA6D;YAC7D,KAAK,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;gBACjC,wDAAwD;gBACxD,MAAM,eAAe,GAAG,IAAI,MAAM,CAChC,+BAA+B,GAAG,2CAA2C,EAC7E,IAAI,CACL,CAAC;gBACF,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;YAC3C,CAAC;YAED,yEAAyE;YACzE,IAAI,aAAa,EAAE,CAAC;gBAClB,MAAM,kBAAkB,GAAG,IAAI,MAAM,CACnC,+BAA+B,aAAa,qDAAqD,aAAa,gBAAgB,EAC9H,IAAI,CACL,CAAC;gBACF,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;YAC9C,CAAC;YAED,6BAA6B;YAC7B,MAAM,OAAO,GAAG;gBACd,iCAAiC;gBACjC,yCAAyC;gBACzC,sCAAsC;gBACtC,0CAA0C;gBAC1C,4BAA4B;gBAC5B,2DAA2D;gBAC3D,GAAG,cAAc,CAAC,GAAG,CACnB,CAAC,GAAG,EAAE,EAAE,CACN,YAAY,GAAG,qDAAqD,GAAG,IAAI,CAC9E;aACF,CAAC;YAEF,6CAA6C;YAC7C,IAAI,aAAa,EAAE,CAAC;gBAClB,OAAO,CAAC,IAAI,CACV,YAAY,aAAa,kCAAkC,aAAa,GAAG,CAC5E,CAAC;YACJ,CAAC;YAED,IAAI,GAAG,CAAC,GAAG,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEzC,6BAA6B;YAC7B,IAAI;gBACF,IAAI;oBACF,oCAAoC;qBACnC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;oBACvB,0DAA0D;qBACzD,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC;oBAC3B,6CAA6C;qBAC5C,OAAO,CAAC,uCAAuC,EAAE,UAAU,CAAC;oBAC7D,0CAA0C;qBACzC,IAAI,EAAE,GAAG,IAAI,CAAC;YAEnB,oBAAoB;YACpB,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAE5E,6BAA6B;YAC7B,IAAI,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAEhD,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,mBAAmB,EAAE,cAAc,CAAC,CAAC;YAE5D,OAAO,IAAI,CAAC;QACd,CAAC;KAAA,CAAC;AACJ,CAAC"}
+{"version":3,"file":"replaceImportStatements.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/realize/utils/replaceImportStatements.ts"],"names":[],"mappings":";;;;;;;;;;;AAMA,0DAoHC;AAtHD,uFAAoF;AAEpF,SAAsB,uBAAuB,CAC3C,GAAyB,EACzB,KAIC;;QAED,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,KAAK,CAAC;QAE/C,MAAM,cAAc,GAAa,KAAK,CAAC,IAAI,CACzC,IAAI,GAAG,CACL,CAAC,SAAS,CAAC,WAAW,EAAE,SAAS,CAAC,YAAY,CAAC;aAC5C,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,IAAI,CAAC;aAC3B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAE,CAAC,CAC3C,CACF,CAAC;QAEF,MAAM,QAAQ,GAAoB,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;QACvD,IAAI,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChD,kEAAkE;QAClE,IAAI,GAAG,IAAI;aACR,OAAO,CACN,qEAAqE,EACrE,EAAE,CACH;aACA,OAAO,CACN,oEAAoE,EACpE,EAAE,CACH;aACA,OAAO,CAAC,wDAAwD,EAAE,EAAE,CAAC;aACrE,OAAO,CACN,oEAAoE,EACpE,EAAE,CACH;aACA,OAAO,CAAC,iDAAiD,EAAE,EAAE,CAAC;aAC9D,OAAO,CACN,oEAAoE,EACpE,EAAE,CACH;aACA,OAAO,CAAC,qDAAqD,EAAE,EAAE,CAAC;aAClE,OAAO,CACN,yFAAyF,EACzF,EAAE,CACH;YACD,0DAA0D;aACzD,OAAO,CAAC,sDAAsD,EAAE,EAAE,CAAC;aACnE,OAAO,CACN,gEAAgE,EAChE,EAAE,CACH,CAAC;QAEJ,4CAA4C;QAC5C,oCAAoC;QACpC,IAAI,GAAG,IAAI,CAAC,OAAO,CACjB,6FAA6F,EAC7F,EAAE,CACH,CAAC;QACF,2DAA2D;QAC3D,IAAI,GAAG,IAAI,CAAC,OAAO,CACjB,mHAAmH,EACnH,EAAE,CACH,CAAC;QAEF,6DAA6D;QAC7D,KAAK,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;YACjC,wDAAwD;YACxD,MAAM,eAAe,GAAG,IAAI,MAAM,CAChC,+BAA+B,GAAG,2CAA2C,EAC7E,IAAI,CACL,CAAC;YACF,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,yEAAyE;QACzE,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,kBAAkB,GAAG,IAAI,MAAM,CACnC,+BAA+B,aAAa,qDAAqD,aAAa,gBAAgB,EAC9H,IAAI,CACL,CAAC;YACF,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;QAC9C,CAAC;QAED,6BAA6B;QAC7B,MAAM,OAAO,GAAG,IAAA,iEAA+B,EAAC,SAAS,CAAC,CAAC;QAE3D,6CAA6C;QAC7C,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,CAAC,IAAI,CACV,YAAY,aAAa,kCAAkC,aAAa,GAAG,CAC5E,CAAC;QACJ,CAAC;QAED,IAAI,GAAG,CAAC,GAAG,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEzC,6BAA6B;QAC7B,IAAI;YACF,IAAI;gBACF,oCAAoC;iBACnC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;gBACvB,0DAA0D;iBACzD,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC;gBAC3B,6CAA6C;iBAC5C,OAAO,CAAC,uCAAuC,EAAE,UAAU,CAAC;gBAC7D,0CAA0C;iBACzC,IAAI,EAAE,GAAG,IAAI,CAAC;QAEnB,oBAAoB;QACpB,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAE5E,6BAA6B;QAC7B,IAAI,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAEhD,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,mBAAmB,EAAE,cAAc,CAAC,CAAC;QAE5D,OAAO,IAAI,CAAC;IACd,CAAC;CAAA"}

package/lib/agent/src/orchestrate/test/compile/completeTestCode.js

@@ -2,23 +2,24 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.completeTestCode = completeTestCode;
 const utils_1 = require("@autobe/utils");
+const getTestImportStatements_1 = require("./getTestImportStatements");
 function completeTestCode(artifacts, code) {
-    const typeReferences = Array.from(new Set(Object.keys(artifacts.document.components.schemas).map((key) => key.split(".")[0])));
-    code = code.replace(/^[ \t]*import\b[\s\S]*?;[ \t]*$/gm, "").trim();
-    code = code.replace(/^[ \t]*import\b[\s\S]*?;[ \t]*$/gm, "").trim();
-    code = code.replaceAll('string & Format<"uuid">', 'string & tags.Format<"uuid">');
-    code = utils_1.StringUtil.trim `
-    import { ArrayUtil, RandomGenerator, TestValidator } from "@nestia/e2e";
-    import { IConnection } from "@nestia/fetcher";
-    import typia, { tags } from "typia";
-    
-    import api from "@ORGANIZATION/PROJECT-api";
-    ${typeReferences
-        .map((ref) => `import type { ${ref} } from "@ORGANIZATION/PROJECT-api/lib/structures/${ref}";`)
-        .join("\n")}
+    code = code
+        .split("\r\n")
+        .join("\n")
+        .split("\n")
+        .filter((str) => str.trim().startsWith("import") === false)
+        .join("\n");
+    // code = code.replace(/^[ \t]*import\b[\s\S]*?;[ \t]*$/gm, "").trim();
+    // code = code.replace(/^[ \t]*import\b[\s\S]*?;[ \t]*$/gm, "").trim();
+    // code = code.replaceAll(
+    //   'string & Format<"uuid">',
+    //   'string & tags.Format<"uuid">',
+    // );
+    return utils_1.StringUtil.trim `
+    ${(0, getTestImportStatements_1.getTestImportStatements)(artifacts.document)}
     
     ${code}
   `;
-    return code;
 }
 //# sourceMappingURL=completeTestCode.js.map

package/lib/agent/src/orchestrate/test/compile/completeTestCode.js.map

@@ -1 +1 @@
-{"version":3,"file":"completeTestCode.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/test/compile/completeTestCode.ts"],"names":[],"mappings":";;AAIA,4CAkCC;AAtCD,yCAA2C;AAI3C,SAAgB,gBAAgB,CAC9B,SAAuC,EACvC,IAAY;IAEZ,MAAM,cAAc,GAAa,KAAK,CAAC,IAAI,CACzC,IAAI,GAAG,CACL,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,GAAG,CACpD,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAE,CAC5B,CACF,CACF,CAAC;IAEF,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,mCAAmC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACpE,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,mCAAmC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACpE,IAAI,GAAG,IAAI,CAAC,UAAU,CACpB,yBAAyB,EACzB,8BAA8B,CAC/B,CAAC;IACF,IAAI,GAAG,kBAAU,CAAC,IAAI,CAAA;;;;;;MAMlB,cAAc;SACb,GAAG,CACF,CAAC,GAAG,EAAE,EAAE,CACN,iBAAiB,GAAG,qDAAqD,GAAG,IAAI,CACnF;SACA,IAAI,CAAC,IAAI,CAAC;;MAEX,IAAI;GACP,CAAC;IACF,OAAO,IAAI,CAAC;AACd,CAAC"}
+{"version":3,"file":"completeTestCode.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/test/compile/completeTestCode.ts"],"names":[],"mappings":";;AAKA,4CAsBC;AA3BD,yCAA2C;AAG3C,uEAAoE;AAEpE,SAAgB,gBAAgB,CAC9B,SAAuC,EACvC,IAAY;IAEZ,IAAI,GAAG,IAAI;SACR,KAAK,CAAC,MAAM,CAAC;SACb,IAAI,CAAC,IAAI,CAAC;SACV,KAAK,CAAC,IAAI,CAAC;SACX,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,KAAK,CAAC;SAC1D,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,uEAAuE;IACvE,uEAAuE;IACvE,0BAA0B;IAC1B,+BAA+B;IAC/B,oCAAoC;IACpC,KAAK;IACL,OAAO,kBAAU,CAAC,IAAI,CAAA;MAClB,IAAA,iDAAuB,EAAC,SAAS,CAAC,QAAQ,CAAC;;MAE3C,IAAI;GACP,CAAC;AACJ,CAAC"}

package/lib/agent/src/orchestrate/test/compile/getTestExternalDeclarations.js

@@ -25,6 +25,6 @@ const singleton = new tstl_1.Singleton((compiler) => __awaiter(void 0, void 0, v
             throw new Error(`File not found: ${location}`);
         return { [location]: content };
     };
-    return Object.assign(Object.assign(Object.assign(Object.assign({}, external("node_modules/@nestia/e2e/lib/ArrayUtil.d.ts")), external("node_modules/@nestia/e2e/lib/RandomGenerator.d.ts")), external("node_modules/@nestia/e2e/lib/TestValidator.d.ts")), external("node_modules/@nestia/fetcher/lib/IConnection.d.ts"));
+    return Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({}, external("node_modules/@nestia/e2e/lib/ArrayUtil.d.ts")), external("node_modules/@nestia/e2e/lib/RandomGenerator.d.ts")), external("node_modules/@nestia/e2e/lib/TestValidator.d.ts")), external("node_modules/@nestia/fetcher/lib/IConnection.d.ts")), external("node_modules/typia/lib/module.d.ts"));
 }));
 //# sourceMappingURL=getTestExternalDeclarations.js.map

package/lib/agent/src/orchestrate/test/compile/getTestExternalDeclarations.js.map

@@ -1 +1 @@
-{"version":3,"file":"getTestExternalDeclarations.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/test/compile/getTestExternalDeclarations.ts"],"names":[],"mappings":";;;;;;;;;;;AAMA,kEAKC;AATD,+BAAiC;AAIjC,SAAsB,2BAA2B,CAE/C,GAAyB;;QACzB,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;QACtC,OAAO,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;CAAA;AAED,MAAM,SAAS,GAAG,IAAI,gBAAS,CAAC,CAAO,QAA6B,EAAE,EAAE;IACtE,MAAM,OAAO,GAA2B,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;IACrE,MAAM,QAAQ,GAAG,CAAC,QAAgB,EAA0B,EAAE;QAC5D,MAAM,OAAO,GAAuB,OAAO,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,OAAO,KAAK,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAC;QAC1E,OAAO,EAAE,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;IACjC,CAAC,CAAC;IACF,OAAO,4DACF,QAAQ,CAAC,6CAA6C,CAAC,GACvD,QAAQ,CAAC,mDAAmD,CAAC,GAC7D,QAAQ,CAAC,iDAAiD,CAAC,GAC3D,QAAQ,CAAC,mDAAmD,CAAC,CAChC,CAAC;AACrC,CAAC,CAAA,CAAC,CAAC"}
+{"version":3,"file":"getTestExternalDeclarations.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/test/compile/getTestExternalDeclarations.ts"],"names":[],"mappings":";;;;;;;;;;;AAMA,kEAKC;AATD,+BAAiC;AAIjC,SAAsB,2BAA2B,CAE/C,GAAyB;;QACzB,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;QACtC,OAAO,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;CAAA;AAED,MAAM,SAAS,GAAG,IAAI,gBAAS,CAAC,CAAO,QAA6B,EAAE,EAAE;IACtE,MAAM,OAAO,GAA2B,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;IACrE,MAAM,QAAQ,GAAG,CAAC,QAAgB,EAA0B,EAAE;QAC5D,MAAM,OAAO,GAAuB,OAAO,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,OAAO,KAAK,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAC;QAC1E,OAAO,EAAE,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;IACjC,CAAC,CAAC;IACF,OAAO,0EACF,QAAQ,CAAC,6CAA6C,CAAC,GACvD,QAAQ,CAAC,mDAAmD,CAAC,GAC7D,QAAQ,CAAC,iDAAiD,CAAC,GAC3D,QAAQ,CAAC,mDAAmD,CAAC,GAC7D,QAAQ,CAAC,oCAAoC,CAAC,CACjB,CAAC;AACrC,CAAC,CAAA,CAAC,CAAC"}

package/lib/agent/src/orchestrate/test/compile/getTestImportStatements.d.ts

@@ -0,0 +1,2 @@
+import { AutoBeOpenApi } from "@autobe/interface";
+export declare const getTestImportStatements: (document: AutoBeOpenApi.IDocument) => string;

package/lib/agent/src/orchestrate/test/compile/getTestImportStatements.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getTestImportStatements = void 0;
+const utils_1 = require("@autobe/utils");
+const getTestImportStatements = (document) => {
+    const typeReferences = Array.from(new Set(Object.keys(document.components.schemas).map((key) => key.split(".")[0]))).sort();
+    return utils_1.StringUtil.trim `
+    import { ArrayUtil, RandomGenerator, TestValidator } from "@nestia/e2e";
+    import { IConnection } from "@nestia/fetcher";
+    import typia, { tags } from "typia";
+    
+    import api from "@ORGANIZATION/PROJECT-api";
+    ${typeReferences
+        .map((ref) => `import type { ${ref} } from "@ORGANIZATION/PROJECT-api/lib/structures/${ref}";`)
+        .join("\n")}
+  `;
+};
+exports.getTestImportStatements = getTestImportStatements;
+//# sourceMappingURL=getTestImportStatements.js.map

package/lib/agent/src/orchestrate/test/compile/getTestImportStatements.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getTestImportStatements.js","sourceRoot":"","sources":["../../../../../../src/orchestrate/test/compile/getTestImportStatements.ts"],"names":[],"mappings":";;;AACA,yCAA2C;AAEpC,MAAM,uBAAuB,GAAG,CACrC,QAAiC,EACzB,EAAE;IACV,MAAM,cAAc,GAAa,KAAK,CAAC,IAAI,CACzC,IAAI,GAAG,CACL,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAE,CAAC,CAC1E,CACF,CAAC,IAAI,EAAE,CAAC;IACT,OAAO,kBAAU,CAAC,IAAI,CAAA;;;;;;MAMlB,cAAc;SACb,GAAG,CACF,CAAC,GAAG,EAAE,EAAE,CACN,iBAAiB,GAAG,qDAAqD,GAAG,IAAI,CACnF;SACA,IAAI,CAAC,IAAI,CAAC;GACd,CAAC;AACJ,CAAC,CAAC;AArBW,QAAA,uBAAuB,2BAqBlC"}

package/lib/agent/src/orchestrate/test/compile/getTestScenarioArtifacts.d.ts

@@ -2,4 +2,4 @@ import { AutoBeTestScenario } from "@autobe/interface";
 import { ILlmSchema } from "@samchon/openapi";
 import { AutoBeContext } from "../../../context/AutoBeContext";
 import { IAutoBeTestScenarioArtifacts } from "../structures/IAutoBeTestScenarioArtifacts";
-export declare function getTestScenarioArtifacts<Model extends ILlmSchema.Model>(ctx: AutoBeContext<Model>, scenario: Pick<AutoBeTestScenario, "endpoint" | "dependencies">): Promise<IAutoBeTestScenarioArtifacts>;
+export declare function getTestScenarioArtifacts<Model extends ILlmSchema.Model>(ctx: AutoBeContext<Model>, scenario: Pick<AutoBeTestScenario, "endpoint" | "dependencies" | "functionName">): Promise<IAutoBeTestScenarioArtifacts>;