attocode 0.1.0

package/dist/src/core/queues/submission-queue.js

@@ -0,0 +1,236 @@
+/**
+ * Bounded async queue for operations from UI to Agent.
+ * Implements backpressure - when full, producers wait until space is available.
+ */
+import { AtomicCounter } from './atomic-counter.js';
+/**
+ * Error thrown when a queue operation times out.
+ */
+export class QueueTimeoutError extends Error {
+    constructor(message = 'Queue operation timed out') {
+        super(message);
+        this.name = 'QueueTimeoutError';
+    }
+}
+/**
+ * Error thrown when attempting to operate on a closed queue.
+ */
+export class QueueClosedError extends Error {
+    constructor(message = 'Queue is closed') {
+        super(message);
+        this.name = 'QueueClosedError';
+    }
+}
+/**
+ * A bounded async queue for operations from UI to Agent.
+ *
+ * Features:
+ * - Bounded capacity with backpressure (producers wait when full)
+ * - Async iterator support for consumption
+ * - Graceful shutdown via close()
+ * - Timeout handling for blocked producers
+ */
+export class SubmissionQueue {
+    maxSize;
+    timeout;
+    counter;
+    queue = [];
+    producerWaiters = [];
+    consumerWaiters = [];
+    closed = false;
+    /**
+     * Creates a new SubmissionQueue.
+     * @param config - Optional configuration
+     */
+    constructor(config) {
+        this.maxSize = config?.maxSize ?? 64;
+        this.timeout = config?.timeout ?? 30000;
+        this.counter = new AtomicCounter();
+    }
+    /**
+     * Submits an operation to the queue.
+     * Blocks if the queue is full (backpressure).
+     *
+     * @param op - The operation to submit
+     * @param correlationId - Optional correlation ID for request/response tracking
+     * @returns The submission ID
+     * @throws QueueClosedError if the queue is closed
+     * @throws QueueTimeoutError if waiting for space times out
+     */
+    async submit(op, correlationId) {
+        if (this.closed) {
+            throw new QueueClosedError('Cannot submit to a closed queue');
+        }
+        // Wait for space if queue is full
+        if (this.queue.length >= this.maxSize) {
+            await this.waitForSpace();
+        }
+        // Re-check closed status after waiting
+        if (this.closed) {
+            throw new QueueClosedError('Queue closed while waiting for space');
+        }
+        const id = this.counter.next();
+        const submission = {
+            id,
+            op,
+            timestamp: Date.now(),
+            ...(correlationId !== undefined && { correlationId }),
+        };
+        this.queue.push(submission);
+        // Wake up a waiting consumer if any
+        this.wakeConsumer(submission);
+        return id;
+    }
+    /**
+     * Takes the next submission from the queue.
+     * Blocks if the queue is empty.
+     *
+     * @returns The next submission, or null if the queue is closed and empty
+     */
+    async take() {
+        // Return immediately if there's an item
+        if (this.queue.length > 0) {
+            const submission = this.queue.shift();
+            this.wakeProducer();
+            return submission;
+        }
+        // If closed and empty, return null
+        if (this.closed) {
+            return null;
+        }
+        // Wait for an item
+        return new Promise((resolve, reject) => {
+            this.consumerWaiters.push({ resolve, reject });
+        });
+    }
+    /**
+     * Non-blocking version of take().
+     * @returns The next submission, or null if the queue is empty
+     */
+    tryTake() {
+        if (this.queue.length === 0) {
+            return null;
+        }
+        const submission = this.queue.shift();
+        this.wakeProducer();
+        return submission;
+    }
+    /**
+     * Returns the current queue size.
+     */
+    size() {
+        return this.queue.length;
+    }
+    /**
+     * Returns true if the queue is empty.
+     */
+    isEmpty() {
+        return this.queue.length === 0;
+    }
+    /**
+     * Returns true if the queue is full.
+     */
+    isFull() {
+        return this.queue.length >= this.maxSize;
+    }
+    /**
+     * Returns true if the queue is closed.
+     */
+    isClosed() {
+        return this.closed;
+    }
+    /**
+     * Closes the queue.
+     * - Rejects all pending producer waiters
+     * - Resolves all pending consumer waiters with null
+     * - No more submissions are accepted
+     */
+    close() {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        // Reject all waiting producers
+        const closedError = new QueueClosedError('Queue was closed');
+        for (const waiter of this.producerWaiters) {
+            waiter.reject(closedError);
+        }
+        this.producerWaiters.length = 0;
+        // Resolve all waiting consumers with null
+        for (const waiter of this.consumerWaiters) {
+            waiter.resolve(null);
+        }
+        this.consumerWaiters.length = 0;
+    }
+    /**
+     * Async iterator for consuming submissions.
+     * Yields submissions until the queue is closed and empty.
+     */
+    async *[Symbol.asyncIterator]() {
+        while (true) {
+            const submission = await this.take();
+            if (submission === null) {
+                return;
+            }
+            yield submission;
+        }
+    }
+    /**
+     * Waits for space to become available in the queue.
+     * @throws QueueTimeoutError if waiting times out
+     * @throws QueueClosedError if queue is closed while waiting
+     */
+    waitForSpace() {
+        return new Promise((resolve, reject) => {
+            const waiter = { resolve, reject };
+            // Set up timeout
+            const timeoutId = setTimeout(() => {
+                const index = this.producerWaiters.indexOf(waiter);
+                if (index !== -1) {
+                    this.producerWaiters.splice(index, 1);
+                    reject(new QueueTimeoutError(`Timed out waiting for space after ${this.timeout}ms`));
+                }
+            }, this.timeout);
+            // Wrap resolve to clear timeout
+            const originalResolve = waiter.resolve;
+            waiter.resolve = (value) => {
+                clearTimeout(timeoutId);
+                originalResolve(value);
+            };
+            // Wrap reject to clear timeout
+            const originalReject = waiter.reject;
+            waiter.reject = (error) => {
+                clearTimeout(timeoutId);
+                originalReject(error);
+            };
+            this.producerWaiters.push(waiter);
+        });
+    }
+    /**
+     * Wakes a waiting producer (called when space becomes available).
+     */
+    wakeProducer() {
+        const waiter = this.producerWaiters.shift();
+        if (waiter) {
+            waiter.resolve();
+        }
+    }
+    /**
+     * Wakes a waiting consumer with a submission.
+     * If no consumers are waiting, the submission is already in the queue.
+     */
+    wakeConsumer(submission) {
+        const waiter = this.consumerWaiters.shift();
+        if (waiter) {
+            // Remove from queue since we're giving it directly to consumer
+            const index = this.queue.indexOf(submission);
+            if (index !== -1) {
+                this.queue.splice(index, 1);
+            }
+            waiter.resolve(submission);
+            // Wake a producer since we freed a slot
+            this.wakeProducer();
+        }
+    }
+}
+//# sourceMappingURL=submission-queue.js.map

package/dist/src/core/queues/submission-queue.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"submission-queue.js","sourceRoot":"","sources":["../../../../src/core/queues/submission-queue.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAoBpD;;GAEG;AACH,MAAM,OAAO,iBAAkB,SAAQ,KAAK;IAC1C,YAAY,UAAkB,2BAA2B;QACvD,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,mBAAmB,CAAC;IAClC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,gBAAiB,SAAQ,KAAK;IACzC,YAAY,UAAkB,iBAAiB;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAC;IACjC,CAAC;CACF;AAED;;;;;;;;GAQG;AACH,MAAM,OAAO,eAAe;IACT,OAAO,CAAS;IAChB,OAAO,CAAS;IAChB,OAAO,CAAgB;IACvB,KAAK,GAAiB,EAAE,CAAC;IACzB,eAAe,GAAmB,EAAE,CAAC;IACrC,eAAe,GAAgC,EAAE,CAAC;IAC3D,MAAM,GAAG,KAAK,CAAC;IAEvB;;;OAGG;IACH,YAAY,MAA8B;QACxC,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,OAAO,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,OAAO,IAAI,KAAK,CAAC;QACxC,IAAI,CAAC,OAAO,GAAG,IAAI,aAAa,EAAE,CAAC;IACrC,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,MAAM,CAAC,EAAa,EAAE,aAAsB;QAChD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,gBAAgB,CAAC,iCAAiC,CAAC,CAAC;QAChE,CAAC;QAED,kCAAkC;QAClC,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC5B,CAAC;QAED,uCAAuC;QACvC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,gBAAgB,CAAC,sCAAsC,CAAC,CAAC;QACrE,CAAC;QAED,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,UAAU,GAAe;YAC7B,EAAE;YACF,EAAE;YACF,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,GAAG,CAAC,aAAa,KAAK,SAAS,IAAI,EAAE,aAAa,EAAE,CAAC;SACtD,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAE5B,oCAAoC;QACpC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAE9B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,IAAI;QACR,wCAAwC;QACxC,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAG,CAAC;YACvC,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,mCAAmC;QACnC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,mBAAmB;QACnB,OAAO,IAAI,OAAO,CAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxD,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,OAAO;QACL,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAG,CAAC;QACvC,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,IAAI;QACF,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,OAAO;QACL,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;;;;OAKG;IACH,KAAK;QACH,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QAEnB,+BAA+B;QAC/B,MAAM,WAAW,GAAG,IAAI,gBAAgB,CAAC,kBAAkB,CAAC,CAAC;QAC7D,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAC7B,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC;QAEhC,0CAA0C;QAC1C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC;IAClC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;QAC3B,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YACrC,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;gBACxB,OAAO;YACT,CAAC;YACD,MAAM,UAAU,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,YAAY;QAClB,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,MAAM,MAAM,GAAiB,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAEjD,iBAAiB;YACjB,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACnD,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;oBACjB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;oBACtC,MAAM,CAAC,IAAI,iBAAiB,CAAC,qCAAqC,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;gBACvF,CAAC;YACH,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YAEjB,gCAAgC;YAChC,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC;YACvC,MAAM,CAAC,OAAO,GAAG,CAAC,KAAW,EAAE,EAAE;gBAC/B,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,eAAe,CAAC,KAAK,CAAC,CAAC;YACzB,CAAC,CAAC;YAEF,+BAA+B;YAC/B,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC;YACrC,MAAM,CAAC,MAAM,GAAG,CAAC,KAAY,EAAE,EAAE;gBAC/B,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,cAAc,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC,CAAC;YAEF,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,YAAY;QAClB,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC5C,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,YAAY,CAAC,UAAsB;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC5C,IAAI,MAAM,EAAE,CAAC;YACX,+DAA+D;YAC/D,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC7C,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;gBACjB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAC9B,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC3B,wCAAwC;YACxC,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;CACF"}

package/dist/src/costs/index.d.ts

@@ -0,0 +1,22 @@
+/**
+ * Cost tracking and model registry module.
+ * Provides model configuration, pricing, and cost calculation utilities.
+ *
+ * @example
+ * ```typescript
+ * import { modelRegistry, type UsageRecord } from './costs/index.js';
+ *
+ * const usage: UsageRecord = {
+ *   modelId: 'claude-3-5-sonnet-20241022',
+ *   promptTokens: 1000,
+ *   completionTokens: 500,
+ *   cacheReadTokens: 200,
+ * };
+ *
+ * const cost = modelRegistry.calculateCost(usage);
+ * console.log(cost.formatted); // '$0.0195'
+ * ```
+ */
+export * from './types.js';
+export * from './model-registry.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/src/costs/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/costs/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AAEH,cAAc,YAAY,CAAC;AAC3B,cAAc,qBAAqB,CAAC"}

package/dist/src/costs/index.js

@@ -0,0 +1,22 @@
+/**
+ * Cost tracking and model registry module.
+ * Provides model configuration, pricing, and cost calculation utilities.
+ *
+ * @example
+ * ```typescript
+ * import { modelRegistry, type UsageRecord } from './costs/index.js';
+ *
+ * const usage: UsageRecord = {
+ *   modelId: 'claude-3-5-sonnet-20241022',
+ *   promptTokens: 1000,
+ *   completionTokens: 500,
+ *   cacheReadTokens: 200,
+ * };
+ *
+ * const cost = modelRegistry.calculateCost(usage);
+ * console.log(cost.formatted); // '$0.0195'
+ * ```
+ */
+export * from './types.js';
+export * from './model-registry.js';
+//# sourceMappingURL=index.js.map

package/dist/src/costs/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/costs/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AAEH,cAAc,YAAY,CAAC;AAC3B,cAAc,qBAAqB,CAAC"}

package/dist/src/costs/model-registry.d.ts

@@ -0,0 +1,80 @@
+import type { ModelConfig, UsageRecord, CostBreakdown } from './types.js';
+/**
+ * Registry for model configurations with caching and cost calculation.
+ * Provides a centralized way to manage model metadata and pricing.
+ */
+export declare class ModelRegistry {
+    private models;
+    private cacheTimestamp;
+    private readonly cacheDurationMs;
+    /**
+     * Creates a new ModelRegistry instance.
+     * @param cacheDurationMs - How long to cache model configs (default: 24 hours)
+     */
+    constructor(cacheDurationMs?: number);
+    /**
+     * Loads default model configurations into the registry.
+     */
+    private loadDefaults;
+    /**
+     * Gets a model configuration by ID.
+     * @param modelId - The unique model identifier
+     * @returns The model config or undefined if not found
+     */
+    getModel(modelId: string): ModelConfig | undefined;
+    /**
+     * Gets all registered models.
+     * @returns Array of all model configurations
+     */
+    getAllModels(): ModelConfig[];
+    /**
+     * Gets models filtered by provider.
+     * @param provider - The provider name to filter by
+     * @returns Array of models from the specified provider
+     */
+    getModelsByProvider(provider: string): ModelConfig[];
+    /**
+     * Adds or updates a model configuration.
+     * @param config - The model configuration to set
+     */
+    setModel(config: ModelConfig): void;
+    /**
+     * Calculates cost breakdown for a usage record.
+     * Uses default pricing if model is not found in registry.
+     * @param usage - The usage record with token counts
+     * @returns Cost breakdown with prompt, completion, cache, and total costs
+     */
+    calculateCost(usage: UsageRecord): CostBreakdown;
+    /**
+     * Checks if the cache needs to be refreshed based on cache duration.
+     * @returns True if cache is stale and needs refresh
+     */
+    needsRefresh(): boolean;
+    /**
+     * Refreshes the cache timestamp.
+     * In future, this could fetch updated pricing from an API.
+     */
+    refreshCache(): void;
+    /**
+     * Exports the current cache state for persistence.
+     * @returns Object containing models array and timestamp
+     */
+    exportCache(): {
+        models: ModelConfig[];
+        timestamp: number;
+    };
+    /**
+     * Imports a previously exported cache.
+     * @param cache - The cache object to import
+     */
+    importCache(cache: {
+        models: ModelConfig[];
+        timestamp: number;
+    }): void;
+}
+/**
+ * Singleton instance of ModelRegistry for global access.
+ * Use this for shared model configuration across the application.
+ */
+export declare const modelRegistry: ModelRegistry;
+//# sourceMappingURL=model-registry.d.ts.map

package/dist/src/costs/model-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry.d.ts","sourceRoot":"","sources":["../../../src/costs/model-registry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AA8G1E;;;GAGG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAuC;IACrD,OAAO,CAAC,cAAc,CAAa;IACnC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IAEzC;;;OAGG;gBACS,eAAe,GAAE,MAA4B;IAKzD;;OAEG;IACH,OAAO,CAAC,YAAY;IAOpB;;;;OAIG;IACH,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAIlD;;;OAGG;IACH,YAAY,IAAI,WAAW,EAAE;IAI7B;;;;OAIG;IACH,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,WAAW,EAAE;IAIpD;;;OAGG;IACH,QAAQ,CAAC,MAAM,EAAE,WAAW,GAAG,IAAI;IAInC;;;;;OAKG;IACH,aAAa,CAAC,KAAK,EAAE,WAAW,GAAG,aAAa;IA+BhD;;;OAGG;IACH,YAAY,IAAI,OAAO;IAIvB;;;OAGG;IACH,YAAY,IAAI,IAAI;IAMpB;;;OAGG;IACH,WAAW,IAAI;QAAE,MAAM,EAAE,WAAW,EAAE,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE;IAO3D;;;OAGG;IACH,WAAW,CAAC,KAAK,EAAE;QAAE,MAAM,EAAE,WAAW,EAAE,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI;CAOvE;AAED;;;GAGG;AACH,eAAO,MAAM,aAAa,eAAsB,CAAC"}

package/dist/src/costs/model-registry.js

@@ -0,0 +1,237 @@
+/**
+ * Default model configurations with current pricing (2024/2025).
+ * Pricing sourced from official provider documentation.
+ */
+const DEFAULT_MODELS = [
+    // Claude models (Anthropic)
+    {
+        modelId: 'claude-3-5-sonnet-20241022',
+        provider: 'anthropic',
+        displayName: 'Claude 3.5 Sonnet',
+        pricing: {
+            promptPerMillion: 3.0,
+            completionPerMillion: 15.0,
+            cacheReadPerMillion: 0.30,
+            cacheWritePerMillion: 3.75,
+        },
+        capabilities: {
+            maxContextTokens: 200000,
+            maxOutputTokens: 8192,
+            supportsVision: true,
+            supportsTools: true,
+            supportsCaching: true,
+        },
+    },
+    {
+        modelId: 'claude-3-5-haiku-20241022',
+        provider: 'anthropic',
+        displayName: 'Claude 3.5 Haiku',
+        pricing: {
+            promptPerMillion: 0.80,
+            completionPerMillion: 4.0,
+            cacheReadPerMillion: 0.08,
+            cacheWritePerMillion: 1.0,
+        },
+        capabilities: {
+            maxContextTokens: 200000,
+            maxOutputTokens: 8192,
+            supportsVision: true,
+            supportsTools: true,
+            supportsCaching: true,
+        },
+    },
+    {
+        modelId: 'claude-3-opus-20240229',
+        provider: 'anthropic',
+        displayName: 'Claude 3 Opus',
+        pricing: {
+            promptPerMillion: 15.0,
+            completionPerMillion: 75.0,
+            cacheReadPerMillion: 1.50,
+            cacheWritePerMillion: 18.75,
+        },
+        capabilities: {
+            maxContextTokens: 200000,
+            maxOutputTokens: 4096,
+            supportsVision: true,
+            supportsTools: true,
+            supportsCaching: true,
+        },
+    },
+    // GPT-4o models (OpenAI)
+    {
+        modelId: 'gpt-4o',
+        provider: 'openai',
+        displayName: 'GPT-4o',
+        pricing: {
+            promptPerMillion: 2.50,
+            completionPerMillion: 10.0,
+        },
+        capabilities: {
+            maxContextTokens: 128000,
+            maxOutputTokens: 16384,
+            supportsVision: true,
+            supportsTools: true,
+        },
+    },
+    {
+        modelId: 'gpt-4o-mini',
+        provider: 'openai',
+        displayName: 'GPT-4o Mini',
+        pricing: {
+            promptPerMillion: 0.15,
+            completionPerMillion: 0.60,
+        },
+        capabilities: {
+            maxContextTokens: 128000,
+            maxOutputTokens: 16384,
+            supportsVision: true,
+            supportsTools: true,
+        },
+    },
+    {
+        modelId: 'gpt-4-turbo',
+        provider: 'openai',
+        displayName: 'GPT-4 Turbo',
+        pricing: {
+            promptPerMillion: 10.0,
+            completionPerMillion: 30.0,
+        },
+        capabilities: {
+            maxContextTokens: 128000,
+            maxOutputTokens: 4096,
+            supportsVision: true,
+            supportsTools: true,
+        },
+    },
+];
+/**
+ * Registry for model configurations with caching and cost calculation.
+ * Provides a centralized way to manage model metadata and pricing.
+ */
+export class ModelRegistry {
+    models = new Map();
+    cacheTimestamp = 0;
+    cacheDurationMs;
+    /**
+     * Creates a new ModelRegistry instance.
+     * @param cacheDurationMs - How long to cache model configs (default: 24 hours)
+     */
+    constructor(cacheDurationMs = 24 * 60 * 60 * 1000) {
+        this.cacheDurationMs = cacheDurationMs;
+        this.loadDefaults();
+    }
+    /**
+     * Loads default model configurations into the registry.
+     */
+    loadDefaults() {
+        for (const model of DEFAULT_MODELS) {
+            this.models.set(model.modelId, model);
+        }
+        this.cacheTimestamp = Date.now();
+    }
+    /**
+     * Gets a model configuration by ID.
+     * @param modelId - The unique model identifier
+     * @returns The model config or undefined if not found
+     */
+    getModel(modelId) {
+        return this.models.get(modelId);
+    }
+    /**
+     * Gets all registered models.
+     * @returns Array of all model configurations
+     */
+    getAllModels() {
+        return Array.from(this.models.values());
+    }
+    /**
+     * Gets models filtered by provider.
+     * @param provider - The provider name to filter by
+     * @returns Array of models from the specified provider
+     */
+    getModelsByProvider(provider) {
+        return this.getAllModels().filter(m => m.provider === provider);
+    }
+    /**
+     * Adds or updates a model configuration.
+     * @param config - The model configuration to set
+     */
+    setModel(config) {
+        this.models.set(config.modelId, config);
+    }
+    /**
+     * Calculates cost breakdown for a usage record.
+     * Uses default pricing if model is not found in registry.
+     * @param usage - The usage record with token counts
+     * @returns Cost breakdown with prompt, completion, cache, and total costs
+     */
+    calculateCost(usage) {
+        const model = this.getModel(usage.modelId);
+        // Default pricing if model not found (Claude 3.5 Sonnet pricing as fallback)
+        const pricing = model?.pricing ?? {
+            promptPerMillion: 3.0,
+            completionPerMillion: 15.0,
+        };
+        const promptCost = (usage.promptTokens / 1_000_000) * pricing.promptPerMillion;
+        const completionCost = (usage.completionTokens / 1_000_000) * pricing.completionPerMillion;
+        let cacheCost = 0;
+        if (usage.cacheReadTokens && pricing.cacheReadPerMillion) {
+            cacheCost += (usage.cacheReadTokens / 1_000_000) * pricing.cacheReadPerMillion;
+        }
+        if (usage.cacheWriteTokens && pricing.cacheWritePerMillion) {
+            cacheCost += (usage.cacheWriteTokens / 1_000_000) * pricing.cacheWritePerMillion;
+        }
+        const totalCost = promptCost + completionCost + cacheCost;
+        return {
+            promptCost,
+            completionCost,
+            cacheCost,
+            totalCost,
+            formatted: `$${totalCost.toFixed(4)}`,
+        };
+    }
+    /**
+     * Checks if the cache needs to be refreshed based on cache duration.
+     * @returns True if cache is stale and needs refresh
+     */
+    needsRefresh() {
+        return Date.now() - this.cacheTimestamp > this.cacheDurationMs;
+    }
+    /**
+     * Refreshes the cache timestamp.
+     * In future, this could fetch updated pricing from an API.
+     */
+    refreshCache() {
+        // In future, this could fetch from an API
+        // For now, just reset the timestamp
+        this.cacheTimestamp = Date.now();
+    }
+    /**
+     * Exports the current cache state for persistence.
+     * @returns Object containing models array and timestamp
+     */
+    exportCache() {
+        return {
+            models: this.getAllModels(),
+            timestamp: this.cacheTimestamp,
+        };
+    }
+    /**
+     * Imports a previously exported cache.
+     * @param cache - The cache object to import
+     */
+    importCache(cache) {
+        this.models.clear();
+        for (const model of cache.models) {
+            this.models.set(model.modelId, model);
+        }
+        this.cacheTimestamp = cache.timestamp;
+    }
+}
+/**
+ * Singleton instance of ModelRegistry for global access.
+ * Use this for shared model configuration across the application.
+ */
+export const modelRegistry = new ModelRegistry();
+//# sourceMappingURL=model-registry.js.map

package/dist/src/costs/model-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry.js","sourceRoot":"","sources":["../../../src/costs/model-registry.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,cAAc,GAAkB;IACpC,4BAA4B;IAC5B;QACE,OAAO,EAAE,4BAA4B;QACrC,QAAQ,EAAE,WAAW;QACrB,WAAW,EAAE,mBAAmB;QAChC,OAAO,EAAE;YACP,gBAAgB,EAAE,GAAG;YACrB,oBAAoB,EAAE,IAAI;YAC1B,mBAAmB,EAAE,IAAI;YACzB,oBAAoB,EAAE,IAAI;SAC3B;QACD,YAAY,EAAE;YACZ,gBAAgB,EAAE,MAAM;YACxB,eAAe,EAAE,IAAI;YACrB,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,eAAe,EAAE,IAAI;SACtB;KACF;IACD;QACE,OAAO,EAAE,2BAA2B;QACpC,QAAQ,EAAE,WAAW;QACrB,WAAW,EAAE,kBAAkB;QAC/B,OAAO,EAAE;YACP,gBAAgB,EAAE,IAAI;YACtB,oBAAoB,EAAE,GAAG;YACzB,mBAAmB,EAAE,IAAI;YACzB,oBAAoB,EAAE,GAAG;SAC1B;QACD,YAAY,EAAE;YACZ,gBAAgB,EAAE,MAAM;YACxB,eAAe,EAAE,IAAI;YACrB,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,eAAe,EAAE,IAAI;SACtB;KACF;IACD;QACE,OAAO,EAAE,wBAAwB;QACjC,QAAQ,EAAE,WAAW;QACrB,WAAW,EAAE,eAAe;QAC5B,OAAO,EAAE;YACP,gBAAgB,EAAE,IAAI;YACtB,oBAAoB,EAAE,IAAI;YAC1B,mBAAmB,EAAE,IAAI;YACzB,oBAAoB,EAAE,KAAK;SAC5B;QACD,YAAY,EAAE;YACZ,gBAAgB,EAAE,MAAM;YACxB,eAAe,EAAE,IAAI;YACrB,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,eAAe,EAAE,IAAI;SACtB;KACF;IACD,yBAAyB;IACzB;QACE,OAAO,EAAE,QAAQ;QACjB,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,QAAQ;QACrB,OAAO,EAAE;YACP,gBAAgB,EAAE,IAAI;YACtB,oBAAoB,EAAE,IAAI;SAC3B;QACD,YAAY,EAAE;YACZ,gBAAgB,EAAE,MAAM;YACxB,eAAe,EAAE,KAAK;YACtB,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACpB;KACF;IACD;QACE,OAAO,EAAE,aAAa;QACtB,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,aAAa;QAC1B,OAAO,EAAE;YACP,gBAAgB,EAAE,IAAI;YACtB,oBAAoB,EAAE,IAAI;SAC3B;QACD,YAAY,EAAE;YACZ,gBAAgB,EAAE,MAAM;YACxB,eAAe,EAAE,KAAK;YACtB,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACpB;KACF;IACD;QACE,OAAO,EAAE,aAAa;QACtB,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,aAAa;QAC1B,OAAO,EAAE;YACP,gBAAgB,EAAE,IAAI;YACtB,oBAAoB,EAAE,IAAI;SAC3B;QACD,YAAY,EAAE;YACZ,gBAAgB,EAAE,MAAM;YACxB,eAAe,EAAE,IAAI;YACrB,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACpB;KACF;CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,OAAO,aAAa;IAChB,MAAM,GAA6B,IAAI,GAAG,EAAE,CAAC;IAC7C,cAAc,GAAW,CAAC,CAAC;IAClB,eAAe,CAAS;IAEzC;;;OAGG;IACH,YAAY,kBAA0B,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI;QACvD,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAED;;OAEG;IACK,YAAY;QAClB,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACnC,CAAC;IAED;;;;OAIG;IACH,QAAQ,CAAC,OAAe;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAED;;;OAGG;IACH,YAAY;QACV,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED;;;;OAIG;IACH,mBAAmB,CAAC,QAAgB;QAClC,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC;IAClE,CAAC;IAED;;;OAGG;IACH,QAAQ,CAAC,MAAmB;QAC1B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED;;;;;OAKG;IACH,aAAa,CAAC,KAAkB;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAE3C,6EAA6E;QAC7E,MAAM,OAAO,GAAG,KAAK,EAAE,OAAO,IAAI;YAChC,gBAAgB,EAAE,GAAG;YACrB,oBAAoB,EAAE,IAAI;SAC3B,CAAC;QAEF,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;QAC/E,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,gBAAgB,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,oBAAoB,CAAC;QAE3F,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,IAAI,KAAK,CAAC,eAAe,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC;YACzD,SAAS,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,mBAAmB,CAAC;QACjF,CAAC;QACD,IAAI,KAAK,CAAC,gBAAgB,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC3D,SAAS,IAAI,CAAC,KAAK,CAAC,gBAAgB,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,oBAAoB,CAAC;QACnF,CAAC;QAED,MAAM,SAAS,GAAG,UAAU,GAAG,cAAc,GAAG,SAAS,CAAC;QAE1D,OAAO;YACL,UAAU;YACV,cAAc;YACd,SAAS;YACT,SAAS;YACT,SAAS,EAAE,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;SACtC,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,YAAY;QACV,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC;IACjE,CAAC;IAED;;;OAGG;IACH,YAAY;QACV,0CAA0C;QAC1C,oCAAoC;QACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACnC,CAAC;IAED;;;OAGG;IACH,WAAW;QACT,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE;YAC3B,SAAS,EAAE,IAAI,CAAC,cAAc;SAC/B,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,WAAW,CAAC,KAAmD;QAC7D,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC;IACxC,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC"}

package/dist/src/costs/types.d.ts

@@ -0,0 +1,50 @@
+/**
+ * Model configuration with pricing information.
+ */
+export interface ModelConfig {
+    /** Unique model identifier (e.g., 'claude-3-5-sonnet-20241022') */
+    modelId: string;
+    /** Provider name (e.g., 'anthropic', 'openai', 'openrouter') */
+    provider: 'anthropic' | 'openai' | 'openrouter' | string;
+    /** Human-readable display name */
+    displayName: string;
+    /** Pricing per million tokens */
+    pricing: {
+        promptPerMillion: number;
+        completionPerMillion: number;
+        /** Cache read pricing (if supported) */
+        cacheReadPerMillion?: number;
+        /** Cache write pricing (if supported) */
+        cacheWritePerMillion?: number;
+    };
+    /** Model capabilities */
+    capabilities: {
+        maxContextTokens: number;
+        maxOutputTokens: number;
+        supportsVision?: boolean;
+        supportsTools?: boolean;
+        supportsCaching?: boolean;
+    };
+}
+/**
+ * Usage record for cost calculation.
+ */
+export interface UsageRecord {
+    modelId: string;
+    promptTokens: number;
+    completionTokens: number;
+    cacheReadTokens?: number;
+    cacheWriteTokens?: number;
+}
+/**
+ * Calculated cost breakdown.
+ */
+export interface CostBreakdown {
+    promptCost: number;
+    completionCost: number;
+    cacheCost: number;
+    totalCost: number;
+    /** Cost formatted as string (e.g., '$0.0123') */
+    formatted: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/src/costs/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/costs/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,mEAAmE;IACnE,OAAO,EAAE,MAAM,CAAC;IAChB,gEAAgE;IAChE,QAAQ,EAAE,WAAW,GAAG,QAAQ,GAAG,YAAY,GAAG,MAAM,CAAC;IACzD,kCAAkC;IAClC,WAAW,EAAE,MAAM,CAAC;IACpB,iCAAiC;IACjC,OAAO,EAAE;QACP,gBAAgB,EAAE,MAAM,CAAC;QACzB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,wCAAwC;QACxC,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B,yCAAyC;QACzC,oBAAoB,CAAC,EAAE,MAAM,CAAC;KAC/B,CAAC;IACF,yBAAyB;IACzB,YAAY,EAAE;QACZ,gBAAgB,EAAE,MAAM,CAAC;QACzB,eAAe,EAAE,MAAM,CAAC;QACxB,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,aAAa,CAAC,EAAE,OAAO,CAAC;QACxB,eAAe,CAAC,EAAE,OAAO,CAAC;KAC3B,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,iDAAiD;IACjD,SAAS,EAAE,MAAM,CAAC;CACnB"}