@autobe/agent 0.28.1 → 0.29.0

package/src/orchestrate/realize/histories/transformRealizeWriteHistories.ts

@@ -1,14 +1,16 @@
-import { IAgenticaHistoryJson } from "@agentica/core";
-import { AutoBeRealizeAuthorization } from "@autobe/interface";
+import { IMicroAgenticaHistoryJson } from "@agentica/core";
+import { AutoBeOpenApi, AutoBeRealizeAuthorization } from "@autobe/interface";
 import { StringUtil } from "@autobe/utils";
 import { v7 } from "uuid";
 
 import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
 import { AutoBeState } from "../../../context/AutoBeState";
+import { IAutoBeOrchestrateHistory } from "../../../structures/IAutoBeOrchestrateHistory";
+import { AutoBePreliminaryController } from "../../common/AutoBePreliminaryController";
 import { IAutoBeRealizeScenarioResult } from "../structures/IAutoBeRealizeScenarioResult";
 import { getRealizeWriteCodeTemplate } from "../utils/getRealizeWriteCodeTemplate";
 import { getRealizeWriteInputType } from "../utils/getRealizeWriteInputType";
-import { transformRealizeWriteAuthorizationsHistories } from "./transformRealizeWriteAuthorizationsHistories";
+import { transformRealizeWriteMembershipHistory } from "./transformRealizeWriteMembershipHistory";
 
 export const transformRealizeWriteHistories = (props: {
   state: AutoBeState;
@@ -16,154 +18,183 @@ export const transformRealizeWriteHistories = (props: {
   authorization: AutoBeRealizeAuthorization | null;
   totalAuthorizations: AutoBeRealizeAuthorization[];
   dto: Record<string, string>;
-}): Array<
-  IAgenticaHistoryJson.IAssistantMessage | IAgenticaHistoryJson.ISystemMessage
-> => {
-  const payloads = Object.fromEntries(
+  preliminary: AutoBePreliminaryController<"prismaSchemas">;
+}): IAutoBeOrchestrateHistory => {
+  if (props.state.analyze === null)
+    return {
+      histories: [
+        {
+          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(" "),
+        },
+      ],
+      userMessage: "Please wait for prerequisites to complete",
+    };
+  else if (props.state.prisma === null)
+    return {
+      histories: [
+        {
+          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(" "),
+        },
+      ],
+      userMessage: "Please wait for prerequisites to complete",
+    };
+  else if (props.state.analyze.step !== props.state.prisma.step)
+    return {
+      histories: [
+        {
+          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(" "),
+        },
+      ],
+      userMessage: "Please wait for prerequisites to complete",
+    };
+  else if (props.state.prisma.compiled.type !== "success")
+    return {
+      histories: [
+        {
+          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(" "),
+        },
+      ],
+      userMessage: "Please wait for prerequisites to complete",
+    };
+  else if (props.state.interface === null)
+    return {
+      histories: [
+        {
+          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(" "),
+        },
+      ],
+      userMessage: "Please wait for prerequisites to complete",
+    };
+
+  const payloads: Record<string, string> = Object.fromEntries(
     props.totalAuthorizations.map((el) => [
       el.payload.location,
       el.payload.content,
     ]),
   );
+  const operation: AutoBeOpenApi.IOperation = props.scenario.operation;
+  const authorizationHistories: IMicroAgenticaHistoryJson[] =
+    operation.authorizationType
+      ? transformRealizeWriteMembershipHistory(operation, payloads)
+      : [];
+  const document: AutoBeOpenApi.IDocument = props.state.interface.document;
 
-  const operation = props.scenario.operation;
-
-  if (props.state.analyze === null)
-    return [
+  return {
+    histories: [
       {
         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: AutoBeSystemPromptConstant.REALIZE_WRITE,
       },
-    ];
-  else if (props.state.prisma === null)
-    return [
+      ...props.preliminary.getHistories(),
+      ...authorizationHistories,
       {
         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: AutoBeSystemPromptConstant.REALIZE_WRITE_ARTIFACT.replaceAll(
+          `{input}`,
+          getRealizeWriteInputType(operation, props.authorization),
+        ).replaceAll(`{artifacts_dto}`, JSON.stringify(props.dto)),
       },
-    ];
-  else if (props.state.analyze.step !== props.state.prisma.step)
-    return [
       {
         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(" "),
+        created_at: new Date().toISOString(),
+        text: getRealizeWriteCodeTemplate({
+          authorization: props.authorization,
+          scenario: props.scenario,
+          schemas: document.components.schemas,
+          operation,
+        }),
       },
-    ];
-  else if (props.state.prisma.compiled.type !== "success")
-    return [
       {
         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(" "),
+        created_at: new Date().toISOString(),
+        text: StringUtil.trim`
+          Write new code based on the following operation.
+
+          \`\`\`json
+          ${JSON.stringify(props.scenario)}
+          \`\`\`
+        `,
       },
-    ];
-  else if (props.state.interface === null)
-    return [
       {
         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(" "),
-      },
-    ];
+        type: "assistantMessage",
+        text: StringUtil.trim`
+          I understand your request.
 
-  // Get authorization-specific histories if authorizationType is set
-  const authorizationHistories = operation.authorizationType
-    ? transformRealizeWriteAuthorizationsHistories(operation, payloads)
-    : [];
-  const document = props.state.interface.document;
+          To summarize:
+          - I must **never use the native \`Date\` type** in any code or type definitions.
+          - Instead, all date and datetime values must be handled as \`string & tags.Format<'date-time'>\`.
+          - This rule is **strict** and applies everywhere, including domain types, API inputs/outputs, and Prisma models.
+          - Even if a library or tool returns a \`Date\`, I must convert it to the correct string format before use.
 
-  return [
-    {
-      id: v7(),
-      created_at: new Date().toISOString(),
-      type: "systemMessage",
-      text: AutoBeSystemPromptConstant.REALIZE_WRITE,
-    },
-    ...authorizationHistories,
-    {
-      id: v7(),
-      created_at: new Date().toISOString(),
-      type: "systemMessage",
-      text: AutoBeSystemPromptConstant.REALIZE_WRITE_ARTIFACT.replaceAll(
-        `{prismaSchemas}`,
-        JSON.stringify(props.state.prisma.schemas),
-      )
-        .replaceAll(
-          `{input}`,
-          getRealizeWriteInputType(operation, props.authorization),
-        )
-        .replaceAll(`{artifacts_dto}`, JSON.stringify(props.dto)),
-    },
-    {
-      id: v7(),
-      type: "systemMessage",
-      created_at: new Date().toISOString(),
-      text: getRealizeWriteCodeTemplate({
-        authorization: props.authorization,
-        scenario: props.scenario,
-        schemas: document.components.schemas,
-        operation,
-      }),
-    },
-    {
-      id: v7(),
-      type: "systemMessage",
-      created_at: new Date().toISOString(),
-      text: StringUtil.trim`
-        Write new code based on the following operation.
-        
-        \`\`\`json
-        ${JSON.stringify(props.scenario)}
-        \`\`\`
-      `,
-    },
-    {
-      id: v7(),
-      created_at: new Date().toISOString(),
-      type: "assistantMessage",
-      text: StringUtil.trim`
-        I understand your request.
+          Especially regarding the \`Date\` type: I understand that using it can lead to type inconsistency and runtime issues, so I will completely avoid it in all circumstances.
+
+          I'll make sure to follow all these rules strictly. Let's proceed with this in mind.
+        `,
+      },
+    ],
+    userMessage: StringUtil.trim`
+      Write implementation for ${operation.method.toUpperCase()} ${operation.path} please.
+
+      Write the complete, production-ready TypeScript code that strictly follows these rules:
 
-        To summarize:
-        - I must **never use the native \`Date\` type** in any code or type definitions.
-        - Instead, all date and datetime values must be handled as \`string & tags.Format<'date-time'>\`.
-        - This rule is **strict** and applies everywhere, including domain types, API inputs/outputs, and Prisma models.
-        - Even if a library or tool returns a \`Date\`, I must convert it to the correct string format before use.
+      DO NOT:
+      - Use the native \`Date\` type anywhere
+      - Use \`as\` for type assertions
 
-        Especially regarding the \`Date\` type: I understand that using it can lead to type inconsistency and runtime issues, so I will completely avoid it in all circumstances.
+      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.
 
-        I'll make sure to follow all these rules strictly. Let's proceed with this in mind.
-      `,
-    },
-  ];
+      Use \`@nestia/e2e\` test structure if relevant.
+    `,
+  };
 };

package/src/orchestrate/realize/histories/{transformRealizeWriteAuthorizationsHistories.ts → transformRealizeWriteMembershipHistory.ts}

@@ -4,13 +4,13 @@ import { v7 } from "uuid";
 
 import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
 
-export const transformRealizeWriteAuthorizationsHistories = (
+export const transformRealizeWriteMembershipHistory = (
   operation: AutoBeOpenApi.IOperation,
   payload: Record<string, string>,
 ): Array<IAgenticaHistoryJson.ISystemMessage> => {
   if (operation.authorizationType === null) return [];
   const text: string = PROMPTS[operation.authorizationType].replace(
-    "${PAYLOAD}",
+    "{{PAYLOAD}}",
     JSON.stringify(payload),
   );
   const history: IAgenticaHistoryJson.ISystemMessage = {

package/src/orchestrate/realize/orchestrateRealize.ts

@@ -17,7 +17,7 @@ import { executeCachedBatch } from "../../utils/executeCachedBatch";
 import { predicateStateMessage } from "../../utils/predicateStateMessage";
 import { IAutoBeFacadeApplicationProps } from "../facade/histories/IAutoBeFacadeApplicationProps";
 import { compileRealizeFiles } from "./internal/compileRealizeFiles";
-import { orchestrateRealizeAuthorization } from "./orchestrateRealizeAuthorization";
+import { orchestrateRealizeAuthorizationWrite } from "./orchestrateRealizeAuthorizationWrite";
 import { orchestrateRealizeCorrect } from "./orchestrateRealizeCorrect";
 import { orchestrateRealizeCorrectCasting } from "./orchestrateRealizeCorrectCasting";
 import { orchestrateRealizeWrite } from "./orchestrateRealizeWrite";
@@ -59,7 +59,7 @@ export const orchestrateRealize =
     // PREPARE ASSETS
     const compiler: IAutoBeCompiler = await ctx.compiler();
     const authorizations: AutoBeRealizeAuthorization[] =
-      await orchestrateRealizeAuthorization(ctx);
+      await orchestrateRealizeAuthorizationWrite(ctx);
 
     const writeProgress: AutoBeProgressEventBase = {
       total: document.operations.length,
@@ -75,6 +75,7 @@ export const orchestrateRealize =
     ): Promise<IBucket> => {
       const writes: AutoBeRealizeWriteEvent[] = (
         await executeCachedBatch(
+          ctx,
           artifacts.map((art) => async (promptCacheKey) => {
             const write = async (): Promise<AutoBeRealizeWriteEvent | null> => {
               try {
@@ -143,7 +144,7 @@ export const orchestrateRealize =
     // SCENARIOS
     const entireScenarios: IAutoBeRealizeScenarioResult[] =
       document.operations.map((operation) =>
-        generateRealizeScenario(ctx, operation, authorizations),
+        generateRealizeScenario(operation, authorizations),
       );
     let bucket: IBucket = await process(entireScenarios);
     for (let i: number = 0; i < 2; ++i) {