@grislabs/agentmeter 0.1.0

package/dist/trace-context.d.ts

@@ -0,0 +1,86 @@
+import type { TraceTags, LLMCallEvent, ToolCallEvent, PaymentEvent, BudgetRule, CostType } from "./types.js";
+type EmittableEvent = LLMCallEvent | ToolCallEvent | PaymentEvent;
+export declare class TraceContext {
+    readonly traceId: string;
+    readonly workflow: string;
+    readonly tags: TraceTags;
+    readonly startTime: number;
+    private _llmCost;
+    private _toolCost;
+    private _paymentCost;
+    private _totalInputTokens;
+    private _totalOutputTokens;
+    private _llmCalls;
+    private _toolCalls;
+    private _payments;
+    private _budgetRule;
+    private _emitEvent;
+    constructor(workflow: string, tags: TraceTags, budgetRule: BudgetRule | undefined, emitEvent: (event: EmittableEvent) => void);
+    get totalCost(): number;
+    get llmCost(): number;
+    get toolCost(): number;
+    get paymentCost(): number;
+    get totalInputTokens(): number;
+    get totalOutputTokens(): number;
+    get llmCallCount(): number;
+    get toolCallCount(): number;
+    get paymentCount(): number;
+    get durationMs(): number;
+    trackLLM(params: {
+        model: string;
+        inputTokens: number;
+        outputTokens: number;
+        cachedInputTokens?: number;
+        durationMs: number;
+        success?: boolean;
+        error?: string;
+    }): void;
+    trackTool(params: {
+        toolName: string;
+        durationMs: number;
+        cost?: number;
+        costType?: CostType;
+        success?: boolean;
+        error?: string;
+    }): void;
+    trackPayment(params: {
+        provider: PaymentEvent["provider"];
+        amount: number;
+        currency?: string;
+        description: string;
+        recipient?: string;
+        transactionId?: string;
+        durationMs: number;
+        success?: boolean;
+        error?: string;
+    }): void;
+    /**
+     * Convenience: wrap an async LLM call, auto-tracking duration and extracting usage.
+     */
+    llm<T>(call: Promise<T>, extract: (response: T) => {
+        model: string;
+        inputTokens: number;
+        outputTokens: number;
+        cachedInputTokens?: number;
+    }): Promise<T>;
+    /**
+     * Convenience: wrap an async tool call with optional cost.
+     */
+    tool<T>(toolName: string, call: Promise<T>, opts?: {
+        cost?: number;
+        costType?: CostType;
+    }): Promise<T>;
+    /**
+     * Convenience: wrap an async payment call.
+     */
+    payment<T>(params: {
+        provider: PaymentEvent["provider"];
+        amount: number;
+        currency?: string;
+        description: string;
+        recipient?: string;
+    }, call: Promise<T>): Promise<T>;
+    private _checkBudget;
+}
+export {};
+//# sourceMappingURL=trace-context.d.ts.map

package/dist/trace-context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trace-context.d.ts","sourceRoot":"","sources":["../src/trace-context.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,SAAS,EACT,YAAY,EACZ,aAAa,EACb,YAAY,EACZ,UAAU,EACV,QAAQ,EACT,MAAM,YAAY,CAAC;AAEpB,KAAK,cAAc,GAAG,YAAY,GAAG,aAAa,GAAG,YAAY,CAAC;AAElE,qBAAa,YAAY;IACvB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;IACzB,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAE3B,OAAO,CAAC,QAAQ,CAAK;IACrB,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,kBAAkB,CAAK;IAC/B,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,UAAU,CAAK;IACvB,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,WAAW,CAAyB;IAC5C,OAAO,CAAC,UAAU,CAAkC;gBAGlD,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,SAAS,EACf,UAAU,EAAE,UAAU,GAAG,SAAS,EAClC,SAAS,EAAE,CAAC,KAAK,EAAE,cAAc,KAAK,IAAI;IAU5C,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,IAAI,OAAO,IAAI,MAAM,CAEpB;IAED,IAAI,QAAQ,IAAI,MAAM,CAErB;IAED,IAAI,WAAW,IAAI,MAAM,CAExB;IAED,IAAI,gBAAgB,IAAI,MAAM,CAE7B;IAED,IAAI,iBAAiB,IAAI,MAAM,CAE9B;IAED,IAAI,YAAY,IAAI,MAAM,CAEzB;IAED,IAAI,aAAa,IAAI,MAAM,CAE1B;IAED,IAAI,YAAY,IAAI,MAAM,CAEzB;IAED,IAAI,UAAU,IAAI,MAAM,CAEvB;IAED,QAAQ,CAAC,MAAM,EAAE;QACf,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,IAAI;IAgCR,SAAS,CAAC,MAAM,EAAE;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,QAAQ,CAAC;QACpB,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,IAAI;IAsBR,YAAY,CAAC,MAAM,EAAE;QACnB,QAAQ,EAAE,YAAY,CAAC,UAAU,CAAC,CAAC;QACnC,MAAM,EAAE,MAAM,CAAC;QACf,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,IAAI;IAyBR;;OAEG;IACG,GAAG,CAAC,CAAC,EACT,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,EAChB,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,KAAK;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B,GACA,OAAO,CAAC,CAAC,CAAC;IAoBb;;OAEG;IACG,IAAI,CAAC,CAAC,EACV,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,EAChB,IAAI,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,QAAQ,CAAA;KAAE,GAC5C,OAAO,CAAC,CAAC,CAAC;IAwBb;;OAEG;IACG,OAAO,CAAC,CAAC,EACb,MAAM,EAAE;QACN,QAAQ,EAAE,YAAY,CAAC,UAAU,CAAC,CAAC;QACnC,MAAM,EAAE,MAAM,CAAC;QACf,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,EACD,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,GACf,OAAO,CAAC,CAAC,CAAC;IAiBb,OAAO,CAAC,YAAY;CAqBrB"}

package/dist/trace-context.js

@@ -0,0 +1,216 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TraceContext = void 0;
+const node_crypto_1 = require("node:crypto");
+const pricing_js_1 = require("./pricing.js");
+class TraceContext {
+    traceId;
+    workflow;
+    tags;
+    startTime;
+    _llmCost = 0;
+    _toolCost = 0;
+    _paymentCost = 0;
+    _totalInputTokens = 0;
+    _totalOutputTokens = 0;
+    _llmCalls = 0;
+    _toolCalls = 0;
+    _payments = 0;
+    _budgetRule;
+    _emitEvent;
+    constructor(workflow, tags, budgetRule, emitEvent) {
+        this.traceId = (0, node_crypto_1.randomUUID)();
+        this.workflow = workflow;
+        this.tags = tags;
+        this.startTime = Date.now();
+        this._budgetRule = budgetRule;
+        this._emitEvent = emitEvent;
+    }
+    get totalCost() {
+        return this._llmCost + this._toolCost + this._paymentCost;
+    }
+    get llmCost() {
+        return this._llmCost;
+    }
+    get toolCost() {
+        return this._toolCost;
+    }
+    get paymentCost() {
+        return this._paymentCost;
+    }
+    get totalInputTokens() {
+        return this._totalInputTokens;
+    }
+    get totalOutputTokens() {
+        return this._totalOutputTokens;
+    }
+    get llmCallCount() {
+        return this._llmCalls;
+    }
+    get toolCallCount() {
+        return this._toolCalls;
+    }
+    get paymentCount() {
+        return this._payments;
+    }
+    get durationMs() {
+        return Date.now() - this.startTime;
+    }
+    trackLLM(params) {
+        const cost = (0, pricing_js_1.calculateCost)(params.model, params.inputTokens, params.outputTokens, params.cachedInputTokens ?? 0);
+        this._llmCost += cost;
+        this._totalInputTokens += params.inputTokens;
+        this._totalOutputTokens += params.outputTokens;
+        this._llmCalls++;
+        this._emitEvent({
+            type: "llm",
+            traceId: this.traceId,
+            workflow: this.workflow,
+            tags: this.tags,
+            model: params.model,
+            inputTokens: params.inputTokens,
+            outputTokens: params.outputTokens,
+            cachedInputTokens: params.cachedInputTokens ?? 0,
+            cost,
+            durationMs: params.durationMs,
+            success: params.success ?? true,
+            error: params.error,
+            timestamp: Date.now(),
+        });
+        this._checkBudget();
+    }
+    trackTool(params) {
+        const cost = params.cost ?? 0;
+        this._toolCost += cost;
+        this._toolCalls++;
+        this._emitEvent({
+            type: "tool",
+            traceId: this.traceId,
+            workflow: this.workflow,
+            tags: this.tags,
+            toolName: params.toolName,
+            cost,
+            costType: params.costType,
+            durationMs: params.durationMs,
+            success: params.success ?? true,
+            error: params.error,
+            timestamp: Date.now(),
+        });
+        if (cost > 0)
+            this._checkBudget();
+    }
+    trackPayment(params) {
+        const amount = params.success !== false ? params.amount : 0;
+        this._paymentCost += amount;
+        this._payments++;
+        this._emitEvent({
+            type: "payment",
+            traceId: this.traceId,
+            workflow: this.workflow,
+            tags: this.tags,
+            provider: params.provider,
+            amount: params.amount,
+            currency: params.currency ?? "usd",
+            description: params.description,
+            recipient: params.recipient,
+            transactionId: params.transactionId,
+            durationMs: params.durationMs,
+            success: params.success ?? true,
+            error: params.error,
+            timestamp: Date.now(),
+        });
+        if (amount > 0)
+            this._checkBudget();
+    }
+    /**
+     * Convenience: wrap an async LLM call, auto-tracking duration and extracting usage.
+     */
+    async llm(call, extract) {
+        const start = Date.now();
+        try {
+            const response = await call;
+            const usage = extract(response);
+            this.trackLLM({ ...usage, durationMs: Date.now() - start });
+            return response;
+        }
+        catch (err) {
+            this.trackLLM({
+                model: "unknown",
+                inputTokens: 0,
+                outputTokens: 0,
+                durationMs: Date.now() - start,
+                success: false,
+                error: err instanceof Error ? err.message : String(err),
+            });
+            throw err;
+        }
+    }
+    /**
+     * Convenience: wrap an async tool call with optional cost.
+     */
+    async tool(toolName, call, opts) {
+        const start = Date.now();
+        try {
+            const result = await call;
+            this.trackTool({
+                toolName,
+                durationMs: Date.now() - start,
+                cost: opts?.cost,
+                costType: opts?.costType,
+            });
+            return result;
+        }
+        catch (err) {
+            this.trackTool({
+                toolName,
+                durationMs: Date.now() - start,
+                cost: opts?.cost,
+                costType: opts?.costType,
+                success: false,
+                error: err instanceof Error ? err.message : String(err),
+            });
+            throw err;
+        }
+    }
+    /**
+     * Convenience: wrap an async payment call.
+     */
+    async payment(params, call) {
+        const start = Date.now();
+        try {
+            const result = await call;
+            this.trackPayment({ ...params, durationMs: Date.now() - start });
+            return result;
+        }
+        catch (err) {
+            this.trackPayment({
+                ...params,
+                durationMs: Date.now() - start,
+                success: false,
+                error: err instanceof Error ? err.message : String(err),
+            });
+            throw err;
+        }
+    }
+    _checkBudget() {
+        if (!this._budgetRule)
+            return;
+        const { perRun, onExceeded } = this._budgetRule;
+        if (perRun && this.totalCost > perRun) {
+            const err = new Error(`AgentMeter: budget exceeded for workflow "${this.workflow}". ` +
+                `Cost $${this.totalCost.toFixed(4)} (LLM: $${this._llmCost.toFixed(4)}, ` +
+                `Tools: $${this._toolCost.toFixed(4)}, Payments: $${this._paymentCost.toFixed(4)}) ` +
+                `exceeds per-run limit $${perRun}. Action: ${onExceeded}`);
+            err.code = "BUDGET_EXCEEDED";
+            err.workflow = this.workflow;
+            err.traceId = this.traceId;
+            err.currentCost = this.totalCost;
+            err.limit = perRun;
+            err.limitType = "perRun";
+            if (onExceeded === "kill")
+                throw err;
+        }
+    }
+}
+exports.TraceContext = TraceContext;
+//# sourceMappingURL=trace-context.js.map

package/dist/trace-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"trace-context.js","sourceRoot":"","sources":["../src/trace-context.ts"],"names":[],"mappings":";;;AAAA,6CAAyC;AACzC,6CAA6C;AAY7C,MAAa,YAAY;IACd,OAAO,CAAS;IAChB,QAAQ,CAAS;IACjB,IAAI,CAAY;IAChB,SAAS,CAAS;IAEnB,QAAQ,GAAG,CAAC,CAAC;IACb,SAAS,GAAG,CAAC,CAAC;IACd,YAAY,GAAG,CAAC,CAAC;IACjB,iBAAiB,GAAG,CAAC,CAAC;IACtB,kBAAkB,GAAG,CAAC,CAAC;IACvB,SAAS,GAAG,CAAC,CAAC;IACd,UAAU,GAAG,CAAC,CAAC;IACf,SAAS,GAAG,CAAC,CAAC;IACd,WAAW,CAAyB;IACpC,UAAU,CAAkC;IAEpD,YACE,QAAgB,EAChB,IAAe,EACf,UAAkC,EAClC,SAA0C;QAE1C,IAAI,CAAC,OAAO,GAAG,IAAA,wBAAU,GAAE,CAAC;QAC5B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;IAC9B,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC;IAC5D,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,IAAI,gBAAgB;QAClB,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED,IAAI,iBAAiB;QACnB,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;IACrC,CAAC;IAED,QAAQ,CAAC,MAQR;QACC,MAAM,IAAI,GAAG,IAAA,0BAAa,EACxB,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,WAAW,EAClB,MAAM,CAAC,YAAY,EACnB,MAAM,CAAC,iBAAiB,IAAI,CAAC,CAC9B,CAAC;QAEF,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC;QACtB,IAAI,CAAC,iBAAiB,IAAI,MAAM,CAAC,WAAW,CAAC;QAC7C,IAAI,CAAC,kBAAkB,IAAI,MAAM,CAAC,YAAY,CAAC;QAC/C,IAAI,CAAC,SAAS,EAAE,CAAC;QAEjB,IAAI,CAAC,UAAU,CAAC;YACd,IAAI,EAAE,KAAK;YACX,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,CAAC;YAChD,IAAI;YACJ,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI;YAC/B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAED,SAAS,CAAC,MAOT;QACC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC;QACvB,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,IAAI,CAAC,UAAU,CAAC;YACd,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,IAAI;YACJ,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI;YAC/B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,IAAI,IAAI,GAAG,CAAC;YAAE,IAAI,CAAC,YAAY,EAAE,CAAC;IACpC,CAAC;IAED,YAAY,CAAC,MAUZ;QACC,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,YAAY,IAAI,MAAM,CAAC;QAC5B,IAAI,CAAC,SAAS,EAAE,CAAC;QAEjB,IAAI,CAAC,UAAU,CAAC;YACd,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,KAAK;YAClC,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,aAAa,EAAE,MAAM,CAAC,aAAa;YACnC,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI;YAC/B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,IAAI,MAAM,GAAG,CAAC;YAAE,IAAI,CAAC,YAAY,EAAE,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,GAAG,CACP,IAAgB,EAChB,OAKC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC;YAC5B,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAChC,IAAI,CAAC,QAAQ,CAAC,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;YAC5D,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,QAAQ,CAAC;gBACZ,KAAK,EAAE,SAAS;gBAChB,WAAW,EAAE,CAAC;gBACd,YAAY,EAAE,CAAC;gBACf,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;gBAC9B,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aACxD,CAAC,CAAC;YACH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CACR,QAAgB,EAChB,IAAgB,EAChB,IAA6C;QAE7C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC;YAC1B,IAAI,CAAC,SAAS,CAAC;gBACb,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;gBAC9B,IAAI,EAAE,IAAI,EAAE,IAAI;gBAChB,QAAQ,EAAE,IAAI,EAAE,QAAQ;aACzB,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,SAAS,CAAC;gBACb,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;gBAC9B,IAAI,EAAE,IAAI,EAAE,IAAI;gBAChB,QAAQ,EAAE,IAAI,EAAE,QAAQ;gBACxB,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aACxD,CAAC,CAAC;YACH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,MAMC,EACD,IAAgB;QAEhB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC;YAC1B,IAAI,CAAC,YAAY,CAAC,EAAE,GAAG,MAAM,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;YACjE,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,YAAY,CAAC;gBAChB,GAAG,MAAM;gBACT,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;gBAC9B,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aACxD,CAAC,CAAC;YACH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,YAAY;QAClB,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,OAAO;QAE9B,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC;QAChD,IAAI,MAAM,IAAI,IAAI,CAAC,SAAS,GAAG,MAAM,EAAE,CAAC;YACtC,MAAM,GAAG,GAAG,IAAI,KAAK,CACnB,6CAA6C,IAAI,CAAC,QAAQ,KAAK;gBAC/D,SAAS,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;gBACzE,WAAW,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,gBAAgB,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;gBACpF,0BAA0B,MAAM,aAAa,UAAU,EAAE,CAC4D,CAAC;YACxH,GAAG,CAAC,IAAI,GAAG,iBAAiB,CAAC;YAC7B,GAAG,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;YAC7B,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YAC3B,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC;YACjC,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC;YACnB,GAAG,CAAC,SAAS,GAAG,QAAQ,CAAC;YAEzB,IAAI,UAAU,KAAK,MAAM;gBAAE,MAAM,GAAG,CAAC;QACvC,CAAC;IACH,CAAC;CACF;AA/RD,oCA+RC"}

package/dist/types.d.ts

@@ -0,0 +1,102 @@
+export interface AgentMeterConfig {
+    apiKey?: string;
+    endpoint?: string;
+    /** Flush events every N ms (default: 10000) */
+    flushIntervalMs?: number;
+    /** Flush when buffer reaches N events (default: 100) */
+    flushSize?: number;
+    /** Enable local-only mode (no API calls, just console output) */
+    local?: boolean;
+    /** Custom model pricing overrides */
+    pricing?: Record<string, {
+        input: number;
+        output: number;
+        cachedInput?: number;
+    }>;
+}
+export interface TraceTags {
+    customerId?: string;
+    feature?: string;
+    userId?: string;
+    environment?: string;
+    [key: string]: string | undefined;
+}
+export type CostType = "mcp" | "api" | "compute" | "other";
+export interface LLMCallEvent {
+    type: "llm";
+    traceId: string;
+    workflow: string;
+    tags: TraceTags;
+    model: string;
+    inputTokens: number;
+    outputTokens: number;
+    cachedInputTokens: number;
+    cost: number;
+    durationMs: number;
+    success: boolean;
+    error?: string;
+    timestamp: number;
+}
+export interface ToolCallEvent {
+    type: "tool";
+    traceId: string;
+    workflow: string;
+    tags: TraceTags;
+    toolName: string;
+    cost: number;
+    costType?: CostType;
+    durationMs: number;
+    success: boolean;
+    error?: string;
+    timestamp: number;
+}
+export interface PaymentEvent {
+    type: "payment";
+    traceId: string;
+    workflow: string;
+    tags: TraceTags;
+    provider: "stripe_mpp" | "x402" | "ap2" | "custom";
+    amount: number;
+    currency: string;
+    description: string;
+    recipient?: string;
+    transactionId?: string;
+    durationMs: number;
+    success: boolean;
+    error?: string;
+    timestamp: number;
+}
+export interface TraceCompleteEvent {
+    type: "trace_complete";
+    traceId: string;
+    workflow: string;
+    tags: TraceTags;
+    totalCost: number;
+    llmCost: number;
+    toolCost: number;
+    paymentCost: number;
+    totalInputTokens: number;
+    totalOutputTokens: number;
+    llmCalls: number;
+    toolCalls: number;
+    payments: number;
+    durationMs: number;
+    success: boolean;
+    error?: string;
+    timestamp: number;
+}
+export type AgentMeterEvent = LLMCallEvent | ToolCallEvent | PaymentEvent | TraceCompleteEvent;
+export interface BudgetRule {
+    perRun?: number;
+    daily?: number;
+    monthly?: number;
+    onExceeded: "alert" | "kill" | "downgrade-model";
+}
+export interface BudgetExceededError extends Error {
+    workflow: string;
+    traceId: string;
+    currentCost: number;
+    limit: number;
+    limitType: "perRun" | "daily" | "monthly";
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,gBAAgB;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,+CAA+C;IAC/C,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,wDAAwD;IACxD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,iEAAiE;IACjE,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,qCAAqC;IACrC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACnF;AAED,MAAM,WAAW,SAAS;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;CACnC;AAED,MAAM,MAAM,QAAQ,GAAG,KAAK,GAAG,KAAK,GAAG,SAAS,GAAG,OAAO,CAAC;AAE3D,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,KAAK,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,SAAS,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,SAAS,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,SAAS,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,SAAS,CAAC;IAChB,QAAQ,EAAE,YAAY,GAAG,MAAM,GAAG,KAAK,GAAG,QAAQ,CAAC;IACnD,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,gBAAgB,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,SAAS,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,MAAM,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,MAAM,eAAe,GAAG,YAAY,GAAG,aAAa,GAAG,YAAY,GAAG,kBAAkB,CAAC;AAE/F,MAAM,WAAW,UAAU;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,OAAO,GAAG,MAAM,GAAG,iBAAiB,CAAC;CAClD;AAED,MAAM,WAAW,mBAAoB,SAAQ,KAAK;IAChD,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,QAAQ,GAAG,OAAO,GAAG,SAAS,CAAC;CAC3C"}

package/dist/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/package.json

@@ -0,0 +1,50 @@
+{
+  "name": "@grislabs/agentmeter",
+  "publishConfig": {
+    "access": "public"
+  },
+  "version": "0.1.0",
+  "description": "Cost intelligence for AI agents. Track per-task, per-workflow, per-customer costs.",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    },
+    "./mcp": {
+      "types": "./dist/mcp.d.ts",
+      "default": "./dist/mcp.js"
+    }
+  },
+  "bin": {
+    "agentmeter": "dist/cli.js"
+  },
+  "files": [
+    "dist",
+    "!dist/__tests__"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "test": "vitest"
+  },
+  "keywords": [
+    "ai",
+    "agent",
+    "cost",
+    "finops",
+    "llm",
+    "observability",
+    "anthropic",
+    "openai",
+    "langchain",
+    "mcp"
+  ],
+  "license": "MIT",
+  "devDependencies": {
+    "@types/node": "^25.5.0",
+    "typescript": "^5.7.0",
+    "vitest": "^3.0.0"
+  }
+}