"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.VercelKVCache = void 0;
const kv_1 = require("@vercel/kv");
const caches_1 = require("@langchain/core/caches");
/**
 * A cache that uses Vercel KV as the backing store.
 * @example
 * ```typescript
 * const cache = new VercelKVCache({
 *   ttl: 3600, // Optional: Cache entries will expire after 1 hour
 * });
 *
 * // Initialize the OpenAI model with Vercel KV cache for caching responses
 * const model = new ChatOpenAI({
 *   model: "gpt-4o-mini",
 *   cache,
 * });
 * await model.invoke("How are you today?");
 * const cachedValues = await cache.lookup("How are you today?", "llmKey");
 * ```
 */
class VercelKVCache extends caches_1.BaseCache {
    constructor(props) {
        super();
        Object.defineProperty(this, "client", {
            enumerable: true,
            configurable: true,
            writable: true,
            value: void 0
        });
        Object.defineProperty(this, "ttl", {
            enumerable: true,
            configurable: true,
            writable: true,
            value: void 0
        });
        const { client, ttl } = props;
        this.client = client ?? kv_1.kv;
        this.ttl = ttl;
    }
    /**
     * Lookup LLM generations in cache by prompt and associated LLM key.
     */
    async lookup(prompt, llmKey) {
        let idx = 0;
        let key = this.keyEncoder(prompt, llmKey, String(idx));
        let value = await this.client.get(key);
        const generations = [];
        while (value) {
            generations.push((0, caches_1.deserializeStoredGeneration)(value));
            idx += 1;
            key = this.keyEncoder(prompt, llmKey, String(idx));
            value = await this.client.get(key);
        }
        return generations.length > 0 ? generations : null;
    }
    /**
     * Update the cache with the given generations.
     *
     * Note this overwrites any existing generations for the given prompt and LLM key.
     */
    async update(prompt, llmKey, value) {
        for (let i = 0; i < value.length; i += 1) {
            const key = this.keyEncoder(prompt, llmKey, String(i));
            const serializedValue = JSON.stringify((0, caches_1.serializeGeneration)(value[i]));
            if (this.ttl) {
                await this.client.set(key, serializedValue, { ex: this.ttl });
            }
            else {
                await this.client.set(key, serializedValue);
            }
        }
    }
}
exports.VercelKVCache = VercelKVCache;
