@langchain/google-cloud-sql-pg 0.0.1 → 1.0.0

package/dist/chat_message_history.js

@@ -1,69 +1,46 @@
 import { BaseChatMessageHistory } from "@langchain/core/chat_history";
-import { AIMessage, HumanMessage, mapStoredMessagesToChatMessages, } from "@langchain/core/messages";
-export class PostgresChatMessageHistory extends BaseChatMessageHistory {
-    constructor({ engine, sessionId, tableName, schemaName = "public", }) {
-        super();
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: [
-                "langchain",
-                "stores",
-                "message",
-                "google-cloud-sql-pg",
-            ]
-        });
-        Object.defineProperty(this, "engine", {
-            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, "tableName", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "schemaName", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.engine = engine;
-        this.sessionId = sessionId;
-        this.tableName = tableName;
-        this.schemaName = schemaName;
-    }
-    /**
-     * Create a new PostgresChatMessageHistory instance.
-     *
-     * @param {PostgresEngine} engine Postgres engine instance to use.
-     * @param {string} sessionId Retrieve the table content with this session ID.
-     * @param {string} tableName Table name that stores that chat message history. Parameter is not escaped. Do not use with end user input.
-     * @param {string} schemaName Schema name for the chat message history table. Default: "public". Parameter is not escaped. Do not use with end user input.
-     * @returns PostgresChatMessageHistory instance.
-     */
-    static async initialize(engine, sessionId, tableName, schemaName = "public") {
-        const query = `SELECT column_name, data_type FROM information_schema.columns WHERE table_name = '${tableName}' AND table_schema = '${schemaName}'`;
-        const { rows } = await engine.pool.raw(query);
-        const columnNames = [];
-        for (const index in rows) {
-            if (Object.prototype.hasOwnProperty.call(rows, index)) {
-                columnNames.push(rows[index].column_name);
-            }
-        }
-        const requiredColumns = ["id", "session_id", "data", "type"];
-        if (!requiredColumns.every((x) => columnNames.includes(x))) {
-            throw new Error(`Table '${schemaName}'.'${tableName}' has incorrect schema.
+import { AIMessage, HumanMessage, mapStoredMessagesToChatMessages } from "@langchain/core/messages";
+
+//#region src/chat_message_history.ts
+var PostgresChatMessageHistory = class PostgresChatMessageHistory extends BaseChatMessageHistory {
+	lc_namespace = [
+		"langchain",
+		"stores",
+		"message",
+		"google-cloud-sql-pg"
+	];
+	engine;
+	sessionId;
+	tableName;
+	schemaName;
+	constructor({ engine, sessionId, tableName, schemaName = "public" }) {
+		super();
+		this.engine = engine;
+		this.sessionId = sessionId;
+		this.tableName = tableName;
+		this.schemaName = schemaName;
+	}
+	/**
+	* Create a new PostgresChatMessageHistory instance.
+	*
+	* @param {PostgresEngine} engine Postgres engine instance to use.
+	* @param {string} sessionId Retrieve the table content with this session ID.
+	* @param {string} tableName Table name that stores that chat message history. Parameter is not escaped. Do not use with end user input.
+	* @param {string} schemaName Schema name for the chat message history table. Default: "public". Parameter is not escaped. Do not use with end user input.
+	* @returns PostgresChatMessageHistory instance.
+	*/
+	static async initialize(engine, sessionId, tableName, schemaName = "public") {
+		const query = `SELECT column_name, data_type FROM information_schema.columns WHERE table_name = '${tableName}' AND table_schema = '${schemaName}'`;
+		const { rows } = await engine.pool.raw(query);
+		const columnNames = [];
+		for (const index in rows) if (Object.prototype.hasOwnProperty.call(rows, index)) columnNames.push(rows[index].column_name);
+		const requiredColumns = [
+			"id",
+			"session_id",
+			"data",
+			"type"
+		];
+		if (!requiredColumns.every((x) => columnNames.includes(x))) throw new Error(`Table '${schemaName}'.'${tableName}' has incorrect schema.
         Got column names ${columnNames} but required column names ${requiredColumns}.
         Please create table with following schema: CREATE TABLE '${schemaName}'.'${tableName}' (
             id SERIAL AUTO_INCREMENT PRIMARY KEY,
@@ -72,71 +49,64 @@ export class PostgresChatMessageHistory extends BaseChatMessageHistory {
             type TEXT NOT NULL
         );
       `);
-        }
-        return new PostgresChatMessageHistory({
-            engine,
-            sessionId,
-            tableName,
-            schemaName,
-        });
-    }
-    addUserMessage(message) {
-        return this.addMessage(new HumanMessage(message));
-    }
-    addAIChatMessage(message) {
-        return this.addMessage(new AIMessage(message));
-    }
-    /**
-     * Returns a list of messages stored in the store.
-     */
-    async getMessages() {
-        const query = `SELECT data, type FROM "${this.schemaName}"."${this.tableName}" WHERE session_id = :session_id ORDER BY id;`;
-        const values = {
-            session_id: this.sessionId,
-        };
-        const items = [];
-        const { rows } = await this.engine.pool.raw(query, values);
-        if (rows.length === 0) {
-            return [];
-        }
-        for (const row of rows) {
-            items.push({
-                data: row.data.data,
-                type: row.type,
-            });
-        }
-        return mapStoredMessagesToChatMessages(items);
-    }
-    /**
-     * Add a message object to the store.
-     * @param {BaseMessage} message Message to be added to the store
-     */
-    async addMessage(message) {
-        const query = `INSERT INTO "${this.schemaName}"."${this.tableName}"("session_id", "data", "type") VALUES (:session_id, :data, :type)`;
-        const values = {
-            session_id: this.sessionId,
-            data: JSON.stringify(message.toDict()),
-            type: message.getType(),
-        };
-        await this.engine.pool.raw(query, values);
-    }
-    /**
-     * Add a list of messages object to the store.
-     * @param {Array<BaseMessage>} messages List of messages to be added to the store
-     */
-    async addMessages(messages) {
-        for (const msg of messages) {
-            await this.addMessage(msg);
-        }
-    }
-    /**
-     * Remove all messages from the store.
-     */
-    async clear() {
-        const query = `DELETE FROM "${this.schemaName}"."${this.tableName}" WHERE session_id = :session_id;`;
-        const values = {
-            session_id: this.sessionId,
-        };
-        await this.engine.pool.raw(query, values);
-    }
-}
+		return new PostgresChatMessageHistory({
+			engine,
+			sessionId,
+			tableName,
+			schemaName
+		});
+	}
+	addUserMessage(message) {
+		return this.addMessage(new HumanMessage(message));
+	}
+	addAIChatMessage(message) {
+		return this.addMessage(new AIMessage(message));
+	}
+	/**
+	* Returns a list of messages stored in the store.
+	*/
+	async getMessages() {
+		const query = `SELECT data, type FROM "${this.schemaName}"."${this.tableName}" WHERE session_id = :session_id ORDER BY id;`;
+		const values = { session_id: this.sessionId };
+		const items = [];
+		const { rows } = await this.engine.pool.raw(query, values);
+		if (rows.length === 0) return [];
+		for (const row of rows) items.push({
+			data: row.data.data,
+			type: row.type
+		});
+		return mapStoredMessagesToChatMessages(items);
+	}
+	/**
+	* Add a message object to the store.
+	* @param {BaseMessage} message Message to be added to the store
+	*/
+	async addMessage(message) {
+		const query = `INSERT INTO "${this.schemaName}"."${this.tableName}"("session_id", "data", "type") VALUES (:session_id, :data, :type)`;
+		const values = {
+			session_id: this.sessionId,
+			data: JSON.stringify(message.toDict()),
+			type: message.getType()
+		};
+		await this.engine.pool.raw(query, values);
+	}
+	/**
+	* Add a list of messages object to the store.
+	* @param {Array<BaseMessage>} messages List of messages to be added to the store
+	*/
+	async addMessages(messages) {
+		for (const msg of messages) await this.addMessage(msg);
+	}
+	/**
+	* Remove all messages from the store.
+	*/
+	async clear() {
+		const query = `DELETE FROM "${this.schemaName}"."${this.tableName}" WHERE session_id = :session_id;`;
+		const values = { session_id: this.sessionId };
+		await this.engine.pool.raw(query, values);
+	}
+};
+
+//#endregion
+export { PostgresChatMessageHistory };
+//# sourceMappingURL=chat_message_history.js.map

package/dist/chat_message_history.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat_message_history.js","names":["engine: PostgresEngine","sessionId: string","tableName: string","schemaName: string","columnNames: string[]","message: string","values: { [key: string]: string }","items: StoredMessage[]","message: BaseMessage","messages: BaseMessage[]"],"sources":["../src/chat_message_history.ts"],"sourcesContent":["import { BaseChatMessageHistory } from \"@langchain/core/chat_history\";\nimport {\n  AIMessage,\n  BaseMessage,\n  HumanMessage,\n  StoredMessage,\n  mapStoredMessagesToChatMessages,\n} from \"@langchain/core/messages\";\nimport PostgresEngine from \"./engine.js\";\n\nexport interface PostgresChatMessageHistoryInput {\n  engine: PostgresEngine;\n  sessionId: string;\n  tableName: string;\n  schemaName: string;\n}\n\nexport class PostgresChatMessageHistory extends BaseChatMessageHistory {\n  lc_namespace: string[] = [\n    \"langchain\",\n    \"stores\",\n    \"message\",\n    \"google-cloud-sql-pg\",\n  ];\n\n  engine: PostgresEngine;\n\n  sessionId: string;\n\n  tableName: string;\n\n  schemaName: string;\n\n  constructor({\n    engine,\n    sessionId,\n    tableName,\n    schemaName = \"public\",\n  }: PostgresChatMessageHistoryInput) {\n    super();\n    this.engine = engine;\n    this.sessionId = sessionId;\n    this.tableName = tableName;\n    this.schemaName = schemaName;\n  }\n\n  /**\n   * Create a new PostgresChatMessageHistory instance.\n   *\n   * @param {PostgresEngine} engine Postgres engine instance to use.\n   * @param {string} sessionId Retrieve the table content with this session ID.\n   * @param {string} tableName Table name that stores that chat message history. Parameter is not escaped. Do not use with end user input.\n   * @param {string} schemaName Schema name for the chat message history table. Default: \"public\". Parameter is not escaped. Do not use with end user input.\n   * @returns PostgresChatMessageHistory instance.\n   */\n  static async initialize(\n    engine: PostgresEngine,\n    sessionId: string,\n    tableName: string,\n    schemaName: string = \"public\"\n  ) {\n    const query = `SELECT column_name, data_type FROM information_schema.columns WHERE table_name = '${tableName}' AND table_schema = '${schemaName}'`;\n    const { rows } = await engine.pool.raw(query);\n    const columnNames: string[] = [];\n\n    for (const index in rows) {\n      if (Object.prototype.hasOwnProperty.call(rows, index)) {\n        columnNames.push(rows[index].column_name);\n      }\n    }\n\n    const requiredColumns = [\"id\", \"session_id\", \"data\", \"type\"];\n\n    if (!requiredColumns.every((x) => columnNames.includes(x))) {\n      throw new Error(\n        `Table '${schemaName}'.'${tableName}' has incorrect schema.\n        Got column names ${columnNames} but required column names ${requiredColumns}.\n        Please create table with following schema: CREATE TABLE '${schemaName}'.'${tableName}' (\n            id SERIAL AUTO_INCREMENT PRIMARY KEY,\n            session_id TEXT NOT NULL,\n            data JSONB NOT NULL,\n            type TEXT NOT NULL\n        );\n      `\n      );\n    }\n\n    return new PostgresChatMessageHistory({\n      engine,\n      sessionId,\n      tableName,\n      schemaName,\n    });\n  }\n\n  addUserMessage(message: string): Promise<void> {\n    return this.addMessage(new HumanMessage(message));\n  }\n\n  addAIChatMessage(message: string): Promise<void> {\n    return this.addMessage(new AIMessage(message));\n  }\n\n  /**\n   * Returns a list of messages stored in the store.\n   */\n  async getMessages(): Promise<BaseMessage[]> {\n    const query = `SELECT data, type FROM \"${this.schemaName}\".\"${this.tableName}\" WHERE session_id = :session_id ORDER BY id;`;\n    const values: { [key: string]: string } = {\n      session_id: this.sessionId,\n    };\n    const items: StoredMessage[] = [];\n    const { rows } = await this.engine.pool.raw(query, values);\n\n    if (rows.length === 0) {\n      return [];\n    }\n\n    for (const row of rows) {\n      items.push({\n        data: row.data.data,\n        type: row.type,\n      });\n    }\n\n    return mapStoredMessagesToChatMessages(items);\n  }\n\n  /**\n   * Add a message object to the store.\n   * @param {BaseMessage} message Message to be added to the store\n   */\n  async addMessage(message: BaseMessage): Promise<void> {\n    const query = `INSERT INTO \"${this.schemaName}\".\"${this.tableName}\"(\"session_id\", \"data\", \"type\") VALUES (:session_id, :data, :type)`;\n    const values: { [key: string]: string } = {\n      session_id: this.sessionId,\n      data: JSON.stringify(message.toDict()),\n      type: message.getType(),\n    };\n\n    await this.engine.pool.raw(query, values);\n  }\n\n  /**\n   * Add a list of messages object to the store.\n   * @param {Array<BaseMessage>} messages List of messages to be added to the store\n   */\n  async addMessages(messages: BaseMessage[]): Promise<void> {\n    for (const msg of messages) {\n      await this.addMessage(msg);\n    }\n  }\n\n  /**\n   * Remove all messages from the store.\n   */\n  async clear(): Promise<void> {\n    const query = `DELETE FROM \"${this.schemaName}\".\"${this.tableName}\" WHERE session_id = :session_id;`;\n    const values: { [key: string]: string } = {\n      session_id: this.sessionId,\n    };\n\n    await this.engine.pool.raw(query, values);\n  }\n}\n"],"mappings":";;;;AAiBA,IAAa,6BAAb,MAAa,mCAAmC,uBAAuB;CACrE,eAAyB;EACvB;EACA;EACA;EACA;CACD;CAED;CAEA;CAEA;CAEA;CAEA,YAAY,EACV,QACA,WACA,WACA,aAAa,UACmB,EAAE;EAClC,OAAO;EACP,KAAK,SAAS;EACd,KAAK,YAAY;EACjB,KAAK,YAAY;EACjB,KAAK,aAAa;CACnB;;;;;;;;;;CAWD,aAAa,WACXA,QACAC,WACAC,WACAC,aAAqB,UACrB;EACA,MAAM,QAAQ,CAAC,kFAAkF,EAAE,UAAU,sBAAsB,EAAE,WAAW,CAAC,CAAC;EAClJ,MAAM,EAAE,MAAM,GAAG,MAAM,OAAO,KAAK,IAAI,MAAM;EAC7C,MAAMC,cAAwB,CAAE;AAEhC,OAAK,MAAM,SAAS,KAClB,KAAI,OAAO,UAAU,eAAe,KAAK,MAAM,MAAM,EACnD,YAAY,KAAK,KAAK,OAAO,YAAY;EAI7C,MAAM,kBAAkB;GAAC;GAAM;GAAc;GAAQ;EAAO;AAE5D,MAAI,CAAC,gBAAgB,MAAM,CAAC,MAAM,YAAY,SAAS,EAAE,CAAC,CACxD,OAAM,IAAI,MACR,CAAC,OAAO,EAAE,WAAW,GAAG,EAAE,UAAU;yBACnB,EAAE,YAAY,2BAA2B,EAAE,gBAAgB;iEACnB,EAAE,WAAW,GAAG,EAAE,UAAU;;;;;;MAMvF,CAAC;AAIH,SAAO,IAAI,2BAA2B;GACpC;GACA;GACA;GACA;EACD;CACF;CAED,eAAeC,SAAgC;AAC7C,SAAO,KAAK,WAAW,IAAI,aAAa,SAAS;CAClD;CAED,iBAAiBA,SAAgC;AAC/C,SAAO,KAAK,WAAW,IAAI,UAAU,SAAS;CAC/C;;;;CAKD,MAAM,cAAsC;EAC1C,MAAM,QAAQ,CAAC,wBAAwB,EAAE,KAAK,WAAW,GAAG,EAAE,KAAK,UAAU,6CAA6C,CAAC;EAC3H,MAAMC,SAAoC,EACxC,YAAY,KAAK,UAClB;EACD,MAAMC,QAAyB,CAAE;EACjC,MAAM,EAAE,MAAM,GAAG,MAAM,KAAK,OAAO,KAAK,IAAI,OAAO,OAAO;AAE1D,MAAI,KAAK,WAAW,EAClB,QAAO,CAAE;AAGX,OAAK,MAAM,OAAO,MAChB,MAAM,KAAK;GACT,MAAM,IAAI,KAAK;GACf,MAAM,IAAI;EACX,EAAC;AAGJ,SAAO,gCAAgC,MAAM;CAC9C;;;;;CAMD,MAAM,WAAWC,SAAqC;EACpD,MAAM,QAAQ,CAAC,aAAa,EAAE,KAAK,WAAW,GAAG,EAAE,KAAK,UAAU,kEAAkE,CAAC;EACrI,MAAMF,SAAoC;GACxC,YAAY,KAAK;GACjB,MAAM,KAAK,UAAU,QAAQ,QAAQ,CAAC;GACtC,MAAM,QAAQ,SAAS;EACxB;EAED,MAAM,KAAK,OAAO,KAAK,IAAI,OAAO,OAAO;CAC1C;;;;;CAMD,MAAM,YAAYG,UAAwC;AACxD,OAAK,MAAM,OAAO,UAChB,MAAM,KAAK,WAAW,IAAI;CAE7B;;;;CAKD,MAAM,QAAuB;EAC3B,MAAM,QAAQ,CAAC,aAAa,EAAE,KAAK,WAAW,GAAG,EAAE,KAAK,UAAU,iCAAiC,CAAC;EACpG,MAAMH,SAAoC,EACxC,YAAY,KAAK,UAClB;EAED,MAAM,KAAK,OAAO,KAAK,IAAI,OAAO,OAAO;CAC1C;AACF"}