@langchain/community/dist/caches/ioredis.js

Third-party integrations for LangChain.js

import { __exportAll } from "../_virtual/_rolldown/runtime.js";
import { BaseCache, deserializeStoredGeneration, serializeGeneration } from "@langchain/core/caches";
//#region src/caches/ioredis.ts
var ioredis_exports = /* @__PURE__ */ __exportAll({ RedisCache: () => RedisCache });
/**
* Cache LLM results using Redis.
* @example
* ```typescript
* const model = new ChatOpenAI({
*   model: "gpt-4o-mini",
*   cache: new RedisCache(new Redis(), { ttl: 60 }),
* });
*
* // Invoke the model with a prompt
* const response = await model.invoke("Do something random!");
* console.log(response);
*
* // Remember to disconnect the Redis client when done
* await redisClient.disconnect();
* ```
*/
var RedisCache = class extends BaseCache {
	redisClient;
	ttl;
	constructor(redisClient, config) {
		super();
		this.redisClient = redisClient;
		this.ttl = config?.ttl;
	}
	/**
	* Retrieves data from the Redis server using a prompt and an LLM key. If
	* the data is not found, it returns null.
	* @param prompt The prompt used to find the data.
	* @param llmKey The LLM key used to find the data.
	* @returns The corresponding data as an array of Generation objects, or null if not found.
	*/
	async lookup(prompt, llmKey) {
		let idx = 0;
		let key = this.keyEncoder(prompt, llmKey, String(idx));
		let value = await this.redisClient.get(key);
		const generations = [];
		while (value) {
			const storedGeneration = JSON.parse(value);
			generations.push(deserializeStoredGeneration(storedGeneration));
			idx += 1;
			key = this.keyEncoder(prompt, llmKey, String(idx));
			value = await this.redisClient.get(key);
		}
		return generations.length > 0 ? generations : null;
	}
	/**
	* Updates the data in the Redis server using a prompt and an LLM key.
	* @param prompt The prompt used to store the data.
	* @param llmKey The LLM key used to store the data.
	* @param value The data to be stored, represented as an array of Generation objects.
	*/
	async update(prompt, llmKey, value) {
		for (let i = 0; i < value.length; i += 1) {
			const key = this.keyEncoder(prompt, llmKey, String(i));
			if (this.ttl !== void 0) await this.redisClient.set(key, JSON.stringify(serializeGeneration(value[i])), "EX", this.ttl);
			else await this.redisClient.set(key, JSON.stringify(serializeGeneration(value[i])));
		}
	}
};
//#endregion
export { RedisCache, ioredis_exports };

//# sourceMappingURL=ioredis.js.map