@autobe/agent 0.18.0 → 0.20.0

package/src/orchestrate/realize/orchestrateRealizeWrite.ts

@@ -1,10 +1,13 @@
 import {
+  AutoBeProgressEventBase,
   AutoBeRealizeAuthorization,
   AutoBeRealizeWriteEvent,
 } from "@autobe/interface";
+import { StringUtil } from "@autobe/utils";
 import { ILlmApplication, ILlmController, ILlmSchema } from "@samchon/openapi";
 import { IPointer } from "tstl";
 import typia from "typia";
+import { v7 } from "uuid";
 
 import { AutoBeContext } from "../../context/AutoBeContext";
 import { assertSchemaModel } from "../../context/assertSchemaModel";
@@ -17,13 +20,17 @@ import { replaceImportStatements } from "./utils/replaceImportStatements";
 
 export async function orchestrateRealizeWrite<Model extends ILlmSchema.Model>(
   ctx: AutoBeContext<Model>,
-  authorization: AutoBeRealizeAuthorization | null,
-  scenario: IAutoBeRealizeScenarioApplication.IProps,
-  progress: IProgress,
+  props: {
+    totalAuthorizations: AutoBeRealizeAuthorization[];
+    authorization: AutoBeRealizeAuthorization | null;
+    scenario: IAutoBeRealizeScenarioApplication.IProps;
+    progress: AutoBeProgressEventBase;
+    promptCacheKey: string;
+  },
 ): Promise<AutoBeRealizeWriteEvent> {
   const artifacts: IAutoBeTestScenarioArtifacts =
     await getTestScenarioArtifacts(ctx, {
-      endpoint: scenario.operation,
+      endpoint: props.scenario.operation,
       dependencies: [],
     });
   const pointer: IPointer<IAutoBeRealizeWriteApplication.IProps | null> = {
@@ -33,9 +40,10 @@ export async function orchestrateRealizeWrite<Model extends ILlmSchema.Model>(
     source: "realizeWrite",
     histories: transformRealizeWriteHistories({
       state: ctx.state(),
-      scenario,
+      scenario: props.scenario,
       artifacts,
-      authorization,
+      authorization: props.authorization,
+      totalAuthorizations: props.totalAuthorizations,
     }),
     controller: createController({
       model: ctx.model,
@@ -44,35 +52,41 @@ export async function orchestrateRealizeWrite<Model extends ILlmSchema.Model>(
       },
     }),
     enforceFunctionCall: true,
-    message: [
-      `Write complete, production-ready TypeScript code that strictly follows these rules:`,
-      "",
-      `1. Do **not** use the native \`Date\` type anywhere.`,
-      `2. All date or datetime values must be written as \`string & tags.Format<'date-time'>\`.`,
-      `3. UUIDs must be generated using \`v4()\` and typed as \`string & tags.Format<'uuid'>\`.`,
-      `4. Do not use \`as\` for type assertions — resolve types properly.`,
-      `5. All functions must be fully typed with clear parameter and return types.`,
-      `6. Do not skip validations or default values where necessary.`,
-      `7. Follow functional, immutable, and consistent code structure.`,
-      "",
-      `Use \`@nestia/e2e\` test structure if relevant.`,
-    ].join("\n"),
+    promptCacheKey: props.promptCacheKey,
+    message: StringUtil.trim`
+      Write complete, production-ready TypeScript code that strictly follows these rules:
+
+      DO NOT:
+      - Use the native \`Date\` type anywhere
+      - Use \`as\` for type assertions
+
+      DO:
+      - Write all date/datetime values as \`string & tags.Format<'date-time'>\`
+      - Generate UUIDs using \`v4()\` and type as \`string & tags.Format<'uuid'>\`
+      - Resolve types properly without assertions
+      - Type all functions with clear parameter and return types
+      6. Do not skip validations or default values where necessary.
+      7. Follow functional, immutable, and consistent code structure.
+
+      Use \`@nestia/e2e\` test structure if relevant.
+    `,
   });
   if (pointer.value === null) throw new Error("Failed to write code.");
 
   pointer.value.implementationCode = await replaceImportStatements(ctx)(
     artifacts,
     pointer.value.implementationCode,
-    authorization?.payload.name,
+    props.authorization?.payload.name,
   );
 
   const event: AutoBeRealizeWriteEvent = {
     type: "realizeWrite",
-    location: scenario.location,
+    id: v7(),
+    location: props.scenario.location,
     content: pointer.value.implementationCode,
     tokenUsage,
-    completed: ++progress.completed,
-    total: progress.total,
+    completed: ++props.progress.completed,
+    total: props.progress.total,
     step: ctx.state().analyze?.step ?? 0,
     created_at: new Date().toISOString(),
   };
@@ -113,8 +127,3 @@ const collection = {
   deepseek: claude,
   "3.1": claude,
 };
-
-export interface IProgress {
-  total: number;
-  completed: number;
-}

package/src/orchestrate/realize/structures/IAutoBeRealizeAuthorizationApplication.ts

@@ -41,7 +41,7 @@ export namespace IAutoBeRealizeAuthorizationApplication {
      * called by the decorator to verify JWT tokens and return authenticated
      * user information for the specified role.
      *
-     * MUST use camelCase naming convention.
+     * DO: Use camelCase naming convention.
      */
     name: string & CamelPattern;
 
@@ -64,7 +64,7 @@ export namespace IAutoBeRealizeAuthorizationApplication {
      * authorize users for the specific role, injecting the authenticated user
      * payload as a method parameter.
      *
-     * MUST use PascalCase naming convention.
+     * DO: Use PascalCase naming convention.
      */
     name: string & PascalPattern;
 
@@ -86,7 +86,7 @@ export namespace IAutoBeRealizeAuthorizationApplication {
      * authenticated user data that will be injected into Controller methods
      * when using the decorator.
      *
-     * MUST use PascalCase naming convention.
+     * DO: Use PascalCase naming convention.
      */
     name: string & PascalPattern;
 

package/src/orchestrate/realize/structures/IAutoBeRealizeWriteApplication.ts

@@ -13,16 +13,18 @@ export namespace IAutoBeRealizeWriteApplication {
      * function according to the Realize Coder Agent specification. Before
      * writing the actual code, think through the logic and structure.
      *
-     * The plan MUST follow MANDATORY SCHEMA-FIRST APPROACH:
+     * The plan follows a SCHEMA-FIRST APPROACH:
      *
-     * 📋 STEP 1 - PRISMA SCHEMA VERIFICATION (MOST CRITICAL):
+     * 📋 STEP 1 - PRISMA SCHEMA VERIFICATION:
      *
-     * - EXAMINE the actual Prisma schema model definition
-     * - LIST EVERY field that exists in the model with exact types
-     * - EXPLICITLY NOTE fields that DO NOT exist (e.g., "Note: deleted_at field
-     *   DOES NOT EXIST")
-     * - Common assumption errors to avoid: `deleted_at`, `created_by`,
-     *   `updated_by`, `is_deleted`, `is_active`
+     * DO:
+     * - Examine the actual Prisma schema model definition
+     * - List EVERY field that exists in the model with exact types
+     * - Explicitly note fields that DO NOT exist
+     * 
+     * DO NOT:
+     * - Assume common fields exist without verification
+     * - Use fields like deleted_at, created_by, updated_by, is_deleted, is_active without checking
      *
      * 📋 STEP 2 - FIELD INVENTORY:
      *
@@ -158,8 +160,9 @@ export namespace IAutoBeRealizeWriteApplication {
      * Draft WITHOUT using native Date type.
      *
      * This is the initial drafting phase where you outline the basic skeleton
-     * of the function. CRITICAL: This draft must NEVER use the native Date
-     * type.
+     * of the function.
+     * 
+     * DO NOT: Use the native Date type.
      *
      * - The function signature must correctly include `user`, `parameters`, and
      *   `body` arguments.
@@ -168,13 +171,13 @@ export namespace IAutoBeRealizeWriteApplication {
      * - Mark any incomplete or missing parts clearly with placeholders (e.g.,
      *   comments or temporary values).
      *
-     * ⚠️ Import rules:
+     * Import rules:
      *
-     * - Do NOT add any new import statements manually.
-     * - All necessary imports are provided globally or by the system
-     *   automatically.
-     * - Writing import statements directly is prohibited and may cause compile
-     *   errors. If import errors occur, check your environment configuration.
+     * DO NOT:
+     * - Add any new import statements manually
+     * - Write import statements directly (this causes compile errors)
+     * 
+     * Note: All necessary imports are provided globally or by the system automatically.
      *
      * ✅ Requirements:
      *

package/src/orchestrate/realize/utils/replaceImportStatements.ts

@@ -46,6 +46,12 @@ export function replaceImportStatements<Model extends ILlmSchema.Model>(
       .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
@@ -81,6 +87,7 @@ export function replaceImportStatements<Model extends ILlmSchema.Model>(
 
     // Build the standard imports
     const imports = [
+      'import jwt from "jsonwebtoken";',
       'import { MyGlobal } from "../MyGlobal";',
       'import typia, { tags } from "typia";',
       'import { Prisma } from "@prisma/client";',

package/src/orchestrate/test/compile/completeTestCode.ts

@@ -1,3 +1,5 @@
+import { StringUtil } from "@autobe/utils";
+
 import { IAutoBeTestScenarioArtifacts } from "../structures/IAutoBeTestScenarioArtifacts";
 
 export function completeTestCode(
@@ -18,18 +20,20 @@ export function completeTestCode(
     'string & Format<"uuid">',
     'string & tags.Format<"uuid">',
   );
-  code = [
-    `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}";`,
-    ),
-    "",
-    code,
-  ].join("\n");
+  code = 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}
+  `;
   return code;
 }

package/src/orchestrate/test/experimental/transformTestCorrectHistories.ast

@@ -1,6 +1,6 @@
 import { IAgenticaHistoryJson } from "@agentica/core";
 import { IAutoBeTypeScriptCompileResult } from "@autobe/interface";
-import { v4 } from "uuid";
+import { v7 } from "uuid";
 
 import { AutoBeSystemPromptConstant } from "../../constants/AutoBeSystemPromptConstant";
 import { IAutoBeTestWriteResult } from "./structures/IAutoBeTestWriteResult";
@@ -14,13 +14,13 @@ export const transformTestCorrectHistories = (
 > => {
   return [
     {
-      id: v4(),
+      id: v7(),
       created_at: new Date().toISOString(),
       type: "systemMessage",
       text: AutoBeSystemPromptConstant.TEST_WRITE,
     },
     {
-      id: v4(),
+      id: v7(),
       created_at: new Date().toISOString(),
       type: "assistantMessage",
       text: [
@@ -43,7 +43,7 @@ export const transformTestCorrectHistories = (
       ].join("\n"),
     },
     {
-      id: v4(),
+      id: v7(),
       created_at: new Date().toISOString(),
       type: "systemMessage",
       text: AutoBeSystemPromptConstant.TEST_CORRECT,

package/src/orchestrate/test/histories/transformTestCorrectHistories.ts

@@ -1,6 +1,7 @@
 import { IAgenticaHistoryJson } from "@agentica/core";
 import { IAutoBeTypeScriptCompileResult } from "@autobe/interface";
-import { v4 } from "uuid";
+import { StringUtil } from "@autobe/utils";
+import { v7 } from "uuid";
 
 import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
 import { IAutoBeTestFunction } from "../structures/IAutoBeTestFunction";
@@ -14,25 +15,25 @@ export const transformTestCorrectHistories = (
 > => [
   ...transformTestWriteHistories(func.scenario, func.artifacts),
   {
-    id: v4(),
+    id: v7(),
     created_at: new Date().toISOString(),
     type: "assistantMessage",
-    text: [
-      "## Generated TypeScript Code",
-      "```typescript",
-      func.script,
-      "```",
-      "",
-      "## Compile Errors",
-      "Fix the compilation error in the provided code.",
-      "",
-      "```json",
-      JSON.stringify(failure.diagnostics),
-      "```",
-    ].join("\n"),
+    text: StringUtil.trim`
+      ## Generated TypeScript Code
+      \`\`\`typescript
+      ${func.script}
+      \`\`\`
+
+      ## Compile Errors
+      Fix the compilation error in the provided code.
+
+      \`\`\`json
+      ${JSON.stringify(failure.diagnostics)}
+      \`\`\`
+    `,
   },
   {
-    id: v4(),
+    id: v7(),
     created_at: new Date().toISOString(),
     type: "systemMessage",
     text: AutoBeSystemPromptConstant.TEST_CORRECT.replace(

package/src/orchestrate/test/histories/transformTestHistories.ts

@@ -1,5 +1,6 @@
 import { IAgenticaHistoryJson } from "@agentica/core";
-import { v4 } from "uuid";
+import { StringUtil } from "@autobe/utils";
+import { v7 } from "uuid";
 
 import { AutoBeState } from "../../../context/AutoBeState";
 
@@ -11,67 +12,67 @@ export const transformTestHistories = (
   if (state.analyze === null)
     return [
       {
-        id: v4(),
+        id: v7(),
         created_at: new Date().toISOString(),
         type: "systemMessage",
-        text: [
-          "Requirement analysis is not yet completed.",
-          "Don't call the any tool function,",
-          "but say to process the requirement analysis.",
-        ].join(" "),
+        text: StringUtil.trim`
+          Requirement analysis is not yet completed.
+          Don't call the any tool function,
+          but say to process the requirement analysis.
+        `,
       },
     ];
   else if (state.prisma === null)
     return [
       {
-        id: v4(),
+        id: v7(),
         created_at: new Date().toISOString(),
         type: "systemMessage",
-        text: [
-          "Prisma DB schema generation is not yet completed.",
-          "Don't call the any tool function,",
-          "but say to process the Prisma DB schema generation.",
-        ].join(" "),
+        text: StringUtil.trim`
+          Prisma DB schema generation is not yet completed.
+          Don't call the any tool function,
+          but say to process the Prisma DB schema generation.
+        `,
       },
     ];
   else if (state.analyze.step !== state.prisma.step)
     return [
       {
-        id: v4(),
+        id: v7(),
         created_at: new Date().toISOString(),
         type: "systemMessage",
-        text: [
-          "Prisma DB schema generation has not been updated",
-          "for the latest requirement analysis.",
-          "Don't call the any tool function,",
-          "but say to re-process the Prisma DB schema generation.",
-        ].join(" "),
+        text: StringUtil.trim`
+          Prisma DB schema generation has not been updated
+          for the latest requirement analysis.
+          Don't call the any tool function,
+          but say to re-process the Prisma DB schema generation.
+        `,
       },
     ];
   else if (state.prisma.compiled.type !== "success")
     return [
       {
-        id: v4(),
+        id: v7(),
         created_at: new Date().toISOString(),
         type: "systemMessage",
-        text: [
-          "Prisma DB schema generation has not been updated",
-          "for the latest requirement analysis.",
-          "Don't call the any tool function,",
-          "but say to re-process the Prisma DB schema generation.",
-        ].join(" "),
+        text: StringUtil.trim`
+          Prisma DB schema generation has not been updated
+          for the latest requirement analysis.
+          Don't call the any tool function,
+          but say to re-process the Prisma DB schema generation.
+        `,
       },
     ];
   return [
     {
-      id: v4(),
+      id: v7(),
       created_at: new Date().toISOString(),
       type: "systemMessage",
-      text: [
-        "Interface generation is not yet completed.",
-        "Don't call the any tool function,",
-        "but say to process the interface generation.",
-      ].join(" "),
+      text: StringUtil.trim`
+        Interface generation is not yet completed.
+        Don't call the any tool function,
+        but say to process the interface generation.
+      `,
     },
   ];
-};
+};

package/src/orchestrate/test/histories/transformTestScenarioHistories.ts

@@ -1,70 +1,126 @@
 import { IAgenticaHistoryJson } from "@agentica/core";
-import { AutoBeOpenApi } from "@autobe/interface";
-import { v4 } from "uuid";
+import { AutoBeInterfaceAuthorization, AutoBeOpenApi } from "@autobe/interface";
+import { MapUtil, StringUtil } from "@autobe/utils";
+import { v7 } from "uuid";
 
 import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
+import { AutoBeState } from "../../../context/AutoBeState";
+import { IAutoBeTestScenarioAuthorizationRole } from "../structures/IAutoBeTestScenarioAuthorizationRole";
 
 export const transformTestScenarioHistories = (
+  state: AutoBeState,
   entire: AutoBeOpenApi.IOperation[],
   include: AutoBeOpenApi.IOperation[],
   exclude: Pick<AutoBeOpenApi.IOperation, "method" | "path">[],
 ): Array<
   IAgenticaHistoryJson.IAssistantMessage | IAgenticaHistoryJson.ISystemMessage
-> => [
-  {
-    id: v4(),
-    created_at: new Date().toISOString(),
-    type: "systemMessage",
-    text: AutoBeSystemPromptConstant.TEST_SCENARIO,
-  } satisfies IAgenticaHistoryJson.ISystemMessage,
-  {
-    id: v4(),
-    created_at: new Date().toISOString(),
-    type: "systemMessage",
-    text: [
-      "# Operations",
-      "Below are the full operations. Please refer to this.",
-      "Your role is to draft all test cases for each given Operation.",
-      "It is also permissible to write multiple test codes on a single endpoint.",
-      "However, rather than meaningless tests, business logic tests should be written and an E2E test situation should be assumed.",
-      "",
-      "Please carefully analyze each operation to identify all dependencies required for testing.",
-      "For example, if you want to test liking and then deleting a post,",
-      "you might think to test post creation, liking, and unlike operations.",
-      "However, even if not explicitly mentioned, user registration and login are essential prerequisites.",
-      "Pay close attention to IDs and related values in the API,",
-      "and ensure you identify all dependencies between endpoints.",
-      "",
-      "```json",
-      JSON.stringify(
-        entire.map((el) => ({
-          ...el,
-          specification: undefined,
-        })),
-      ),
-      "```",
-    ].join("\n"),
-  } satisfies IAgenticaHistoryJson.ISystemMessage,
-  {
-    id: v4(),
-    created_at: new Date().toISOString(),
-    type: "systemMessage",
-    text: [
-      "# Included in Test Plan",
-      include
-        .map(
-          (el) =>
-            `- ${el.method.toUpperCase()}: ${el.path} ${el.authorizationRole ? `(Role: ${el.authorizationRole})` : ""}`,
-        )
-        .join("\n"),
-      "",
-      "# Excluded from Test Plan",
-      "These are the endpoints that have already been used in test codes generated as part of a plan group.",
-      "These endpoints do not need to be tested again.",
-      "However, it is allowed to reference or depend on these endpoints when writing test codes for other purposes.",
-      exclude
-        .map((el) => `- ${el.method.toUpperCase()}: ${el.path}`)
-        .join("\n"),
-    ].join("\n"),
-  } satisfies IAgenticaHistoryJson.ISystemMessage,
-];
+> => {
+  const authorizations: AutoBeInterfaceAuthorization[] =
+    state.interface?.authorizations ?? [];
+
+  const authorizationRoles: Map<string, IAutoBeTestScenarioAuthorizationRole> =
+    new Map();
+
+  for (const authorization of authorizations) {
+    for (const op of authorization.operations) {
+      if (op.authorizationType === null) continue;
+      const value: IAutoBeTestScenarioAuthorizationRole = MapUtil.take(
+        authorizationRoles,
+        authorization.role,
+        () => ({
+          name: authorization.role,
+          join: null,
+          login: null,
+        }),
+      );
+      if (op.authorizationType === "join") value.join = op;
+      else if (op.authorizationType === "login") value.login = op;
+    }
+  }
+
+  return [
+    {
+      id: v7(),
+      created_at: new Date().toISOString(),
+      type: "systemMessage",
+      text: AutoBeSystemPromptConstant.TEST_SCENARIO,
+    } satisfies IAgenticaHistoryJson.ISystemMessage,
+    {
+      id: v7(),
+      created_at: new Date().toISOString(),
+      type: "systemMessage",
+      text: StringUtil.trim`
+        # Operations
+
+        Below are the full operations. Please refer to this.
+        Your role is to draft all test cases for each given Operation.
+        It is also permissible to write multiple test codes on a single endpoint.
+        However, rather than meaningless tests, business logic tests should be written and an E2E test situation should be assumed.
+
+        Please carefully analyze each operation to identify all dependencies required for testing.
+        For example, if you want to test liking and then deleting a post,
+        you might think to test post creation, liking, and unlike operations.
+        However, even if not explicitly mentioned, user registration or login are essential prerequisites.
+        Pay close attention to IDs and related values in the API,
+        and ensure you identify all dependencies between endpoints.
+
+        \`\`\`json
+        ${JSON.stringify(
+          entire.map((el) => ({
+            ...el,
+            specification: undefined,
+          })),
+        )}
+        \`\`\`
+      `,
+    } satisfies IAgenticaHistoryJson.ISystemMessage,
+    {
+      id: v7(),
+      created_at: new Date().toISOString(),
+      type: "systemMessage",
+      text: StringUtil.trim`
+        # Included in Test Plan
+
+        Below are the endpoints that have been included in the test plan.
+        Each endpoint shows its authentication requirements and related authentication APIs.
+        When testing endpoints that require authentication, ensure you include the corresponding join/login operations in your test scenario to establish proper authentication context.
+
+        ${include
+          .map((el, i) => {
+            const roles = Array.from(authorizationRoles.values()).filter(
+              (role) => role.name === el.authorizationRole,
+            );
+            return StringUtil.trim`
+              ## ${i + 1}. ${el.method.toUpperCase()} ${el.path}
+
+              Related Authentication APIs:
+
+              ${
+                roles.length > 0
+                  ? roles
+                      .map((role) => {
+                        return StringUtil.trim`
+                          - ${role.join?.method.toUpperCase()}: ${role.join?.path}
+                          - ${role.login?.method.toUpperCase()}: ${role.login?.path}
+                        `;
+                      })
+                      .join("\n")
+                  : "- None"
+              }
+            `;
+          })
+          .join("\n")}
+
+        # Excluded from Test Plan
+
+        These are the endpoints that have already been used in test codes generated as part of a plan group.
+        These endpoints do not need to be tested again.
+        However, it is allowed to reference or depend on these endpoints when writing test codes for other purposes.
+
+        ${exclude
+          .map((el) => `- ${el.method.toUpperCase()}: ${el.path}`)
+          .join("\n")}
+      `,
+    } satisfies IAgenticaHistoryJson.ISystemMessage,
+  ];
+};