@xoxoai/checkmate 0.4.0

package/dist/ai/message-history.js

@@ -0,0 +1,48 @@
+export class MessageHistory {
+    static SNAPSHOT_IDENTIFIER = 'this is a current page snapshot';
+    static SCREENSHOT_IDENTIFIER = 'this is a current screenshot of the page';
+    static TOOL_EXECUTION_SUMMARY_IDENTIFIER = 'tool execution summary';
+    buildInitialMessages(config) {
+        return [
+            {
+                role: 'system',
+                content: [{ type: 'text', text: config.systemPrompt }],
+            },
+            {
+                role: 'user',
+                content: [{ type: 'text', text: config.userPrompt }],
+            },
+            ...(config.contextMessages ?? []),
+        ];
+    }
+    createSnapshotMessage(snapshotContent) {
+        return {
+            role: 'user',
+            content: [{ type: 'text', text: `${MessageHistory.SNAPSHOT_IDENTIFIER}:\n${snapshotContent}` }],
+        };
+    }
+    createScreenshotMessage(base64Data, mimeType = 'image/png') {
+        return {
+            role: 'user',
+            content: [
+                { type: 'text', text: MessageHistory.SCREENSHOT_IDENTIFIER },
+                { type: 'image_url', image_url: { url: `data:${mimeType};base64,${base64Data}`, detail: 'high' } },
+            ],
+        };
+    }
+    removeEphemeralStateMessages(aiClient) {
+        aiClient.replaceHistory(aiClient.getMessages().filter((message) => !this.isEphemeralStateMessage(message)));
+    }
+    isEphemeralStateMessage(message) {
+        if (message.role !== 'user' || !Array.isArray(message.content) || message.content.length === 0) {
+            return false;
+        }
+        const firstPart = message.content[0];
+        if (!('text' in firstPart) || typeof firstPart.text !== 'string') {
+            return false;
+        }
+        return (firstPart.text.startsWith(`${MessageHistory.SNAPSHOT_IDENTIFIER}:`) ||
+            firstPart.text === MessageHistory.SCREENSHOT_IDENTIFIER);
+    }
+}
+//# sourceMappingURL=message-history.js.map

package/dist/ai/message-history.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-history.js","sourceRoot":"","sources":["../../src/ai/message-history.ts"],"names":[],"mappings":"AASA,MAAM,OAAO,cAAc;IAC1B,MAAM,CAAU,mBAAmB,GAAG,iCAAiC,CAAA;IACvE,MAAM,CAAU,qBAAqB,GAAG,0CAA0C,CAAA;IAClF,MAAM,CAAU,iCAAiC,GAAG,wBAAwB,CAAA;IAE5E,oBAAoB,CAAC,MAAgC;QACpD,OAAO;YACN;gBACC,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC;aACtD;YACD;gBACC,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC;aACpD;YACD,GAAG,CAAC,MAAM,CAAC,eAAe,IAAI,EAAE,CAAC;SACjC,CAAA;IACF,CAAC;IAED,qBAAqB,CAAC,eAA8B;QACnD,OAAO;YACN,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,cAAc,CAAC,mBAAmB,MAAM,eAAe,EAAE,EAAE,CAAC;SAC/F,CAAA;IACF,CAAC;IAED,uBAAuB,CAAC,UAAkB,EAAE,WAAmB,WAAW;QACzE,OAAO;YACN,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE;gBACR,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,CAAC,qBAAqB,EAAE;gBAC5D,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,QAAQ,QAAQ,WAAW,UAAU,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE;aAClG;SACD,CAAA;IACF,CAAC;IAED,4BAA4B,CAAC,QAAkB;QAC9C,QAAQ,CAAC,cAAc,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;IAC5G,CAAC;IAEO,uBAAuB,CAAC,OAAmC;QAClE,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChG,OAAO,KAAK,CAAA;QACb,CAAC;QAED,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;QACpC,IAAI,CAAC,CAAC,MAAM,IAAI,SAAS,CAAC,IAAI,OAAO,SAAS,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClE,OAAO,KAAK,CAAA;QACb,CAAC;QAED,OAAO,CACN,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,cAAc,CAAC,mBAAmB,GAAG,CAAC;YACnE,SAAS,CAAC,IAAI,KAAK,cAAc,CAAC,qBAAqB,CACvD,CAAA;IACF,CAAC"}

package/dist/ai/prompts.d.ts

@@ -0,0 +1,4 @@
+import { Step } from '../runtime/types';
+export declare const STEP_SYSTEM_PROMPT: (instructions?: string[]) => string;
+export declare const STEP_START_USER_PROMPT: (step: Step) => string;
+//# sourceMappingURL=prompts.d.ts.map

package/dist/ai/prompts.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompts.d.ts","sourceRoot":"","sources":["../../src/ai/prompts.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAA;AAGvC,eAAO,MAAM,kBAAkB,GAAI,eAAc,MAAM,EAAO,WAkB7D,CAAA;AAED,eAAO,MAAM,sBAAsB,GAAI,MAAM,IAAI,WAIhD,CAAA"}

package/dist/ai/prompts.js

@@ -0,0 +1,24 @@
+import { StepResultTool } from '../tools/step/result-tool';
+export const STEP_SYSTEM_PROMPT = (instructions = []) => {
+    const extensionInstructions = instructions.length > 0
+        ? `
+
+# EXTENSIONS:
+${instructions.join('\n')}`
+        : '';
+    return `You are a quality assurance assistant executing test steps.
+
+# RULES:
+Be objective when asserting if expected results are met and focus on executing all the actions.
+Never ask for user input. Use only tools to proceed with the step.
+If you cannot proceed, fail the step and provide the reason as the actual result.
+If the step fails, call '${StepResultTool.TOOL_FAIL_TEST_STEP}' with the actual result.
+If the step succeeds, call '${StepResultTool.TOOL_PASS_TEST_STEP}' with the actual result.
+${extensionInstructions}`;
+};
+export const STEP_START_USER_PROMPT = (step) => `Here is the test step I want you to execute.
+
+# STEP:
+${JSON.stringify(step, null, 2)}
+`;
+//# sourceMappingURL=prompts.js.map

package/dist/ai/prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompts.js","sourceRoot":"","sources":["../../src/ai/prompts.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAA;AAE1D,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,eAAyB,EAAE,EAAE,EAAE;IACjE,MAAM,qBAAqB,GAC1B,YAAY,CAAC,MAAM,GAAG,CAAC;QACtB,CAAC,CAAC;;;EAGH,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;QACxB,CAAC,CAAC,EAAE,CAAA;IAEN,OAAO;;;;;;2BAMmB,cAAc,CAAC,mBAAmB;8BAC/B,cAAc,CAAC,mBAAmB;EAC9D,qBAAqB,EAAE,CAAA;AACzB,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,IAAU,EAAE,EAAE,CAAC;;;EAGpD,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;CAC9B,CAAA"}

package/dist/ai/rate-limit-policy.d.ts

@@ -0,0 +1,7 @@
+import { RuntimeConfig } from '../config/runtime-config';
+export declare class RateLimitPolicy {
+    private readonly config;
+    constructor(config?: RuntimeConfig);
+    wait(): Promise<void>;
+}
+//# sourceMappingURL=rate-limit-policy.d.ts.map

package/dist/ai/rate-limit-policy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limit-policy.d.ts","sourceRoot":"","sources":["../../src/ai/rate-limit-policy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAA;AAGxD,qBAAa,eAAe;IACf,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAN,MAAM,gBAAsB;IAEnD,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;CAS3B"}

package/dist/ai/rate-limit-policy.js

@@ -0,0 +1,17 @@
+import { RuntimeConfig } from '../config/runtime-config';
+import { logger } from '../logging';
+export class RateLimitPolicy {
+    config;
+    constructor(config = new RuntimeConfig()) {
+        this.config = config;
+    }
+    async wait() {
+        const delay = this.config.getApiRateLimitDelayMs();
+        if (delay <= 0) {
+            return;
+        }
+        logger.warn(`waiting: ${Math.floor(delay / 1000)} seconds to avoid rate limit`);
+        await new Promise((resolve) => setTimeout(resolve, delay));
+    }
+}
+//# sourceMappingURL=rate-limit-policy.js.map

package/dist/ai/rate-limit-policy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limit-policy.js","sourceRoot":"","sources":["../../src/ai/rate-limit-policy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAA;AACxD,OAAO,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AAEnC,MAAM,OAAO,eAAe;IACE;IAA7B,YAA6B,SAAS,IAAI,aAAa,EAAE;QAA5B,WAAM,GAAN,MAAM,CAAsB;IAAG,CAAC;IAE7D,KAAK,CAAC,IAAI;QACT,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,sBAAsB,EAAE,CAAA;QAClD,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YAChB,OAAM;QACP,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,8BAA8B,CAAC,CAAA;QAC/E,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAA;IAC3D,CAAC;CACD"}

package/dist/ai/response-processor.d.ts

@@ -0,0 +1,20 @@
+import { ChatCompletion } from 'openai/resources/chat/completions';
+import { Step, ResolveStepResult } from '../runtime/types';
+import { AiClient } from './client';
+import { ExtensionHost } from '../runtime/extension';
+export type ResponseProcessorDependencies = {
+    aiClient: AiClient;
+    extensionHost: ExtensionHost;
+};
+export declare class ResponseProcessor {
+    private readonly tokenTracker;
+    private readonly toolDispatcher;
+    private readonly toolResponseHandler;
+    private readonly rateLimitPolicy;
+    private readonly messageHandler;
+    private readonly aiClient;
+    constructor({ aiClient, extensionHost }: ResponseProcessorDependencies);
+    resetStepTokens(): void;
+    handleResponse(response: ChatCompletion, step: Step, resolveStepResult: ResolveStepResult): Promise<void>;
+}
+//# sourceMappingURL=response-processor.d.ts.map

package/dist/ai/response-processor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"response-processor.d.ts","sourceRoot":"","sources":["../../src/ai/response-processor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,mCAAmC,CAAA;AAClE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAA;AAI1D,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA;AAMnC,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAA;AAEpD,MAAM,MAAM,6BAA6B,GAAG;IAC3C,QAAQ,EAAE,QAAQ,CAAA;IAClB,aAAa,EAAE,aAAa,CAAA;CAC5B,CAAA;AAED,qBAAa,iBAAiB;IAC7B,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAqB;IAClD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAgB;IAC/C,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAqB;IACzD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAwB;IACxD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAgB;IAC/C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAU;gBAEvB,EAAE,QAAQ,EAAE,aAAa,EAAE,EAAE,6BAA6B;IAOtE,eAAe,IAAI,IAAI;IAIjB,cAAc,CAAC,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE,IAAI,EAAE,iBAAiB,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;CA+C/G"}

package/dist/ai/response-processor.js

@@ -0,0 +1,62 @@
+import { ToolDispatcher } from '../tools/dispatcher';
+import { MessageHandler } from './message-handler';
+import { MessageHistory } from './message-history';
+import { RateLimitPolicy } from './rate-limit-policy';
+import { TokenTracker } from './token-tracker';
+import { ToolResponseHandler } from './tool-response-handler';
+export class ResponseProcessor {
+    tokenTracker = new TokenTracker();
+    toolDispatcher;
+    toolResponseHandler;
+    rateLimitPolicy = new RateLimitPolicy();
+    messageHandler;
+    aiClient;
+    constructor({ aiClient, extensionHost }) {
+        this.aiClient = aiClient;
+        this.toolDispatcher = new ToolDispatcher(aiClient.getToolRegistry());
+        this.toolResponseHandler = new ToolResponseHandler(aiClient, new MessageHistory(), this, extensionHost);
+        this.messageHandler = new MessageHandler(aiClient, this);
+    }
+    resetStepTokens() {
+        this.tokenTracker.resetStep();
+    }
+    async handleResponse(response, step, resolveStepResult) {
+        await this.rateLimitPolicy.wait();
+        const historyTokenCount = this.aiClient.countHistoryTokens();
+        this.tokenTracker.log(response, historyTokenCount, this.aiClient.getRuntimeConfig().getModel());
+        if (!response.choices || response.choices.length === 0) {
+            throw new Error(`No choices found in response:\n${JSON.stringify(response, null, 2)}`);
+        }
+        for (const choice of response.choices) {
+            if (!choice.message.tool_calls || choice.message.tool_calls.length === 0) {
+                await this.messageHandler.handle(choice, step, resolveStepResult);
+                continue;
+            }
+            const toolResponses = [];
+            let dispatchedToolCall = false;
+            for (const toolCall of choice.message.tool_calls) {
+                if (toolCall.type !== 'function') {
+                    continue;
+                }
+                dispatchedToolCall = true;
+                const parsedToolCall = {
+                    name: toolCall.function.name,
+                    arguments: JSON.parse(toolCall.function.arguments || '{}'),
+                };
+                const toolResponse = await this.toolDispatcher.dispatch(parsedToolCall, { step, resolveStepResult });
+                if (toolResponse) {
+                    toolResponses.push({ toolCallId: toolCall.id, toolCall: parsedToolCall, toolResponse });
+                }
+            }
+            if (toolResponses.length > 0) {
+                await this.toolResponseHandler.handleMultiple(toolResponses, step, resolveStepResult);
+                return;
+            }
+            if (dispatchedToolCall) {
+                return;
+            }
+            await this.messageHandler.handle(choice, step, resolveStepResult);
+        }
+    }
+}
+//# sourceMappingURL=response-processor.js.map

package/dist/ai/response-processor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"response-processor.js","sourceRoot":"","sources":["../../src/ai/response-processor.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAA;AAIpD,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAA;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAA;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAA;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAA;AAC9C,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAA;AAQ7D,MAAM,OAAO,iBAAiB;IACZ,YAAY,GAAG,IAAI,YAAY,EAAE,CAAA;IACjC,cAAc,CAAgB;IAC9B,mBAAmB,CAAqB;IACxC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAA;IACvC,cAAc,CAAgB;IAC9B,QAAQ,CAAU;IAEnC,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAiC;QACrE,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAA;QACxB,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,CAAA;QACpE,IAAI,CAAC,mBAAmB,GAAG,IAAI,mBAAmB,CAAC,QAAQ,EAAE,IAAI,cAAc,EAAE,EAAE,IAAI,EAAE,aAAa,CAAC,CAAA;QACvG,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;IACzD,CAAC;IAED,eAAe;QACd,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAA;IAC9B,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAwB,EAAE,IAAU,EAAE,iBAAoC;QAC9F,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAA;QAEjC,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,EAAE,CAAA;QAC5D,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,iBAAiB,EAAE,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAA;QAE/F,IAAI,CAAC,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,kCAAkC,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAA;QACvF,CAAC;QAED,KAAK,MAAM,MAAM,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1E,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE,iBAAiB,CAAC,CAAA;gBACjE,SAAQ;YACT,CAAC;YAED,MAAM,aAAa,GAAkF,EAAE,CAAA;YACvG,IAAI,kBAAkB,GAAG,KAAK,CAAA;YAE9B,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;gBAClD,IAAI,QAAQ,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAClC,SAAQ;gBACT,CAAC;gBAED,kBAAkB,GAAG,IAAI,CAAA;gBACzB,MAAM,cAAc,GAAa;oBAChC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;oBAC5B,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC;iBAC1D,CAAA;gBACD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAA;gBACpG,IAAI,YAAY,EAAE,CAAC;oBAClB,aAAa,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,EAAE,QAAQ,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,CAAA;gBACxF,CAAC;YACF,CAAC;YAED,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,MAAM,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,aAAa,EAAE,IAAI,EAAE,iBAAiB,CAAC,CAAA;gBACrF,OAAM;YACP,CAAC;YAED,IAAI,kBAAkB,EAAE,CAAC;gBACxB,OAAM;YACP,CAAC;YAED,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE,iBAAiB,CAAC,CAAA;QAClE,CAAC;IACF,CAAC;CACD"}

package/dist/ai/token-pricing.d.ts

@@ -0,0 +1,13 @@
+/**
+ * TokenPricing estimates provider token costs in USD.
+ */
+export declare class TokenPricing {
+    private static roundToCents;
+    private static inputPricePerMillionUSD;
+    private static outputPricePerMillionUSD;
+    static inputPriceUSD(model: string, tokens: number): number;
+    static cachedInputPriceUSD(model: string, tokens: number): number;
+    static outputPriceUSD(model: string, tokens: number): number;
+    static totalPriceUSD(model: string, inputTokens: number, outputTokens: number, cachedInputTokens?: number): number;
+}
+//# sourceMappingURL=token-pricing.d.ts.map

package/dist/ai/token-pricing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-pricing.d.ts","sourceRoot":"","sources":["../../src/ai/token-pricing.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,qBAAa,YAAY;IACxB,OAAO,CAAC,MAAM,CAAC,YAAY;IAI3B,OAAO,CAAC,MAAM,CAAC,uBAAuB;IAqKtC,OAAO,CAAC,MAAM,CAAC,wBAAwB;IAiKvC,MAAM,CAAC,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,MAAM;IAI3D,MAAM,CAAC,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,MAAM;IAIjE,MAAM,CAAC,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,MAAM;IAI5D,MAAM,CAAC,aAAa,CACnB,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,EACpB,iBAAiB,GAAE,MAAU,GAC3B,MAAM;CAQT"}

package/dist/ai/token-pricing.js

@@ -0,0 +1,348 @@
+/**
+ * TokenPricing estimates provider token costs in USD.
+ */
+export class TokenPricing {
+    static roundToCents(price) {
+        return Math.round(price * 1000) / 1000;
+    }
+    static inputPricePerMillionUSD(model) {
+        switch (model) {
+            case 'gpt-5.4':
+                return this.roundToCents(2.5);
+            case 'gpt-5.3-instant':
+            case 'gpt-5.3-codex':
+                return this.roundToCents(1.5);
+            case 'gpt-5.2':
+                return this.roundToCents(1.75);
+            case 'gpt-5.2-pro':
+                return this.roundToCents(21.0);
+            case 'gpt-5-mini':
+                return this.roundToCents(0.25);
+            case 'gpt-5.1':
+                return this.roundToCents(1.25);
+            case 'gpt-5-nano':
+                return this.roundToCents(0.05);
+            case 'gpt-5-pro':
+                return this.roundToCents(15.0);
+            case 'gpt-4.1':
+                return this.roundToCents(3.0);
+            case 'gpt-4.1-mini':
+                return this.roundToCents(0.8);
+            case 'gpt-4.1-nano':
+                return this.roundToCents(0.2);
+            case 'gpt-4o':
+            case 'gpt-4o-2024-11-20':
+                return this.roundToCents(2.5);
+            case 'gpt-4o-mini':
+            case 'gpt-4o-mini-2024-07-18':
+                return this.roundToCents(0.15);
+            case 'gpt-4-turbo':
+            case 'gpt-4-turbo-preview':
+                return this.roundToCents(10.0);
+            case 'gpt-4':
+                return this.roundToCents(30.0);
+            case 'gpt-3.5-turbo':
+                return this.roundToCents(0.5);
+            case 'o4-mini':
+                return this.roundToCents(4.0);
+            case 'o3-mini':
+                return this.roundToCents(1.1);
+            case 'o1':
+            case 'o1-2024-12-17':
+                return this.roundToCents(15.0);
+            case 'o1-mini':
+            case 'o1-mini-2024-09-12':
+                return this.roundToCents(3.0);
+            case 'claude-3-5-sonnet-20241022':
+            case 'claude-3-5-sonnet-latest':
+            case 'claude-4-sonnet':
+            case 'claude-4.5-sonnet':
+            case 'claude-4.6-sonnet':
+                return this.roundToCents(3.0);
+            case 'claude-3-5-haiku-20241022':
+            case 'claude-3-5-haiku-latest':
+                return this.roundToCents(0.25);
+            case 'claude-4.5-haiku':
+                return this.roundToCents(1.0);
+            case 'claude-3-opus-20240229':
+            case 'claude-3-opus-latest':
+                return this.roundToCents(15.0);
+            case 'claude-4.6-opus':
+                return this.roundToCents(5.0);
+            case 'claude-opus-4.5':
+            case 'claude-opus-4-5':
+                return this.roundToCents(5.0);
+            case 'claude-opus-4.1':
+            case 'claude-opus-4-1':
+            case 'claude-opus-4':
+                return this.roundToCents(15.0);
+            case 'claude-sonnet-4.5':
+            case 'claude-sonnet-4-5':
+            case 'claude-sonnet-4':
+            case 'claude-sonnet-3.7':
+                return this.roundToCents(3.0);
+            case 'claude-haiku-4.5':
+            case 'claude-haiku-4-5':
+                return this.roundToCents(1.0);
+            case 'claude-haiku-3.5':
+                return this.roundToCents(0.8);
+            case 'claude-haiku-3':
+                return this.roundToCents(0.25);
+            case 'gemini-3.1-pro-preview':
+                return this.roundToCents(2.0);
+            case 'gemini-3.1-flash-lite-preview':
+                return this.roundToCents(0.25);
+            case 'gemini-3.1-flash-image-preview':
+                return this.roundToCents(0.5);
+            case 'gemini-3-pro-preview':
+                return this.roundToCents(2.0);
+            case 'gemini-2.5-pro':
+                return this.roundToCents(1.25);
+            case 'gemini-2.5-flash':
+            case 'gemini-2.5-flash-preview-09-2025':
+                return this.roundToCents(0.3);
+            case 'gemini-2.5-flash-lite':
+            case 'gemini-2.5-flash-lite-preview-09-2025':
+                return this.roundToCents(0.1);
+            case 'gemini-3-flash-preview':
+            case 'gemini-3-flash':
+                return this.roundToCents(0.5);
+            case 'gemini-3-pro-image-preview':
+            case 'gemini-3-pro-image':
+                return this.roundToCents(2.0);
+            case 'gemini-2.0-flash':
+                return this.roundToCents(0.1);
+            case 'gemini-2.0-flash-lite':
+                return this.roundToCents(0.075);
+            case 'grok-4.20':
+            case 'grok-4.20-latest':
+                return this.roundToCents(3.0);
+            case 'grok-4':
+            case 'grok-4-0709':
+            case 'grok-4-latest':
+                return this.roundToCents(3.0);
+            case 'grok-4-1-fast':
+            case 'grok-4-1-fast-reasoning':
+            case 'grok-4-1-fast-reasoning-latest':
+            case 'grok-4-1-fast-non-reasoning':
+            case 'grok-4-fast-reasoning':
+            case 'grok-4-fast-non-reasoning':
+                return this.roundToCents(0.2);
+            case 'grok-code-fast-1':
+                return this.roundToCents(0.2);
+            case 'grok-2-vision-1212':
+                return this.roundToCents(2.0);
+            case 'grok-3':
+            case 'grok-3-latest':
+            case 'grok-3-beta':
+            case 'grok-3-fast':
+            case 'grok-3-fast-latest':
+            case 'grok-3-fast-beta':
+                return this.roundToCents(3.0);
+            case 'grok-3-mini':
+            case 'grok-3-mini-latest':
+            case 'grok-3-mini-beta':
+            case 'grok-3-mini-fast':
+            case 'grok-3-mini-fast-latest':
+            case 'grok-3-mini-fast-beta':
+                return this.roundToCents(0.3);
+            case 'openai/gpt-oss-20b':
+            case 'openai/gpt-oss-safeguard-20b':
+                return this.roundToCents(0.075);
+            case 'openai/gpt-oss-120b':
+                return this.roundToCents(0.15);
+            case 'moonshotai/kimi-k2-instruct-0905':
+                return this.roundToCents(1.0);
+            case 'meta-llama/llama-4-scout-17b-16e-instruct':
+                return this.roundToCents(0.11);
+            case 'meta-llama/llama-4-maverick-17b-128e-instruct':
+                return this.roundToCents(0.2);
+            case 'meta-llama/llama-guard-4-12b':
+                return this.roundToCents(0.2);
+            case 'qwen/qwen3-32b':
+                return this.roundToCents(0.29);
+            case 'llama-3.3-70b-versatile':
+                return this.roundToCents(0.59);
+            case 'llama-3.1-8b-instant':
+                return this.roundToCents(0.05);
+            default:
+                return this.roundToCents(1.0);
+        }
+    }
+    static outputPricePerMillionUSD(model) {
+        switch (model) {
+            case 'gpt-5.4':
+                return this.roundToCents(10.0);
+            case 'gpt-5.3-instant':
+            case 'gpt-5.3-codex':
+                return this.roundToCents(6.0);
+            case 'gpt-5.2':
+                return this.roundToCents(7.0);
+            case 'gpt-5.2-pro':
+                return this.roundToCents(84.0);
+            case 'gpt-5-mini':
+                return this.roundToCents(2.0);
+            case 'gpt-5.1':
+                return this.roundToCents(10.0);
+            case 'gpt-5-nano':
+                return this.roundToCents(0.4);
+            case 'gpt-5-pro':
+                return this.roundToCents(120.0);
+            case 'gpt-4.1':
+                return this.roundToCents(12.0);
+            case 'gpt-4.1-mini':
+                return this.roundToCents(3.2);
+            case 'gpt-4.1-nano':
+                return this.roundToCents(0.8);
+            case 'gpt-4o':
+            case 'gpt-4o-2024-11-20':
+                return this.roundToCents(10.0);
+            case 'gpt-4o-mini':
+            case 'gpt-4o-mini-2024-07-18':
+                return this.roundToCents(0.6);
+            case 'gpt-4-turbo':
+            case 'gpt-4-turbo-preview':
+                return this.roundToCents(30.0);
+            case 'gpt-4':
+                return this.roundToCents(60.0);
+            case 'gpt-3.5-turbo':
+                return this.roundToCents(1.5);
+            case 'o4-mini':
+                return this.roundToCents(16.0);
+            case 'o3-mini':
+                return this.roundToCents(4.4);
+            case 'o1':
+            case 'o1-2024-12-17':
+                return this.roundToCents(60.0);
+            case 'o1-mini':
+            case 'o1-mini-2024-09-12':
+                return this.roundToCents(12.0);
+            case 'claude-3-5-sonnet-20241022':
+            case 'claude-3-5-sonnet-latest':
+            case 'claude-4-sonnet':
+            case 'claude-4.5-sonnet':
+            case 'claude-4.6-sonnet':
+            case 'claude-sonnet-4.5':
+            case 'claude-sonnet-4-5':
+            case 'claude-sonnet-4':
+            case 'claude-sonnet-3.7':
+                return this.roundToCents(15.0);
+            case 'claude-3-5-haiku-20241022':
+            case 'claude-3-5-haiku-latest':
+                return this.roundToCents(1.25);
+            case 'claude-4.5-haiku':
+            case 'claude-haiku-4.5':
+            case 'claude-haiku-4-5':
+                return this.roundToCents(5.0);
+            case 'claude-haiku-3.5':
+                return this.roundToCents(4.0);
+            case 'claude-haiku-3':
+                return this.roundToCents(1.25);
+            case 'claude-3-opus-20240229':
+            case 'claude-3-opus-latest':
+                return this.roundToCents(75.0);
+            case 'claude-4.6-opus':
+            case 'claude-opus-4.5':
+            case 'claude-opus-4-5':
+                return this.roundToCents(25.0);
+            case 'claude-opus-4.1':
+            case 'claude-opus-4-1':
+            case 'claude-opus-4':
+                return this.roundToCents(75.0);
+            case 'gemini-3.1-pro-preview':
+                return this.roundToCents(10.0);
+            case 'gemini-3.1-flash-lite-preview':
+                return this.roundToCents(1.0);
+            case 'gemini-3.1-flash-image-preview':
+                return this.roundToCents(2.0);
+            case 'gemini-3-pro-preview':
+                return this.roundToCents(10.0);
+            case 'gemini-2.5-pro':
+                return this.roundToCents(10.0);
+            case 'gemini-2.5-flash':
+            case 'gemini-2.5-flash-preview-09-2025':
+                return this.roundToCents(2.5);
+            case 'gemini-2.5-flash-lite':
+            case 'gemini-2.5-flash-lite-preview-09-2025':
+                return this.roundToCents(0.4);
+            case 'gemini-3-flash-preview':
+            case 'gemini-3-flash':
+                return this.roundToCents(2.0);
+            case 'gemini-3-pro-image-preview':
+            case 'gemini-3-pro-image':
+                return this.roundToCents(10.0);
+            case 'gemini-2.0-flash':
+                return this.roundToCents(0.4);
+            case 'gemini-2.0-flash-lite':
+                return this.roundToCents(0.3);
+            case 'grok-4.20':
+            case 'grok-4.20-latest':
+                return this.roundToCents(15.0);
+            case 'grok-4':
+            case 'grok-4-0709':
+            case 'grok-4-latest':
+                return this.roundToCents(15.0);
+            case 'grok-4-1-fast':
+            case 'grok-4-1-fast-reasoning':
+            case 'grok-4-1-fast-reasoning-latest':
+            case 'grok-4-1-fast-non-reasoning':
+            case 'grok-4-fast-reasoning':
+            case 'grok-4-fast-non-reasoning':
+            case 'grok-code-fast-1':
+                return this.roundToCents(0.5);
+            case 'grok-2-vision-1212':
+                return this.roundToCents(10.0);
+            case 'grok-3':
+            case 'grok-3-latest':
+            case 'grok-3-beta':
+            case 'grok-3-fast':
+            case 'grok-3-fast-latest':
+            case 'grok-3-fast-beta':
+                return this.roundToCents(15.0);
+            case 'grok-3-mini':
+            case 'grok-3-mini-latest':
+            case 'grok-3-mini-beta':
+            case 'grok-3-mini-fast':
+            case 'grok-3-mini-fast-latest':
+            case 'grok-3-mini-fast-beta':
+                return this.roundToCents(0.5);
+            case 'openai/gpt-oss-20b':
+            case 'openai/gpt-oss-safeguard-20b':
+                return this.roundToCents(0.3);
+            case 'openai/gpt-oss-120b':
+                return this.roundToCents(0.6);
+            case 'moonshotai/kimi-k2-instruct-0905':
+                return this.roundToCents(3.0);
+            case 'meta-llama/llama-4-scout-17b-16e-instruct':
+                return this.roundToCents(0.34);
+            case 'meta-llama/llama-4-maverick-17b-128e-instruct':
+                return this.roundToCents(0.6);
+            case 'meta-llama/llama-guard-4-12b':
+                return this.roundToCents(0.6);
+            case 'qwen/qwen3-32b':
+                return this.roundToCents(0.59);
+            case 'llama-3.3-70b-versatile':
+                return this.roundToCents(0.79);
+            case 'llama-3.1-8b-instant':
+                return this.roundToCents(0.08);
+            default:
+                return this.roundToCents(2.0);
+        }
+    }
+    static inputPriceUSD(model, tokens) {
+        return this.roundToCents((this.inputPricePerMillionUSD(model) * tokens) / 1_000_000);
+    }
+    static cachedInputPriceUSD(model, tokens) {
+        return this.roundToCents(this.inputPriceUSD(model, tokens) / 10);
+    }
+    static outputPriceUSD(model, tokens) {
+        return this.roundToCents((this.outputPricePerMillionUSD(model) * tokens) / 1_000_000);
+    }
+    static totalPriceUSD(model, inputTokens, outputTokens, cachedInputTokens = 0) {
+        const uncachedInputTokens = Math.max(inputTokens - cachedInputTokens, 0);
+        return this.roundToCents(this.inputPriceUSD(model, uncachedInputTokens) +
+            this.cachedInputPriceUSD(model, cachedInputTokens) +
+            this.outputPriceUSD(model, outputTokens));
+    }
+}
+//# sourceMappingURL=token-pricing.js.map

package/dist/ai/token-pricing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-pricing.js","sourceRoot":"","sources":["../../src/ai/token-pricing.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,OAAO,YAAY;IAChB,MAAM,CAAC,YAAY,CAAC,KAAa;QACxC,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,IAAI,CAAA;IACvC,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,KAAa;QACnD,QAAQ,KAAK,EAAE,CAAC;YACf,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,iBAAiB,CAAC;YACvB,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,aAAa;gBACjB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,YAAY;gBAChB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,YAAY;gBAChB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,WAAW;gBACf,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,cAAc;gBAClB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,cAAc;gBAClB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,QAAQ,CAAC;YACd,KAAK,mBAAmB;gBACvB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,aAAa,CAAC;YACnB,KAAK,wBAAwB;gBAC5B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,aAAa,CAAC;YACnB,KAAK,qBAAqB;gBACzB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,OAAO;gBACX,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,IAAI,CAAC;YACV,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,SAAS,CAAC;YACf,KAAK,oBAAoB;gBACxB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,4BAA4B,CAAC;YAClC,KAAK,0BAA0B,CAAC;YAChC,KAAK,iBAAiB,CAAC;YACvB,KAAK,mBAAmB,CAAC;YACzB,KAAK,mBAAmB;gBACvB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,2BAA2B,CAAC;YACjC,KAAK,yBAAyB;gBAC7B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,wBAAwB,CAAC;YAC9B,KAAK,sBAAsB;gBAC1B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,iBAAiB;gBACrB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,iBAAiB,CAAC;YACvB,KAAK,iBAAiB;gBACrB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,iBAAiB,CAAC;YACvB,KAAK,iBAAiB,CAAC;YACvB,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,mBAAmB,CAAC;YACzB,KAAK,mBAAmB,CAAC;YACzB,KAAK,iBAAiB,CAAC;YACvB,KAAK,mBAAmB;gBACvB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,kBAAkB,CAAC;YACxB,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,gBAAgB;gBACpB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,wBAAwB;gBAC5B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,+BAA+B;gBACnC,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,gCAAgC;gBACpC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,sBAAsB;gBAC1B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,gBAAgB;gBACpB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,kBAAkB,CAAC;YACxB,KAAK,kCAAkC;gBACtC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,uBAAuB,CAAC;YAC7B,KAAK,uCAAuC;gBAC3C,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,wBAAwB,CAAC;YAC9B,KAAK,gBAAgB;gBACpB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,4BAA4B,CAAC;YAClC,KAAK,oBAAoB;gBACxB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,uBAAuB;gBAC3B,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAA;YAChC,KAAK,WAAW,CAAC;YACjB,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,QAAQ,CAAC;YACd,KAAK,aAAa,CAAC;YACnB,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,eAAe,CAAC;YACrB,KAAK,yBAAyB,CAAC;YAC/B,KAAK,gCAAgC,CAAC;YACtC,KAAK,6BAA6B,CAAC;YACnC,KAAK,uBAAuB,CAAC;YAC7B,KAAK,2BAA2B;gBAC/B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,oBAAoB;gBACxB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,QAAQ,CAAC;YACd,KAAK,eAAe,CAAC;YACrB,KAAK,aAAa,CAAC;YACnB,KAAK,aAAa,CAAC;YACnB,KAAK,oBAAoB,CAAC;YAC1B,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,aAAa,CAAC;YACnB,KAAK,oBAAoB,CAAC;YAC1B,KAAK,kBAAkB,CAAC;YACxB,KAAK,kBAAkB,CAAC;YACxB,KAAK,yBAAyB,CAAC;YAC/B,KAAK,uBAAuB;gBAC3B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,oBAAoB,CAAC;YAC1B,KAAK,8BAA8B;gBAClC,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAA;YAChC,KAAK,qBAAqB;gBACzB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,kCAAkC;gBACtC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,2CAA2C;gBAC/C,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,+CAA+C;gBACnD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,8BAA8B;gBAClC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,gBAAgB;gBACpB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,yBAAyB;gBAC7B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,sBAAsB;gBAC1B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B;gBACC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;QAC/B,CAAC;IACF,CAAC;IAEO,MAAM,CAAC,wBAAwB,CAAC,KAAa;QACpD,QAAQ,KAAK,EAAE,CAAC;YACf,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,iBAAiB,CAAC;YACvB,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,aAAa;gBACjB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,YAAY;gBAChB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,YAAY;gBAChB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,WAAW;gBACf,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAA;YAChC,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,cAAc;gBAClB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,cAAc;gBAClB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,QAAQ,CAAC;YACd,KAAK,mBAAmB;gBACvB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,aAAa,CAAC;YACnB,KAAK,wBAAwB;gBAC5B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,aAAa,CAAC;YACnB,KAAK,qBAAqB;gBACzB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,OAAO;gBACX,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,SAAS;gBACb,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,IAAI,CAAC;YACV,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,SAAS,CAAC;YACf,KAAK,oBAAoB;gBACxB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,4BAA4B,CAAC;YAClC,KAAK,0BAA0B,CAAC;YAChC,KAAK,iBAAiB,CAAC;YACvB,KAAK,mBAAmB,CAAC;YACzB,KAAK,mBAAmB,CAAC;YACzB,KAAK,mBAAmB,CAAC;YACzB,KAAK,mBAAmB,CAAC;YACzB,KAAK,iBAAiB,CAAC;YACvB,KAAK,mBAAmB;gBACvB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,2BAA2B,CAAC;YACjC,KAAK,yBAAyB;gBAC7B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,kBAAkB,CAAC;YACxB,KAAK,kBAAkB,CAAC;YACxB,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,gBAAgB;gBACpB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,wBAAwB,CAAC;YAC9B,KAAK,sBAAsB;gBAC1B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,iBAAiB,CAAC;YACvB,KAAK,iBAAiB,CAAC;YACvB,KAAK,iBAAiB;gBACrB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,iBAAiB,CAAC;YACvB,KAAK,iBAAiB,CAAC;YACvB,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,wBAAwB;gBAC5B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,+BAA+B;gBACnC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,gCAAgC;gBACpC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,sBAAsB;gBAC1B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,gBAAgB;gBACpB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,kBAAkB,CAAC;YACxB,KAAK,kCAAkC;gBACtC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,uBAAuB,CAAC;YAC7B,KAAK,uCAAuC;gBAC3C,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,wBAAwB,CAAC;YAC9B,KAAK,gBAAgB;gBACpB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,4BAA4B,CAAC;YAClC,KAAK,oBAAoB;gBACxB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,uBAAuB;gBAC3B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,WAAW,CAAC;YACjB,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,QAAQ,CAAC;YACd,KAAK,aAAa,CAAC;YACnB,KAAK,eAAe;gBACnB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,eAAe,CAAC;YACrB,KAAK,yBAAyB,CAAC;YAC/B,KAAK,gCAAgC,CAAC;YACtC,KAAK,6BAA6B,CAAC;YACnC,KAAK,uBAAuB,CAAC;YAC7B,KAAK,2BAA2B,CAAC;YACjC,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,oBAAoB;gBACxB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,QAAQ,CAAC;YACd,KAAK,eAAe,CAAC;YACrB,KAAK,aAAa,CAAC;YACnB,KAAK,aAAa,CAAC;YACnB,KAAK,oBAAoB,CAAC;YAC1B,KAAK,kBAAkB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,aAAa,CAAC;YACnB,KAAK,oBAAoB,CAAC;YAC1B,KAAK,kBAAkB,CAAC;YACxB,KAAK,kBAAkB,CAAC;YACxB,KAAK,yBAAyB,CAAC;YAC/B,KAAK,uBAAuB;gBAC3B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,oBAAoB,CAAC;YAC1B,KAAK,8BAA8B;gBAClC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,qBAAqB;gBACzB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,kCAAkC;gBACtC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,2CAA2C;gBAC/C,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,+CAA+C;gBACnD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,8BAA8B;gBAClC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,KAAK,gBAAgB;gBACpB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,yBAAyB;gBAC7B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B,KAAK,sBAAsB;gBAC1B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YAC/B;gBACC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;QAC/B,CAAC;IACF,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,KAAa,EAAE,MAAc;QACjD,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,GAAG,SAAS,CAAC,CAAA;IACrF,CAAC;IAED,MAAM,CAAC,mBAAmB,CAAC,KAAa,EAAE,MAAc;QACvD,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAA;IACjE,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,KAAa,EAAE,MAAc;QAClD,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,GAAG,SAAS,CAAC,CAAA;IACtF,CAAC;IAED,MAAM,CAAC,aAAa,CACnB,KAAa,EACb,WAAmB,EACnB,YAAoB,EACpB,oBAA4B,CAAC;QAE7B,MAAM,mBAAmB,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,GAAG,iBAAiB,EAAE,CAAC,CAAC,CAAA;QACxE,OAAO,IAAI,CAAC,YAAY,CACvB,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,mBAAmB,CAAC;YAC7C,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,iBAAiB,CAAC;YAClD,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,YAAY,CAAC,CACzC,CAAA;IACF,CAAC;CACD"}

package/dist/ai/token-tracker.d.ts

@@ -0,0 +1,33 @@
+import { ChatCompletion } from 'openai/resources/chat/completions';
+import { RuntimeConfig } from '../config/runtime-config';
+export declare class TokenTracker {
+    private config;
+    private _inputTokensUsedForTest;
+    private _inputTokensUsedForStep;
+    private _cachedInputTokensUsedForTest;
+    private _cachedInputTokensUsedForStep;
+    private _cacheMetricsUnavailableForTest;
+    private _cacheMetricsUnavailableForStep;
+    private _outputTokensUsedForTest;
+    private _outputTokensUsedForStep;
+    constructor(config?: RuntimeConfig);
+    private get inputTokensUsedForTest();
+    private set inputTokensUsedForTest(value);
+    private get inputTokensUsedForStep();
+    private set inputTokensUsedForStep(value);
+    private get cachedInputTokensUsedForTest();
+    private set cachedInputTokensUsedForTest(value);
+    private get cachedInputTokensUsedForStep();
+    private set cachedInputTokensUsedForStep(value);
+    private get outputTokensUsedForTest();
+    private set outputTokensUsedForTest(value);
+    private get outputTokensUsedForStep();
+    private set outputTokensUsedForStep(value);
+    private checkBudget;
+    resetStep(): void;
+    log(response: ChatCompletion, historyTokenCount: number, model: string): void;
+    private inputCostUSD;
+    private formatCachedUsage;
+    private formatUSD;
+}
+//# sourceMappingURL=token-tracker.d.ts.map

package/dist/ai/token-tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-tracker.d.ts","sourceRoot":"","sources":["../../src/ai/token-tracker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,mCAAmC,CAAA;AAClE,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAA;AAIxD,qBAAa,YAAY;IAUZ,OAAO,CAAC,MAAM;IAT1B,OAAO,CAAC,uBAAuB,CAAI;IACnC,OAAO,CAAC,uBAAuB,CAAI;IACnC,OAAO,CAAC,6BAA6B,CAAI;IACzC,OAAO,CAAC,6BAA6B,CAAI;IACzC,OAAO,CAAC,+BAA+B,CAAQ;IAC/C,OAAO,CAAC,+BAA+B,CAAQ;IAC/C,OAAO,CAAC,wBAAwB,CAAI;IACpC,OAAO,CAAC,wBAAwB,CAAI;gBAEhB,MAAM,gBAAsB;IAEhD,OAAO,KAAK,sBAAsB,GAEjC;IAED,OAAO,KAAK,sBAAsB,QAGjC;IAED,OAAO,KAAK,sBAAsB,GAEjC;IAED,OAAO,KAAK,sBAAsB,QAGjC;IAED,OAAO,KAAK,4BAA4B,GAEvC;IAED,OAAO,KAAK,4BAA4B,QAGvC;IAED,OAAO,KAAK,4BAA4B,GAEvC;IAED,OAAO,KAAK,4BAA4B,QAGvC;IAED,OAAO,KAAK,uBAAuB,GAElC;IAED,OAAO,KAAK,uBAAuB,QAGlC;IAED,OAAO,KAAK,uBAAuB,GAElC;IAED,OAAO,KAAK,uBAAuB,QAGlC;IAED,OAAO,CAAC,WAAW;IA6BnB,SAAS,IAAI,IAAI;IAOjB,GAAG,CAAC,QAAQ,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAuE7E,OAAO,CAAC,YAAY;IAIpB,OAAO,CAAC,iBAAiB;IAKzB,OAAO,CAAC,SAAS;CAGjB"}