@autobe/agent 0.9.2 → 0.10.1

package/src/context/AutoBeTokenUsage.ts

@@ -1,3 +1,87 @@
 import { AgenticaTokenUsage } from "@agentica/core";
+import { IAutoBeTokenUsageJson } from "@autobe/interface";
 
-export import AutoBeTokenUsage = AgenticaTokenUsage;
+import { IAutoBeApplication } from "./IAutoBeApplication";
+
+export class AutoBeTokenUsage {
+  public readonly facade: AgenticaTokenUsage;
+  public readonly analyze: AgenticaTokenUsage;
+  public readonly prisma: AgenticaTokenUsage;
+  public readonly interface: AgenticaTokenUsage;
+  public readonly test: AgenticaTokenUsage;
+  public readonly realize: AgenticaTokenUsage;
+
+  public constructor(props?: IAutoBeTokenUsageJson) {
+    if (props === undefined) {
+      this.facade = new AgenticaTokenUsage();
+      this.analyze = new AgenticaTokenUsage();
+      this.prisma = new AgenticaTokenUsage();
+      this.interface = new AgenticaTokenUsage();
+      this.test = new AgenticaTokenUsage();
+      this.realize = new AgenticaTokenUsage();
+      return;
+    }
+
+    this.facade = new AgenticaTokenUsage(props.facade);
+    this.analyze = new AgenticaTokenUsage(props.analyze);
+    this.prisma = new AgenticaTokenUsage(props.prisma);
+    this.interface = new AgenticaTokenUsage(props.interface);
+    this.test = new AgenticaTokenUsage(props.test);
+    this.realize = new AgenticaTokenUsage(props.realize);
+  }
+
+  public record(
+    usage: AgenticaTokenUsage,
+    additionalStages: (keyof IAutoBeApplication)[] = [],
+  ) {
+    this.facade.increment(usage);
+    additionalStages.forEach((stage) => {
+      this[stage].increment(usage);
+    });
+  }
+
+  public increment(usage: AutoBeTokenUsage) {
+    AutoBeTokenUsage.keys().forEach((key) => {
+      this[key].increment(usage[key]);
+    });
+    return this;
+  }
+
+  public static plus(usageA: AutoBeTokenUsage, usageB: AutoBeTokenUsage) {
+    return new AutoBeTokenUsage({
+      facade: AgenticaTokenUsage.plus(usageA.facade, usageB.facade),
+      analyze: AgenticaTokenUsage.plus(usageA.analyze, usageB.analyze),
+      prisma: AgenticaTokenUsage.plus(usageA.prisma, usageB.prisma),
+      interface: AgenticaTokenUsage.plus(usageA.interface, usageB.interface),
+      test: AgenticaTokenUsage.plus(usageA.test, usageB.test),
+      realize: AgenticaTokenUsage.plus(usageA.realize, usageB.realize),
+    });
+  }
+
+  public toJSON(): IAutoBeTokenUsageJson {
+    return {
+      facade: this.facade.toJSON(),
+      analyze: this.analyze.toJSON(),
+      prisma: this.prisma.toJSON(),
+      interface: this.interface.toJSON(),
+      test: this.test.toJSON(),
+      realize: this.realize.toJSON(),
+    };
+  }
+
+  /** @internal */
+  private static keys(): ("facade" | keyof IAutoBeApplication)[] {
+    return ["facade", "analyze", "prisma", "interface", "test", "realize"];
+  }
+}
+
+// /** Type check statements */
+// 1 as unknown as AutoBeTokenUsage satisfies {
+//   [key in "facade" | keyof IAutoBeApplication]: AgenticaTokenUsage;
+// };
+
+// 1 as unknown as IAutoBeTokenUsageJson satisfies {
+//   [key in "facade" | keyof IAutoBeApplication]: IAutoBeInternalTokenUsageJson;
+// };
+
+// 1 as unknown as IAutoBeInternalTokenUsageJson satisfies IAgenticaTokenUsageJson;

package/src/factory/createAutoBeApplication.ts

@@ -1,4 +1,5 @@
 import { IAgenticaController } from "@agentica/core";
+import { StringUtil } from "@autobe/utils";
 import { ILlmApplication, ILlmSchema } from "@samchon/openapi";
 import typia from "typia";
 
@@ -7,10 +8,9 @@ import { IAutoBeApplication } from "../context/IAutoBeApplication";
 import { assertSchemaModel } from "../context/assertSchemaModel";
 import { orchestrateAnalyze } from "../orchestrate/analyze/orchestrateAnalyze";
 import { orchestrateInterface } from "../orchestrate/interface/orchestrateInterface";
-import { orchestrateRealize } from "../orchestrate/orchestrateRealize";
 import { orchestratePrisma } from "../orchestrate/prisma/orchestratePrisma";
+import { orchestrateRealize } from "../orchestrate/realize/orchestrateRealize";
 import { orchestrateTest } from "../orchestrate/test/orchestrateTest";
-import { StringUtil } from "../utils/StringUtil";
 
 export const createAutoBeController = <Model extends ILlmSchema.Model>(props: {
   model: Model;
@@ -128,5 +128,4 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<IAutoBeApplication, "3.0">(),
 };

package/src/orchestrate/analyze/AutoBeAnalyzeAgent.ts

@@ -49,7 +49,6 @@ export class AutoBeAnalyzeAgent<Model extends ILlmSchema.Model> {
             describe: null,
           },
         },
-        tokenUsage: ctx.usage(),
         histories: [
           {
             id: v4(),
@@ -99,7 +98,12 @@ export class AutoBeAnalyzeAgent<Model extends ILlmSchema.Model> {
       return "Abort due to excess retry count";
     }
 
-    const response = await this.createAnalyzeAgent().conversate(content);
+    const agent = this.createAnalyzeAgent();
+    const response = await agent.conversate(content);
+
+    const tokenUsage = agent.getTokenUsage();
+    this.ctx.usage().record(tokenUsage, ["analyze"]);
+
     const lastMessage = response[response.length - 1]!;
 
     if ("text" in lastMessage) {
@@ -134,6 +138,8 @@ export class AutoBeAnalyzeAgent<Model extends ILlmSchema.Model> {
       const filenames = Object.keys(this.fileMap).join(",");
       const command = `Please proceed with the review of these files only.: ${filenames}`;
       const response = await reviewer.conversate(command);
+      this.ctx.usage().record(reviewer.getTokenUsage(), ["analyze"]);
+
       const review = response.find((el) => el.type === "assistantMessage");
 
       if (review) {
@@ -210,5 +216,4 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<AutoBeAnalyzeFileSystem, "3.0">(),
 };

package/src/orchestrate/analyze/AutoBeAnalyzeReviewer.ts

@@ -43,7 +43,6 @@ export const AutoBeAnalyzeReviewer = <Model extends ILlmSchema.Model>(
         ].join("\n"),
       },
     ],
-    tokenUsage: ctx.usage(),
   });
 
   return agent;

package/src/orchestrate/analyze/orchestrateAnalyze.ts

@@ -57,13 +57,15 @@ export const orchestrateAnalyze =
             (el) => el.type === "assistantMessage" || el.type === "userMessage",
           ),
       ],
-      tokenUsage: ctx.usage(),
     });
     enforceToolCall(agentica);
 
-    const determined = await agentica.conversate(
-      "Design a complete list of documents for that document",
-    );
+    const determined = await agentica
+      .conversate("Design a complete list of documents for that document")
+      .finally(() => {
+        const tokenUsage = agentica.getTokenUsage();
+        ctx.usage().record(tokenUsage, ["analyze"]);
+      });
 
     const lastMessage = determined[determined.length - 1]!;
     if (lastMessage.type === "assistantMessage") {
@@ -261,5 +263,4 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<DeterminingFiles, "3.0">(),
 };

package/src/orchestrate/facade/transformFacadeStateMessage.ts

@@ -1,6 +1,7 @@
+import { StringUtil } from "@autobe/utils";
+
 import { AutoBeSystemPromptConstant } from "../../constants/AutoBeSystemPromptConstant";
 import { AutoBeState } from "../../context/AutoBeState";
-import { StringUtil } from "../../utils/StringUtil";
 
 export function transformFacadeStateMessage(state: AutoBeState): string {
   const currentState: ICurrentState = getCurrentState(state);

package/src/orchestrate/index.ts

@@ -1,5 +1,5 @@
 export { orchestrateAnalyze as analyze } from "./analyze/orchestrateAnalyze";
 export { orchestrateInterface as interface } from "./interface/orchestrateInterface";
-export { orchestrateRealize as realize } from "./orchestrateRealize";
-export { orchestrateTest as test } from "./test/orchestrateTest";
 export { orchestratePrisma as prisma } from "./prisma/orchestratePrisma";
+export { orchestrateRealize as realize } from "./realize/orchestrateRealize";
+export { orchestrateTest as test } from "./test/orchestrateTest";

package/src/orchestrate/interface/orchestrateInterfaceComplement.ts

@@ -75,7 +75,6 @@ async function step<Model extends ILlmSchema.Model>(
         ].join("\n"),
       },
     ],
-    tokenUsage: ctx.usage(),
     controllers: [
       createApplication({
         model: ctx.model,
@@ -96,7 +95,10 @@ async function step<Model extends ILlmSchema.Model>(
   });
   enforceToolCall(agentica);
 
-  await agentica.conversate("Fill missing schema types please");
+  await agentica.conversate("Fill missing schema types please").finally(() => {
+    const tokenUsage = agentica.getTokenUsage();
+    ctx.usage().record(tokenUsage, ["interface"]);
+  });
   if (pointer.value === null) {
     // unreachable
     throw new Error(
@@ -186,7 +188,6 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<IApplication, "3.0">(),
 };
 
 interface IApplication {

package/src/orchestrate/interface/orchestrateInterfaceComponents.ts

@@ -127,7 +127,6 @@ async function process<Model extends ILlmSchema.Model>(
         ].join("\n"),
       },
     ],
-    tokenUsage: ctx.usage(),
     controllers: [
       createApplication({
         model: ctx.model,
@@ -145,27 +144,32 @@ async function process<Model extends ILlmSchema.Model>(
   enforceToolCall(agentica);
 
   const already: string[] = Object.keys(oldbie.schemas);
-  await agentica.conversate(
-    [
-      "Make type components please.",
-      "",
-      "Here is the list of request/response bodies' type names from",
-      "OpenAPI operations. Make type components of them. If more object",
-      "types are required during making the components, please make them",
-      "too.",
-      "",
-      ...Array.from(remained).map((k) => `- \`${k}\``),
-      ...(already.length !== 0
-        ? [
-            "",
-            "> By the way, here is the list of components schemas what you've",
-            "> already made. So, you don't need to make them again.",
-            ">",
-            ...already.map((k) => `> - \`${k}\``),
-          ]
-        : []),
-    ].join("\n"),
-  );
+  await agentica
+    .conversate(
+      [
+        "Make type components please.",
+        "",
+        "Here is the list of request/response bodies' type names from",
+        "OpenAPI operations. Make type components of them. If more object",
+        "types are required during making the components, please make them",
+        "too.",
+        "",
+        ...Array.from(remained).map((k) => `- \`${k}\``),
+        ...(already.length !== 0
+          ? [
+              "",
+              "> By the way, here is the list of components schemas what you've",
+              "> already made. So, you don't need to make them again.",
+              ">",
+              ...already.map((k) => `> - \`${k}\``),
+            ]
+          : []),
+      ].join("\n"),
+    )
+    .finally(() => {
+      const tokenUsage = agentica.getTokenUsage();
+      ctx.usage().record(tokenUsage, ["interface"]);
+    });
   if (pointer.value === null) {
     // never be happened
     throw new Error("Failed to create components.");
@@ -212,7 +216,6 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<IApplication, "3.0">(),
 };
 
 interface IApplication {

package/src/orchestrate/interface/orchestrateInterfaceEndpoints.ts

@@ -43,7 +43,6 @@ export async function orchestrateInterfaceEndpoints<
       ctx.state(),
       AutoBeSystemPromptConstant.INTERFACE_ENDPOINT,
     ),
-    tokenUsage: ctx.usage(),
     controllers: [
       createApplication({
         model: ctx.model,
@@ -55,8 +54,12 @@ export async function orchestrateInterfaceEndpoints<
     ],
   });
 
-  const histories: MicroAgenticaHistory<Model>[] =
-    await agentica.conversate(content);
+  const histories: MicroAgenticaHistory<Model>[] = await agentica
+    .conversate(content)
+    .finally(() => {
+      const tokenUsage = agentica.getTokenUsage();
+      ctx.usage().record(tokenUsage, ["interface"]);
+    });
   if (histories.at(-1)?.type === "assistantMessage")
     return {
       ...(histories.at(-1)! as AgenticaAssistantMessageHistory),
@@ -114,7 +117,6 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<IApplication, "3.0">(),
 };
 
 interface IApplication {

package/src/orchestrate/interface/orchestrateInterfaceOperations.ts

@@ -101,7 +101,6 @@ async function process<Model extends ILlmSchema.Model>(
       ctx.state(),
       AutoBeSystemPromptConstant.INTERFACE_ENDPOINT,
     ),
-    tokenUsage: ctx.usage(),
     controllers: [
       createApplication({
         model: ctx.model,
@@ -113,15 +112,20 @@ async function process<Model extends ILlmSchema.Model>(
     ],
   });
   enforceToolCall(agentica);
-  await agentica.conversate(
-    [
-      "Make API operations for below endpoints:",
-      "",
-      "```json",
-      JSON.stringify(Array.from(endpoints), null, 2),
-      "```",
-    ].join("\n"),
-  );
+  await agentica
+    .conversate(
+      [
+        "Make API operations for below endpoints:",
+        "",
+        "```json",
+        JSON.stringify(Array.from(endpoints), null, 2),
+        "```",
+      ].join("\n"),
+    )
+    .finally(() => {
+      const tokenUsage = agentica.getTokenUsage();
+      ctx.usage().record(tokenUsage, ["interface"]);
+    });
   if (pointer.value === null) throw new Error("Failed to create operations."); // never be happened
   return pointer.value;
 }
@@ -185,7 +189,6 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<IApplication, "3.0">(),
 };
 
 interface IApplication {

package/src/orchestrate/interface/transformInterfaceHistories.ts

@@ -124,6 +124,8 @@ export const transformInterfaceHistories = (
       type: "assistantMessage",
       text: [
         "Database schema and entity relationship diagrams are ready.",
+        "You should also look at this and consider logic including membership/login and token issuance.",
+        "You can use table's name to define role in operations.",
         "",
         "## Prisma DB Schema",
         "```json",

package/src/orchestrate/prisma/orchestratePrismaComponent.ts

@@ -33,9 +33,11 @@ export async function orchestratePrismaComponents<
     vendor: ctx.vendor,
     config: {
       ...(ctx.config ?? {}),
+      executor: {
+        describe: null,
+      },
     },
     histories: transformPrismaComponentsHistories(ctx.state(), prefix),
-    tokenUsage: ctx.usage(),
     controllers: [
       createApplication({
         model: ctx.model,
@@ -49,8 +51,12 @@ export async function orchestratePrismaComponents<
     ],
   });
 
-  const histories: MicroAgenticaHistory<Model>[] =
-    await agentica.conversate(content);
+  const histories: MicroAgenticaHistory<Model>[] = await agentica
+    .conversate(content)
+    .finally(() => {
+      const tokenUsage = agentica.getTokenUsage();
+      ctx.usage().record(tokenUsage, ["prisma"]);
+    });
   if (histories.at(-1)?.type === "assistantMessage")
     return {
       ...(histories.at(-1)! as AgenticaAssistantMessageHistory),
@@ -105,7 +111,6 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<IApplication, "3.0">(),
 };
 
 interface IApplication {
@@ -154,7 +159,7 @@ interface IExtractComponentsProps {
    * {
    *   {
    *     filename: "schema-01-users.prisma",
-   *     tables: ["user", "user_profile", "user_settings"]
+   *     tables: ["user", "admin", "moderator", "user_profile", "user_settings"]
    *   },
    *   {
    *     filename: "schema-02-articles.prisma",

package/src/orchestrate/prisma/orchestratePrismaCorrect.ts

@@ -60,9 +60,11 @@ async function step<Model extends ILlmSchema.Model>(
     vendor: ctx.vendor,
     config: {
       ...(ctx.config ?? {}),
+      executor: {
+        describe: null,
+      },
     },
     histories: transformPrismaCorrectHistories(result),
-    tokenUsage: ctx.usage(),
     controllers: [
       createApplication({
         model: ctx.model,
@@ -75,9 +77,14 @@ async function step<Model extends ILlmSchema.Model>(
   enforceToolCall(agentica);
 
   // REQUEST CORRECTION
-  await agentica.conversate(
-    "Resolve the compilation errors in the provided Prisma schema files.",
-  );
+  await agentica
+    .conversate(
+      "Resolve the compilation errors in the provided Prisma schema files.",
+    )
+    .finally(() => {
+      const tokenUsage = agentica.getTokenUsage();
+      ctx.usage().record(tokenUsage, ["prisma"]);
+    });
   if (pointer.value === null) {
     console.error(
       "Unreachable error: PrismaCompilerAgent.pointer.value is null",
@@ -147,7 +154,6 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<IApplication, "3.0">(),
 };
 
 interface IApplication {

package/src/orchestrate/prisma/orchestratePrismaSchema.ts

@@ -8,6 +8,7 @@ import typia from "typia";
 import { AutoBeContext } from "../../context/AutoBeContext";
 import { assertSchemaModel } from "../../context/assertSchemaModel";
 import { enforceToolCall } from "../../utils/enforceToolCall";
+import { forceRetry } from "../../utils/forceRetry";
 import { transformPrismaSchemaHistories } from "./transformPrismaSchemaHistories";
 
 export async function orchestratePrismaSchemas<Model extends ILlmSchema.Model>(
@@ -23,11 +24,13 @@ export async function orchestratePrismaSchemas<Model extends ILlmSchema.Model>(
   let i: number = 0;
   return await Promise.all(
     components.map(async (c) => {
-      const result: IMakePrismaSchemaFileProps = await process(ctx, {
-        filename: c.filename,
-        tables: c.tables,
-        entireTables,
-      });
+      const result: IMakePrismaSchemaFileProps = await forceRetry(() =>
+        process(ctx, {
+          filename: c.filename,
+          tables: c.tables,
+          entireTables,
+        }),
+      );
       const event: AutoBePrismaSchemasEvent = {
         type: "prismaSchemas",
         created_at: start.toISOString(),
@@ -58,9 +61,11 @@ async function process<Model extends ILlmSchema.Model>(
     vendor: ctx.vendor,
     config: {
       ...(ctx.config ?? {}),
+      executor: {
+        describe: null,
+      },
     },
     histories: transformPrismaSchemaHistories(ctx.state().analyze!, component),
-    tokenUsage: ctx.usage(),
     controllers: [
       createApplication({
         model: ctx.model,
@@ -78,7 +83,11 @@ async function process<Model extends ILlmSchema.Model>(
     ],
   });
   enforceToolCall(agentica);
-  await agentica.conversate("Make prisma schema file please");
+  await agentica.conversate("Make prisma schema file please").finally(() => {
+    const tokenUsage = agentica.getTokenUsage();
+    ctx.usage().record(tokenUsage, ["prisma"]);
+  });
+
   if (pointer.value === null)
     throw new Error("Unreachable code: Prisma Schema not generated");
   return pointer.value;
@@ -119,7 +128,6 @@ const collection = {
   llama: claude,
   deepseek: claude,
   "3.1": claude,
-  "3.0": typia.llm.application<IApplication, "3.0">(),
 };
 
 interface IApplication {

package/src/orchestrate/prisma/transformPrismaComponentsHistories.ts

@@ -68,6 +68,15 @@ export const transformPrismaComponentsHistories = (
         "For example:",
         "",
         "* `mv_shopping_daily_stats`",
+        "",
+        "## User Role Handling",
+        "",
+        "If the Requirement Analysis Report contains User Role information, **do not normalize** user roles into a single table.",
+        "Instead, create separate tables for each distinct role mentioned in the requirements.",
+        "",
+        "For example, if the requirements mention User, Admin, and Moderator roles:",
+        "",
+        "* Create separate tables: `User`, `Admin`, `Moderator` (or with prefix: `shopping_user`, `shopping_admin`, `shopping_moderator`)",
       ].join("\n"),
     },
   ];

package/src/orchestrate/prisma/transformPrismaSchemaHistories.ts

@@ -59,6 +59,14 @@ export const transformPrismaSchemaHistories = (
         ...component.tables.map((table) => `    - ${table}`),
         `  - Entire tables you can reference:`,
         ...component.entireTables.map((table) => `    - ${table}`),
+        "",
+        "## User Role Rules",
+        "",
+        "**CRITICAL**: For User Role tables (User, Admin, Moderator, etc.):",
+        "1. **Separate tables** - Each role = independent table (User, Admin, Moderator tables)",
+        "2. **No `role` field** - Never include role enum fields in any table",
+        "",
+        "You can use TPT (Table Per Type) pattern with FK relationships if needed, or completely separate tables.",
       ].join("\n"),
     },
   ];