120 lines
4.3 KiB
JavaScript
120 lines
4.3 KiB
JavaScript
"use strict";
|
|
/* eslint-disable @typescript-eslint/no-explicit-any */
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.ConvexChatMessageHistory = void 0;
|
|
// eslint-disable-next-line import/no-extraneous-dependencies
|
|
const server_1 = require("convex/server");
|
|
const chat_history_1 = require("@langchain/core/chat_history");
|
|
const messages_1 = require("@langchain/core/messages");
|
|
class ConvexChatMessageHistory extends chat_history_1.BaseListChatMessageHistory {
|
|
constructor(config) {
|
|
super();
|
|
Object.defineProperty(this, "lc_namespace", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: ["langchain", "stores", "message", "convex"]
|
|
});
|
|
Object.defineProperty(this, "ctx", {
|
|
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, "table", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: void 0
|
|
});
|
|
Object.defineProperty(this, "index", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: void 0
|
|
});
|
|
Object.defineProperty(this, "sessionIdField", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: void 0
|
|
});
|
|
Object.defineProperty(this, "messageTextFieldName", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: void 0
|
|
});
|
|
Object.defineProperty(this, "insert", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: void 0
|
|
});
|
|
Object.defineProperty(this, "lookup", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: void 0
|
|
});
|
|
Object.defineProperty(this, "deleteMany", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: void 0
|
|
});
|
|
this.ctx = config.ctx;
|
|
this.sessionId = config.sessionId;
|
|
this.table = config.table ?? "messages";
|
|
this.index = config.index ?? "bySessionId";
|
|
this.sessionIdField =
|
|
config.sessionIdField ?? "sessionId";
|
|
this.messageTextFieldName =
|
|
config.messageTextFieldName ?? "message";
|
|
this.insert =
|
|
config.insert ?? (0, server_1.makeFunctionReference)("langchain/db:insert");
|
|
this.lookup =
|
|
config.lookup ?? (0, server_1.makeFunctionReference)("langchain/db:lookup");
|
|
this.deleteMany =
|
|
config.deleteMany ??
|
|
(0, server_1.makeFunctionReference)("langchain/db:deleteMany");
|
|
}
|
|
async getMessages() {
|
|
const convexDocuments = await this.ctx.runQuery(this.lookup, {
|
|
table: this.table,
|
|
index: this.index,
|
|
keyField: this.sessionIdField,
|
|
key: this.sessionId,
|
|
});
|
|
return (0, messages_1.mapStoredMessagesToChatMessages)(convexDocuments.map((doc) => doc[this.messageTextFieldName]));
|
|
}
|
|
async addMessage(message) {
|
|
const messages = (0, messages_1.mapChatMessagesToStoredMessages)([message]);
|
|
// TODO: Remove chunking when Convex handles the concurrent requests correctly
|
|
const PAGE_SIZE = 16;
|
|
for (let i = 0; i < messages.length; i += PAGE_SIZE) {
|
|
await Promise.all(messages.slice(i, i + PAGE_SIZE).map((message) => this.ctx.runMutation(this.insert, {
|
|
table: this.table,
|
|
document: {
|
|
[this.sessionIdField]: this.sessionId,
|
|
[this.messageTextFieldName]: message,
|
|
},
|
|
})));
|
|
}
|
|
}
|
|
async clear() {
|
|
await this.ctx.runMutation(this.deleteMany, {
|
|
table: this.table,
|
|
index: this.index,
|
|
keyField: this.sessionIdField,
|
|
key: this.sessionId,
|
|
});
|
|
}
|
|
}
|
|
exports.ConvexChatMessageHistory = ConvexChatMessageHistory;
|