@autobe/agent 0.9.1 → 0.10.0

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/context/IAutoBeApplicationProps.ts

@@ -1,66 +1,4 @@
 export interface IAutoBeApplicationProps {
   /** The reason of the function call. */
   reason: string;
-
-  /**
-   * # Define prompts to translate user planning requirements into messages for internal agents
-   *
-   * This prompt defines how to convert a user's planning-oriented requirements
-   * into a structured message for an internal agent.
-   *
-   * All content the user provides must be included in the message. However, if
-   * some parts of the user's input are inappropriate or insufficient from a
-   * planning standpoint, you are allowed to add **supplementary remarks**—but
-   * only under strict rules.
-   *
-   * # Supplementary Remark Rules
-   *
-   * 1. **Definition** A supplementary remark is additional information that may
-   *    differ from the user's original intent. Because of this, **you must
-   *    clearly indicate that it is _not_ part of the user’s thinking**.
-   * 2. **When to Supplement**
-   *
-   * - If the user's input reveals a lack of technical understanding (e.g.,
-   *   suggesting "put all data into one table"), and the plan is not an MVP or
-   *   PoC, it's encouraged to make reasonable additions for a more scalable or
-   *   robust structure.
-   * - If there are clear gaps in the user's planning logic, you may supplement
-   *   the content to ensure completeness.
-   *
-   * 3. **When Not to Supplement**
-   *
-   * - If the user's input is vague or ambiguous, **do not assume or add extra
-   *   details**. Instead, it’s better to ask the user follow-up questions to
-   *   clarify their intent.
-   * - If the user has made no comment on design, **do not impose design-related
-   *   decisions** (e.g., colors, fonts, tone). However, you may state
-   *   explicitly that no design requirements were provided.
-   * - Generic advice like "UX should be good" can be omitted unless it adds
-   *   value, as such goals are assumed in all services.
-   *
-   * # Style Guidelines
-   *
-   * This prompt is delivered to the sub-agent, and several are created for
-   * parallel processing of the sub-agent. Additionally, there should be a guide
-   * to style, since sub-agents cannot create different styles of documents due
-   * to the disconnection of their conversations with each other.
-   *
-   * For example, there should be a hyperlink to the previous document, the next
-   * document, before or after the document, or there should be no more than N
-   * headings. The entire content of the document will have requirements, such
-   * as maintaining informal or formal language.
-   *
-   * The style guide should include conventions for Markdown formatting elements
-   * such as headings, lists, and tables. Additionally, it should define
-   * expectations regarding document length and overall composition. When
-   * describing structural guidelines, include a template to illustrate the
-   * recommended format.
-   *
-   * # Limiting the volume of a document
-   *
-   * However, do not go beyond the volume guide; each agent only needs to create
-   * one page because the agent receiving this document will be created as many
-   * as the number of pages.
-   */
-  userPlanningRequirements?: string;
 }

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

@@ -23,18 +23,11 @@ export const orchestrateAnalyze =
   async (
     props: IAutoBeApplicationProps,
   ): Promise<AutoBeAssistantMessageHistory | AutoBeAnalyzeHistory> => {
-    const userPlanningRequirements = props.userPlanningRequirements;
-    if (!userPlanningRequirements) {
-      throw new Error(
-        `Unable to prepare a proposal because there is no user requirement`,
-      );
-    }
-
     const step = ctx.state().analyze?.step ?? 0;
     const created_at = new Date().toISOString();
     ctx.dispatch({
       type: "analyzeStart",
-      reason: userPlanningRequirements,
+      reason: props.reason,
       step,
       created_at,
     });
@@ -67,14 +60,12 @@ export const orchestrateAnalyze =
     });
     enforceToolCall(agentica);
 
-    const determined = await agentica.conversate(
-      [
-        "Design a complete list of documents for that document",
-        "```md",
-        userPlanningRequirements,
-        "```",
-      ].join("\n"),
-    );
+    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") {
@@ -94,21 +85,6 @@ export const orchestrateAnalyze =
     }
 
     const described = determined.find((el) => el.type === "describe");
-    // const determinedOutput = Array.from(
-    //   new Set(
-    //     described
-    //       ? described.executes
-    //           .map((el) => {
-    //             if (el.protocol === "class") {
-    //               return el.arguments as unknown as IDeterminingInput;
-    //             }
-    //             return null;
-    //           })
-    //           .filter((el) => el !== null)
-    //       : [],
-    //   ),
-    // );
-
     const determinedOutput = described?.executes.find(
       (el) => el.protocol === "class" && typia.is<IDeterminingInput>(el.value),
     )?.value as IDeterminingInput;
@@ -152,10 +128,6 @@ export const orchestrateAnalyze =
             `Only write this document named '${filename}'.`,
             "Never write other documents.",
             "",
-            "# User Planning Requirements",
-            "```md",
-            JSON.stringify(userPlanningRequirements),
-            "```",
             "The reason why this document needs to be written is as follows.",
             `- reason: ${reason}`,
           ].join("\n"),
@@ -175,7 +147,7 @@ export const orchestrateAnalyze =
       const history: AutoBeAnalyzeHistory = {
         id: v4(),
         type: "analyze",
-        reason: userPlanningRequirements,
+        reason: props.reason,
         prefix,
         files: files,
         step,
@@ -291,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/orchestrateInterface.ts

@@ -22,6 +22,13 @@ export const orchestrateInterface =
   ): Promise<AutoBeAssistantMessageHistory | AutoBeInterfaceHistory> => {
     // ENDPOINTS
     const start: Date = new Date();
+    ctx.dispatch({
+      type: "interfaceStart",
+      created_at: start.toISOString(),
+      reason: props.reason,
+      step: ctx.state().analyze?.step ?? 0,
+    });
+
     const init: AutoBeAssistantMessageHistory | AutoBeInterfaceEndpointsEvent =
       await orchestrateInterfaceEndpoints(ctx);
     if (init.type === "assistantMessage") {

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 {