@autobe/agent 0.18.0 → 0.20.0

package/src/orchestrate/prisma/orchestratePrismaComponent.ts

@@ -1,13 +1,9 @@
-import {
-  AgenticaAssistantMessageHistory,
-  IAgenticaController,
-} from "@agentica/core";
-import { AutoBeAssistantMessageHistory } from "@autobe/interface";
+import { IAgenticaController } from "@agentica/core";
 import { AutoBePrismaComponentsEvent } from "@autobe/interface/src/events/AutoBePrismaComponentsEvent";
 import { ILlmApplication, ILlmSchema } from "@samchon/openapi";
 import { IPointer } from "tstl";
 import typia from "typia";
-import { v4 } from "uuid";
+import { v7 } from "uuid";
 
 import { AutoBeContext } from "../../context/AutoBeContext";
 import { assertSchemaModel } from "../../context/assertSchemaModel";
@@ -18,14 +14,14 @@ export async function orchestratePrismaComponents<
   Model extends ILlmSchema.Model,
 >(
   ctx: AutoBeContext<Model>,
-  message: string = "Please extract files and tables from the given documents.",
-): Promise<AutoBeAssistantMessageHistory | AutoBePrismaComponentsEvent> {
+  message: string = "Design database from the given requirement analysis documents.",
+): Promise<AutoBePrismaComponentsEvent> {
   const start: Date = new Date();
   const pointer: IPointer<IAutoBePrismaComponentApplication.IProps | null> = {
     value: null,
   };
   const prefix: string | null = ctx.state().analyze?.prefix ?? null;
-  const { histories, tokenUsage } = await ctx.conversate({
+  const { tokenUsage } = await ctx.conversate({
     source: "prismaComponents",
     histories: transformPrismaComponentsHistories(ctx.state(), prefix),
     controller: createController({
@@ -34,21 +30,14 @@ export async function orchestratePrismaComponents<
         pointer.value = next;
       },
     }),
-    enforceFunctionCall: false,
+    enforceFunctionCall: true,
     message,
   });
-  if (histories.at(-1)?.type === "assistantMessage")
-    return {
-      ...(histories.at(-1)! as AgenticaAssistantMessageHistory),
-      created_at: start.toISOString(),
-      completed_at: new Date().toISOString(),
-      id: v4(),
-    } satisfies AutoBeAssistantMessageHistory;
-  else if (pointer.value === null) {
+  if (pointer.value === null)
     throw new Error("Failed to extract files and tables."); // unreachable
-  }
   return {
     type: "prismaComponents",
+    id: v7(),
     created_at: start.toISOString(),
     thinking: pointer.value.thinking,
     review: pointer.value.review,

package/src/orchestrate/prisma/orchestratePrismaCorrect.ts

@@ -8,6 +8,7 @@ import {
 import { ILlmApplication, 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";
@@ -49,6 +50,7 @@ async function iterate<Model extends ILlmSchema.Model>(
   );
   ctx.dispatch({
     type: "prismaValidate",
+    id: v7(),
     result,
     schemas,
     compiled: await compiler.prisma.compile({
@@ -144,6 +146,7 @@ async function execute<Model extends ILlmSchema.Model>(
   };
   ctx.dispatch({
     type: "prismaCorrect",
+    id: v7(),
     failure,
     planning: pointer.value.planning,
     correction: correction,

package/src/orchestrate/prisma/orchestratePrismaReview.ts

@@ -1,13 +1,14 @@
 import { IAgenticaController } from "@agentica/core";
-import { AutoBePrisma } from "@autobe/interface";
+import { AutoBePrisma, AutoBeProgressEventBase } from "@autobe/interface";
 import { AutoBePrismaReviewEvent } from "@autobe/interface/src/events/AutoBePrismaReviewEvent";
 import { ILlmApplication, 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";
-import { IProgress } from "../internal/IProgress";
+import { executeCachedBatch } from "../../utils/executeCachedBatch";
 import { transformPrismaReviewHistories } from "./histories/transformPrismaReviewHistories";
 import { IAutoBePrismaReviewApplication } from "./structures/IAutoBePrismaReviewApplication";
 
@@ -17,23 +18,39 @@ export async function orchestratePrismaReview<Model extends ILlmSchema.Model>(
   schemas: Record<string, string>,
   componentList: AutoBePrisma.IComponent[],
 ): Promise<AutoBePrismaReviewEvent[]> {
-  const progress: IProgress = {
+  const progress: AutoBeProgressEventBase = {
     completed: 0,
     total: componentList.length,
   };
-  return await Promise.all(
-    componentList.map((component) =>
-      step(ctx, application, schemas, component, progress),
-    ),
-  );
+  return (
+    await executeCachedBatch(
+      componentList.map((component) => async (promptCacheKey) => {
+        try {
+          return await step(ctx, {
+            application,
+            schemas,
+            component,
+            progress,
+            promptCacheKey,
+          });
+        } catch {
+          ++progress.completed;
+          return null;
+        }
+      }),
+    )
+  ).filter((v) => v !== null);
 }
 
 async function step<Model extends ILlmSchema.Model>(
   ctx: AutoBeContext<Model>,
-  application: AutoBePrisma.IApplication,
-  schemas: Record<string, string>,
-  component: AutoBePrisma.IComponent,
-  progress: IProgress,
+  props: {
+    application: AutoBePrisma.IApplication;
+    schemas: Record<string, string>;
+    component: AutoBePrisma.IComponent;
+    progress: AutoBeProgressEventBase;
+    promptCacheKey: string;
+  },
 ): Promise<AutoBePrismaReviewEvent> {
   const start: Date = new Date();
   const pointer: IPointer<IAutoBePrismaReviewApplication.IProps | null> = {
@@ -49,9 +66,9 @@ async function step<Model extends ILlmSchema.Model>(
           .reduce((acc, cur) => {
             return Object.assign(acc, cur);
           }, {}) ?? {},
-      application,
-      schemas,
-      component,
+      application: props.application,
+      schemas: props.schemas,
+      component: props.component,
     }),
     controller: createController(ctx, {
       build: (next) => {
@@ -59,6 +76,7 @@ async function step<Model extends ILlmSchema.Model>(
       },
     }),
     enforceFunctionCall: true,
+    promptCacheKey: props.promptCacheKey,
     message: "Please review the Prisma schema file.",
   });
   if (pointer.value === null)
@@ -66,14 +84,15 @@ async function step<Model extends ILlmSchema.Model>(
 
   const event: AutoBePrismaReviewEvent = {
     type: "prismaReview",
+    id: v7(),
     created_at: start.toISOString(),
-    filename: component.filename,
+    filename: props.component.filename,
     review: pointer.value.review,
     plan: pointer.value.plan,
     modifications: pointer.value.modifications,
     tokenUsage,
-    completed: ++progress.completed,
-    total: progress.total,
+    completed: ++props.progress.completed,
+    total: props.progress.total,
     step: ctx.state().analyze?.step ?? 0,
   };
   ctx.dispatch(event);

package/src/orchestrate/prisma/orchestratePrismaSchemas.ts

@@ -1,12 +1,15 @@
 import { IAgenticaController } from "@agentica/core";
 import { AutoBePrisma } from "@autobe/interface";
 import { AutoBePrismaSchemasEvent } from "@autobe/interface/src/events/AutoBePrismaSchemasEvent";
+import { StringUtil } from "@autobe/utils";
 import { ILlmApplication, ILlmSchema, IValidation } 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";
+import { executeCachedBatch } from "../../utils/executeCachedBatch";
 import { transformPrismaSchemaHistories } from "./histories/transformPrismaSchemaHistories";
 import { IAutoBePrismaSchemaApplication } from "./structures/IAutoBePrismaSchemaApplication";
 
@@ -19,8 +22,8 @@ export async function orchestratePrismaSchemas<Model extends ILlmSchema.Model>(
     .map((c) => c.tables.length)
     .reduce((x, y) => x + y, 0);
   const completed: IPointer<number> = { value: 0 };
-  return await Promise.all(
-    componentList.map(async (component) => {
+  return await executeCachedBatch(
+    componentList.map((component) => async (promptCacheKey) => {
       const otherTables: string[] = componentList
         .filter((y) => component !== y)
         .map((c) => c.tables)
@@ -31,6 +34,7 @@ export async function orchestratePrismaSchemas<Model extends ILlmSchema.Model>(
         start,
         total,
         completed,
+        promptCacheKey,
       });
       ctx.dispatch(event);
       return event;
@@ -46,6 +50,7 @@ async function process<Model extends ILlmSchema.Model>(
     start: Date;
     total: number;
     completed: IPointer<number>;
+    promptCacheKey: string;
   },
 ): Promise<AutoBePrismaSchemasEvent> {
   const pointer: IPointer<IAutoBePrismaSchemaApplication.IProps | null> = {
@@ -71,12 +76,14 @@ async function process<Model extends ILlmSchema.Model>(
       },
     }),
     enforceFunctionCall: true,
+    promptCacheKey: props.promptCacheKey,
     message: "Make prisma schema file please",
   });
   if (pointer.value === null)
     throw new Error("Unreachable code: Prisma Schema not generated");
   return {
     type: "prismaSchemas",
+    id: v7(),
     created_at: props.start.toISOString(),
     plan: pointer.value.plan,
     models: pointer.value.models,
@@ -118,6 +125,7 @@ function createController<Model extends ILlmSchema.Model>(
 
     ctx.dispatch({
       type: "prismaInsufficient",
+      id: v7(),
       created_at: new Date().toISOString(),
       component: props.targetComponent,
       actual,
@@ -131,26 +139,26 @@ function createController<Model extends ILlmSchema.Model>(
           path: "$input.models",
           value: result.data.models,
           expected: `Array<AutoBePrisma.IModel>`,
-          description: [
-            "You missed some tables from the current domain's component.",
-            "",
-            "Look at the following details to fix the schemas. Never forget to",
-            "compose the `missed` tables at the next function calling.",
-            "",
-            "- filename: current domain's filename",
-            "- namespace: current domain's namespace",
-            "- expected: expected tables in the current domain",
-            "- actual: actual tables you made",
-            "- missed: tables you have missed, and you have to compose again",
-            "",
-            JSON.stringify({
+          description: StringUtil.trim`
+            You missed some tables from the current domain's component.
+
+            Look at the following details to fix the schemas. Never forget to
+            compose the \`missed\` tables at the next function calling.
+
+            - filename: current domain's filename
+            - namespace: current domain's namespace
+            - expected: expected tables in the current domain
+            - actual: actual tables you made
+            - missed: tables you have missed, and you have to compose again
+
+            ${JSON.stringify({
               filename: props.targetComponent.filename,
               namespace: props.targetComponent.namespace,
               expected,
               actual,
               missed,
-            }),
-          ].join("\n"),
+            })}
+          `,
         },
       ],
     };

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

@@ -1,7 +1,8 @@
 import { IAgenticaHistoryJson } from "@agentica/core";
 import { AutoBeAnalyzeRole } from "@autobe/interface";
+import { StringUtil } from "@autobe/utils";
 import { ILlmSchema } from "@samchon/openapi";
-import { v4 } from "uuid";
+import { v7 } from "uuid";
 
 import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
 import { AutoBeContext } from "../../../context/AutoBeContext";
@@ -16,33 +17,35 @@ export const transformRealizeAuthorizationHistories = <
 > => {
   return [
     {
-      id: v4(),
+      id: v7(),
       created_at: new Date().toISOString(),
       type: "systemMessage",
       text: AutoBeSystemPromptConstant.REALIZE_AUTHORIZATION,
     },
     {
-      id: v4(),
+      id: v7(),
       created_at: new Date().toISOString(),
       type: "systemMessage",
-      text: [
-        "## Role",
-        "```json",
-        JSON.stringify(role),
-        "```",
-        "",
-        "## Prisma Schema",
-        "",
-        JSON.stringify(ctx.state().prisma?.schemas, null, 2),
-        "",
-        "## Component Naming Convention",
-        "",
-        "Please follow this naming convention for the authorization components:",
-        "",
-        `- Provider Name: ${role.name}Authorize (e.g. ${role.name}Authorize)`,
-        `- Decorator Name: ${role.name.charAt(0).toUpperCase() + role.name.slice(1)}Auth (e.g. ${role.name.charAt(0).toUpperCase() + role.name.slice(1)}Auth)`,
-        `- Payload Name: ${role.name.charAt(0).toUpperCase() + role.name.slice(1)}Payload (e.g. ${role.name.charAt(0).toUpperCase() + role.name.slice(1)}Payload)`,
-      ].join("\n"),
+      text: StringUtil.trim`
+        ## Role
+        \`\`\`json
+        ${JSON.stringify(role)}
+        \`\`\`
+
+        ## Prisma Schema
+
+        \`\`\`json
+        ${JSON.stringify(ctx.state().prisma?.schemas)}
+        \`\`\`
+
+        ## Component Naming Convention
+
+        Please follow this naming convention for the authorization components:
+
+        - Provider Name: ${role.name}Authorize (e.g. ${role.name}Authorize)
+        - Decorator Name: ${role.name.charAt(0).toUpperCase() + role.name.slice(1)}Auth (e.g. ${role.name.charAt(0).toUpperCase() + role.name.slice(1)}Auth)
+        - Payload Name: ${role.name.charAt(0).toUpperCase() + role.name.slice(1)}Payload (e.g. ${role.name.charAt(0).toUpperCase() + role.name.slice(1)}Payload)
+      `,
     },
   ];
 };

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

@@ -3,8 +3,9 @@ import {
   AutoBeRealizeAuthorization,
   IAutoBeTypeScriptCompileResult,
 } from "@autobe/interface";
+import { StringUtil } from "@autobe/utils";
 import { ILlmSchema } from "@samchon/openapi";
-import { v4 } from "uuid";
+import { v7 } from "uuid";
 
 import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
 import { AutoBeContext } from "../../../context/AutoBeContext";
@@ -21,77 +22,74 @@ export const transformRealizeAuthorizationCorrectHistories = <
 > => {
   return [
     {
-      id: v4(),
+      id: v7(),
       created_at: new Date().toISOString(),
       type: "systemMessage",
       text: AutoBeSystemPromptConstant.REALIZE_AUTHORIZATION,
     },
     {
-      id: v4(),
+      id: v7(),
       created_at: new Date().toISOString(),
       type: "systemMessage",
       text: AutoBeSystemPromptConstant.REALIZE_AUTHORIZATION_CORRECT,
     },
     {
-      id: v4(),
+      id: v7(),
       created_at: new Date().toISOString(),
       type: "assistantMessage",
-      text: [
-        "## Generated TypeScript Code",
-        "",
-        "```json",
-        `${JSON.stringify(
-          {
-            provider: {
-              location: auth.provider.location,
-              name: auth.provider.name,
-              content: auth.provider.content,
-            },
-            decorator: {
-              location: auth.decorator.location,
-              name: auth.decorator.name,
-              content: auth.decorator.content,
-            },
-            payload: {
-              location: auth.payload.location,
-              name: auth.payload.name,
-              content: auth.payload.content,
-            },
-          } satisfies Omit<AutoBeRealizeAuthorization, "role">,
-          null,
-          2,
-        )}`,
-        "```",
-        "",
-        "## Prisma Schema",
-        "",
-        "```json",
-        `${JSON.stringify(ctx.state().prisma?.schemas, null, 2)}`,
-        "```",
-        "",
-        "## File Paths",
-        "",
-        Object.keys(templateFiles)
+      text: StringUtil.trim`
+        ## Generated TypeScript Code
+
+        \`\`\`json
+        ${JSON.stringify({
+          provider: {
+            location: auth.provider.location,
+            name: auth.provider.name,
+            content: auth.provider.content,
+          },
+          decorator: {
+            location: auth.decorator.location,
+            name: auth.decorator.name,
+            content: auth.decorator.content,
+          },
+          payload: {
+            location: auth.payload.location,
+            name: auth.payload.name,
+            content: auth.payload.content,
+          },
+        } satisfies Omit<AutoBeRealizeAuthorization, "role">)}
+        \`\`\`
+
+        ## Prisma Schema
+
+        \`\`\`json
+        ${JSON.stringify(ctx.state().prisma?.schemas)}
+        \`\`\`
+
+        ## File Paths
+
+        ${Object.keys(templateFiles)
           .map((path) => `- ${path}`)
-          .join("\n"),
-        "",
-        "## Compile Errors",
-        "",
-        "Fix the compilation error in the provided code.",
-        "",
-        "```json",
-        JSON.stringify(diagnostics, null, 2),
-        "```",
-        "## Component Naming Convention",
-        "",
-        "If the name of the component is not correct, please correct it.",
-        "",
-        "Please follow this naming convention for the authorization components:",
-        "",
-        `- Provider Name: ${auth.role.name.toLowerCase()}Authorize (e.g. ${auth.role.name.toLowerCase()}Authorize)`,
-        `- Decorator Name: ${auth.role.name.charAt(0).toUpperCase() + auth.role.name.slice(1).toLowerCase()}Auth (e.g. ${auth.role.name.charAt(0).toUpperCase() + auth.role.name.slice(1).toLowerCase()}Auth)`,
-        `- Payload Name: ${auth.role.name.charAt(0).toUpperCase() + auth.role.name.slice(1).toLowerCase()}Payload (e.g. ${auth.role.name.charAt(0).toUpperCase() + auth.role.name.slice(1).toLowerCase()}Payload)`,
-      ].join("\n"),
+          .join("\n")}
+
+        ## Compile Errors
+
+        Fix the compilation error in the provided code.
+
+        \`\`\`json
+        ${JSON.stringify(diagnostics)}
+        \`\`\`
+        
+        ## Component Naming Convention
+
+        If the name of the component is not correct, please correct it.
+
+        Please follow this naming convention for the authorization components:
+
+        - Provider Name: ${auth.role.name.toLowerCase()}Authorize (e.g. ${auth.role.name.toLowerCase()}Authorize)
+        - Decorator Name: ${auth.role.name.charAt(0).toUpperCase() + auth.role.name.slice(1).toLowerCase()}Auth (e.g. ${auth.role.name.charAt(0).toUpperCase() + auth.role.name.slice(1).toLowerCase()}Auth)
+        - Payload Name: ${auth.role.name.charAt(0).toUpperCase() + auth.role.name.slice(1).toLowerCase()}Payload (e.g. ${auth.role.name.charAt(0).toUpperCase() + auth.role.name.slice(1).toLowerCase()}Payload)
+      `,
     },
   ];
 };

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

@@ -3,8 +3,10 @@ import {
   AutoBeRealizeAuthorization,
   IAutoBeTypeScriptCompileResult,
 } from "@autobe/interface";
-import { v4 } from "uuid";
+import { StringUtil } from "@autobe/utils";
+import { v7 } from "uuid";
 
+import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
 import { AutoBeState } from "../../../context/AutoBeState";
 import { IAutoBeTestScenarioArtifacts } from "../../test/structures/IAutoBeTestScenarioArtifacts";
 import { IAutoBeRealizeScenarioApplication } from "../structures/IAutoBeRealizeScenarioApplication";
@@ -15,6 +17,7 @@ export function transformRealizeCorrectHistories(props: {
   scenario: IAutoBeRealizeScenarioApplication.IProps;
   artifacts: IAutoBeTestScenarioArtifacts;
   authorization: AutoBeRealizeAuthorization | null;
+  totalAuthorizations: AutoBeRealizeAuthorization[];
   code: string;
   diagnostic: IAutoBeTypeScriptCompileResult.IDiagnostic;
 }): Array<
@@ -23,24 +26,27 @@ export function transformRealizeCorrectHistories(props: {
   return [
     ...transformRealizeWriteHistories(props),
     {
-      id: v4(),
+      id: v7(),
       type: "assistantMessage",
-      text: [
-        `Below is the code you made before. It's also something to review.`,
-        "```typescript",
-        props.code,
-        "```",
-        `The code has a compilation error:`,
-        "```json",
-        JSON.stringify(props.diagnostic, null, 2),
-        "",
-      ].join("\n"),
+      text: StringUtil.trim`
+        Below is the code you made before. It's also something to review.
+
+        \`\`\`typescript
+        ${props.code}
+        \`\`\`
+
+        The code has a compilation error:
+        
+        \`\`\`json
+        ${JSON.stringify(props.diagnostic)}
+        \`\`\`
+      `,
       created_at: new Date().toISOString(),
     },
     {
-      id: v4(),
+      id: v7(),
       type: "systemMessage",
-      text: `You are an expert TypeScript developer. You will review the code and correct`,
+      text: AutoBeSystemPromptConstant.REALIZE_CORRECT,
       created_at: new Date().toISOString(),
     },
   ];