@graphext/cuery 0.3.0 → 0.5.1

package/script/src/providers/openai.js

@@ -0,0 +1,181 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIProvider = void 0;
+/**
+ * OpenAI provider implementation.
+ */
+const dntShim = __importStar(require("../../_dnt.shims.js"));
+const openai_1 = __importDefault(require("openai"));
+const index_js_1 = require("../../deps/jsr.io/@zod/zod/4.3.6/src/index.js");
+const zodTextFormatCache = new Map();
+class ZodValidationError extends Error {
+}
+/**
+ * Sanitizes a JSON schema for OpenAI compatibility.
+ * OpenAI's structured output has strict requirements:
+ * - additionalProperties must be false (not empty object or missing)
+ * - propertyNames is not permitted
+ */
+function sanitizeSchemaForOpenAI(schema) {
+    const result = {};
+    for (const [key, value] of Object.entries(schema)) {
+        // Remove unsupported keys
+        if (key === 'propertyNames') {
+            continue;
+        }
+        // Handle additionalProperties - set to false for strict mode
+        if (key === 'additionalProperties') {
+            result[key] = false;
+            continue;
+        }
+        // Recursively sanitize nested objects
+        if (value && typeof value === 'object' && !Array.isArray(value)) {
+            result[key] = sanitizeSchemaForOpenAI(value);
+        }
+        else if (Array.isArray(value)) {
+            result[key] = value.map(item => item && typeof item === 'object' && !Array.isArray(item)
+                ? sanitizeSchemaForOpenAI(item)
+                : item);
+        }
+        else {
+            result[key] = value;
+        }
+    }
+    // For object types without additionalProperties, add it as false
+    if (result['type'] === 'object' && !('additionalProperties' in result)) {
+        result['additionalProperties'] = false;
+    }
+    return result;
+}
+function getCachedZodTextFormat(zodObject, name) {
+    const cached = zodTextFormatCache.get(zodObject);
+    if (cached && cached.name === name) {
+        return cached;
+    }
+    const format = zodTextFormat(zodObject, name);
+    zodTextFormatCache.set(zodObject, format);
+    return format;
+}
+function zodTextFormat(zodObject, name) {
+    const rawSchema = index_js_1.z.toJSONSchema(zodObject, { target: 'draft-7' });
+    const sanitizedSchema = sanitizeSchemaForOpenAI(rawSchema);
+    return {
+        type: 'json_schema',
+        name,
+        strict: true,
+        schema: sanitizedSchema,
+        $brand: 'auto-parseable-response-format',
+        $parseRaw: (content) => {
+            try {
+                return zodObject.parse(JSON.parse(content));
+            }
+            catch (error) {
+                return new ZodValidationError(error instanceof Error ? error.message : String(error));
+            }
+        },
+        __output: undefined,
+    };
+}
+/**
+ * OpenAI LLM provider.
+ */
+class OpenAIProvider {
+    name = 'openai';
+    client;
+    constructor(apiKey) {
+        const fetchOptions = {};
+        const abortSignal = dntShim.dntGlobalThis.abortSignal;
+        if (abortSignal) {
+            fetchOptions.signal = abortSignal;
+        }
+        this.client = new openai_1.default({
+            apiKey: apiKey ?? dntShim.Deno.env.get('OPENAI_API_KEY'),
+            fetchOptions,
+        });
+    }
+    async complete(messages, model, schema, params) {
+        try {
+            const response = await this.client.responses.parse({
+                ...params,
+                model,
+                input: messages.map((m) => ({
+                    role: m.role,
+                    content: m.content,
+                })),
+                ...(schema != null
+                    ? {
+                        text: {
+                            format: getCachedZodTextFormat(schema, 'response'),
+                        },
+                    }
+                    : {}),
+            });
+            const usage = response.usage
+                ? {
+                    inputTokens: response.usage.input_tokens,
+                    outputTokens: response.usage.output_tokens,
+                    totalTokens: response.usage.total_tokens,
+                }
+                : null;
+            if (response.output_parsed instanceof Error) {
+                return {
+                    parsed: null,
+                    text: response.output_text,
+                    usage,
+                    error: response.output_parsed,
+                };
+            }
+            return {
+                parsed: schema != null ? response.output_parsed : response.output_text,
+                text: response.output_text,
+                usage,
+                error: null,
+            };
+        }
+        catch (error) {
+            return {
+                parsed: null,
+                text: null,
+                usage: null,
+                error: error instanceof Error ? error : new Error(String(error)),
+            };
+        }
+    }
+}
+exports.OpenAIProvider = OpenAIProvider;

package/script/src/providers/pricing.d.ts

@@ -0,0 +1,72 @@
+/**
+ * Model pricing lookup and cost calculation.
+ *
+ * Pricing data is bundled from models.dev at build time.
+ * Run `deno task update-models` to refresh the data.
+ */
+import type { TokenUsage, UsageCost } from '../response.js';
+/**
+ * Pricing information for a model (USD per 1M tokens).
+ */
+export interface ModelPricing {
+    input: number;
+    output: number;
+    cacheRead?: number;
+    cacheWrite?: number;
+    reasoning?: number;
+}
+/**
+ * Model information including pricing and limits.
+ */
+export interface ModelInfo {
+    id: string;
+    name: string;
+    provider: string;
+    pricing: ModelPricing | null;
+    contextLimit: number | null;
+    outputLimit: number | null;
+    capabilities: {
+        structuredOutput: boolean;
+        toolCall: boolean;
+        reasoning: boolean;
+    };
+}
+interface ModelEntry {
+    id: string;
+    name: string;
+    cost?: {
+        input?: number;
+        output?: number;
+        cache_read?: number;
+        cache_write?: number;
+        reasoning?: number;
+    };
+    limit?: {
+        context?: number;
+        output?: number;
+    };
+    structured_output?: boolean;
+    tool_call?: boolean;
+    reasoning?: boolean;
+}
+/**
+ * Look up a model entry by ID (with normalization fallback).
+ */
+export declare function lookupModel(modelId: string): {
+    provider: string;
+    model: ModelEntry;
+} | null;
+/**
+ * Get pricing information for a model.
+ */
+export declare function getModelPricing(modelId: string): ModelPricing | null;
+/**
+ * Get full model information including pricing and capabilities.
+ */
+export declare function getModelInfo(modelId: string): ModelInfo | null;
+/**
+ * Calculate cost for token usage.
+ */
+export declare function calculateCost(modelId: string, usage: TokenUsage): UsageCost | null;
+export {};
+//# sourceMappingURL=pricing.d.ts.map

package/script/src/providers/pricing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.d.ts","sourceRoot":"","sources":["../../../src/src/providers/pricing.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAE5D;;GAEG;AACH,MAAM,WAAW,YAAY;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,SAAS;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,YAAY,GAAG,IAAI,CAAC;IAC7B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,YAAY,EAAE;QACb,gBAAgB,EAAE,OAAO,CAAC;QAC1B,QAAQ,EAAE,OAAO,CAAC;QAClB,SAAS,EAAE,OAAO,CAAC;KACnB,CAAC;CACF;AAGD,UAAU,UAAU;IACnB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE;QACN,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,KAAK,CAAC,EAAE;QACP,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,MAAM,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,SAAS,CAAC,EAAE,OAAO,CAAC;CACpB;AA4BD;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,UAAU,CAAA;CAAE,GAAG,IAAI,CAE3F;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,YAAY,GAAG,IAAI,CAoBpE;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI,CAsB9D;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,SAAS,GAAG,IAAI,CAgBlF"}

package/script/src/providers/pricing.js

@@ -0,0 +1,97 @@
+"use strict";
+/**
+ * Model pricing lookup and cost calculation.
+ *
+ * Pricing data is bundled from models.dev at build time.
+ * Run `deno task update-models` to refresh the data.
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.lookupModel = lookupModel;
+exports.getModelPricing = getModelPricing;
+exports.getModelInfo = getModelInfo;
+exports.calculateCost = calculateCost;
+const models_js_1 = __importDefault(require("../assets/models.js"));
+// Build flat index: model ID -> { provider, model }
+const modelIndex = new Map();
+for (const [providerId, provider] of Object.entries(models_js_1.default)) {
+    if (!provider.models)
+        continue;
+    for (const [modelId, model] of Object.entries(provider.models)) {
+        modelIndex.set(modelId, { provider: providerId, model });
+    }
+}
+/**
+ * Normalize a model ID by removing date suffixes and common variations.
+ */
+function normalizeModel(modelId) {
+    return modelId.replace(/-\d{4}-\d{2}-\d{2}$/, '');
+}
+/**
+ * Look up a model entry by ID (with normalization fallback).
+ */
+function lookupModel(modelId) {
+    return modelIndex.get(modelId) ?? modelIndex.get(normalizeModel(modelId)) ?? null;
+}
+/**
+ * Get pricing information for a model.
+ */
+function getModelPricing(modelId) {
+    const entry = lookupModel(modelId);
+    if (!entry?.model.cost) {
+        return null;
+    }
+    const cost = entry.model.cost;
+    if (cost.input === 0 && cost.output === 0) {
+        return null;
+    }
+    return {
+        input: cost.input ?? 0,
+        output: cost.output ?? 0,
+        cacheRead: cost.cache_read,
+        cacheWrite: cost.cache_write,
+        reasoning: cost.reasoning,
+    };
+}
+/**
+ * Get full model information including pricing and capabilities.
+ */
+function getModelInfo(modelId) {
+    const entry = lookupModel(modelId);
+    if (!entry) {
+        return null;
+    }
+    const { provider, model: m } = entry;
+    return {
+        id: m.id,
+        name: m.name,
+        provider,
+        pricing: getModelPricing(modelId),
+        contextLimit: m.limit?.context ?? null,
+        outputLimit: m.limit?.output ?? null,
+        capabilities: {
+            structuredOutput: m.structured_output ?? false,
+            toolCall: m.tool_call ?? false,
+            reasoning: m.reasoning ?? false,
+        },
+    };
+}
+/**
+ * Calculate cost for token usage.
+ */
+function calculateCost(modelId, usage) {
+    const pricing = getModelPricing(modelId);
+    if (!pricing) {
+        return null;
+    }
+    const inputCost = (usage.inputTokens / 1_000_000) * pricing.input;
+    const outputCost = (usage.outputTokens / 1_000_000) * pricing.output;
+    return {
+        inputCost,
+        outputCost,
+        totalCost: inputCost + outputCost,
+        currency: 'USD',
+    };
+}

package/script/src/providers/registry.d.ts

@@ -0,0 +1,20 @@
+/**
+ * Provider registry - instantiates LLM providers by name or model ID.
+ */
+import type { LLMProvider } from './types.js';
+/**
+ * Get a provider by name.
+ * Creates a fresh provider instance each time to support different API keys.
+ * @param name - Provider name ('openai', 'google')
+ * @param apiKey - Optional API key. If not provided, uses environment variable.
+ */
+export declare function getProvider(name: string, apiKey?: string): LLMProvider;
+/**
+ * Get the appropriate provider for a model ID.
+ * Infers provider from model name prefix.
+ * Creates a fresh provider instance each time to support different API keys.
+ * @param modelId - Model identifier (e.g., 'gpt-4.1', 'gemini-2.0-flash')
+ * @param apiKey - Optional API key. If not provided, uses environment variable.
+ */
+export declare function getProviderForModel(modelId: string, apiKey?: string): LLMProvider;
+//# sourceMappingURL=registry.d.ts.map

package/script/src/providers/registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../../src/src/providers/registry.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAI9C;;;;;GAKG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,WAAW,CAQtE;AAED;;;;;;GAMG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,WAAW,CASjF"}

package/script/src/providers/registry.js

@@ -0,0 +1,39 @@
+"use strict";
+/**
+ * Provider registry - instantiates LLM providers by name or model ID.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getProvider = getProvider;
+exports.getProviderForModel = getProviderForModel;
+const openai_js_1 = require("./openai.js");
+const google_js_1 = require("./google.js");
+/**
+ * Get a provider by name.
+ * Creates a fresh provider instance each time to support different API keys.
+ * @param name - Provider name ('openai', 'google')
+ * @param apiKey - Optional API key. If not provided, uses environment variable.
+ */
+function getProvider(name, apiKey) {
+    switch (name) {
+        case 'google':
+            return new google_js_1.GoogleProvider(apiKey);
+        case 'openai':
+        default:
+            return new openai_js_1.OpenAIProvider(apiKey);
+    }
+}
+/**
+ * Get the appropriate provider for a model ID.
+ * Infers provider from model name prefix.
+ * Creates a fresh provider instance each time to support different API keys.
+ * @param modelId - Model identifier (e.g., 'gpt-4.1', 'gemini-2.0-flash')
+ * @param apiKey - Optional API key. If not provided, uses environment variable.
+ */
+function getProviderForModel(modelId, apiKey) {
+    const modelLower = modelId.toLowerCase();
+    if (modelLower.startsWith('gemini')) {
+        return new google_js_1.GoogleProvider(apiKey);
+    }
+    // Default to OpenAI for gpt-*, o1-*, o3-*, and unknown models
+    return new openai_js_1.OpenAIProvider(apiKey);
+}

package/script/src/providers/types.d.ts

@@ -0,0 +1,49 @@
+/**
+ * Core LLM types - provider-agnostic interfaces.
+ *
+ * This module is the foundation of the provider system.
+ * It has no dependencies on other modules to avoid circular imports.
+ */
+import type { z } from '../../deps/jsr.io/@zod/zod/4.3.6/src/index.js';
+import type { TokenUsage } from '../response.js';
+/**
+ * A message in an LLM conversation.
+ */
+export interface Message {
+    role: 'system' | 'user' | 'assistant';
+    content: string;
+}
+/**
+ * Response from an LLM provider.
+ */
+export interface LLMResponse<T> {
+    parsed: T | null;
+    text: string | null;
+    usage: TokenUsage | null;
+    error: Error | null;
+}
+/**
+ * Provider-specific parameters passed through to the underlying API.
+ */
+export type ProviderParams = Record<string, unknown>;
+/**
+ * Interface for LLM providers.
+ */
+export interface LLMProvider {
+    /** Provider name (e.g., 'openai', 'gemini') */
+    readonly name: string;
+    /**
+     * Make a completion request to the LLM.
+     *
+     * @param messages - The conversation messages
+     * @param model - The model identifier
+     * @param schema - Optional Zod schema for structured output
+     * @param params - Provider-specific parameters
+     */
+    complete<T>(messages: Message[], model: string, schema: z.ZodType<T> | null, params?: ProviderParams): Promise<LLMResponse<T>>;
+}
+/**
+ * Conversation type (array of messages).
+ */
+export type LLMConversation = Message[];
+//# sourceMappingURL=types.d.ts.map

package/script/src/providers/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/src/providers/types.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,+CAA+C,CAAC;AACvE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAEjD;;GAEG;AACH,MAAM,WAAW,OAAO;IACvB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,WAAW,CAAC,CAAC;IAC7B,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC;IACjB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,KAAK,EAAE,UAAU,GAAG,IAAI,CAAC;IACzB,KAAK,EAAE,KAAK,GAAG,IAAI,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAErD;;GAEG;AACH,MAAM,WAAW,WAAW;IAC3B,+CAA+C;IAC/C,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAEtB;;;;;;;OAOG;IACH,QAAQ,CAAC,CAAC,EACT,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,IAAI,EAC3B,MAAM,CAAC,EAAE,cAAc,GACrB,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,OAAO,EAAE,CAAC"}

package/script/src/providers/types.js

@@ -0,0 +1,8 @@
+"use strict";
+/**
+ * Core LLM types - provider-agnostic interfaces.
+ *
+ * This module is the foundation of the provider system.
+ * It has no dependencies on other modules to avoid circular imports.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });

package/script/src/response.d.ts

@@ -0,0 +1,74 @@
+/**
+ * BatchResponse - A wrapper for batch LLM results with usage tracking.
+ *
+ * Provides an array-like interface for backwards compatibility while
+ * adding usage aggregation capabilities.
+ */
+/**
+ * Token usage from a single LLM call.
+ */
+export interface TokenUsage {
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+}
+/**
+ * Cost calculation result in USD.
+ */
+export interface UsageCost {
+    inputCost: number;
+    outputCost: number;
+    totalCost: number;
+    currency: 'USD';
+}
+/**
+ * Aggregated usage across multiple LLM calls.
+ */
+export interface AggregatedUsage {
+    tokens: TokenUsage;
+    cost: UsageCost | null;
+    callCount: number;
+    model: string | null;
+}
+/**
+ * A batch of LLM results with usage tracking.
+ *
+ * Implements Iterable<T> and provides array-like methods for backwards
+ * compatibility with code expecting Array<T>.
+ */
+export declare class BatchResponse<T> implements Iterable<T> {
+    readonly results: ReadonlyArray<T>;
+    private readonly _tokenUsages;
+    private readonly _model;
+    constructor(results: T[], tokenUsages?: Array<TokenUsage | null | undefined>, model?: string);
+    get length(): number;
+    [Symbol.iterator](): Iterator<T>;
+    at(index: number): T | undefined;
+    map<U>(fn: (value: T, index: number) => U): BatchResponse<U>;
+    filter(fn: (value: T, index: number) => boolean): BatchResponse<T>;
+    forEach(fn: (value: T, index: number) => void): void;
+    find(fn: (value: T, index: number) => boolean): T | undefined;
+    findIndex(fn: (value: T, index: number) => boolean): number;
+    some(fn: (value: T, index: number) => boolean): boolean;
+    every(fn: (value: T, index: number) => boolean): boolean;
+    reduce<U>(fn: (acc: U, value: T, index: number) => U, initial: U): U;
+    /**
+     * Convert to a plain array.
+     */
+    toArray(): T[];
+    /**
+     * Get aggregated usage across all calls in this batch.
+     * Cost is calculated lazily only if model and costCalculator were provided.
+     * Returns null if usage tracking was not enabled.
+     */
+    usage(): AggregatedUsage | null;
+    /**
+     * Get token usage for a specific index.
+     */
+    usageAt(index: number): TokenUsage | undefined;
+    /**
+     * Check if any usage info is available.
+     */
+    hasUsage(): boolean;
+}
+//# sourceMappingURL=response.d.ts.map

package/script/src/response.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"response.d.ts","sourceRoot":"","sources":["../../src/src/response.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH;;GAEG;AACH,MAAM,WAAW,UAAU;IAC1B,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,SAAS;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,KAAK,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC/B,MAAM,EAAE,UAAU,CAAC;IACnB,IAAI,EAAE,SAAS,GAAG,IAAI,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CACrB;AAED;;;;;GAKG;AACH,qBAAa,aAAa,CAAC,CAAC,CAAE,YAAW,QAAQ,CAAC,CAAC,CAAC;IACnD,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAuC;IACpE,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgB;gBAGtC,OAAO,EAAE,CAAC,EAAE,EACZ,WAAW,CAAC,EAAE,KAAK,CAAC,UAAU,GAAG,IAAI,GAAG,SAAS,CAAC,EAClD,KAAK,CAAC,EAAE,MAAM;IAUf,IAAI,MAAM,IAAI,MAAM,CAEnB;IAED,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC;IAIhC,EAAE,CAAC,KAAK,EAAE,MAAM,GAAG,CAAC,GAAG,SAAS;IAIhC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC;IAK5D,MAAM,CAAC,EAAE,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,GAAG,aAAa,CAAC,CAAC,CAAC;IAclE,OAAO,CAAC,EAAE,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAIpD,IAAI,CAAC,EAAE,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,GAAG,CAAC,GAAG,SAAS;IAI7D,SAAS,CAAC,EAAE,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,GAAG,MAAM;IAI3D,IAAI,CAAC,EAAE,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,GAAG,OAAO;IAIvD,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,GAAG,OAAO;IAIxD,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC;IAIpE;;OAEG;IACH,OAAO,IAAI,CAAC,EAAE;IAMd;;;;OAIG;IACH,KAAK,IAAI,eAAe,GAAG,IAAI;IAwB/B;;OAEG;IACH,OAAO,CAAC,KAAK,EAAE,MAAM,GAAG,UAAU,GAAG,SAAS;IAI9C;;OAEG;IACH,QAAQ,IAAI,OAAO;CAGnB"}

package/script/src/response.js

@@ -0,0 +1,114 @@
+"use strict";
+/**
+ * BatchResponse - A wrapper for batch LLM results with usage tracking.
+ *
+ * Provides an array-like interface for backwards compatibility while
+ * adding usage aggregation capabilities.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BatchResponse = void 0;
+const pricing_js_1 = require("./providers/pricing.js");
+/**
+ * A batch of LLM results with usage tracking.
+ *
+ * Implements Iterable<T> and provides array-like methods for backwards
+ * compatibility with code expecting Array<T>.
+ */
+class BatchResponse {
+    results;
+    _tokenUsages;
+    _model;
+    constructor(results, tokenUsages, model) {
+        this.results = results;
+        // Normalize null to undefined for internal storage, or null if not provided
+        this._tokenUsages = tokenUsages ? tokenUsages.map(u => u ?? undefined) : null;
+        this._model = model ?? null;
+    }
+    // Array-like interface
+    get length() {
+        return this.results.length;
+    }
+    [Symbol.iterator]() {
+        return this.results[Symbol.iterator]();
+    }
+    at(index) {
+        return this.results.at(index);
+    }
+    map(fn) {
+        const mapped = this.results.map((item, i) => fn(item, i));
+        return new BatchResponse(mapped, this._tokenUsages ?? undefined, this._model ?? undefined);
+    }
+    filter(fn) {
+        const filtered = [];
+        const usages = this._tokenUsages ? [] : undefined;
+        this.results.forEach((item, i) => {
+            if (fn(item, i)) {
+                filtered.push(item);
+                usages?.push(this._tokenUsages[i]);
+            }
+        });
+        return new BatchResponse(filtered, usages, this._model ?? undefined);
+    }
+    forEach(fn) {
+        this.results.forEach(fn);
+    }
+    find(fn) {
+        return this.results.find(fn);
+    }
+    findIndex(fn) {
+        return this.results.findIndex(fn);
+    }
+    some(fn) {
+        return this.results.some(fn);
+    }
+    every(fn) {
+        return this.results.every(fn);
+    }
+    reduce(fn, initial) {
+        return this.results.reduce(fn, initial);
+    }
+    /**
+     * Convert to a plain array.
+     */
+    toArray() {
+        return [...this.results];
+    }
+    // Usage tracking
+    /**
+     * Get aggregated usage across all calls in this batch.
+     * Cost is calculated lazily only if model and costCalculator were provided.
+     * Returns null if usage tracking was not enabled.
+     */
+    usage() {
+        if (!this._tokenUsages) {
+            return null;
+        }
+        const valid = this._tokenUsages.filter((u) => u !== undefined);
+        const aggregatedTokens = {
+            inputTokens: valid.reduce((sum, u) => sum + u.inputTokens, 0),
+            outputTokens: valid.reduce((sum, u) => sum + u.outputTokens, 0),
+            totalTokens: valid.reduce((sum, u) => sum + u.totalTokens, 0),
+        };
+        // Calculate cost if we have a model
+        const cost = this._model ? (0, pricing_js_1.calculateCost)(this._model, aggregatedTokens) : null;
+        return {
+            tokens: aggregatedTokens,
+            cost,
+            callCount: valid.length,
+            model: this._model,
+        };
+    }
+    /**
+     * Get token usage for a specific index.
+     */
+    usageAt(index) {
+        return this._tokenUsages?.[index];
+    }
+    /**
+     * Check if any usage info is available.
+     */
+    hasUsage() {
+        return this._tokenUsages?.some((u) => u !== undefined) ?? false;
+    }
+}
+exports.BatchResponse = BatchResponse;

package/script/src/schemas/brand.schema.d.ts

@@ -1,4 +1,4 @@
-import { z } from '../../deps/jsr.io/@zod/zod/4.1.12/src/index.js';
+import { z } from '../../deps/jsr.io/@zod/zod/4.3.6/src/index.js';
 import type { SearchOptions } from './search.schema.js';
 /**
  * Zod schemas for brand information.