langchain 0.0.151 → 0.0.152

package/dist/stores/message/cloudflare_d1.js

@@ -0,0 +1,130 @@
+import { v4 } from "uuid";
+import { BaseListChatMessageHistory, } from "../../schema/index.js";
+import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
+/**
+ * Class for storing and retrieving chat message history from a
+ * Cloudflare D1 database. Extends the BaseListChatMessageHistory class.
+ */
+export class CloudflareD1MessageHistory extends 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 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 = mapChatMessagesToStoredMessages([message]);
+        const query = `INSERT INTO ${this.tableName} (id, session_id, type, content, role, name, additional_kwargs) VALUES(?, ?, ?, ?, ?, ?, ?)`;
+        const id = 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();
+    }
+}

package/dist/types/openai-types.d.ts

@@ -1,5 +1,7 @@
 import type { OpenAI as OpenAIClient } from "openai";
+import { TiktokenModel } from "js-tiktoken/lite";
 import { BaseLanguageModelCallOptions } from "../base_language/index.js";
+export type { TiktokenModel };
 export declare interface OpenAIBaseInput {
     /** Sampling temperature to use */
     temperature: number;

package/document_loaders/web/pdf.cjs

@@ -0,0 +1 @@
+module.exports = require('../../dist/document_loaders/web/pdf.cjs');

package/document_loaders/web/pdf.d.ts

@@ -0,0 +1 @@
+export * from '../../dist/document_loaders/web/pdf.js'

package/document_loaders/web/pdf.js

@@ -0,0 +1 @@
+export * from '../../dist/document_loaders/web/pdf.js'

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.151",
+  "version": "0.0.152",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {
@@ -313,6 +313,9 @@
     "document_loaders/web/notionapi.cjs",
     "document_loaders/web/notionapi.js",
     "document_loaders/web/notionapi.d.ts",
+    "document_loaders/web/pdf.cjs",
+    "document_loaders/web/pdf.js",
+    "document_loaders/web/pdf.d.ts",
     "document_loaders/web/recursive_url.cjs",
     "document_loaders/web/recursive_url.js",
     "document_loaders/web/recursive_url.d.ts",
@@ -511,6 +514,9 @@
     "cache.cjs",
     "cache.js",
     "cache.d.ts",
+    "cache/cloudflare_kv.cjs",
+    "cache/cloudflare_kv.js",
+    "cache/cloudflare_kv.d.ts",
     "cache/momento.cjs",
     "cache/momento.js",
     "cache/momento.d.ts",
@@ -535,6 +541,9 @@
     "stores/file/node.cjs",
     "stores/file/node.js",
     "stores/file/node.d.ts",
+    "stores/message/cloudflare_d1.cjs",
+    "stores/message/cloudflare_d1.js",
+    "stores/message/cloudflare_d1.d.ts",
     "stores/message/in_memory.cjs",
     "stores/message/in_memory.js",
     "stores/message/in_memory.d.ts",
@@ -646,6 +655,7 @@
     "@aws-sdk/types": "^3.357.0",
     "@azure/storage-blob": "^12.15.0",
     "@clickhouse/client": "^0.0.14",
+    "@cloudflare/workers-types": "^4.20230904.0",
     "@elastic/elasticsearch": "^8.4.0",
     "@faker-js/faker": "^7.6.0",
     "@getmetal/metal-sdk": "^4.0.0",
@@ -765,6 +775,7 @@
     "@aws-sdk/signature-v4": "^3.374.0",
     "@azure/storage-blob": "^12.15.0",
     "@clickhouse/client": "^0.0.14",
+    "@cloudflare/workers-types": "^4.20230904.0",
     "@elastic/elasticsearch": "^8.4.0",
     "@getmetal/metal-sdk": "*",
     "@getzep/zep-js": "^0.7.0",
@@ -871,6 +882,9 @@
     "@clickhouse/client": {
       "optional": true
     },
+    "@cloudflare/workers-types": {
+      "optional": true
+    },
     "@elastic/elasticsearch": {
       "optional": true
     },
@@ -1623,6 +1637,11 @@
       "import": "./document_loaders/web/notionapi.js",
       "require": "./document_loaders/web/notionapi.cjs"
     },
+    "./document_loaders/web/pdf": {
+      "types": "./document_loaders/web/pdf.d.ts",
+      "import": "./document_loaders/web/pdf.js",
+      "require": "./document_loaders/web/pdf.cjs"
+    },
     "./document_loaders/web/recursive_url": {
       "types": "./document_loaders/web/recursive_url.d.ts",
       "import": "./document_loaders/web/recursive_url.js",
@@ -1953,6 +1972,11 @@
       "import": "./cache.js",
       "require": "./cache.cjs"
     },
+    "./cache/cloudflare_kv": {
+      "types": "./cache/cloudflare_kv.d.ts",
+      "import": "./cache/cloudflare_kv.js",
+      "require": "./cache/cloudflare_kv.cjs"
+    },
     "./cache/momento": {
       "types": "./cache/momento.d.ts",
       "import": "./cache/momento.js",
@@ -1993,6 +2017,11 @@
       "import": "./stores/file/node.js",
       "require": "./stores/file/node.cjs"
     },
+    "./stores/message/cloudflare_d1": {
+      "types": "./stores/message/cloudflare_d1.d.ts",
+      "import": "./stores/message/cloudflare_d1.js",
+      "require": "./stores/message/cloudflare_d1.cjs"
+    },
     "./stores/message/in_memory": {
       "types": "./stores/message/in_memory.d.ts",
       "import": "./stores/message/in_memory.js",

package/stores/message/cloudflare_d1.cjs

@@ -0,0 +1 @@
+module.exports = require('../../dist/stores/message/cloudflare_d1.cjs');

package/stores/message/cloudflare_d1.d.ts

@@ -0,0 +1 @@
+export * from '../../dist/stores/message/cloudflare_d1.js'

package/stores/message/cloudflare_d1.js

@@ -0,0 +1 @@
+export * from '../../dist/stores/message/cloudflare_d1.js'