langchain 0.0.167 → 0.0.169

package/dist/llms/cloudflare_workersai.js

@@ -0,0 +1,120 @@
+import { LLM } from "./base.js";
+import { getEnvironmentVariable } from "../util/env.js";
+/**
+ * Class representing the CloudflareWorkersAI language model. It extends the LLM (Large
+ * Language Model) class, providing a standard interface for interacting
+ * with the CloudflareWorkersAI language model.
+ */
+export class CloudflareWorkersAI extends LLM {
+    static lc_name() {
+        return "CloudflareWorkersAI";
+    }
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "model", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "@cf/meta/llama-2-7b-chat-int8"
+        });
+        Object.defineProperty(this, "cloudflareAccountId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "cloudflareApiToken", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "baseUrl", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "lc_serializable", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        this.model = fields?.model ?? this.model;
+        this.cloudflareAccountId =
+            fields?.cloudflareAccountId ??
+                getEnvironmentVariable("CLOUDFLARE_ACCOUNT_ID");
+        this.cloudflareApiToken =
+            fields?.cloudflareApiToken ??
+                getEnvironmentVariable("CLOUDFLARE_API_TOKEN");
+        this.baseUrl =
+            fields?.baseUrl ??
+                `https://api.cloudflare.com/client/v4/accounts/${this.cloudflareAccountId}/ai/run`;
+        if (this.baseUrl.endsWith("/")) {
+            this.baseUrl = this.baseUrl.slice(0, -1);
+        }
+    }
+    /**
+     * Method to validate the environment.
+     */
+    validateEnvironment() {
+        if (this.baseUrl === undefined) {
+            if (!this.cloudflareAccountId) {
+                throw new Error(`No Cloudflare account ID found. Please provide it when instantiating the CloudflareWorkersAI class, or set it as "CLOUDFLARE_ACCOUNT_ID" in your environment variables.`);
+            }
+            if (!this.cloudflareApiToken) {
+                throw new Error(`No Cloudflare API key found. Please provide it when instantiating the CloudflareWorkersAI class, or set it as "CLOUDFLARE_API_KEY" in your environment variables.`);
+            }
+        }
+    }
+    /** Get the identifying parameters for this LLM. */
+    get identifyingParams() {
+        return { model: this.model };
+    }
+    /**
+     * Get the parameters used to invoke the model
+     */
+    invocationParams() {
+        return {
+            model: this.model,
+        };
+    }
+    /** Get the type of LLM. */
+    _llmType() {
+        return "cloudflare";
+    }
+    /** Call out to CloudflareWorkersAI's complete endpoint.
+     Args:
+         prompt: The prompt to pass into the model.
+         Returns:
+     The string generated by the model.
+     Example:
+     let response = CloudflareWorkersAI.call("Tell me a joke.");
+     */
+    async _call(prompt, options) {
+        this.validateEnvironment();
+        const url = `${this.baseUrl}/${this.model}`;
+        const headers = {
+            Authorization: `Bearer ${this.cloudflareApiToken}`,
+            "Content-Type": "application/json",
+        };
+        const data = { prompt };
+        const responseData = await this.caller.call(async () => {
+            const response = await fetch(url, {
+                method: "POST",
+                headers,
+                body: JSON.stringify(data),
+                signal: options.signal,
+            });
+            if (!response.ok) {
+                const error = new Error(`Cloudflare LLM call failed with status code ${response.status}`);
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                error.response = response;
+                throw error;
+            }
+            return response.json();
+        });
+        return responseData.result.response;
+    }
+}

package/dist/llms/fake.cjs

@@ -0,0 +1,82 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FakeListLLM = void 0;
+const base_js_1 = require("./base.cjs");
+const index_js_1 = require("../schema/index.cjs");
+/**
+ * A fake LLM that returns a predefined list of responses. It can be used for
+ * testing purposes.
+ */
+class FakeListLLM extends base_js_1.LLM {
+    static lc_name() {
+        return "FakeListLLM";
+    }
+    constructor({ responses, sleep }) {
+        super({});
+        Object.defineProperty(this, "responses", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "i", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 0
+        });
+        Object.defineProperty(this, "sleep", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.responses = responses;
+        this.sleep = sleep;
+    }
+    _llmType() {
+        return "fake-list";
+    }
+    async _call(_prompt, _options, _runManager) {
+        const response = this._currentResponse();
+        this._incrementResponse();
+        await this._sleepIfRequested();
+        return response;
+    }
+    _currentResponse() {
+        return this.responses[this.i];
+    }
+    _incrementResponse() {
+        if (this.i < this.responses.length - 1) {
+            this.i += 1;
+        }
+        else {
+            this.i = 0;
+        }
+    }
+    async *_streamResponseChunks(_input, _options, _runManager) {
+        const response = this._currentResponse();
+        this._incrementResponse();
+        for await (const text of response) {
+            await this._sleepIfRequested();
+            yield this._createResponseChunk(text);
+        }
+    }
+    async _sleepIfRequested() {
+        if (this.sleep !== undefined) {
+            await this._sleep();
+        }
+    }
+    async _sleep() {
+        return new Promise((resolve) => {
+            setTimeout(() => resolve(), this.sleep);
+        });
+    }
+    _createResponseChunk(text) {
+        return new index_js_1.GenerationChunk({
+            text,
+            generationInfo: {},
+        });
+    }
+}
+exports.FakeListLLM = FakeListLLM;

package/dist/llms/fake.d.ts

@@ -0,0 +1,31 @@
+import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
+import { LLM, BaseLLMParams } from "./base.js";
+import { GenerationChunk } from "../schema/index.js";
+/**
+ * Interface for the input parameters specific to the Fake List model.
+ */
+export interface FakeListInput extends BaseLLMParams {
+    /** Responses to return */
+    responses: string[];
+    /** Time to sleep in milliseconds between responses */
+    sleep?: number;
+}
+/**
+ * A fake LLM that returns a predefined list of responses. It can be used for
+ * testing purposes.
+ */
+export declare class FakeListLLM extends LLM {
+    static lc_name(): string;
+    responses: string[];
+    i: number;
+    sleep?: number;
+    constructor({ responses, sleep }: FakeListInput);
+    _llmType(): string;
+    _call(_prompt: string, _options: this["ParsedCallOptions"], _runManager?: CallbackManagerForLLMRun): Promise<string>;
+    _currentResponse(): string;
+    _incrementResponse(): void;
+    _streamResponseChunks(_input: string, _options: this["ParsedCallOptions"], _runManager?: CallbackManagerForLLMRun): AsyncGenerator<GenerationChunk>;
+    _sleepIfRequested(): Promise<void>;
+    _sleep(): Promise<void>;
+    _createResponseChunk(text: string): GenerationChunk;
+}

package/dist/llms/fake.js

@@ -0,0 +1,78 @@
+import { LLM } from "./base.js";
+import { GenerationChunk } from "../schema/index.js";
+/**
+ * A fake LLM that returns a predefined list of responses. It can be used for
+ * testing purposes.
+ */
+export class FakeListLLM extends LLM {
+    static lc_name() {
+        return "FakeListLLM";
+    }
+    constructor({ responses, sleep }) {
+        super({});
+        Object.defineProperty(this, "responses", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "i", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 0
+        });
+        Object.defineProperty(this, "sleep", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.responses = responses;
+        this.sleep = sleep;
+    }
+    _llmType() {
+        return "fake-list";
+    }
+    async _call(_prompt, _options, _runManager) {
+        const response = this._currentResponse();
+        this._incrementResponse();
+        await this._sleepIfRequested();
+        return response;
+    }
+    _currentResponse() {
+        return this.responses[this.i];
+    }
+    _incrementResponse() {
+        if (this.i < this.responses.length - 1) {
+            this.i += 1;
+        }
+        else {
+            this.i = 0;
+        }
+    }
+    async *_streamResponseChunks(_input, _options, _runManager) {
+        const response = this._currentResponse();
+        this._incrementResponse();
+        for await (const text of response) {
+            await this._sleepIfRequested();
+            yield this._createResponseChunk(text);
+        }
+    }
+    async _sleepIfRequested() {
+        if (this.sleep !== undefined) {
+            await this._sleep();
+        }
+    }
+    async _sleep() {
+        return new Promise((resolve) => {
+            setTimeout(() => resolve(), this.sleep);
+        });
+    }
+    _createResponseChunk(text) {
+        return new GenerationChunk({
+            text,
+            generationInfo: {},
+        });
+    }
+}

package/dist/llms/sagemaker_endpoint.cjs

@@ -131,18 +131,18 @@ class SageMakerEndpoint extends base_js_1.LLM {
      * Calls the SageMaker endpoint and retrieves the result.
      * @param {string} prompt The input prompt.
      * @param {this["ParsedCallOptions"]} options Parsed call options.
-     * @param {CallbackManagerForLLMRun} _runManager Optional run manager.
+     * @param {CallbackManagerForLLMRun} runManager Optional run manager.
      * @returns {Promise<string>} A promise that resolves to the generated string.
      */
     /** @ignore */
-    async _call(prompt, options, _runManager) {
+    async _call(prompt, options, runManager) {
         return this.streaming
-            ? await this.streamingCall(prompt, options)
+            ? await this.streamingCall(prompt, options, runManager)
             : await this.noStreamingCall(prompt, options);
     }
-    async streamingCall(prompt, options) {
+    async streamingCall(prompt, options, runManager) {
         const chunks = [];
-        for await (const chunk of this._streamResponseChunks(prompt, options)) {
+        for await (const chunk of this._streamResponseChunks(prompt, options, runManager)) {
             chunks.push(chunk.text);
         }
         return chunks.join("");
@@ -168,7 +168,7 @@ class SageMakerEndpoint extends base_js_1.LLM {
      * @param {this["ParsedCallOptions"]} options Parsed call options.
      * @returns {AsyncGenerator<GenerationChunk>} An asynchronous generator yielding generation chunks.
      */
-    async *_streamResponseChunks(prompt, options) {
+    async *_streamResponseChunks(prompt, options, runManager) {
         const body = await this.contentHandler.transformInput(prompt, this.modelKwargs ?? {});
         const { contentType, accepts } = this.contentHandler;
         const stream = await this.caller.call(() => this.client.send(new client_sagemaker_runtime_1.InvokeEndpointWithResponseStreamCommand({
@@ -183,13 +183,15 @@ class SageMakerEndpoint extends base_js_1.LLM {
         }
         for await (const chunk of stream.Body) {
             if (chunk.PayloadPart && chunk.PayloadPart.Bytes) {
+                const text = await this.contentHandler.transformOutput(chunk.PayloadPart.Bytes);
                 yield new index_js_1.GenerationChunk({
-                    text: await this.contentHandler.transformOutput(chunk.PayloadPart.Bytes),
+                    text,
                     generationInfo: {
                         ...chunk,
                         response: undefined,
                     },
                 });
+                await runManager?.handleLLMNewToken(text);
             }
             else if (chunk.InternalStreamFailure) {
                 throw new Error(chunk.InternalStreamFailure.message);

package/dist/llms/sagemaker_endpoint.d.ts

@@ -104,11 +104,11 @@ export declare class SageMakerEndpoint extends LLM<BaseLLMCallOptions> {
      * Calls the SageMaker endpoint and retrieves the result.
      * @param {string} prompt The input prompt.
      * @param {this["ParsedCallOptions"]} options Parsed call options.
-     * @param {CallbackManagerForLLMRun} _runManager Optional run manager.
+     * @param {CallbackManagerForLLMRun} runManager Optional run manager.
      * @returns {Promise<string>} A promise that resolves to the generated string.
      */
     /** @ignore */
-    _call(prompt: string, options: this["ParsedCallOptions"], _runManager?: CallbackManagerForLLMRun): Promise<string>;
+    _call(prompt: string, options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<string>;
     private streamingCall;
     private noStreamingCall;
     /**
@@ -117,5 +117,5 @@ export declare class SageMakerEndpoint extends LLM<BaseLLMCallOptions> {
      * @param {this["ParsedCallOptions"]} options Parsed call options.
      * @returns {AsyncGenerator<GenerationChunk>} An asynchronous generator yielding generation chunks.
      */
-    _streamResponseChunks(prompt: string, options: this["ParsedCallOptions"]): AsyncGenerator<GenerationChunk>;
+    _streamResponseChunks(prompt: string, options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<GenerationChunk>;
 }

package/dist/llms/sagemaker_endpoint.js

@@ -127,18 +127,18 @@ export class SageMakerEndpoint extends LLM {
      * Calls the SageMaker endpoint and retrieves the result.
      * @param {string} prompt The input prompt.
      * @param {this["ParsedCallOptions"]} options Parsed call options.
-     * @param {CallbackManagerForLLMRun} _runManager Optional run manager.
+     * @param {CallbackManagerForLLMRun} runManager Optional run manager.
      * @returns {Promise<string>} A promise that resolves to the generated string.
      */
     /** @ignore */
-    async _call(prompt, options, _runManager) {
+    async _call(prompt, options, runManager) {
         return this.streaming
-            ? await this.streamingCall(prompt, options)
+            ? await this.streamingCall(prompt, options, runManager)
             : await this.noStreamingCall(prompt, options);
     }
-    async streamingCall(prompt, options) {
+    async streamingCall(prompt, options, runManager) {
         const chunks = [];
-        for await (const chunk of this._streamResponseChunks(prompt, options)) {
+        for await (const chunk of this._streamResponseChunks(prompt, options, runManager)) {
             chunks.push(chunk.text);
         }
         return chunks.join("");
@@ -164,7 +164,7 @@ export class SageMakerEndpoint extends LLM {
      * @param {this["ParsedCallOptions"]} options Parsed call options.
      * @returns {AsyncGenerator<GenerationChunk>} An asynchronous generator yielding generation chunks.
      */
-    async *_streamResponseChunks(prompt, options) {
+    async *_streamResponseChunks(prompt, options, runManager) {
         const body = await this.contentHandler.transformInput(prompt, this.modelKwargs ?? {});
         const { contentType, accepts } = this.contentHandler;
         const stream = await this.caller.call(() => this.client.send(new InvokeEndpointWithResponseStreamCommand({
@@ -179,13 +179,15 @@ export class SageMakerEndpoint extends LLM {
         }
         for await (const chunk of stream.Body) {
             if (chunk.PayloadPart && chunk.PayloadPart.Bytes) {
+                const text = await this.contentHandler.transformOutput(chunk.PayloadPart.Bytes);
                 yield new GenerationChunk({
-                    text: await this.contentHandler.transformOutput(chunk.PayloadPart.Bytes),
+                    text,
                     generationInfo: {
                         ...chunk,
                         response: undefined,
                     },
                 });
+                await runManager?.handleLLMNewToken(text);
             }
             else if (chunk.InternalStreamFailure) {
                 throw new Error(chunk.InternalStreamFailure.message);

package/dist/load/import_constants.cjs

@@ -43,6 +43,8 @@ exports.optionalImportEntrypoints = [
     "langchain/vectorstores/cassandra",
     "langchain/vectorstores/elasticsearch",
     "langchain/vectorstores/cloudflare_vectorize",
+    "langchain/vectorstores/closevector/web",
+    "langchain/vectorstores/closevector/node",
     "langchain/vectorstores/chroma",
     "langchain/vectorstores/googlevertexai",
     "langchain/vectorstores/hnswlib",

package/dist/load/import_constants.js

@@ -40,6 +40,8 @@ export const optionalImportEntrypoints = [
     "langchain/vectorstores/cassandra",
     "langchain/vectorstores/elasticsearch",
     "langchain/vectorstores/cloudflare_vectorize",
+    "langchain/vectorstores/closevector/web",
+    "langchain/vectorstores/closevector/node",
     "langchain/vectorstores/chroma",
     "langchain/vectorstores/googlevertexai",
     "langchain/vectorstores/hnswlib",

package/dist/load/import_map.cjs

@@ -24,8 +24,8 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__fireworks = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__web__searchapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__yandex = exports.llms__fireworks = exports.llms__ollama = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
-exports.runnables__remote = exports.evaluation = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__time = exports.util__math = exports.storage__in_memory = exports.storage__encoder_backed = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__embeddings_filter = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__tavily_search_api = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__chaindesk = exports.retrievers__remote = void 0;
+exports.schema__query_constructor = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__fake = exports.chat_models__yandex = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__fireworks = exports.chat_models__cloudflare_workersai = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__web__searchapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__fake = exports.llms__yandex = exports.llms__fireworks = exports.llms__ollama = exports.llms__cloudflare_workersai = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
+exports.runnables__remote = exports.evaluation = exports.experimental__chains__violation_of_expectations = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__time = exports.util__math = exports.storage__in_memory = exports.storage__encoder_backed = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__embeddings_filter = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__tavily_search_api = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__chaindesk = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = void 0;
 exports.load__serializable = __importStar(require("../load/serializable.cjs"));
 exports.agents = __importStar(require("../agents/index.cjs"));
 exports.agents__toolkits = __importStar(require("../agents/toolkits/index.cjs"));
@@ -43,9 +43,11 @@ exports.llms__base = __importStar(require("../llms/base.cjs"));
 exports.llms__openai = __importStar(require("../llms/openai.cjs"));
 exports.llms__ai21 = __importStar(require("../llms/ai21.cjs"));
 exports.llms__aleph_alpha = __importStar(require("../llms/aleph_alpha.cjs"));
+exports.llms__cloudflare_workersai = __importStar(require("../llms/cloudflare_workersai.cjs"));
 exports.llms__ollama = __importStar(require("../llms/ollama.cjs"));
 exports.llms__fireworks = __importStar(require("../llms/fireworks.cjs"));
 exports.llms__yandex = __importStar(require("../llms/yandex.cjs"));
+exports.llms__fake = __importStar(require("../llms/fake.cjs"));
 exports.prompts = __importStar(require("../prompts/index.cjs"));
 exports.vectorstores__base = __importStar(require("../vectorstores/base.cjs"));
 exports.vectorstores__memory = __importStar(require("../vectorstores/memory.cjs"));
@@ -63,10 +65,13 @@ exports.document_transformers__openai_functions = __importStar(require("../docum
 exports.chat_models__base = __importStar(require("../chat_models/base.cjs"));
 exports.chat_models__openai = __importStar(require("../chat_models/openai.cjs"));
 exports.chat_models__anthropic = __importStar(require("../chat_models/anthropic.cjs"));
+exports.chat_models__cloudflare_workersai = __importStar(require("../chat_models/cloudflare_workersai.cjs"));
 exports.chat_models__fireworks = __importStar(require("../chat_models/fireworks.cjs"));
 exports.chat_models__baiduwenxin = __importStar(require("../chat_models/baiduwenxin.cjs"));
 exports.chat_models__ollama = __importStar(require("../chat_models/ollama.cjs"));
 exports.chat_models__minimax = __importStar(require("../chat_models/minimax.cjs"));
+exports.chat_models__yandex = __importStar(require("../chat_models/yandex.cjs"));
+exports.chat_models__fake = __importStar(require("../chat_models/fake.cjs"));
 exports.schema = __importStar(require("../schema/index.cjs"));
 exports.schema__document = __importStar(require("../schema/document.cjs"));
 exports.schema__output_parser = __importStar(require("../schema/output_parser.cjs"));
@@ -104,5 +109,6 @@ exports.experimental__babyagi = __importStar(require("../experimental/babyagi/in
 exports.experimental__generative_agents = __importStar(require("../experimental/generative_agents/index.cjs"));
 exports.experimental__plan_and_execute = __importStar(require("../experimental/plan_and_execute/index.cjs"));
 exports.experimental__chat_models__bittensor = __importStar(require("../experimental/chat_models/bittensor.cjs"));
+exports.experimental__chains__violation_of_expectations = __importStar(require("../experimental/chains/violation_of_expectations/index.cjs"));
 exports.evaluation = __importStar(require("../evaluation/index.cjs"));
 exports.runnables__remote = __importStar(require("../runnables/remote.cjs"));

package/dist/load/import_map.d.ts

@@ -15,9 +15,11 @@ export * as llms__base from "../llms/base.js";
 export * as llms__openai from "../llms/openai.js";
 export * as llms__ai21 from "../llms/ai21.js";
 export * as llms__aleph_alpha from "../llms/aleph_alpha.js";
+export * as llms__cloudflare_workersai from "../llms/cloudflare_workersai.js";
 export * as llms__ollama from "../llms/ollama.js";
 export * as llms__fireworks from "../llms/fireworks.js";
 export * as llms__yandex from "../llms/yandex.js";
+export * as llms__fake from "../llms/fake.js";
 export * as prompts from "../prompts/index.js";
 export * as vectorstores__base from "../vectorstores/base.js";
 export * as vectorstores__memory from "../vectorstores/memory.js";
@@ -35,10 +37,13 @@ export * as document_transformers__openai_functions from "../document_transforme
 export * as chat_models__base from "../chat_models/base.js";
 export * as chat_models__openai from "../chat_models/openai.js";
 export * as chat_models__anthropic from "../chat_models/anthropic.js";
+export * as chat_models__cloudflare_workersai from "../chat_models/cloudflare_workersai.js";
 export * as chat_models__fireworks from "../chat_models/fireworks.js";
 export * as chat_models__baiduwenxin from "../chat_models/baiduwenxin.js";
 export * as chat_models__ollama from "../chat_models/ollama.js";
 export * as chat_models__minimax from "../chat_models/minimax.js";
+export * as chat_models__yandex from "../chat_models/yandex.js";
+export * as chat_models__fake from "../chat_models/fake.js";
 export * as schema from "../schema/index.js";
 export * as schema__document from "../schema/document.js";
 export * as schema__output_parser from "../schema/output_parser.js";
@@ -76,5 +81,6 @@ export * as experimental__babyagi from "../experimental/babyagi/index.js";
 export * as experimental__generative_agents from "../experimental/generative_agents/index.js";
 export * as experimental__plan_and_execute from "../experimental/plan_and_execute/index.js";
 export * as experimental__chat_models__bittensor from "../experimental/chat_models/bittensor.js";
+export * as experimental__chains__violation_of_expectations from "../experimental/chains/violation_of_expectations/index.js";
 export * as evaluation from "../evaluation/index.js";
 export * as runnables__remote from "../runnables/remote.js";

package/dist/load/import_map.js

@@ -16,9 +16,11 @@ export * as llms__base from "../llms/base.js";
 export * as llms__openai from "../llms/openai.js";
 export * as llms__ai21 from "../llms/ai21.js";
 export * as llms__aleph_alpha from "../llms/aleph_alpha.js";
+export * as llms__cloudflare_workersai from "../llms/cloudflare_workersai.js";
 export * as llms__ollama from "../llms/ollama.js";
 export * as llms__fireworks from "../llms/fireworks.js";
 export * as llms__yandex from "../llms/yandex.js";
+export * as llms__fake from "../llms/fake.js";
 export * as prompts from "../prompts/index.js";
 export * as vectorstores__base from "../vectorstores/base.js";
 export * as vectorstores__memory from "../vectorstores/memory.js";
@@ -36,10 +38,13 @@ export * as document_transformers__openai_functions from "../document_transforme
 export * as chat_models__base from "../chat_models/base.js";
 export * as chat_models__openai from "../chat_models/openai.js";
 export * as chat_models__anthropic from "../chat_models/anthropic.js";
+export * as chat_models__cloudflare_workersai from "../chat_models/cloudflare_workersai.js";
 export * as chat_models__fireworks from "../chat_models/fireworks.js";
 export * as chat_models__baiduwenxin from "../chat_models/baiduwenxin.js";
 export * as chat_models__ollama from "../chat_models/ollama.js";
 export * as chat_models__minimax from "../chat_models/minimax.js";
+export * as chat_models__yandex from "../chat_models/yandex.js";
+export * as chat_models__fake from "../chat_models/fake.js";
 export * as schema from "../schema/index.js";
 export * as schema__document from "../schema/document.js";
 export * as schema__output_parser from "../schema/output_parser.js";
@@ -77,5 +82,6 @@ export * as experimental__babyagi from "../experimental/babyagi/index.js";
 export * as experimental__generative_agents from "../experimental/generative_agents/index.js";
 export * as experimental__plan_and_execute from "../experimental/plan_and_execute/index.js";
 export * as experimental__chat_models__bittensor from "../experimental/chat_models/bittensor.js";
+export * as experimental__chains__violation_of_expectations from "../experimental/chains/violation_of_expectations/index.js";
 export * as evaluation from "../evaluation/index.js";
 export * as runnables__remote from "../runnables/remote.js";

package/dist/retrievers/zep.cjs

@@ -47,8 +47,29 @@ class ZepRetriever extends retriever_js_1.BaseRetriever {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "searchType", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "mmrLambda", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "filter", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.sessionId = config.sessionId;
         this.topK = config.topK;
+        this.searchType = config.searchType;
+        this.mmrLambda = config.mmrLambda;
+        this.filter = config.filter;
         this.zepClientPromise = zep_js_1.ZepClient.init(config.url, config.apiKey);
     }
     /**
@@ -59,9 +80,9 @@ class ZepRetriever extends retriever_js_1.BaseRetriever {
     searchResultToDoc(results) {
         return results
             .filter((r) => r.message)
-            .map(({ message: { content } = {}, ...metadata }, dist) => new document_js_1.Document({
+            .map(({ message: { content, metadata: messageMetadata } = {}, dist, ...rest }) => new document_js_1.Document({
             pageContent: content ?? "",
-            metadata: { score: dist, ...metadata },
+            metadata: { score: dist, ...messageMetadata, ...rest },
         }));
     }
     /**
@@ -70,7 +91,12 @@ class ZepRetriever extends retriever_js_1.BaseRetriever {
      *  @returns {Promise<Document[]>} A promise that resolves to an array of relevant Document objects.
      */
     async _getRelevantDocuments(query) {
-        const payload = { text: query, metadata: {} };
+        const payload = {
+            text: query,
+            metadata: this.filter,
+            search_type: this.searchType,
+            mmr_lambda: this.mmrLambda,
+        };
         // Wait for ZepClient to be initialized
         const zepClient = await this.zepClientPromise;
         if (!zepClient) {