@autobe/agent 0.17.1 → 0.18.0

package/lib/orchestrate/test/structures/IAutoBeTestWriteApplication.d.ts

@@ -1,3 +1,4 @@
+import { SnakePattern } from "@autobe/interface";
 export interface IAutoBeTestWriteApplication {
     /**
      * Main entry point for AI Function Call - generates complete E2E test code.
@@ -37,7 +38,7 @@ export declare namespace IAutoBeTestWriteApplication {
          * Workflow: Scenario analysis → Domain identification → Test organization
          * structure
          */
-        domain: string;
+        domain: string & SnakePattern;
         /**
          * Step 3: Initial TypeScript E2E test code implementation.
          *

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autobe/agent",
-  "version": "0.17.1",
+  "version": "0.18.0",
   "description": "AI backend server code generator",
   "main": "lib/index.js",
   "author": "Wrtn Technologies",
@@ -20,15 +20,15 @@
     "README.md"
   ],
   "dependencies": {
-    "@agentica/core": "^0.30.8",
+    "@agentica/core": "^0.31.1",
     "@prisma/internals": "^6.7.0",
-    "@samchon/openapi": "^4.6.0",
-    "openai": "^5.2.0",
+    "@samchon/openapi": "^4.7.1",
+    "openai": "^5.12.2",
     "tstl": "^3.0.0",
-    "typia": "^9.6.1",
+    "typia": "^9.7.1",
     "uuid": "^11.1.0",
-    "@autobe/utils": "^0.17.1",
-    "@autobe/interface": "^0.17.1"
+    "@autobe/interface": "^0.18.0",
+    "@autobe/utils": "^0.18.0"
   },
   "devDependencies": {
     "@rollup/plugin-json": "^6.1.0",
@@ -40,8 +40,8 @@
     "ts-node": "^10.9.2",
     "ts-patch": "^3.3.0",
     "typescript": "~5.9.2",
-    "@autobe/compiler": "^0.17.1",
-    "@autobe/filesystem": "^0.17.1"
+    "@autobe/compiler": "^0.18.0",
+    "@autobe/filesystem": "^0.18.0"
   },
   "keywords": [
     "ai",

package/src/AutoBeAgent.ts

@@ -1,4 +1,8 @@
-import { IAgenticaVendor, MicroAgentica } from "@agentica/core";
+import {
+  IAgenticaTokenUsageJson,
+  IAgenticaVendor,
+  MicroAgentica,
+} from "@agentica/core";
 import {
   AutoBeAssistantMessageHistory,
   AutoBeHistory,
@@ -165,6 +169,34 @@ export class AutoBeAgent<Model extends ILlmSchema.Model>
         )
         .filter((h) => h !== null),
     );
+
+    // TRACE FACADE TOKEN USAGE
+    let previous: IAgenticaTokenUsageJson.IComponent = this.agentica_
+      .getTokenUsage()
+      .toJSON().aggregate;
+    const increment = () => {
+      const current: IAgenticaTokenUsageJson.IComponent = this.agentica_
+        .getTokenUsage()
+        .toJSON().aggregate;
+      this.usage_.facade.increment({
+        total: current.total - previous.total,
+        input: {
+          total: current.input.total - previous.input.total,
+          cached: current.input.cached - previous.input.cached,
+        },
+        output: {
+          total: current.output.total - previous.output.total,
+          reasoning: current.output.reasoning - previous.output.reasoning,
+          accepted_prediction:
+            current.output.accepted_prediction -
+            previous.output.accepted_prediction,
+          rejected_prediction:
+            current.output.rejected_prediction -
+            previous.output.rejected_prediction,
+        },
+      });
+      previous = current;
+    };
     this.agentica_.on("assistantMessage", async (message) => {
       const start = new Date();
       const history: AutoBeAssistantMessageHistory = {
@@ -174,6 +206,7 @@ export class AutoBeAgent<Model extends ILlmSchema.Model>
         created_at: start.toISOString(),
         completed_at: new Date().toISOString(),
       };
+      increment();
       this.histories_.push(history);
       this.dispatch({
         type: "assistantMessage",
@@ -181,6 +214,9 @@ export class AutoBeAgent<Model extends ILlmSchema.Model>
         created_at: history.created_at,
       }).catch(() => {});
     });
+    this.agentica_.on("call", async () => {
+      increment();
+    });
     this.agentica_.on("request", (e) => {
       if (e.body.parallel_tool_calls !== undefined)
         delete e.body.parallel_tool_calls;
@@ -222,8 +258,6 @@ export class AutoBeAgent<Model extends ILlmSchema.Model>
     this.dispatch(userMessageHistory).catch(() => {});
 
     await this.agentica_.conversate(content);
-
-    this.usage_.facade.increment(this.agentica_.getTokenUsage().aggregate);
     return this.histories_.slice(index);
   }
 

package/src/AutoBeMockAgent.ts

@@ -88,7 +88,24 @@ export class AutoBeMockAgent extends AutoBeAgentBase implements IAutoBeAgent {
     const take = async (
       type: "analyze" | "prisma" | "interface" | "test" | "realize",
     ): Promise<void> => {
-      for (const s of this.getEventSnapshots(type)) {
+      const snapshots: AutoBeEventSnapshot[] | null =
+        this.getEventSnapshots(type);
+      if (snapshots === null) {
+        this.histories_.push(userMessage);
+        this.histories_.push({
+          id: v4(),
+          type: "assistantMessage",
+          text: [
+            "The histories are prepared until current state.",
+            "",
+            "Thanks for using AutoBE!",
+          ].join("\n"),
+          created_at: new Date().toISOString(),
+          completed_at: new Date().toISOString(),
+        });
+        return;
+      }
+      for (const s of snapshots) {
         const time: number = sleepMap[s.event.type] ?? 500;
         await sleep_for(randint(time * 0.2, time * 1.8));
         void this.dispatch(s.event).catch(() => {});
@@ -117,8 +134,8 @@ export class AutoBeMockAgent extends AutoBeAgentBase implements IAutoBeAgent {
 
   private getEventSnapshots(
     state: "analyze" | "prisma" | "interface" | "test" | "realize",
-  ): AutoBeEventSnapshot[] {
-    return this.props_.preset[state];
+  ): AutoBeEventSnapshot[] | null {
+    return this.props_.preset[state] ?? null;
   }
 }
 export namespace AutoBeMockAgent {
@@ -130,17 +147,20 @@ export namespace AutoBeMockAgent {
   }
   export interface IPreset {
     histories: AutoBeHistory[];
-    analyze: AutoBeEventSnapshot[];
-    prisma: AutoBeEventSnapshot[];
-    interface: AutoBeEventSnapshot[];
-    test: AutoBeEventSnapshot[];
-    realize: AutoBeEventSnapshot[];
+    analyze?: AutoBeEventSnapshot[] | null;
+    prisma?: AutoBeEventSnapshot[] | null;
+    interface?: AutoBeEventSnapshot[] | null;
+    test?: AutoBeEventSnapshot[] | null;
+    realize?: AutoBeEventSnapshot[] | null;
   }
 }
 
-const sleepMap: Partial<Record<AutoBeEvent.Type, number>> = {
+const sleepMap: Record<AutoBeEvent.Type, number> = {
+  userMessage: 1_000,
+  assistantMessage: 1_000,
   // ANALYZE
   analyzeStart: 1_000,
+  analyzeScenario: 1_000,
   analyzeWrite: 500,
   analyzeReview: 300,
   analyzeComplete: 1_000,
@@ -148,20 +168,25 @@ const sleepMap: Partial<Record<AutoBeEvent.Type, number>> = {
   prismaStart: 1_000,
   prismaComponents: 1_000,
   prismaSchemas: 500,
+  prismaInsufficient: 1_000,
+  prismaReview: 500,
   prismaValidate: 2_000,
   prismaCorrect: 500,
-  prismaInsufficient: 1_000,
   prismaComplete: 1_000,
   // INTERFACE
   interfaceStart: 1_000,
+  interfaceGroups: 1_000,
   interfaceEndpoints: 1_000,
   interfaceOperations: 400,
+  interfaceOperationsReview: 400,
+  interfaceAuthorization: 400,
   interfaceSchemas: 400,
+  interfaceSchemasReview: 400,
   interfaceComplement: 2_000,
   interfaceComplete: 1_000,
   // TEST
   testStart: 1_000,
-  testScenario: 1_000,
+  testScenarios: 40,
   testWrite: 40,
   testValidate: 100,
   testCorrect: 100,
@@ -175,5 +200,10 @@ const sleepMap: Partial<Record<AutoBeEvent.Type, number>> = {
   realizeAuthorizationStart: 1_000,
   realizeAuthorizationWrite: 200,
   realizeAuthorizationValidate: 200,
+  realizeAuthorizationCorrect: 200,
   realizeAuthorizationComplete: 1_000,
+  realizeTestStart: 1_000,
+  realizeTestReset: 2_500,
+  realizeTestOperation: 400,
+  realizeTestComplete: 1_000,
 };