@autobe/agent 0.16.1 → 0.17.1

package/lib/context/AutoBeTokenUsage.d.ts

@@ -1,16 +1,173 @@
-import { AgenticaTokenUsage } from "@agentica/core";
 import { IAutoBeTokenUsageJson } from "@autobe/interface";
+import { AutoBeTokenUsageComponent } from "./AutoBeTokenUsageComponent";
 import { IAutoBeApplication } from "./IAutoBeApplication";
-export declare class AutoBeTokenUsage {
-    readonly facade: AgenticaTokenUsage;
-    readonly analyze: AgenticaTokenUsage;
-    readonly prisma: AgenticaTokenUsage;
-    readonly interface: AgenticaTokenUsage;
-    readonly test: AgenticaTokenUsage;
-    readonly realize: AgenticaTokenUsage;
-    constructor(props?: IAutoBeTokenUsageJson);
-    record(usage: AgenticaTokenUsage, additionalStages?: (keyof IAutoBeApplication)[]): void;
+/**
+ * Comprehensive token usage tracker for the AutoBe vibe coding system.
+ *
+ * This class provides centralized tracking and management of token consumption
+ * across all AI agents in the automated development pipeline. It captures
+ * detailed token usage statistics for each processing phase - from initial
+ * requirements analysis through final implementation - enabling cost
+ * monitoring, performance optimization, and resource utilization analysis.
+ *
+ * The token usage data includes both input tokens (with cache efficiency) and
+ * output tokens (with generation type breakdowns), providing insights into AI
+ * processing efficiency and helping optimize the balance between computational
+ * costs and output quality.
+ *
+ * @author SunRabbit123
+ */
+export declare class AutoBeTokenUsage implements IAutoBeTokenUsageJson {
+    /**
+     * Token usage for the facade agent orchestrating the entire pipeline.
+     *
+     * Tracks tokens consumed by the initial agent that coordinates and manages
+     * the overall vibe coding process, including request parsing and response
+     * orchestration.
+     */
+    readonly facade: AutoBeTokenUsageComponent;
+    /**
+     * Token usage for the requirements analysis agent.
+     *
+     * Captures tokens used during the analysis phase where user requirements are
+     * processed, understood, and transformed into structured specifications for
+     * subsequent development phases.
+     */
+    readonly analyze: AutoBeTokenUsageComponent;
+    /**
+     * Token usage for the Prisma database schema generation agent.
+     *
+     * Records tokens consumed while designing and generating database schemas,
+     * including entity relationships, field definitions, and database-specific
+     * optimizations.
+     */
+    readonly prisma: AutoBeTokenUsageComponent;
+    /**
+     * Token usage for the API interface specification agent.
+     *
+     * Tracks tokens used in creating OpenAPI/Swagger specifications, defining
+     * endpoints, request/response structures, and API documentation.
+     */
+    readonly interface: AutoBeTokenUsageComponent;
+    /**
+     * Token usage for the test code generation agent.
+     *
+     * Monitors tokens consumed during automated test creation, including scenario
+     * planning, test case generation, and end-to-end test implementation.
+     */
+    readonly test: AutoBeTokenUsageComponent;
+    /**
+     * Token usage for the implementation realization agent.
+     *
+     * Captures tokens used in the final implementation phase where actual
+     * business logic, controllers, services, and integration code are generated.
+     */
+    readonly realize: AutoBeTokenUsageComponent;
+    /**
+     * Aggregated token usage statistics across all agents.
+     *
+     * Provides a unified view of token consumption by combining data from all
+     * processing phases in the vibe coding pipeline. This computed property
+     * dynamically calculates the sum of all agent components (facade, analyze,
+     * prisma, interface, test, realize) whenever accessed, ensuring the aggregate
+     * always reflects the current state of token usage.
+     *
+     * The aggregation performs element-wise addition across all token metrics,
+     * including total counts, input breakdowns with cache statistics, and output
+     * categorizations by generation type. This comprehensive view enables overall
+     * cost assessment and resource utilization analysis for the entire automated
+     * development session.
+     */
+    get aggregate(): IAutoBeTokenUsageJson.IComponent;
+    /**
+     * Default Constructor.
+     *
+     * Creates a new token usage tracker with all agent phases initialized to
+     * empty components. Each component starts with zero values for all token
+     * counters, providing a clean slate for tracking token consumption from the
+     * beginning of a vibe coding session.
+     */
+    constructor();
+    /**
+     * Binding Constructor.
+     *
+     * Creates a new instance that shares component references with the provided
+     * AutoBeTokenUsage instance. This establishes a bound relationship where both
+     * instances point to the same component objects. Any modifications to token
+     * counts through either instance will be reflected in both, enabling shared
+     * tracking across different contexts.
+     *
+     * @param bind Existing AutoBeTokenUsage instance to bind to
+     */
+    constructor(bind: AutoBeTokenUsage);
+    /**
+     * Initializer Constructor.
+     *
+     * Reconstructs a token usage tracker from serialized JSON data. Creates new
+     * AutoBeTokenUsageComponent instances for each agent phase, initializing them
+     * with the corresponding values from the JSON structure. This enables
+     * restoration of token usage state from persisted data or transmission
+     * between different parts of the system.
+     *
+     * @param props Token usage data in JSON format
+     */
+    constructor(props?: Omit<IAutoBeTokenUsageJson, "aggregate">);
+    /**
+     * Serialize token usage data to JSON format.
+     *
+     * Converts the internal token usage representation to the standardized
+     * IAutoBeTokenUsageJson format, suitable for persistence, transmission, or
+     * external analysis. The serialized data maintains the complete structure
+     * including all agent phases and detailed token breakdowns.
+     *
+     * @returns JSON representation of token usage statistics
+     */
+    toJSON(): IAutoBeTokenUsageJson;
+    /**
+     * Record token usage for specific processing stages.
+     *
+     * Updates token consumption statistics for one or more agent phases based on
+     * the provided usage data. This method allows selective recording of token
+     * usage for specific stages, enabling fine-grained tracking during
+     * multi-phase processing or when certain agents are invoked multiple times.
+     *
+     * @example
+     *   ```ts
+     *   tokenUsage.record(
+     *     { total: 150, input: { total: 100, cached: 20 }, output: { total: 50, ... } },
+     *     ['analyze', 'prisma']
+     *   );
+     *   ```;
+     *
+     * @param usage - Token usage component data to record
+     * @param additionalStages - Array of stage names to update with the usage
+     *   data
+     */
+    record(usage: IAutoBeTokenUsageJson.IComponent, additionalStages?: (keyof IAutoBeApplication)[]): void;
+    /**
+     * Increment current token usage with data from another instance.
+     *
+     * Adds token usage statistics from another AutoBeTokenUsage instance to this
+     * one, updating all agent phases simultaneously. This method is useful for
+     * combining token usage from parallel processing, multiple runs, or when
+     * aggregating statistics from distributed agent executions.
+     *
+     * @param usage - AutoBeTokenUsage instance to add to current statistics
+     * @returns This instance for method chaining
+     */
     increment(usage: AutoBeTokenUsage): this;
+    /**
+     * Create a new instance combining token usage from two sources.
+     *
+     * Performs element-wise addition of token usage statistics from two
+     * AutoBeTokenUsage instances, creating a new instance with the combined
+     * totals. This static method is useful for aggregating token usage across
+     * multiple vibe coding sessions or when merging statistics from different
+     * execution contexts.
+     *
+     * @param usageA - First token usage instance
+     * @param usageB - Second token usage instance
+     * @returns New instance with combined token usage statistics
+     */
     static plus(usageA: AutoBeTokenUsage, usageB: AutoBeTokenUsage): AutoBeTokenUsage;
-    toJSON(): IAutoBeTokenUsageJson;
 }

package/lib/context/AutoBeTokenUsage.js

@@ -1,60 +1,180 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AutoBeTokenUsage = void 0;
-const core_1 = require("@agentica/core");
+const AutoBeTokenUsageComponent_1 = require("./AutoBeTokenUsageComponent");
+/**
+ * Comprehensive token usage tracker for the AutoBe vibe coding system.
+ *
+ * This class provides centralized tracking and management of token consumption
+ * across all AI agents in the automated development pipeline. It captures
+ * detailed token usage statistics for each processing phase - from initial
+ * requirements analysis through final implementation - enabling cost
+ * monitoring, performance optimization, and resource utilization analysis.
+ *
+ * The token usage data includes both input tokens (with cache efficiency) and
+ * output tokens (with generation type breakdowns), providing insights into AI
+ * processing efficiency and helping optimize the balance between computational
+ * costs and output quality.
+ *
+ * @author SunRabbit123
+ */
 class AutoBeTokenUsage {
+    /**
+     * Aggregated token usage statistics across all agents.
+     *
+     * Provides a unified view of token consumption by combining data from all
+     * processing phases in the vibe coding pipeline. This computed property
+     * dynamically calculates the sum of all agent components (facade, analyze,
+     * prisma, interface, test, realize) whenever accessed, ensuring the aggregate
+     * always reflects the current state of token usage.
+     *
+     * The aggregation performs element-wise addition across all token metrics,
+     * including total counts, input breakdowns with cache statistics, and output
+     * categorizations by generation type. This comprehensive view enables overall
+     * cost assessment and resource utilization analysis for the entire automated
+     * development session.
+     */
+    get aggregate() {
+        return AutoBeTokenUsage.keys()
+            .reduce((acc, cur) => AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent.plus(acc, this[cur]), new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent())
+            .toJSON();
+    }
     constructor(props) {
         if (props === undefined) {
-            this.facade = new core_1.AgenticaTokenUsage();
-            this.analyze = new core_1.AgenticaTokenUsage();
-            this.prisma = new core_1.AgenticaTokenUsage();
-            this.interface = new core_1.AgenticaTokenUsage();
-            this.test = new core_1.AgenticaTokenUsage();
-            this.realize = new core_1.AgenticaTokenUsage();
+            this.facade = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent();
+            this.analyze = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent();
+            this.prisma = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent();
+            this.interface = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent();
+            this.test = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent();
+            this.realize = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent();
             return;
         }
-        this.facade = new core_1.AgenticaTokenUsage(props.facade);
-        this.analyze = new core_1.AgenticaTokenUsage(props.analyze);
-        this.prisma = new core_1.AgenticaTokenUsage(props.prisma);
-        this.interface = new core_1.AgenticaTokenUsage(props.interface);
-        this.test = new core_1.AgenticaTokenUsage(props.test);
-        this.realize = new core_1.AgenticaTokenUsage(props.realize);
+        else if (props instanceof AutoBeTokenUsage) {
+            this.facade = props.facade;
+            this.analyze = props.analyze;
+            this.prisma = props.prisma;
+            this.interface = props.interface;
+            this.test = props.test;
+            this.realize = props.realize;
+        }
+        else {
+            this.facade = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent(props.facade);
+            this.analyze = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent(props.analyze);
+            this.prisma = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent(props.prisma);
+            this.interface = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent(props.interface);
+            this.test = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent(props.test);
+            this.realize = new AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent(props.realize);
+        }
     }
+    /**
+     * Serialize token usage data to JSON format.
+     *
+     * Converts the internal token usage representation to the standardized
+     * IAutoBeTokenUsageJson format, suitable for persistence, transmission, or
+     * external analysis. The serialized data maintains the complete structure
+     * including all agent phases and detailed token breakdowns.
+     *
+     * @returns JSON representation of token usage statistics
+     */
+    toJSON() {
+        return {
+            aggregate: this.aggregate,
+            facade: this.facade.toJSON(),
+            analyze: this.analyze.toJSON(),
+            prisma: this.prisma.toJSON(),
+            interface: this.interface.toJSON(),
+            test: this.test.toJSON(),
+            realize: this.realize.toJSON(),
+        };
+    }
+    /* -----------------------------------------------------------
+      OPERATORS
+    ----------------------------------------------------------- */
+    /**
+     * Record token usage for specific processing stages.
+     *
+     * Updates token consumption statistics for one or more agent phases based on
+     * the provided usage data. This method allows selective recording of token
+     * usage for specific stages, enabling fine-grained tracking during
+     * multi-phase processing or when certain agents are invoked multiple times.
+     *
+     * @example
+     *   ```ts
+     *   tokenUsage.record(
+     *     { total: 150, input: { total: 100, cached: 20 }, output: { total: 50, ... } },
+     *     ['analyze', 'prisma']
+     *   );
+     *   ```;
+     *
+     * @param usage - Token usage component data to record
+     * @param additionalStages - Array of stage names to update with the usage
+     *   data
+     */
     record(usage, additionalStages = []) {
-        this.facade.increment(usage);
         additionalStages.forEach((stage) => {
             this[stage].increment(usage);
         });
     }
+    /**
+     * Increment current token usage with data from another instance.
+     *
+     * Adds token usage statistics from another AutoBeTokenUsage instance to this
+     * one, updating all agent phases simultaneously. This method is useful for
+     * combining token usage from parallel processing, multiple runs, or when
+     * aggregating statistics from distributed agent executions.
+     *
+     * @param usage - AutoBeTokenUsage instance to add to current statistics
+     * @returns This instance for method chaining
+     */
     increment(usage) {
         AutoBeTokenUsage.keys().forEach((key) => {
             this[key].increment(usage[key]);
         });
         return this;
     }
+    /**
+     * Create a new instance combining token usage from two sources.
+     *
+     * Performs element-wise addition of token usage statistics from two
+     * AutoBeTokenUsage instances, creating a new instance with the combined
+     * totals. This static method is useful for aggregating token usage across
+     * multiple vibe coding sessions or when merging statistics from different
+     * execution contexts.
+     *
+     * @param usageA - First token usage instance
+     * @param usageB - Second token usage instance
+     * @returns New instance with combined token usage statistics
+     */
     static plus(usageA, usageB) {
         return new AutoBeTokenUsage({
-            facade: core_1.AgenticaTokenUsage.plus(usageA.facade, usageB.facade),
-            analyze: core_1.AgenticaTokenUsage.plus(usageA.analyze, usageB.analyze),
-            prisma: core_1.AgenticaTokenUsage.plus(usageA.prisma, usageB.prisma),
-            interface: core_1.AgenticaTokenUsage.plus(usageA.interface, usageB.interface),
-            test: core_1.AgenticaTokenUsage.plus(usageA.test, usageB.test),
-            realize: core_1.AgenticaTokenUsage.plus(usageA.realize, usageB.realize),
+            facade: AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent.plus(usageA.facade, usageB.facade),
+            analyze: AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent.plus(usageA.analyze, usageB.analyze),
+            prisma: AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent.plus(usageA.prisma, usageB.prisma),
+            interface: AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent.plus(usageA.interface, usageB.interface),
+            test: AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent.plus(usageA.test, usageB.test),
+            realize: AutoBeTokenUsageComponent_1.AutoBeTokenUsageComponent.plus(usageA.realize, usageB.realize),
         });
     }
-    toJSON() {
-        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 */
+    /**
+     * Get all agent phase keys for iteration.
+     *
+     * Returns a readonly array of all agent phase names used in the vibe coding
+     * system. This internal utility method ensures consistent iteration over all
+     * token usage components when performing operations like aggregation or
+     * serialization.
+     *
+     * @returns Readonly array of agent phase keys
+     * @internal
+     */
     static keys() {
-        return ["facade", "analyze", "prisma", "interface", "test", "realize"];
+        return [
+            "facade",
+            "analyze",
+            "prisma",
+            "interface",
+            "test",
+            "realize",
+        ];
     }
 }
 exports.AutoBeTokenUsage = AutoBeTokenUsage;

package/lib/context/AutoBeTokenUsage.js.map

@@ -1 +1 @@
-{"version":3,"file":"AutoBeTokenUsage.js","sourceRoot":"","sources":["../../src/context/AutoBeTokenUsage.ts"],"names":[],"mappings":";;;AAAA,yCAAoD;AAKpD,MAAa,gBAAgB;IAQ3B,YAAmB,KAA6B;QAC9C,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC,MAAM,GAAG,IAAI,yBAAkB,EAAE,CAAC;YACvC,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAkB,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,GAAG,IAAI,yBAAkB,EAAE,CAAC;YACvC,IAAI,CAAC,SAAS,GAAG,IAAI,yBAAkB,EAAE,CAAC;YAC1C,IAAI,CAAC,IAAI,GAAG,IAAI,yBAAkB,EAAE,CAAC;YACrC,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAkB,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,yBAAkB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACnD,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAkB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACrD,IAAI,CAAC,MAAM,GAAG,IAAI,yBAAkB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACnD,IAAI,CAAC,SAAS,GAAG,IAAI,yBAAkB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACzD,IAAI,CAAC,IAAI,GAAG,IAAI,yBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAkB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;IAEM,MAAM,CACX,KAAyB,EACzB,mBAAiD,EAAE;QAEnD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC7B,gBAAgB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YACjC,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,SAAS,CAAC,KAAuB;QACtC,gBAAgB,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACtC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,MAAM,CAAC,IAAI,CAAC,MAAwB,EAAE,MAAwB;QACnE,OAAO,IAAI,gBAAgB,CAAC;YAC1B,MAAM,EAAE,yBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;YAC7D,OAAO,EAAE,yBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC;YAChE,MAAM,EAAE,yBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;YAC7D,SAAS,EAAE,yBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC;YACtE,IAAI,EAAE,yBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC;YACvD,OAAO,EAAE,yBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC;SACjE,CAAC,CAAC;IACL,CAAC;IAEM,MAAM;QACX,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;YAC5B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YAC9B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;YAC5B,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE;YAClC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YACxB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;SAC/B,CAAC;IACJ,CAAC;IAED,gBAAgB;IACR,MAAM,CAAC,IAAI;QACjB,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IACzE,CAAC;CACF;AAtED,4CAsEC"}
+{"version":3,"file":"AutoBeTokenUsage.js","sourceRoot":"","sources":["../../src/context/AutoBeTokenUsage.ts"],"names":[],"mappings":";;;AAEA,2EAAwE;AAGxE;;;;;;;;;;;;;;;GAeG;AACH,MAAa,gBAAgB;IAoD3B;;;;;;;;;;;;;;OAcG;IACH,IAAW,SAAS;QAClB,OAAO,gBAAgB,CAAC,IAAI,EAAE;aAC3B,MAAM,CACL,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,qDAAyB,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAC5D,IAAI,qDAAyB,EAAE,CAChC;aACA,MAAM,EAAE,CAAC;IACd,CAAC;IAyCD,YACE,KAA4D;QAE5D,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC,MAAM,GAAG,IAAI,qDAAyB,EAAE,CAAC;YAC9C,IAAI,CAAC,OAAO,GAAG,IAAI,qDAAyB,EAAE,CAAC;YAC/C,IAAI,CAAC,MAAM,GAAG,IAAI,qDAAyB,EAAE,CAAC;YAC9C,IAAI,CAAC,SAAS,GAAG,IAAI,qDAAyB,EAAE,CAAC;YACjD,IAAI,CAAC,IAAI,GAAG,IAAI,qDAAyB,EAAE,CAAC;YAC5C,IAAI,CAAC,OAAO,GAAG,IAAI,qDAAyB,EAAE,CAAC;YAC/C,OAAO;QACT,CAAC;aAAM,IAAI,KAAK,YAAY,gBAAgB,EAAE,CAAC;YAC7C,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;YAC3B,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;YAC7B,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;YAC3B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;YACjC,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;YACvB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,GAAG,IAAI,qDAAyB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1D,IAAI,CAAC,OAAO,GAAG,IAAI,qDAAyB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5D,IAAI,CAAC,MAAM,GAAG,IAAI,qDAAyB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1D,IAAI,CAAC,SAAS,GAAG,IAAI,qDAAyB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAChE,IAAI,CAAC,IAAI,GAAG,IAAI,qDAAyB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACtD,IAAI,CAAC,OAAO,GAAG,IAAI,qDAAyB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAED;;;;;;;;;OASG;IACI,MAAM;QACX,OAAO;YACL,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;YAC5B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YAC9B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;YAC5B,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE;YAClC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YACxB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;SAC/B,CAAC;IACJ,CAAC;IAED;;kEAE8D;IAC9D;;;;;;;;;;;;;;;;;;;OAmBG;IACI,MAAM,CACX,KAAuC,EACvC,mBAAiD,EAAE;QAEnD,gBAAgB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YACjC,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;OAUG;IACI,SAAS,CAAC,KAAuB;QACtC,gBAAgB,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACtC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;;;;;;;OAYG;IACI,MAAM,CAAC,IAAI,CAAC,MAAwB,EAAE,MAAwB;QACnE,OAAO,IAAI,gBAAgB,CAAC;YAC1B,MAAM,EAAE,qDAAyB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;YACpE,OAAO,EAAE,qDAAyB,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC;YACvE,MAAM,EAAE,qDAAyB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;YACpE,SAAS,EAAE,qDAAyB,CAAC,IAAI,CACvC,MAAM,CAAC,SAAS,EAChB,MAAM,CAAC,SAAS,CACjB;YACD,IAAI,EAAE,qDAAyB,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC;YAC9D,OAAO,EAAE,qDAAyB,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC;SACxE,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;OAUG;IACK,MAAM,CAAC,IAAI;QACjB,OAAO;YACL,QAAQ;YACR,SAAS;YACT,QAAQ;YACR,WAAW;YACX,MAAM;YACN,SAAS;SACD,CAAC;IACb,CAAC;CACF;AAvQD,4CAuQC"}

package/lib/context/AutoBeTokenUsageComponent.d.ts

@@ -0,0 +1,116 @@
+import { IAutoBeTokenUsageJson } from "@autobe/interface";
+/**
+ * Token usage component for individual AI agents in the vibe coding pipeline.
+ *
+ * Represents detailed token consumption statistics for a specific processing
+ * phase (facade, analyze, prisma, interface, test, or realize). This class
+ * tracks both input and output token usage with granular breakdowns, enabling
+ * precise cost analysis and performance optimization for each agent.
+ *
+ * The component structure includes:
+ *
+ * - Total token count for quick cost calculations
+ * - Input token breakdown with cache efficiency metrics
+ * - Output token categorization by generation type
+ *
+ * This granular tracking helps identify optimization opportunities and
+ * understand the computational characteristics of each agent phase.
+ *
+ * @author SunRabbit123
+ */
+export declare class AutoBeTokenUsageComponent implements IAutoBeTokenUsageJson.IComponent {
+    /**
+     * Detailed breakdown of input token consumption.
+     *
+     * Tracks how many tokens were processed as input to the AI agent, including:
+     *
+     * - Total input tokens processed
+     * - Cached tokens that were reused from previous operations
+     *
+     * The cache efficiency (cached/total ratio) indicates how well the system is
+     * reusing context across multiple invocations.
+     */
+    readonly input: IAutoBeTokenUsageJson.IInput;
+    /**
+     * Detailed breakdown of output token generation.
+     *
+     * Categorizes generated tokens by their purpose and acceptance status:
+     *
+     * - Total output tokens generated
+     * - Reasoning tokens (internal processing)
+     * - Accepted prediction tokens (efficient generation)
+     * - Rejected prediction tokens (quality control overhead)
+     *
+     * These metrics help understand the AI's generation efficiency and the
+     * effectiveness of its predictive mechanisms.
+     */
+    readonly output: IAutoBeTokenUsageJson.IOutput;
+    /**
+     * Total token count combining all input and output tokens.
+     *
+     * Represents the complete token consumption for this component, providing a
+     * single metric for overall resource utilization. This value is critical for
+     * cost calculations and comparing efficiency across different agents or
+     * processing phases.
+     */
+    get total(): number;
+    /**
+     * Default Constructor.
+     *
+     * Creates a new token usage component with all counters initialized to zero.
+     * Constructs fresh input and output objects with default values, providing a
+     * clean starting point for tracking token consumption in an agent phase.
+     */
+    constructor();
+    /**
+     * Initializer Constructor.
+     *
+     * Creates a new component populated with existing token usage data. Directly
+     * assigns the provided values to instance properties, preserving the exact
+     * token counts and structure from the source data for accurate tracking
+     * continuation.
+     *
+     * @param props - Token usage data to initialize the component
+     */
+    constructor(props: Omit<IAutoBeTokenUsageJson.IComponent, "total">);
+    /**
+     * Export token usage data as JSON.
+     *
+     * Converts the component's token usage statistics to the standardized
+     * IAutoBeTokenUsageJson.IComponent format. This serialization maintains the
+     * complete structure including total counts and detailed breakdowns for both
+     * input and output tokens.
+     *
+     * @returns JSON representation of the token usage component
+     */
+    toJSON(): IAutoBeTokenUsageJson.IComponent;
+    /**
+     * Add token usage data to current statistics.
+     *
+     * Increments all token counters in this component by the corresponding values
+     * from the provided component data. This method performs in-place updates,
+     * modifying the current instance rather than creating a new one.
+     *
+     * Updates include:
+     *
+     * - Total token count
+     * - Input tokens (both total and cached)
+     * - Output tokens (reasoning, accepted/rejected predictions)
+     *
+     * @param props - Token usage component data to add to current values
+     */
+    increment(props: IAutoBeTokenUsageJson.IComponent): void;
+    /**
+     * Create new component combining two token usage statistics.
+     *
+     * Performs element-wise addition of all token counters from two components,
+     * creating a new AutoBeTokenUsageComponent instance with the combined totals.
+     * This static method is useful for aggregating token usage across multiple
+     * agent invocations or combining statistics from parallel processing.
+     *
+     * @param a - First token usage component
+     * @param b - Second token usage component
+     * @returns New component with combined token statistics
+     */
+    static plus(a: AutoBeTokenUsageComponent, b: AutoBeTokenUsageComponent): AutoBeTokenUsageComponent;
+}

package/lib/context/AutoBeTokenUsageComponent.js

@@ -0,0 +1,120 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AutoBeTokenUsageComponent = void 0;
+/**
+ * Token usage component for individual AI agents in the vibe coding pipeline.
+ *
+ * Represents detailed token consumption statistics for a specific processing
+ * phase (facade, analyze, prisma, interface, test, or realize). This class
+ * tracks both input and output token usage with granular breakdowns, enabling
+ * precise cost analysis and performance optimization for each agent.
+ *
+ * The component structure includes:
+ *
+ * - Total token count for quick cost calculations
+ * - Input token breakdown with cache efficiency metrics
+ * - Output token categorization by generation type
+ *
+ * This granular tracking helps identify optimization opportunities and
+ * understand the computational characteristics of each agent phase.
+ *
+ * @author SunRabbit123
+ */
+class AutoBeTokenUsageComponent {
+    /**
+     * Total token count combining all input and output tokens.
+     *
+     * Represents the complete token consumption for this component, providing a
+     * single metric for overall resource utilization. This value is critical for
+     * cost calculations and comparing efficiency across different agents or
+     * processing phases.
+     */
+    get total() {
+        return this.input.total + this.output.total;
+    }
+    constructor(props) {
+        if (props === undefined) {
+            this.input = { total: 0, cached: 0 };
+            this.output = {
+                total: 0,
+                reasoning: 0,
+                accepted_prediction: 0,
+                rejected_prediction: 0,
+            };
+            return;
+        }
+        this.input = props.input;
+        this.output = props.output;
+    }
+    /**
+     * Export token usage data as JSON.
+     *
+     * Converts the component's token usage statistics to the standardized
+     * IAutoBeTokenUsageJson.IComponent format. This serialization maintains the
+     * complete structure including total counts and detailed breakdowns for both
+     * input and output tokens.
+     *
+     * @returns JSON representation of the token usage component
+     */
+    toJSON() {
+        return {
+            total: this.total,
+            input: this.input,
+            output: this.output,
+        };
+    }
+    /* -----------------------------------------------------------
+      OPERATORS
+    ----------------------------------------------------------- */
+    /**
+     * Add token usage data to current statistics.
+     *
+     * Increments all token counters in this component by the corresponding values
+     * from the provided component data. This method performs in-place updates,
+     * modifying the current instance rather than creating a new one.
+     *
+     * Updates include:
+     *
+     * - Total token count
+     * - Input tokens (both total and cached)
+     * - Output tokens (reasoning, accepted/rejected predictions)
+     *
+     * @param props - Token usage component data to add to current values
+     */
+    increment(props) {
+        this.input.total += props.input.total;
+        this.input.cached += props.input.cached;
+        this.output.total += props.output.total;
+        this.output.reasoning += props.output.reasoning;
+        this.output.accepted_prediction += props.output.accepted_prediction;
+        this.output.rejected_prediction += props.output.rejected_prediction;
+    }
+    /**
+     * Create new component combining two token usage statistics.
+     *
+     * Performs element-wise addition of all token counters from two components,
+     * creating a new AutoBeTokenUsageComponent instance with the combined totals.
+     * This static method is useful for aggregating token usage across multiple
+     * agent invocations or combining statistics from parallel processing.
+     *
+     * @param a - First token usage component
+     * @param b - Second token usage component
+     * @returns New component with combined token statistics
+     */
+    static plus(a, b) {
+        return new AutoBeTokenUsageComponent({
+            input: {
+                total: a.input.total + b.input.total,
+                cached: a.input.cached + b.input.cached,
+            },
+            output: {
+                total: a.output.total + b.output.total,
+                reasoning: a.output.reasoning + b.output.reasoning,
+                accepted_prediction: a.output.accepted_prediction + b.output.accepted_prediction,
+                rejected_prediction: a.output.rejected_prediction + b.output.rejected_prediction,
+            },
+        });
+    }
+}
+exports.AutoBeTokenUsageComponent = AutoBeTokenUsageComponent;
+//# sourceMappingURL=AutoBeTokenUsageComponent.js.map

package/lib/context/AutoBeTokenUsageComponent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AutoBeTokenUsageComponent.js","sourceRoot":"","sources":["../../src/context/AutoBeTokenUsageComponent.ts"],"names":[],"mappings":";;;AAEA;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAa,yBAAyB;IA+BpC;;;;;;;OAOG;IACH,IAAW,KAAK;QACd,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;IAC9C,CAAC;IA0BD,YAAmB,KAAuD;QACxE,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC,KAAK,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;YACrC,IAAI,CAAC,MAAM,GAAG;gBACZ,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,CAAC;gBACZ,mBAAmB,EAAE,CAAC;gBACtB,mBAAmB,EAAE,CAAC;aACvB,CAAC;YACF,OAAO;QACT,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED;;;;;;;;;OASG;IACI,MAAM;QACX,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;IACJ,CAAC;IAED;;kEAE8D;IAC9D;;;;;;;;;;;;;;OAcG;IACI,SAAS,CAAC,KAAuC;QACtD,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC;QACtC,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,mBAAmB,IAAI,KAAK,CAAC,MAAM,CAAC,mBAAmB,CAAC;QACpE,IAAI,CAAC,MAAM,CAAC,mBAAmB,IAAI,KAAK,CAAC,MAAM,CAAC,mBAAmB,CAAC;IACtE,CAAC;IAED;;;;;;;;;;;OAWG;IACI,MAAM,CAAC,IAAI,CAChB,CAA4B,EAC5B,CAA4B;QAE5B,OAAO,IAAI,yBAAyB,CAAC;YACnC,KAAK,EAAE;gBACL,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,KAAK;gBACpC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM;aACxC;YACD,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK;gBACtC,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,SAAS,GAAG,CAAC,CAAC,MAAM,CAAC,SAAS;gBAClD,mBAAmB,EACjB,CAAC,CAAC,MAAM,CAAC,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC,mBAAmB;gBAC7D,mBAAmB,EACjB,CAAC,CAAC,MAAM,CAAC,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC,mBAAmB;aAC9D;SACF,CAAC,CAAC;IACL,CAAC;CACF;AA9JD,8DA8JC"}

package/lib/factory/createAutoBeApplication.js

@@ -80,7 +80,7 @@ const createAutoBeController = (props) => {
                     };
             }),
             prisma: (next) => __awaiter(void 0, void 0, void 0, function* () {
-                const r = yield (0, orchestratePrisma_1.orchestratePrisma)(props.context)(next);
+                const r = yield (0, orchestratePrisma_1.orchestratePrisma)(props.context, next);
                 if (r.type === "prisma")
                     return {
                         type: r.compiled.type,