langchain 0.0.179 → 0.0.180

package/dist/llms/{bedrock.cjs → bedrock/web.cjs}

@@ -2,15 +2,14 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Bedrock = void 0;
 const signature_v4_1 = require("@smithy/signature-v4");
-const credential_provider_node_1 = require("@aws-sdk/credential-provider-node");
 const protocol_http_1 = require("@smithy/protocol-http");
 const eventstream_codec_1 = require("@smithy/eventstream-codec");
 const util_utf8_1 = require("@smithy/util-utf8");
 const sha256_js_1 = require("@aws-crypto/sha256-js");
-const bedrock_js_1 = require("../util/bedrock.cjs");
-const env_js_1 = require("../util/env.cjs");
-const base_js_1 = require("./base.cjs");
-const index_js_1 = require("../schema/index.cjs");
+const bedrock_js_1 = require("../../util/bedrock.cjs");
+const env_js_1 = require("../../util/env.cjs");
+const base_js_1 = require("../base.cjs");
+const index_js_1 = require("../../schema/index.cjs");
 /**
  * A type of Large Language Model (LLM) that interacts with the Bedrock
  * service. It extends the base `LLM` class and implements the
@@ -21,8 +20,20 @@ const index_js_1 = require("../schema/index.cjs");
  * region, and the maximum number of tokens to generate.
  */
 class Bedrock extends base_js_1.LLM {
+    get lc_aliases() {
+        return {
+            model: "model_id",
+            region: "region_name",
+        };
+    }
     get lc_secrets() {
-        return {};
+        return {
+            "credentials.accessKeyId": "BEDROCK_AWS_ACCESS_KEY_ID",
+            "credentials.secretAccessKey": "BEDROCK_AWS_SECRET_ACCESS_KEY",
+        };
+    }
+    get lc_attributes() {
+        return { region: this.region };
     }
     _llmType() {
         return "bedrock";
@@ -74,6 +85,7 @@ class Bedrock extends base_js_1.LLM {
             writable: true,
             value: void 0
         });
+        /** @deprecated */
         Object.defineProperty(this, "stopSequences", {
             enumerable: true,
             configurable: true,
@@ -98,8 +110,14 @@ class Bedrock extends base_js_1.LLM {
             writable: true,
             value: false
         });
+        Object.defineProperty(this, "lc_serializable", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         this.model = fields?.model ?? this.model;
-        const allowedModels = ["ai21", "anthropic", "amazon"];
+        const allowedModels = ["ai21", "anthropic", "amazon", "cohere"];
         if (!allowedModels.includes(this.model.split(".")[0])) {
             throw new Error(`Unknown model: '${this.model}', only these are supported: ${allowedModels}`);
         }
@@ -108,7 +126,11 @@ class Bedrock extends base_js_1.LLM {
             throw new Error("Please set the AWS_DEFAULT_REGION environment variable or pass it to the constructor as the region field.");
         }
         this.region = region;
-        this.credentials = fields?.credentials ?? (0, credential_provider_node_1.defaultProvider)();
+        const credentials = fields?.credentials;
+        if (!credentials) {
+            throw new Error("Please set the AWS credentials in the 'credentials' field.");
+        }
+        this.credentials = credentials;
         this.temperature = fields?.temperature ?? this.temperature;
         this.maxTokens = fields?.maxTokens ?? this.maxTokens;
         this.fetchFn = fields?.fetchFn ?? fetch;
@@ -158,7 +180,7 @@ class Bedrock extends base_js_1.LLM {
     }
     async _signedFetch(prompt, options, fields) {
         const { bedrockMethod, endpointHost, provider } = fields;
-        const inputBody = bedrock_js_1.BedrockLLMInputOutputAdapter.prepareInput(provider, prompt, this.maxTokens, this.temperature, this.stopSequences, this.modelKwargs);
+        const inputBody = bedrock_js_1.BedrockLLMInputOutputAdapter.prepareInput(provider, prompt, this.maxTokens, this.temperature, options.stop ?? this.stopSequences, this.modelKwargs, fields.bedrockMethod);
         const url = new URL(`https://${endpointHost}/model/${this.model}/${bedrockMethod}`);
         const request = new protocol_http_1.HttpRequest({
             hostname: url.hostname,
@@ -189,9 +211,21 @@ class Bedrock extends base_js_1.LLM {
         }));
         return response;
     }
+    invocationParams(options) {
+        return {
+            model: this.model,
+            region: this.region,
+            temperature: this.temperature,
+            maxTokens: this.maxTokens,
+            stop: options?.stop ?? this.stopSequences,
+            modelKwargs: this.modelKwargs,
+        };
+    }
     async *_streamResponseChunks(prompt, options, runManager) {
         const provider = this.model.split(".")[0];
-        const bedrockMethod = provider === "anthropic" ? "invoke-with-response-stream" : "invoke";
+        const bedrockMethod = provider === "anthropic" || provider === "cohere"
+            ? "invoke-with-response-stream"
+            : "invoke";
         const service = "bedrock-runtime";
         const endpointHost = this.endpointHost ?? `${service}.${this.region}.amazonaws.com`;
         // Send request to AWS using the low-level fetch API
@@ -203,7 +237,7 @@ class Bedrock extends base_js_1.LLM {
         if (response.status < 200 || response.status >= 300) {
             throw Error(`Failed to access underlying url '${endpointHost}': got ${response.status} ${response.statusText}: ${await response.text()}`);
         }
-        if (provider === "anthropic") {
+        if (provider === "anthropic" || provider === "cohere") {
             const reader = response.body?.getReader();
             const decoder = new TextDecoder();
             for await (const chunk of this._readChunks(reader)) {
@@ -218,7 +252,7 @@ class Bedrock extends base_js_1.LLM {
                     throw new Error(body.message);
                 }
                 if (body.bytes !== undefined) {
-                    const chunkResult = JSON.parse(Buffer.from(body.bytes, "base64").toString());
+                    const chunkResult = JSON.parse(decoder.decode(Uint8Array.from(atob(body.bytes), (m) => m.codePointAt(0) ?? 0)));
                     const text = bedrock_js_1.BedrockLLMInputOutputAdapter.prepareOutput(provider, chunkResult);
                     yield new index_js_1.GenerationChunk({
                         text,

package/dist/llms/{bedrock.d.ts → bedrock/web.d.ts}

@@ -1,8 +1,9 @@
 import { EventStreamCodec } from "@smithy/eventstream-codec";
-import { BaseBedrockInput, type CredentialType } from "../util/bedrock.js";
-import { LLM, BaseLLMParams } from "./base.js";
-import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
-import { GenerationChunk } from "../schema/index.js";
+import { BaseBedrockInput, type CredentialType } from "../../util/bedrock.js";
+import { LLM, BaseLLMParams } from "../base.js";
+import { CallbackManagerForLLMRun } from "../../callbacks/manager.js";
+import { GenerationChunk } from "../../schema/index.js";
+import { SerializedFields } from "../../load/map_keys.js";
 /**
  * A type of Large Language Model (LLM) that interacts with the Bedrock
  * service. It extends the base `LLM` class and implements the
@@ -20,13 +21,17 @@ export declare class Bedrock extends LLM implements BaseBedrockInput {
     maxTokens?: number | undefined;
     fetchFn: typeof fetch;
     endpointHost?: string;
+    /** @deprecated */
     stopSequences?: string[];
     modelKwargs?: Record<string, unknown>;
     codec: EventStreamCodec;
     streaming: boolean;
+    lc_serializable: boolean;
+    get lc_aliases(): Record<string, string>;
     get lc_secrets(): {
         [key: string]: string;
     } | undefined;
+    get lc_attributes(): SerializedFields | undefined;
     _llmType(): string;
     static lc_name(): string;
     constructor(fields?: Partial<BaseBedrockInput> & BaseLLMParams);
@@ -46,6 +51,14 @@ export declare class Bedrock extends LLM implements BaseBedrockInput {
         endpointHost: string;
         provider: string;
     }): Promise<Response>;
+    invocationParams(options?: this["ParsedCallOptions"]): {
+        model: string;
+        region: string;
+        temperature: number | undefined;
+        maxTokens: number | undefined;
+        stop: string[] | undefined;
+        modelKwargs: Record<string, unknown> | undefined;
+    };
     _streamResponseChunks(prompt: string, options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<GenerationChunk>;
     _readChunks(reader: any): {
         [Symbol.asyncIterator](): AsyncGenerator<any, void, unknown>;

package/dist/llms/{bedrock.js → bedrock/web.js}

@@ -1,13 +1,12 @@
 import { SignatureV4 } from "@smithy/signature-v4";
-import { defaultProvider } from "@aws-sdk/credential-provider-node";
 import { HttpRequest } from "@smithy/protocol-http";
 import { EventStreamCodec } from "@smithy/eventstream-codec";
 import { fromUtf8, toUtf8 } from "@smithy/util-utf8";
 import { Sha256 } from "@aws-crypto/sha256-js";
-import { BedrockLLMInputOutputAdapter, } from "../util/bedrock.js";
-import { getEnvironmentVariable } from "../util/env.js";
-import { LLM } from "./base.js";
-import { GenerationChunk } from "../schema/index.js";
+import { BedrockLLMInputOutputAdapter, } from "../../util/bedrock.js";
+import { getEnvironmentVariable } from "../../util/env.js";
+import { LLM } from "../base.js";
+import { GenerationChunk } from "../../schema/index.js";
 /**
  * A type of Large Language Model (LLM) that interacts with the Bedrock
  * service. It extends the base `LLM` class and implements the
@@ -18,8 +17,20 @@ import { GenerationChunk } from "../schema/index.js";
  * region, and the maximum number of tokens to generate.
  */
 export class Bedrock extends LLM {
+    get lc_aliases() {
+        return {
+            model: "model_id",
+            region: "region_name",
+        };
+    }
     get lc_secrets() {
-        return {};
+        return {
+            "credentials.accessKeyId": "BEDROCK_AWS_ACCESS_KEY_ID",
+            "credentials.secretAccessKey": "BEDROCK_AWS_SECRET_ACCESS_KEY",
+        };
+    }
+    get lc_attributes() {
+        return { region: this.region };
     }
     _llmType() {
         return "bedrock";
@@ -71,6 +82,7 @@ export class Bedrock extends LLM {
             writable: true,
             value: void 0
         });
+        /** @deprecated */
         Object.defineProperty(this, "stopSequences", {
             enumerable: true,
             configurable: true,
@@ -95,8 +107,14 @@ export class Bedrock extends LLM {
             writable: true,
             value: false
         });
+        Object.defineProperty(this, "lc_serializable", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         this.model = fields?.model ?? this.model;
-        const allowedModels = ["ai21", "anthropic", "amazon"];
+        const allowedModels = ["ai21", "anthropic", "amazon", "cohere"];
         if (!allowedModels.includes(this.model.split(".")[0])) {
             throw new Error(`Unknown model: '${this.model}', only these are supported: ${allowedModels}`);
         }
@@ -105,7 +123,11 @@ export class Bedrock extends LLM {
             throw new Error("Please set the AWS_DEFAULT_REGION environment variable or pass it to the constructor as the region field.");
         }
         this.region = region;
-        this.credentials = fields?.credentials ?? defaultProvider();
+        const credentials = fields?.credentials;
+        if (!credentials) {
+            throw new Error("Please set the AWS credentials in the 'credentials' field.");
+        }
+        this.credentials = credentials;
         this.temperature = fields?.temperature ?? this.temperature;
         this.maxTokens = fields?.maxTokens ?? this.maxTokens;
         this.fetchFn = fields?.fetchFn ?? fetch;
@@ -155,7 +177,7 @@ export class Bedrock extends LLM {
     }
     async _signedFetch(prompt, options, fields) {
         const { bedrockMethod, endpointHost, provider } = fields;
-        const inputBody = BedrockLLMInputOutputAdapter.prepareInput(provider, prompt, this.maxTokens, this.temperature, this.stopSequences, this.modelKwargs);
+        const inputBody = BedrockLLMInputOutputAdapter.prepareInput(provider, prompt, this.maxTokens, this.temperature, options.stop ?? this.stopSequences, this.modelKwargs, fields.bedrockMethod);
         const url = new URL(`https://${endpointHost}/model/${this.model}/${bedrockMethod}`);
         const request = new HttpRequest({
             hostname: url.hostname,
@@ -186,9 +208,21 @@ export class Bedrock extends LLM {
         }));
         return response;
     }
+    invocationParams(options) {
+        return {
+            model: this.model,
+            region: this.region,
+            temperature: this.temperature,
+            maxTokens: this.maxTokens,
+            stop: options?.stop ?? this.stopSequences,
+            modelKwargs: this.modelKwargs,
+        };
+    }
     async *_streamResponseChunks(prompt, options, runManager) {
         const provider = this.model.split(".")[0];
-        const bedrockMethod = provider === "anthropic" ? "invoke-with-response-stream" : "invoke";
+        const bedrockMethod = provider === "anthropic" || provider === "cohere"
+            ? "invoke-with-response-stream"
+            : "invoke";
         const service = "bedrock-runtime";
         const endpointHost = this.endpointHost ?? `${service}.${this.region}.amazonaws.com`;
         // Send request to AWS using the low-level fetch API
@@ -200,7 +234,7 @@ export class Bedrock extends LLM {
         if (response.status < 200 || response.status >= 300) {
             throw Error(`Failed to access underlying url '${endpointHost}': got ${response.status} ${response.statusText}: ${await response.text()}`);
         }
-        if (provider === "anthropic") {
+        if (provider === "anthropic" || provider === "cohere") {
             const reader = response.body?.getReader();
             const decoder = new TextDecoder();
             for await (const chunk of this._readChunks(reader)) {
@@ -215,7 +249,7 @@ export class Bedrock extends LLM {
                     throw new Error(body.message);
                 }
                 if (body.bytes !== undefined) {
-                    const chunkResult = JSON.parse(Buffer.from(body.bytes, "base64").toString());
+                    const chunkResult = JSON.parse(decoder.decode(Uint8Array.from(atob(body.bytes), (m) => m.codePointAt(0) ?? 0)));
                     const text = BedrockLLMInputOutputAdapter.prepareOutput(provider, chunkResult);
                     yield new GenerationChunk({
                         text,

package/dist/load/import_constants.cjs

@@ -36,6 +36,7 @@ exports.optionalImportEntrypoints = [
     "langchain/llms/googlepalm",
     "langchain/llms/sagemaker_endpoint",
     "langchain/llms/bedrock",
+    "langchain/llms/bedrock/web",
     "langchain/llms/llama_cpp",
     "langchain/llms/writer",
     "langchain/llms/portkey",
@@ -111,6 +112,7 @@ exports.optionalImportEntrypoints = [
     "langchain/document_transformers/mozilla_readability",
     "langchain/chat_models/portkey",
     "langchain/chat_models/bedrock",
+    "langchain/chat_models/bedrock/web",
     "langchain/chat_models/googlevertexai",
     "langchain/chat_models/googlevertexai/web",
     "langchain/chat_models/googlepalm",
@@ -137,6 +139,7 @@ exports.optionalImportEntrypoints = [
     "langchain/cache/upstash_redis",
     "langchain/stores/doc/gcs",
     "langchain/stores/file/node",
+    "langchain/stores/message/cassandra",
     "langchain/stores/message/convex",
     "langchain/stores/message/cloudflare_d1",
     "langchain/stores/message/dynamodb",

package/dist/load/import_constants.js

@@ -33,6 +33,7 @@ export const optionalImportEntrypoints = [
     "langchain/llms/googlepalm",
     "langchain/llms/sagemaker_endpoint",
     "langchain/llms/bedrock",
+    "langchain/llms/bedrock/web",
     "langchain/llms/llama_cpp",
     "langchain/llms/writer",
     "langchain/llms/portkey",
@@ -108,6 +109,7 @@ export const optionalImportEntrypoints = [
     "langchain/document_transformers/mozilla_readability",
     "langchain/chat_models/portkey",
     "langchain/chat_models/bedrock",
+    "langchain/chat_models/bedrock/web",
     "langchain/chat_models/googlevertexai",
     "langchain/chat_models/googlevertexai/web",
     "langchain/chat_models/googlepalm",
@@ -134,6 +136,7 @@ export const optionalImportEntrypoints = [
     "langchain/cache/upstash_redis",
     "langchain/stores/doc/gcs",
     "langchain/stores/file/node",
+    "langchain/stores/message/cassandra",
     "langchain/stores/message/convex",
     "langchain/stores/message/cloudflare_d1",
     "langchain/stores/message/dynamodb",

package/dist/stores/message/cassandra.cjs

@@ -0,0 +1,135 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CassandraChatMessageHistory = void 0;
+const cassandra_driver_1 = require("cassandra-driver");
+const index_js_1 = require("../../schema/index.cjs");
+const utils_js_1 = require("./utils.cjs");
+/**
+ * Class for storing chat message history within Cassandra. It extends the
+ * BaseListChatMessageHistory class and provides methods to get, add, and
+ * clear messages.
+ */
+class CassandraChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
+    constructor(options) {
+        super();
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "stores", "message", "cassandra"]
+        });
+        Object.defineProperty(this, "keyspace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "table", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "client", {
+            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, "tableExists", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "options", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "queries", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.client = new cassandra_driver_1.Client(options);
+        this.keyspace = options.keyspace;
+        this.table = options.table;
+        this.sessionId = options.sessionId;
+        this.tableExists = false;
+        this.options = options;
+    }
+    /**
+     * Method to get all the messages stored in the Cassandra database.
+     * @returns Array of stored BaseMessage instances.
+     */
+    async getMessages() {
+        await this.ensureTable();
+        const resultSet = await this.client.execute(this.queries.select, [this.sessionId], { prepare: true });
+        const storedMessages = resultSet.rows.map((row) => ({
+            type: row.message_type,
+            data: JSON.parse(row.data),
+        }));
+        const baseMessages = (0, utils_js_1.mapStoredMessagesToChatMessages)(storedMessages);
+        return baseMessages;
+    }
+    /**
+     * Method to add a new message to the Cassandra database.
+     * @param message The BaseMessage instance to add.
+     * @returns A promise that resolves when the message has been added.
+     */
+    async addMessage(message) {
+        await this.ensureTable();
+        const messages = (0, utils_js_1.mapChatMessagesToStoredMessages)([message]);
+        const { type, data } = messages[0];
+        return this.client
+            .execute(this.queries.insert, [this.sessionId, type, JSON.stringify(data)], { prepare: true, ...this.options })
+            .then(() => { });
+    }
+    /**
+     * Method to clear all the messages from the Cassandra database.
+     * @returns A promise that resolves when all messages have been cleared.
+     */
+    async clear() {
+        await this.ensureTable();
+        return this.client
+            .execute(this.queries.delete, [this.sessionId], {
+            prepare: true,
+            ...this.options,
+        })
+            .then(() => { });
+    }
+    /**
+     * Method to initialize the Cassandra database.
+     * @returns Promise that resolves when the database has been initialized.
+     */
+    async ensureTable() {
+        if (this.tableExists) {
+            return;
+        }
+        await this.client.execute(`
+    CREATE TABLE IF NOT EXISTS ${this.keyspace}.${this.table} (
+        session_id text,
+        message_ts timestamp,
+        message_type text,
+        data text,
+        PRIMARY KEY ((session_id), message_ts)
+      );
+    `);
+        this.queries = {
+            insert: `INSERT INTO ${this.keyspace}.${this.table} (session_id, message_ts, message_type, data) VALUES (?, toTimestamp(now()), ?, ?);`,
+            select: `SELECT message_type, data FROM ${this.keyspace}.${this.table} WHERE session_id = ?;`,
+            delete: `DELETE FROM ${this.keyspace}.${this.table} WHERE session_id = ?;`,
+        };
+        this.tableExists = true;
+    }
+}
+exports.CassandraChatMessageHistory = CassandraChatMessageHistory;

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

@@ -0,0 +1,44 @@
+import { DseClientOptions } from "cassandra-driver";
+import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
+export interface CassandraChatMessageHistoryOptions extends DseClientOptions {
+    keyspace: string;
+    table: string;
+    sessionId: string;
+}
+/**
+ * Class for storing chat message history within Cassandra. It extends the
+ * BaseListChatMessageHistory class and provides methods to get, add, and
+ * clear messages.
+ */
+export declare class CassandraChatMessageHistory extends BaseListChatMessageHistory {
+    lc_namespace: string[];
+    private keyspace;
+    private table;
+    private client;
+    private sessionId;
+    private tableExists;
+    private options;
+    private queries;
+    constructor(options: CassandraChatMessageHistoryOptions);
+    /**
+     * Method to get all the messages stored in the Cassandra database.
+     * @returns Array of stored BaseMessage instances.
+     */
+    getMessages(): Promise<BaseMessage[]>;
+    /**
+     * Method to add a new message to the Cassandra database.
+     * @param message The BaseMessage instance to add.
+     * @returns A promise that resolves when the message has been added.
+     */
+    addMessage(message: BaseMessage): Promise<void>;
+    /**
+     * Method to clear all the messages from the Cassandra database.
+     * @returns A promise that resolves when all messages have been cleared.
+     */
+    clear(): Promise<void>;
+    /**
+     * Method to initialize the Cassandra database.
+     * @returns Promise that resolves when the database has been initialized.
+     */
+    private ensureTable;
+}

package/dist/stores/message/cassandra.js

@@ -0,0 +1,131 @@
+import { Client } from "cassandra-driver";
+import { BaseListChatMessageHistory, } from "../../schema/index.js";
+import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
+/**
+ * Class for storing chat message history within Cassandra. It extends the
+ * BaseListChatMessageHistory class and provides methods to get, add, and
+ * clear messages.
+ */
+export class CassandraChatMessageHistory extends BaseListChatMessageHistory {
+    constructor(options) {
+        super();
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "stores", "message", "cassandra"]
+        });
+        Object.defineProperty(this, "keyspace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "table", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "client", {
+            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, "tableExists", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "options", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "queries", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.client = new Client(options);
+        this.keyspace = options.keyspace;
+        this.table = options.table;
+        this.sessionId = options.sessionId;
+        this.tableExists = false;
+        this.options = options;
+    }
+    /**
+     * Method to get all the messages stored in the Cassandra database.
+     * @returns Array of stored BaseMessage instances.
+     */
+    async getMessages() {
+        await this.ensureTable();
+        const resultSet = await this.client.execute(this.queries.select, [this.sessionId], { prepare: true });
+        const storedMessages = resultSet.rows.map((row) => ({
+            type: row.message_type,
+            data: JSON.parse(row.data),
+        }));
+        const baseMessages = mapStoredMessagesToChatMessages(storedMessages);
+        return baseMessages;
+    }
+    /**
+     * Method to add a new message to the Cassandra database.
+     * @param message The BaseMessage instance to add.
+     * @returns A promise that resolves when the message has been added.
+     */
+    async addMessage(message) {
+        await this.ensureTable();
+        const messages = mapChatMessagesToStoredMessages([message]);
+        const { type, data } = messages[0];
+        return this.client
+            .execute(this.queries.insert, [this.sessionId, type, JSON.stringify(data)], { prepare: true, ...this.options })
+            .then(() => { });
+    }
+    /**
+     * Method to clear all the messages from the Cassandra database.
+     * @returns A promise that resolves when all messages have been cleared.
+     */
+    async clear() {
+        await this.ensureTable();
+        return this.client
+            .execute(this.queries.delete, [this.sessionId], {
+            prepare: true,
+            ...this.options,
+        })
+            .then(() => { });
+    }
+    /**
+     * Method to initialize the Cassandra database.
+     * @returns Promise that resolves when the database has been initialized.
+     */
+    async ensureTable() {
+        if (this.tableExists) {
+            return;
+        }
+        await this.client.execute(`
+    CREATE TABLE IF NOT EXISTS ${this.keyspace}.${this.table} (
+        session_id text,
+        message_ts timestamp,
+        message_type text,
+        data text,
+        PRIMARY KEY ((session_id), message_ts)
+      );
+    `);
+        this.queries = {
+            insert: `INSERT INTO ${this.keyspace}.${this.table} (session_id, message_ts, message_type, data) VALUES (?, toTimestamp(now()), ?, ?);`,
+            select: `SELECT message_type, data FROM ${this.keyspace}.${this.table} WHERE session_id = ?;`,
+            delete: `DELETE FROM ${this.keyspace}.${this.table} WHERE session_id = ?;`,
+        };
+        this.tableExists = true;
+    }
+}