langchain 0.0.151 → 0.0.152

package/dist/document_loaders/web/pdf.js

@@ -0,0 +1,83 @@
+import { getDocument, version, } from "pdf-parse/lib/pdf.js/v1.10.100/build/pdf.js";
+import { Document } from "../../document.js";
+import { BaseDocumentLoader } from "../base.js";
+/**
+ * A document loader for loading data from PDFs.
+ */
+export class WebPDFLoader extends BaseDocumentLoader {
+    constructor(blob, { splitPages = true } = {}) {
+        super();
+        Object.defineProperty(this, "blob", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "splitPages", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        this.blob = blob;
+        this.splitPages = splitPages ?? this.splitPages;
+    }
+    /**
+     * Loads the contents of the PDF as documents.
+     * @returns An array of Documents representing the retrieved data.
+     */
+    async load() {
+        const parsedPdf = await getDocument({
+            data: new Uint8Array(await this.blob.arrayBuffer()),
+            useWorkerFetch: false,
+            isEvalSupported: false,
+            useSystemFonts: true,
+        }).promise;
+        const meta = await parsedPdf.getMetadata().catch(() => null);
+        const documents = [];
+        for (let i = 1; i <= parsedPdf.numPages; i += 1) {
+            const page = await parsedPdf.getPage(i);
+            const content = await page.getTextContent();
+            if (content.items.length === 0) {
+                continue;
+            }
+            const text = content.items
+                .map((item) => item.str)
+                .join("\n");
+            documents.push(new Document({
+                pageContent: text,
+                metadata: {
+                    pdf: {
+                        version,
+                        info: meta?.info,
+                        metadata: meta?.metadata,
+                        totalPages: parsedPdf.numPages,
+                    },
+                    loc: {
+                        pageNumber: i,
+                    },
+                },
+            }));
+        }
+        if (this.splitPages) {
+            return documents;
+        }
+        if (documents.length === 0) {
+            return [];
+        }
+        return [
+            new Document({
+                pageContent: documents.map((doc) => doc.pageContent).join("\n\n"),
+                metadata: {
+                    pdf: {
+                        version,
+                        info: meta?.info,
+                        metadata: meta?.metadata,
+                        totalPages: parsedPdf.numPages,
+                    },
+                },
+            }),
+        ];
+        return documents;
+    }
+}

package/dist/evaluation/agents/prompt.cjs

@@ -90,8 +90,7 @@ The model did not use the appropriate tools to answer the question.\
 Judgment: Given the good reasoning in the final answer but otherwise poor performance, we give the model a score of 2.
 
 Score: 2`;
-exports.EVAL_CHAT_PROMPT = 
-/* #__PURE__ */ index_js_1.ChatPromptTemplate.fromPromptMessages([
+exports.EVAL_CHAT_PROMPT = index_js_1.ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ index_js_1.SystemMessagePromptTemplate.fromTemplate("You are a helpful assistant that evaluates language models."),
     /* #__PURE__ */ index_js_1.HumanMessagePromptTemplate.fromTemplate(EXAMPLE_INPUT),
     /* #__PURE__ */ index_js_1.AIMessagePromptTemplate.fromTemplate(EXAMPLE_OUTPUT),
@@ -124,7 +123,7 @@ i. Is the final answer helpful?
     iv. Does the AI language model use too many steps to answer the question?
     v. Are the appropriate tools used to answer the question?`;
 exports.TOOL_FREE_EVAL_CHAT_PROMPT = 
-/* #__PURE__ */ index_js_1.ChatPromptTemplate.fromPromptMessages([
+/* #__PURE__ */ index_js_1.ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ index_js_1.SystemMessagePromptTemplate.fromTemplate("You are a helpful assistant that evaluates language models."),
     /* #__PURE__ */ index_js_1.HumanMessagePromptTemplate.fromTemplate(EXAMPLE_INPUT),
     /* #__PURE__ */ index_js_1.AIMessagePromptTemplate.fromTemplate(EXAMPLE_OUTPUT),

package/dist/evaluation/agents/prompt.js

@@ -87,8 +87,7 @@ The model did not use the appropriate tools to answer the question.\
 Judgment: Given the good reasoning in the final answer but otherwise poor performance, we give the model a score of 2.
 
 Score: 2`;
-export const EVAL_CHAT_PROMPT = 
-/* #__PURE__ */ ChatPromptTemplate.fromPromptMessages([
+export const EVAL_CHAT_PROMPT = /* #__PURE__ */ ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ SystemMessagePromptTemplate.fromTemplate("You are a helpful assistant that evaluates language models."),
     /* #__PURE__ */ HumanMessagePromptTemplate.fromTemplate(EXAMPLE_INPUT),
     /* #__PURE__ */ AIMessagePromptTemplate.fromTemplate(EXAMPLE_OUTPUT),
@@ -121,7 +120,7 @@ i. Is the final answer helpful?
     iv. Does the AI language model use too many steps to answer the question?
     v. Are the appropriate tools used to answer the question?`;
 export const TOOL_FREE_EVAL_CHAT_PROMPT = 
-/* #__PURE__ */ ChatPromptTemplate.fromPromptMessages([
+/* #__PURE__ */ ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ SystemMessagePromptTemplate.fromTemplate("You are a helpful assistant that evaluates language models."),
     /* #__PURE__ */ HumanMessagePromptTemplate.fromTemplate(EXAMPLE_INPUT),
     /* #__PURE__ */ AIMessagePromptTemplate.fromTemplate(EXAMPLE_OUTPUT),

package/dist/experimental/plan_and_execute/prompt.cjs

@@ -14,7 +14,7 @@ exports.PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE = [
     `At the end of your plan, say "<END_OF_PLAN>"`,
 ].join(" ");
 exports.PLANNER_CHAT_PROMPT = 
-/* #__PURE__ */ chat_js_1.ChatPromptTemplate.fromPromptMessages([
+/* #__PURE__ */ chat_js_1.ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ chat_js_1.SystemMessagePromptTemplate.fromTemplate(exports.PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE),
     /* #__PURE__ */ chat_js_1.HumanMessagePromptTemplate.fromTemplate(`{input}`),
 ]);

package/dist/experimental/plan_and_execute/prompt.js

@@ -11,7 +11,7 @@ export const PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE = [
     `At the end of your plan, say "<END_OF_PLAN>"`,
 ].join(" ");
 export const PLANNER_CHAT_PROMPT = 
-/* #__PURE__ */ ChatPromptTemplate.fromPromptMessages([
+/* #__PURE__ */ ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ SystemMessagePromptTemplate.fromTemplate(PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE),
     /* #__PURE__ */ HumanMessagePromptTemplate.fromTemplate(`{input}`),
 ]);

package/dist/llms/llama_cpp.cjs

@@ -99,6 +99,12 @@ class LlamaCpp extends base_js_1.LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "_session", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.batchSize = inputs.batchSize;
         this.contextSize = inputs.contextSize;
         this.embedding = inputs.embedding;
@@ -113,19 +119,19 @@ class LlamaCpp extends base_js_1.LLM {
         this.vocabOnly = inputs.vocabOnly;
         this._model = new node_llama_cpp_1.LlamaModel(inputs);
         this._context = new node_llama_cpp_1.LlamaContext({ model: this._model });
+        this._session = new node_llama_cpp_1.LlamaChatSession({ context: this._context });
     }
     _llmType() {
         return "llama2_cpp";
     }
     /** @ignore */
     async _call(prompt, options) {
-        const session = new node_llama_cpp_1.LlamaChatSession({ context: this._context });
         try {
-            const compleation = await session.prompt(prompt, options);
-            return compleation;
+            const completion = await this._session.prompt(prompt, options);
+            return completion;
         }
         catch (e) {
-            throw new Error("Error getting prompt compleation.");
+            throw new Error("Error getting prompt completion.");
         }
     }
 }

package/dist/llms/llama_cpp.d.ts

@@ -1,4 +1,4 @@
-import { LlamaModel, LlamaContext } from "node-llama-cpp";
+import { LlamaModel, LlamaContext, LlamaChatSession } from "node-llama-cpp";
 import { LLM, BaseLLMCallOptions, BaseLLMParams } from "./base.js";
 /**
  * Note that the modelPath is the only required parameter. For testing you
@@ -65,6 +65,7 @@ export declare class LlamaCpp extends LLM<LlamaCppCallOptions> {
     modelPath: string;
     _model: LlamaModel;
     _context: LlamaContext;
+    _session: LlamaChatSession;
     static lc_name(): string;
     constructor(inputs: LlamaCppInputs);
     _llmType(): string;

package/dist/llms/llama_cpp.js

@@ -96,6 +96,12 @@ export class LlamaCpp extends LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "_session", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.batchSize = inputs.batchSize;
         this.contextSize = inputs.contextSize;
         this.embedding = inputs.embedding;
@@ -110,19 +116,19 @@ export class LlamaCpp extends LLM {
         this.vocabOnly = inputs.vocabOnly;
         this._model = new LlamaModel(inputs);
         this._context = new LlamaContext({ model: this._model });
+        this._session = new LlamaChatSession({ context: this._context });
     }
     _llmType() {
         return "llama2_cpp";
     }
     /** @ignore */
     async _call(prompt, options) {
-        const session = new LlamaChatSession({ context: this._context });
         try {
-            const compleation = await session.prompt(prompt, options);
-            return compleation;
+            const completion = await this._session.prompt(prompt, options);
+            return completion;
         }
         catch (e) {
-            throw new Error("Error getting prompt compleation.");
+            throw new Error("Error getting prompt completion.");
         }
     }
 }

package/dist/llms/ollama.cjs

@@ -287,8 +287,8 @@ class Ollama extends base_js_1.LLM {
             },
         };
     }
-    async *_streamResponseChunks(input, options, runManager) {
-        const stream = await this.caller.call(async () => (0, ollama_js_1.createOllamaStream)(this.baseUrl, { ...this.invocationParams(options), prompt: input }, options));
+    async *_streamResponseChunks(prompt, options, runManager) {
+        const stream = await this.caller.call(async () => (0, ollama_js_1.createOllamaStream)(this.baseUrl, { ...this.invocationParams(options), prompt }, options));
         for await (const chunk of stream) {
             yield new index_js_1.GenerationChunk({
                 text: chunk.response,
@@ -301,11 +301,10 @@ class Ollama extends base_js_1.LLM {
         }
     }
     /** @ignore */
-    async _call(prompt, options) {
-        const stream = await this.caller.call(async () => (0, ollama_js_1.createOllamaStream)(this.baseUrl, { ...this.invocationParams(options), prompt }, options));
+    async _call(prompt, options, runManager) {
         const chunks = [];
-        for await (const chunk of stream) {
-            chunks.push(chunk.response);
+        for await (const chunk of this._streamResponseChunks(prompt, options, runManager)) {
+            chunks.push(chunk.text);
         }
         return chunks.join("");
     }

package/dist/llms/ollama.d.ts

@@ -79,7 +79,7 @@ export declare class Ollama extends LLM implements OllamaInput {
             vocab_only: boolean | undefined;
         };
     };
-    _streamResponseChunks(input: string, options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<GenerationChunk>;
+    _streamResponseChunks(prompt: string, options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<GenerationChunk>;
     /** @ignore */
-    _call(prompt: string, options: this["ParsedCallOptions"]): Promise<string>;
+    _call(prompt: string, options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<string>;
 }

package/dist/llms/ollama.js

@@ -284,8 +284,8 @@ export class Ollama extends LLM {
             },
         };
     }
-    async *_streamResponseChunks(input, options, runManager) {
-        const stream = await this.caller.call(async () => createOllamaStream(this.baseUrl, { ...this.invocationParams(options), prompt: input }, options));
+    async *_streamResponseChunks(prompt, options, runManager) {
+        const stream = await this.caller.call(async () => createOllamaStream(this.baseUrl, { ...this.invocationParams(options), prompt }, options));
         for await (const chunk of stream) {
             yield new GenerationChunk({
                 text: chunk.response,
@@ -298,11 +298,10 @@ export class Ollama extends LLM {
         }
     }
     /** @ignore */
-    async _call(prompt, options) {
-        const stream = await this.caller.call(async () => createOllamaStream(this.baseUrl, { ...this.invocationParams(options), prompt }, options));
+    async _call(prompt, options, runManager) {
         const chunks = [];
-        for await (const chunk of stream) {
-            chunks.push(chunk.response);
+        for await (const chunk of this._streamResponseChunks(prompt, options, runManager)) {
+            chunks.push(chunk.text);
         }
         return chunks.join("");
     }

package/dist/llms/openai.cjs

@@ -57,9 +57,8 @@ class OpenAI extends base_js_1.BaseLLM {
     /** @deprecated */
     configuration) {
         if ((fields?.modelName?.startsWith("gpt-3.5-turbo") ||
-            fields?.modelName?.startsWith("gpt-4") ||
-            fields?.modelName?.startsWith("gpt-4-32k")) &&
-            !fields?.modelName.endsWith("-instruct")) {
+            fields?.modelName?.startsWith("gpt-4")) &&
+            !fields?.modelName?.includes("-instruct")) {
             // eslint-disable-next-line no-constructor-return, @typescript-eslint/no-explicit-any
             return new openai_chat_js_1.OpenAIChat(fields, configuration);
         }

package/dist/llms/openai.js

@@ -54,9 +54,8 @@ export class OpenAI extends BaseLLM {
     /** @deprecated */
     configuration) {
         if ((fields?.modelName?.startsWith("gpt-3.5-turbo") ||
-            fields?.modelName?.startsWith("gpt-4") ||
-            fields?.modelName?.startsWith("gpt-4-32k")) &&
-            !fields?.modelName.endsWith("-instruct")) {
+            fields?.modelName?.startsWith("gpt-4")) &&
+            !fields?.modelName?.includes("-instruct")) {
             // eslint-disable-next-line no-constructor-return, @typescript-eslint/no-explicit-any
             return new OpenAIChat(fields, configuration);
         }

package/dist/load/import_constants.cjs

@@ -75,6 +75,7 @@ exports.optionalImportEntrypoints = [
     "langchain/document_loaders/web/github",
     "langchain/document_loaders/web/notiondb",
     "langchain/document_loaders/web/notionapi",
+    "langchain/document_loaders/web/pdf",
     "langchain/document_loaders/web/recursive_url",
     "langchain/document_loaders/web/s3",
     "langchain/document_loaders/web/sonix_audio",
@@ -110,12 +111,14 @@ exports.optionalImportEntrypoints = [
     "langchain/retrievers/self_query/pinecone",
     "langchain/retrievers/self_query/supabase",
     "langchain/retrievers/self_query/weaviate",
+    "langchain/cache/cloudflare_kv",
     "langchain/cache/momento",
     "langchain/cache/redis",
     "langchain/cache/ioredis",
     "langchain/cache/upstash_redis",
     "langchain/stores/doc/gcs",
     "langchain/stores/file/node",
+    "langchain/stores/message/cloudflare_d1",
     "langchain/stores/message/dynamodb",
     "langchain/stores/message/firestore",
     "langchain/stores/message/momento",

package/dist/load/import_constants.js

@@ -72,6 +72,7 @@ export const optionalImportEntrypoints = [
     "langchain/document_loaders/web/github",
     "langchain/document_loaders/web/notiondb",
     "langchain/document_loaders/web/notionapi",
+    "langchain/document_loaders/web/pdf",
     "langchain/document_loaders/web/recursive_url",
     "langchain/document_loaders/web/s3",
     "langchain/document_loaders/web/sonix_audio",
@@ -107,12 +108,14 @@ export const optionalImportEntrypoints = [
     "langchain/retrievers/self_query/pinecone",
     "langchain/retrievers/self_query/supabase",
     "langchain/retrievers/self_query/weaviate",
+    "langchain/cache/cloudflare_kv",
     "langchain/cache/momento",
     "langchain/cache/redis",
     "langchain/cache/ioredis",
     "langchain/cache/upstash_redis",
     "langchain/stores/doc/gcs",
     "langchain/stores/file/node",
+    "langchain/stores/message/cloudflare_d1",
     "langchain/stores/message/dynamodb",
     "langchain/stores/message/firestore",
     "langchain/stores/message/momento",

package/dist/prompts/chat.cjs

@@ -356,8 +356,14 @@ class ChatPromptTemplate extends BaseChatPromptTemplate {
         };
         return new ChatPromptTemplate(promptDict);
     }
+    /**
+     * Create a chat model-specific prompt from individual chat messages
+     * or message-like tuples.
+     * @param promptMessages Messages to be passed to the chat model
+     * @returns A new ChatPromptTemplate
+     */
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    static fromPromptMessages(promptMessages) {
+    static fromMessages(promptMessages) {
         const flattenedMessages = promptMessages.reduce((acc, promptMessage) => acc.concat(
         // eslint-disable-next-line no-instanceof/no-instanceof
         promptMessage instanceof ChatPromptTemplate
@@ -386,5 +392,10 @@ class ChatPromptTemplate extends BaseChatPromptTemplate {
             partialVariables: flattenedPartialVariables,
         });
     }
+    /** @deprecated Renamed to .fromMessages */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    static fromPromptMessages(promptMessages) {
+        return this.fromMessages(promptMessages);
+    }
 }
 exports.ChatPromptTemplate = ChatPromptTemplate;

package/dist/prompts/chat.d.ts

@@ -169,5 +169,13 @@ export declare class ChatPromptTemplate<RunInput extends InputValues = any, Part
     _getPromptType(): "chat";
     formatMessages(values: TypedPromptInputValues<RunInput>): Promise<BaseMessage[]>;
     partial<NewPartialVariableName extends string>(values: PartialValues<NewPartialVariableName>): Promise<ChatPromptTemplate<InputValues<Exclude<Extract<keyof RunInput, string>, NewPartialVariableName>>, any>>;
+    /**
+     * Create a chat model-specific prompt from individual chat messages
+     * or message-like tuples.
+     * @param promptMessages Messages to be passed to the chat model
+     * @returns A new ChatPromptTemplate
+     */
+    static fromMessages<RunInput extends InputValues = any>(promptMessages: (ChatPromptTemplate<InputValues, string> | BaseMessagePromptTemplateLike)[]): ChatPromptTemplate<RunInput>;
+    /** @deprecated Renamed to .fromMessages */
     static fromPromptMessages<RunInput extends InputValues = any>(promptMessages: (ChatPromptTemplate<InputValues, string> | BaseMessagePromptTemplateLike)[]): ChatPromptTemplate<RunInput>;
 }

package/dist/prompts/chat.js

@@ -344,8 +344,14 @@ export class ChatPromptTemplate extends BaseChatPromptTemplate {
         };
         return new ChatPromptTemplate(promptDict);
     }
+    /**
+     * Create a chat model-specific prompt from individual chat messages
+     * or message-like tuples.
+     * @param promptMessages Messages to be passed to the chat model
+     * @returns A new ChatPromptTemplate
+     */
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    static fromPromptMessages(promptMessages) {
+    static fromMessages(promptMessages) {
         const flattenedMessages = promptMessages.reduce((acc, promptMessage) => acc.concat(
         // eslint-disable-next-line no-instanceof/no-instanceof
         promptMessage instanceof ChatPromptTemplate
@@ -374,4 +380,9 @@ export class ChatPromptTemplate extends BaseChatPromptTemplate {
             partialVariables: flattenedPartialVariables,
         });
     }
+    /** @deprecated Renamed to .fromMessages */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    static fromPromptMessages(promptMessages) {
+        return this.fromMessages(promptMessages);
+    }
 }

package/dist/stores/message/cloudflare_d1.cjs

@@ -0,0 +1,134 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CloudflareD1MessageHistory = void 0;
+const uuid_1 = require("uuid");
+const index_js_1 = require("../../schema/index.cjs");
+const utils_js_1 = require("./utils.cjs");
+/**
+ * Class for storing and retrieving chat message history from a
+ * Cloudflare D1 database. Extends the BaseListChatMessageHistory class.
+ */
+class CloudflareD1MessageHistory extends index_js_1.BaseListChatMessageHistory {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "stores", "message", "cloudflare_d1"]
+        });
+        Object.defineProperty(this, "database", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "tableName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "sessionId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "tableInitialized", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        const { sessionId, database, tableName } = fields;
+        if (database) {
+            this.database = database;
+        }
+        else {
+            throw new Error("Either a client or config must be provided to CloudflareD1MessageHistory");
+        }
+        this.tableName = tableName || "langchain_chat_histories";
+        this.tableInitialized = false;
+        this.sessionId = sessionId;
+    }
+    /**
+     * Private method to ensure that the necessary table exists in the
+     * Cloudflare D1 database before performing any operations. If the table
+     * does not exist, it is created.
+     * @returns Promise that resolves to void.
+     */
+    async ensureTable() {
+        if (this.tableInitialized) {
+            return;
+        }
+        const query = `CREATE TABLE IF NOT EXISTS ${this.tableName} (id TEXT PRIMARY KEY, session_id TEXT, type TEXT, content TEXT, role TEXT, name TEXT, additional_kwargs TEXT);`;
+        await this.database.prepare(query).bind().all();
+        const idIndexQuery = `CREATE INDEX IF NOT EXISTS id_index ON ${this.tableName} (id);`;
+        await this.database.prepare(idIndexQuery).bind().all();
+        const sessionIdIndexQuery = `CREATE INDEX IF NOT EXISTS session_id_index ON ${this.tableName} (session_id);`;
+        await this.database.prepare(sessionIdIndexQuery).bind().all();
+        this.tableInitialized = true;
+    }
+    /**
+     * Method to retrieve all messages from the Cloudflare D1 database for the
+     * current session.
+     * @returns Promise that resolves to an array of BaseMessage objects.
+     */
+    async getMessages() {
+        await this.ensureTable();
+        const query = `SELECT * FROM ${this.tableName} WHERE session_id = ?`;
+        const rawStoredMessages = await this.database
+            .prepare(query)
+            .bind(this.sessionId)
+            .all();
+        const storedMessagesObject = rawStoredMessages.results;
+        const orderedMessages = storedMessagesObject.map((message) => {
+            const data = {
+                content: message.content,
+                additional_kwargs: JSON.parse(message.additional_kwargs),
+            };
+            if (message.role) {
+                data.role = message.role;
+            }
+            if (message.name) {
+                data.name = message.name;
+            }
+            return {
+                type: message.type,
+                data,
+            };
+        });
+        return (0, utils_js_1.mapStoredMessagesToChatMessages)(orderedMessages);
+    }
+    /**
+     * Method to add a new message to the Cloudflare D1 database for the current
+     * session.
+     * @param message The BaseMessage object to be added to the database.
+     * @returns Promise that resolves to void.
+     */
+    async addMessage(message) {
+        await this.ensureTable();
+        const messageToAdd = (0, utils_js_1.mapChatMessagesToStoredMessages)([message]);
+        const query = `INSERT INTO ${this.tableName} (id, session_id, type, content, role, name, additional_kwargs) VALUES(?, ?, ?, ?, ?, ?, ?)`;
+        const id = (0, uuid_1.v4)();
+        await this.database
+            .prepare(query)
+            .bind(id, this.sessionId, messageToAdd[0].type || null, messageToAdd[0].data.content || null, messageToAdd[0].data.role || null, messageToAdd[0].data.name || null, JSON.stringify(messageToAdd[0].data.additional_kwargs))
+            .all();
+    }
+    /**
+     * Method to delete all messages from the Cloudflare D1 database for the
+     * current session.
+     * @returns Promise that resolves to void.
+     */
+    async clear() {
+        await this.ensureTable();
+        const query = `DELETE FROM ? WHERE session_id = ? `;
+        await this.database
+            .prepare(query)
+            .bind(this.tableName, this.sessionId)
+            .all();
+    }
+}
+exports.CloudflareD1MessageHistory = CloudflareD1MessageHistory;

package/dist/stores/message/cloudflare_d1.d.ts

@@ -0,0 +1,49 @@
+import type { D1Database } from "@cloudflare/workers-types";
+import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
+/**
+ * Type definition for the input parameters required when instantiating a
+ * CloudflareD1MessageHistory object.
+ */
+export type CloudflareD1MessageHistoryInput = {
+    tableName?: string;
+    sessionId: string;
+    database?: D1Database;
+};
+/**
+ * Class for storing and retrieving chat message history from a
+ * Cloudflare D1 database. Extends the BaseListChatMessageHistory class.
+ */
+export declare class CloudflareD1MessageHistory extends BaseListChatMessageHistory {
+    lc_namespace: string[];
+    database: D1Database;
+    private tableName;
+    private sessionId;
+    private tableInitialized;
+    constructor(fields: CloudflareD1MessageHistoryInput);
+    /**
+     * Private method to ensure that the necessary table exists in the
+     * Cloudflare D1 database before performing any operations. If the table
+     * does not exist, it is created.
+     * @returns Promise that resolves to void.
+     */
+    private ensureTable;
+    /**
+     * Method to retrieve all messages from the Cloudflare D1 database for the
+     * current session.
+     * @returns Promise that resolves to an array of BaseMessage objects.
+     */
+    getMessages(): Promise<BaseMessage[]>;
+    /**
+     * Method to add a new message to the Cloudflare D1 database for the current
+     * session.
+     * @param message The BaseMessage object to be added to the database.
+     * @returns Promise that resolves to void.
+     */
+    addMessage(message: BaseMessage): Promise<void>;
+    /**
+     * Method to delete all messages from the Cloudflare D1 database for the
+     * current session.
+     * @returns Promise that resolves to void.
+     */
+    clear(): Promise<void>;
+}