@polka-codes/core 0.10.11 → 0.10.16

package/dist/Agent/backoff.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Utility to compute exponential backoff delays for rate-limit handling. generated by polka.codes
+ *
+ * The backoff starts at baseSeconds and doubles each time, capped at capSeconds.
+ * Example with base=2, cap=60: 2, 4, 8, 16, 32, 60, 60, ...
+ */
+export declare function computeRateLimitBackoffSeconds(count: number, baseSeconds?: number, capSeconds?: number): number;

package/dist/Agent/backoff.js

@@ -0,0 +1,21 @@
+/**
+ * Utility to compute exponential backoff delays for rate-limit handling. generated by polka.codes
+ *
+ * The backoff starts at baseSeconds and doubles each time, capped at capSeconds.
+ * Example with base=2, cap=60: 2, 4, 8, 16, 32, 60, 60, ...
+ */
+export function computeRateLimitBackoffSeconds(count, baseSeconds = 2, capSeconds = 60) {
+    if (!Number.isFinite(count) || count <= 0) {
+        // Treat non-positive or invalid counts as first occurrence
+        count = 1;
+    }
+    if (!Number.isFinite(baseSeconds) || baseSeconds <= 0) {
+        baseSeconds = 2;
+    }
+    if (!Number.isFinite(capSeconds) || capSeconds <= 0) {
+        capSeconds = 60;
+    }
+    const delay = baseSeconds * 2 ** (count - 1);
+    return Math.min(delay, capSeconds);
+}
+//# sourceMappingURL=backoff.js.map

package/dist/Agent/backoff.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"backoff.js","sourceRoot":"","sources":["../../src/Agent/backoff.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,MAAM,UAAU,8BAA8B,CAAC,KAAa,EAAE,WAAW,GAAG,CAAC,EAAE,UAAU,GAAG,EAAE;IAC5F,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QAC1C,2DAA2D;QAC3D,KAAK,GAAG,CAAC,CAAA;IACX,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,WAAW,IAAI,CAAC,EAAE,CAAC;QACtD,WAAW,GAAG,CAAC,CAAA;IACjB,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;QACpD,UAAU,GAAG,EAAE,CAAA;IACjB,CAAC;IAED,MAAM,KAAK,GAAG,WAAW,GAAG,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAA;IAC5C,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAA;AACpC,CAAC"}

package/dist/Agent/index.d.ts

@@ -0,0 +1,2 @@
+export * from './parseJsonFromMarkdown';
+export * from './prompts';

package/dist/Agent/index.js

@@ -0,0 +1,3 @@
+export * from './parseJsonFromMarkdown';
+export * from './prompts';
+//# sourceMappingURL=index.js.map

package/dist/Agent/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/Agent/index.ts"],"names":[],"mappings":"AAAA,cAAc,yBAAyB,CAAA;AACvC,cAAc,WAAW,CAAA"}

package/dist/Agent/parseJsonFromMarkdown.d.ts

@@ -0,0 +1,8 @@
+export type ParseOutputResult<T> = {
+    success: true;
+    data: T;
+} | {
+    success: false;
+    error?: string;
+};
+export declare const parseJsonFromMarkdown: (markdown: string) => ParseOutputResult<any>;

package/dist/Agent/parseJsonFromMarkdown.js

@@ -0,0 +1,34 @@
+export const parseJsonFromMarkdown = (markdown) => {
+    const jsonRegex = /```(?:json)?\n([\s\S]*?)\n```/;
+    const match = markdown.match(jsonRegex);
+    const tryParse = (str) => {
+        try {
+            let parsed = JSON.parse(str);
+            // The model sometimes returns a stringified JSON object within the JSON.
+            if (typeof parsed === 'string') {
+                try {
+                    parsed = JSON.parse(parsed);
+                }
+                catch {
+                    // ignore if the inner string is not a valid JSON
+                }
+            }
+            return { success: true, data: parsed };
+        }
+        catch (e) {
+            const error = e instanceof Error ? e.message : String(e);
+            return { success: false, error: `Failed to parse JSON: ${error}` };
+        }
+    };
+    if (match?.[1]) {
+        const content = match[1].trim();
+        return tryParse(content);
+    }
+    // Fallback for cases where the output is just the JSON string without markdown fences
+    const parseResult = tryParse(markdown);
+    if (parseResult.success) {
+        return parseResult;
+    }
+    return { success: false, error: 'No JSON object found in the string.' };
+};
+//# sourceMappingURL=parseJsonFromMarkdown.js.map

package/dist/Agent/parseJsonFromMarkdown.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parseJsonFromMarkdown.js","sourceRoot":"","sources":["../../src/Agent/parseJsonFromMarkdown.ts"],"names":[],"mappings":"AAGA,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,QAAgB,EAA0B,EAAE;IAChF,MAAM,SAAS,GAAG,+BAA+B,CAAA;IACjD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAA;IAEvC,MAAM,QAAQ,GAAG,CAAC,GAAW,EAAE,EAAE;QAC/B,IAAI,CAAC;YACH,IAAI,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;YAC5B,yEAAyE;YACzE,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;gBAC7B,CAAC;gBAAC,MAAM,CAAC;oBACP,iDAAiD;gBACnD,CAAC;YACH,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAW,CAAA;QACjD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;YACxD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,yBAAyB,KAAK,EAAE,EAAW,CAAA;QAC7E,CAAC;IACH,CAAC,CAAA;IAED,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;QAC/B,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAA;IAC1B,CAAC;IAED,sFAAsF;IACtF,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAA;IACtC,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;QACxB,OAAO,WAAW,CAAA;IACpB,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,qCAAqC,EAAW,CAAA;AAClF,CAAC,CAAA"}

package/dist/Agent/parseJsonFromMarkdown.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/Agent/parseJsonFromMarkdown.test.js

@@ -0,0 +1,70 @@
+// generated by polka.codes
+import { describe, expect, it } from 'bun:test';
+import { parseJsonFromMarkdown } from './parseJsonFromMarkdown';
+describe('parseJsonFromMarkdown', () => {
+    it('should parse JSON from a markdown code block', () => {
+        const markdown = 'Here is the JSON:\n\n```json\n{"key":"value"}\n```';
+        expect(parseJsonFromMarkdown(markdown)).toEqual({
+            success: true,
+            data: { key: 'value' },
+        });
+    });
+    it('should return an error if no JSON object is found', () => {
+        const markdown = 'This is just a string.';
+        expect(parseJsonFromMarkdown(markdown)).toEqual({
+            success: false,
+            error: 'No JSON object found in the string.',
+        });
+    });
+    it('should handle JSON without markdown fences', () => {
+        const json = '{"key":"value"}';
+        expect(parseJsonFromMarkdown(json)).toEqual({
+            success: true,
+            data: { key: 'value' },
+        });
+    });
+    it('should return an error for malformed JSON in a code block', () => {
+        const markdown = '```json\n{"key":"value",,}\n```';
+        const result = parseJsonFromMarkdown(markdown);
+        expect(result.success).toBe(false);
+        if (!result.success) {
+            expect(result.error).toMatch(/Failed to parse JSON/);
+        }
+    });
+    it('should parse JSON from a markdown code block without the json identifier', () => {
+        const markdown = 'Here is the JSON:\n\n```\n{"key":"value"}\n```';
+        expect(parseJsonFromMarkdown(markdown)).toEqual({
+            success: true,
+            data: { key: 'value' },
+        });
+    });
+    it('should parse an escaped JSON string from a markdown code block', () => {
+        const markdown = '```json\n"{\\"key\\":\\"value\\"}"\n```';
+        expect(parseJsonFromMarkdown(markdown)).toEqual({
+            success: true,
+            data: { key: 'value' },
+        });
+    });
+    it('should parse an escaped JSON string from a markdown code block without the json identifier', () => {
+        const markdown = '```\n"{\\"key\\":\\"value\\"}"\n```';
+        expect(parseJsonFromMarkdown(markdown)).toEqual({
+            success: true,
+            data: { key: 'value' },
+        });
+    });
+    it('should parse a JSON array from a markdown code block', () => {
+        const markdown = '```json\n[{"key":"value"}]\n```';
+        expect(parseJsonFromMarkdown(markdown)).toEqual({
+            success: true,
+            data: [{ key: 'value' }],
+        });
+    });
+    it('should parse a JSON array from a markdown code block without the json identifier', () => {
+        const markdown = '```\n[{"key":"value"}]\n```';
+        expect(parseJsonFromMarkdown(markdown)).toEqual({
+            success: true,
+            data: [{ key: 'value' }],
+        });
+    });
+});
+//# sourceMappingURL=parseJsonFromMarkdown.test.js.map

package/dist/Agent/parseJsonFromMarkdown.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parseJsonFromMarkdown.test.js","sourceRoot":"","sources":["../../src/Agent/parseJsonFromMarkdown.test.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,UAAU,CAAA;AAC/C,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAA;AAE/D,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAG,oDAAoD,CAAA;QACrE,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9C,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;SACvB,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,QAAQ,GAAG,wBAAwB,CAAA;QACzC,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9C,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,qCAAqC;SAC7C,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,IAAI,GAAG,iBAAiB,CAAA;QAC9B,MAAM,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1C,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;SACvB,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,QAAQ,GAAG,iCAAiC,CAAA;QAClD,MAAM,MAAM,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAA;QAC9C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAClC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAA;QACtD,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;QAClF,MAAM,QAAQ,GAAG,gDAAgD,CAAA;QACjE,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9C,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;SACvB,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,QAAQ,GAAG,yCAAyC,CAAA;QAC1D,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9C,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;SACvB,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4FAA4F,EAAE,GAAG,EAAE;QACpG,MAAM,QAAQ,GAAG,qCAAqC,CAAA;QACtD,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9C,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;SACvB,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,QAAQ,GAAG,iCAAiC,CAAA;QAClD,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9C,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;SACzB,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;QAC1F,MAAM,QAAQ,GAAG,6BAA6B,CAAA;QAC9C,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9C,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;SACzB,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/Agent/prompts.d.ts

@@ -0,0 +1,9 @@
+import type { LanguageModelV3ToolResultOutput } from '@ai-sdk/provider';
+import type { FilePart, ImagePart, TextPart } from 'ai';
+export declare const responsePrompts: {
+    readonly errorInvokeTool: (tool: string, error: unknown) => string;
+    readonly requireUseTool: "Error: No tool use detected. You MUST use a tool before proceeding.\ne.g. <tool_tool_name>tool_name</tool_tool_name>\n\nEnsure the opening and closing tags are correctly nested and closed, and that you are using the correct tool name.\nAvoid unnecessary text or symbols before or after the tool use.\nAvoid unnecessary escape characters or special characters.\n";
+    readonly requireUseToolNative: "Error: No tool use detected. You MUST use a tool before proceeding.\n";
+    readonly toolResults: (tool: string, result: LanguageModelV3ToolResultOutput) => Array<TextPart | ImagePart | FilePart>;
+    readonly commandResult: (command: string, exitCode: number, stdout: string, stderr: string) => string;
+};

package/dist/Agent/prompts.js

@@ -0,0 +1,107 @@
+const NO_REASON_PROVIDED = 'No reason provided';
+export const responsePrompts = {
+    errorInvokeTool: (tool, error) => `An error occurred while invoking the tool "${tool}": ${error}`,
+    requireUseTool: `Error: No tool use detected. You MUST use a tool before proceeding.
+e.g. <tool_tool_name>tool_name</tool_tool_name>
+
+Ensure the opening and closing tags are correctly nested and closed, and that you are using the correct tool name.
+Avoid unnecessary text or symbols before or after the tool use.
+Avoid unnecessary escape characters or special characters.
+`,
+    requireUseToolNative: `Error: No tool use detected. You MUST use a tool before proceeding.
+`,
+    toolResults: (tool, result) => {
+        switch (result.type) {
+            case 'text':
+                return [
+                    {
+                        type: 'text',
+                        text: `<tool_response name=${tool}>${result.value}</tool_response>`,
+                    },
+                ];
+            case 'error-text':
+                return [
+                    {
+                        type: 'text',
+                        text: `<tool_response_error name=${tool}>${result.value}</tool_response_error>`,
+                    },
+                ];
+            case 'json':
+                return [
+                    {
+                        type: 'text',
+                        text: `<tool_response_json name=${tool}>${JSON.stringify(result.value)}</tool_response_json>`,
+                    },
+                ];
+            case 'error-json':
+                return [
+                    {
+                        type: 'text',
+                        text: `<tool_response_error_json name=${tool}>${JSON.stringify(result.value)}</tool_response_error_json>`,
+                    },
+                ];
+            case 'content':
+                return [
+                    {
+                        type: 'text',
+                        text: `<tool_response name=${tool}>`,
+                    },
+                    ...result.value.map((part) => {
+                        // Handle text parts directly
+                        if (part.type === 'text') {
+                            return part;
+                        }
+                        // Handle image-data parts
+                        if ('mediaType' in part && 'data' in part && part.mediaType.startsWith('image/')) {
+                            return {
+                                type: 'image',
+                                mediaType: part.mediaType,
+                                image: part.data,
+                            };
+                        }
+                        // Handle file-data parts
+                        if ('mediaType' in part && 'data' in part) {
+                            return {
+                                type: 'file',
+                                mediaType: part.mediaType,
+                                data: part.data,
+                            };
+                        }
+                        // Fallback for URL-based and other part types
+                        return {
+                            type: 'text',
+                            text: JSON.stringify(part),
+                        };
+                    }),
+                    {
+                        type: 'text',
+                        text: '</tool_response>',
+                    },
+                ];
+            case 'execution-denied':
+                return [
+                    {
+                        type: 'text',
+                        text: `<tool_response_error name=${tool}>Execution denied: ${result.reason ?? NO_REASON_PROVIDED}</tool_response_error>`,
+                    },
+                ];
+            default: {
+                return [
+                    {
+                        type: 'text',
+                        text: `<tool_response_error name=${tool}>Unknown result type: ${JSON.stringify(result)}</tool_response_error>`,
+                    },
+                ];
+            }
+        }
+    },
+    commandResult: (command, exitCode, stdout, stderr) => `<command>${command}</command>
+<command_exit_code>${exitCode}</command_exit_code>
+<command_stdout>
+${stdout}
+</command_stdout>
+<command_stderr>
+${stderr}
+</command_stderr>`,
+};
+//# sourceMappingURL=prompts.js.map

package/dist/Agent/prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompts.js","sourceRoot":"","sources":["../../src/Agent/prompts.ts"],"names":[],"mappings":"AAGA,MAAM,kBAAkB,GAAG,oBAAoB,CAAA;AAE/C,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,eAAe,EAAE,CAAC,IAAY,EAAE,KAAc,EAAE,EAAE,CAAC,8CAA8C,IAAI,MAAM,KAAK,EAAE;IAClH,cAAc,EAAE;;;;;;CAMjB;IACC,oBAAoB,EAAE;CACvB;IACC,WAAW,EAAE,CAAC,IAAY,EAAE,MAAuC,EAA0C,EAAE;QAC7G,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;YACpB,KAAK,MAAM;gBACT,OAAO;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,uBAAuB,IAAI,IAAI,MAAM,CAAC,KAAK,kBAAkB;qBACpE;iBACF,CAAA;YACH,KAAK,YAAY;gBACf,OAAO;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,6BAA6B,IAAI,IAAI,MAAM,CAAC,KAAK,wBAAwB;qBAChF;iBACF,CAAA;YACH,KAAK,MAAM;gBACT,OAAO;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,4BAA4B,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB;qBAC9F;iBACF,CAAA;YACH,KAAK,YAAY;gBACf,OAAO;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,kCAAkC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B;qBAC1G;iBACF,CAAA;YACH,KAAK,SAAS;gBACZ,OAAO;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,uBAAuB,IAAI,GAAG;qBACrC;oBACD,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;wBAC3B,6BAA6B;wBAC7B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BACzB,OAAO,IAAI,CAAA;wBACb,CAAC;wBACD,0BAA0B;wBAC1B,IAAI,WAAW,IAAI,IAAI,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;4BACjF,OAAO;gCACL,IAAI,EAAE,OAAO;gCACb,SAAS,EAAE,IAAI,CAAC,SAAS;gCACzB,KAAK,EAAE,IAAI,CAAC,IAAI;6BACR,CAAA;wBACZ,CAAC;wBACD,yBAAyB;wBACzB,IAAI,WAAW,IAAI,IAAI,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;4BAC1C,OAAO;gCACL,IAAI,EAAE,MAAM;gCACZ,SAAS,EAAE,IAAI,CAAC,SAAS;gCACzB,IAAI,EAAE,IAAI,CAAC,IAAI;6BACP,CAAA;wBACZ,CAAC;wBACD,8CAA8C;wBAC9C,OAAO;4BACL,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;yBACR,CAAA;oBACtB,CAAC,CAAC;oBACF;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,kBAAkB;qBACzB;iBACF,CAAA;YACH,KAAK,kBAAkB;gBACrB,OAAO;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,6BAA6B,IAAI,sBAAsB,MAAM,CAAC,MAAM,IAAI,kBAAkB,wBAAwB;qBACzH;iBACF,CAAA;YACH,OAAO,CAAC,CAAC,CAAC;gBACR,OAAO;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,6BAA6B,IAAI,yBAAyB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,wBAAwB;qBAC/G;iBACF,CAAA;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,aAAa,EAAE,CAAC,OAAe,EAAE,QAAgB,EAAE,MAAc,EAAE,MAAc,EAAE,EAAE,CAAC,YAAY,OAAO;qBACtF,QAAQ;;EAE3B,MAAM;;;EAGN,MAAM;kBACU;CACR,CAAA"}

package/dist/UsageMeter.d.ts

@@ -0,0 +1,101 @@
+/**
+ * UsageMeter class for tracking API usage metrics
+ * Generated by polka.codes
+ */
+import type { LanguageModelV3, LanguageModelV3Usage } from '@ai-sdk/provider';
+import type { LanguageModelUsage } from 'ai';
+import type { PricingService } from './pricing/index';
+export type ModelInfo = {
+    inputPrice: number;
+    outputPrice: number;
+    cacheWritesPrice: number;
+    cacheReadsPrice: number;
+};
+type Totals = {
+    input: number;
+    output: number;
+    cachedRead: number;
+    cost: number;
+    messageCount: number;
+};
+type ProviderMetadataEntry = {
+    provider: string;
+    model: string;
+    metadata: Record<string, unknown>;
+    timestamp: number;
+};
+/**
+ * Tracks token / cost usage across any mix of LLM models.
+ * Supports optional caps on total messages and total cost.
+ */
+export declare class UsageMeter {
+    #private;
+    constructor(modelInfos?: Record<string, Record<string, Partial<ModelInfo>>>, opts?: {
+        maxMessages?: number;
+        maxCost?: number;
+        pricingService?: PricingService;
+    });
+    addUsage(llm: LanguageModelV3, resp: {
+        usage: LanguageModelV3Usage;
+        providerMetadata?: Record<string, unknown>;
+    } | {
+        totalUsage: LanguageModelV3Usage;
+        providerMetadata?: Record<string, unknown>;
+    } | {
+        totalUsage: LanguageModelUsage;
+        providerMetadata?: Record<string, unknown>;
+    }, options?: {
+        modelInfo?: ModelInfo;
+    }): Promise<void>;
+    /** Override the running totals (e.g., restore from saved state). */
+    setUsage(newUsage: Partial<Totals>, options?: {
+        clearMetadata?: boolean;
+    }): void;
+    /** Manually bump the message count (useful if you record some messages without token info). */
+    incrementMessageCount(n?: number): void;
+    /** Reset the running totals. */
+    resetUsage(): void;
+    /** Return true once either messages or cost exceed the configured caps. */
+    isLimitExceeded(): {
+        messageCount: boolean;
+        maxMessages: number;
+        cost: boolean;
+        maxCost: number;
+        result: boolean;
+    };
+    /** Same as isLimitExceeded but throws an error if a limit is hit. */
+    checkLimit(): void;
+    /** Getter for the aggregated totals (immutable copy). */
+    get usage(): {
+        input: number;
+        output: number;
+        cachedRead: number;
+        cost: number;
+        messageCount: number;
+    };
+    /** Getter for provider metadata entries (immutable copy). */
+    get providerMetadata(): ProviderMetadataEntry[];
+    /** Calculate cache statistics from stored metadata entries. */
+    get cacheStats(): {
+        totalCachedTokens: number;
+        totalRequests: number;
+        requestsWithCache: number;
+        cacheHitRate: number;
+        entries: ProviderMetadataEntry[];
+    };
+    /** Clear stored provider metadata entries. */
+    clearProviderMetadata(): void;
+    /** Merge another UsageMeter's totals into this one. */
+    merge(other: UsageMeter): void;
+    /** Wait for all pending usage updates to complete. */
+    waitForPending(): Promise<void>;
+    getUsageText(): string;
+    onFinishHandler(llm: LanguageModelV3): (evt: {
+        totalUsage: LanguageModelUsage;
+        providerMetadata?: Record<string, unknown>;
+    } | {
+        totalUsage: LanguageModelV3Usage;
+        providerMetadata?: Record<string, unknown>;
+    }) => Promise<void>;
+}
+export {};