@langchain/neo4j 0.0.1

package/dist/stores/message/neo4j.cjs

@@ -0,0 +1,114 @@
+const require_runtime = require("../../_virtual/_rolldown/runtime.cjs");
+let neo4j_driver = require("neo4j-driver");
+neo4j_driver = require_runtime.__toESM(neo4j_driver);
+let uuid = require("uuid");
+let _langchain_core_chat_history = require("@langchain/core/chat_history");
+let _langchain_core_messages = require("@langchain/core/messages");
+//#region src/stores/message/neo4j.ts
+const defaultConfig = {
+	sessionNodeLabel: "ChatSession",
+	messageNodeLabel: "ChatMessage",
+	windowSize: 3
+};
+var Neo4jChatMessageHistory = class Neo4jChatMessageHistory extends _langchain_core_chat_history.BaseListChatMessageHistory {
+	lc_namespace = [
+		"langchain",
+		"stores",
+		"message",
+		"neo4j"
+	];
+	sessionId;
+	sessionNodeLabel;
+	messageNodeLabel;
+	windowSize;
+	driver;
+	constructor({ sessionId = (0, uuid.v4)(), sessionNodeLabel = defaultConfig.sessionNodeLabel, messageNodeLabel = defaultConfig.messageNodeLabel, url, username, password, windowSize = defaultConfig.windowSize }) {
+		super();
+		this.sessionId = sessionId;
+		this.sessionNodeLabel = sessionNodeLabel;
+		this.messageNodeLabel = messageNodeLabel;
+		this.windowSize = windowSize;
+		if (url && username && password) try {
+			this.driver = neo4j_driver.default.driver(url, neo4j_driver.auth.basic(username, password));
+		} catch (e) {
+			const error = e;
+			throw new Error(`Could not create a Neo4j driver instance. Please check the connection details.\nCause: ${error.message}`);
+		}
+		else throw new Error("Neo4j connection details not provided.");
+	}
+	static async initialize(props) {
+		const instance = new Neo4jChatMessageHistory(props);
+		try {
+			await instance.verifyConnectivity();
+		} catch (e) {
+			const error = e;
+			throw new Error(`Could not verify connection to the Neo4j database.\nCause: ${error.message}`);
+		}
+		return instance;
+	}
+	async verifyConnectivity() {
+		return await this.driver.getServerInfo();
+	}
+	async getMessages() {
+		const getMessagesCypherQuery = `
+      MERGE (chatSession:${this.sessionNodeLabel} {id: $sessionId})
+      WITH chatSession
+      MATCH (chatSession)-[:LAST_MESSAGE]->(lastMessage)
+      MATCH p=(lastMessage)<-[:NEXT*0..${this.windowSize * 2 - 1}]-()
+      WITH p, length(p) AS length
+      ORDER BY length DESC LIMIT 1
+      UNWIND reverse(nodes(p)) AS node
+      RETURN {data:{content: node.content}, type:node.type} AS result
+    `;
+		try {
+			const { records } = await this.driver.executeQuery(getMessagesCypherQuery, { sessionId: this.sessionId });
+			return (0, _langchain_core_messages.mapStoredMessagesToChatMessages)(records.map((record) => record.get("result")));
+		} catch (e) {
+			const error = e;
+			throw new Error(`Ohno! Couldn't get messages.\nCause: ${error.message}`);
+		}
+	}
+	async addMessage(message) {
+		const addMessageCypherQuery = `
+      MERGE (chatSession:${this.sessionNodeLabel} {id: $sessionId})
+      WITH chatSession
+      OPTIONAL MATCH (chatSession)-[lastMessageRel:LAST_MESSAGE]->(lastMessage)
+      CREATE (chatSession)-[:LAST_MESSAGE]->(newLastMessage:${this.messageNodeLabel})
+      SET newLastMessage += {type:$type, content:$content}
+      WITH newLastMessage, lastMessageRel, lastMessage
+      WHERE lastMessage IS NOT NULL
+      CREATE (lastMessage)-[:NEXT]->(newLastMessage)
+      DELETE lastMessageRel
+    `;
+		try {
+			await this.driver.executeQuery(addMessageCypherQuery, {
+				sessionId: this.sessionId,
+				type: message.getType(),
+				content: message.content
+			});
+		} catch (e) {
+			const error = e;
+			throw new Error(`Ohno! Couldn't add message.\nCause: ${error.message}`);
+		}
+	}
+	async clear() {
+		const clearMessagesCypherQuery = `
+      MATCH p=(chatSession:${this.sessionNodeLabel} {id: $sessionId})-[:LAST_MESSAGE]->(lastMessage)<-[:NEXT*0..]-()
+      UNWIND nodes(p) as node
+      DETACH DELETE node
+    `;
+		try {
+			await this.driver.executeQuery(clearMessagesCypherQuery, { sessionId: this.sessionId });
+		} catch (e) {
+			const error = e;
+			throw new Error(`Ohno! Couldn't clear chat history.\nCause: ${error.message}`);
+		}
+	}
+	async close() {
+		await this.driver.close();
+	}
+};
+//#endregion
+exports.Neo4jChatMessageHistory = Neo4jChatMessageHistory;
+
+//# sourceMappingURL=neo4j.cjs.map

package/dist/stores/message/neo4j.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"neo4j.cjs","names":["BaseListChatMessageHistory","neo4j","auth"],"sources":["../../../src/stores/message/neo4j.ts"],"sourcesContent":["import neo4j, { Driver, Record, auth, type ServerInfo } from \"neo4j-driver\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport { BaseListChatMessageHistory } from \"@langchain/core/chat_history\";\nimport {\n  BaseMessage,\n  mapStoredMessagesToChatMessages,\n} from \"@langchain/core/messages\";\n\nexport type Neo4jChatMessageHistoryConfigInput = {\n  sessionId?: string | number;\n  sessionNodeLabel?: string;\n  messageNodeLabel?: string;\n  url: string;\n  username: string;\n  password: string;\n  windowSize?: number;\n};\n\nconst defaultConfig = {\n  sessionNodeLabel: \"ChatSession\",\n  messageNodeLabel: \"ChatMessage\",\n  windowSize: 3,\n};\n\nexport class Neo4jChatMessageHistory extends BaseListChatMessageHistory {\n  lc_namespace: string[] = [\"langchain\", \"stores\", \"message\", \"neo4j\"];\n\n  sessionId: string | number;\n\n  sessionNodeLabel: string;\n\n  messageNodeLabel: string;\n\n  windowSize: number;\n\n  private driver: Driver;\n\n  constructor({\n    sessionId = uuidv4(),\n    sessionNodeLabel = defaultConfig.sessionNodeLabel,\n    messageNodeLabel = defaultConfig.messageNodeLabel,\n    url,\n    username,\n    password,\n    windowSize = defaultConfig.windowSize,\n  }: Neo4jChatMessageHistoryConfigInput) {\n    super();\n\n    this.sessionId = sessionId;\n    this.sessionNodeLabel = sessionNodeLabel;\n    this.messageNodeLabel = messageNodeLabel;\n    this.windowSize = windowSize;\n\n    if (url && username && password) {\n      try {\n        this.driver = neo4j.driver(url, auth.basic(username, password));\n      } catch (e) {\n        const error = e as Error;\n        throw new Error(\n          `Could not create a Neo4j driver instance. Please check the connection details.\\nCause: ${error.message}`\n        );\n      }\n    } else {\n      throw new Error(\"Neo4j connection details not provided.\");\n    }\n  }\n\n  static async initialize(\n    props: Neo4jChatMessageHistoryConfigInput\n  ): Promise<Neo4jChatMessageHistory> {\n    const instance = new Neo4jChatMessageHistory(props);\n\n    try {\n      await instance.verifyConnectivity();\n    } catch (e) {\n      const error = e as Error;\n      throw new Error(\n        `Could not verify connection to the Neo4j database.\\nCause: ${error.message}`\n      );\n    }\n\n    return instance;\n  }\n\n  async verifyConnectivity(): Promise<ServerInfo> {\n    const connectivity = await this.driver.getServerInfo();\n    return connectivity;\n  }\n\n  async getMessages(): Promise<BaseMessage[]> {\n    const getMessagesCypherQuery = `\n      MERGE (chatSession:${this.sessionNodeLabel} {id: $sessionId})\n      WITH chatSession\n      MATCH (chatSession)-[:LAST_MESSAGE]->(lastMessage)\n      MATCH p=(lastMessage)<-[:NEXT*0..${this.windowSize * 2 - 1}]-()\n      WITH p, length(p) AS length\n      ORDER BY length DESC LIMIT 1\n      UNWIND reverse(nodes(p)) AS node\n      RETURN {data:{content: node.content}, type:node.type} AS result\n    `;\n\n    try {\n      const { records } = await this.driver.executeQuery(\n        getMessagesCypherQuery,\n        {\n          sessionId: this.sessionId,\n        }\n      );\n      const results = records.map((record: Record) => record.get(\"result\"));\n\n      return mapStoredMessagesToChatMessages(results);\n    } catch (e) {\n      const error = e as Error;\n      throw new Error(`Ohno! Couldn't get messages.\\nCause: ${error.message}`);\n    }\n  }\n\n  async addMessage(message: BaseMessage): Promise<void> {\n    const addMessageCypherQuery = `\n      MERGE (chatSession:${this.sessionNodeLabel} {id: $sessionId})\n      WITH chatSession\n      OPTIONAL MATCH (chatSession)-[lastMessageRel:LAST_MESSAGE]->(lastMessage)\n      CREATE (chatSession)-[:LAST_MESSAGE]->(newLastMessage:${this.messageNodeLabel})\n      SET newLastMessage += {type:$type, content:$content}\n      WITH newLastMessage, lastMessageRel, lastMessage\n      WHERE lastMessage IS NOT NULL\n      CREATE (lastMessage)-[:NEXT]->(newLastMessage)\n      DELETE lastMessageRel\n    `;\n\n    try {\n      await this.driver.executeQuery(addMessageCypherQuery, {\n        sessionId: this.sessionId,\n        type: message.getType(),\n        content: message.content,\n      });\n    } catch (e) {\n      const error = e as Error;\n      throw new Error(`Ohno! Couldn't add message.\\nCause: ${error.message}`);\n    }\n  }\n\n  async clear() {\n    const clearMessagesCypherQuery = `\n      MATCH p=(chatSession:${this.sessionNodeLabel} {id: $sessionId})-[:LAST_MESSAGE]->(lastMessage)<-[:NEXT*0..]-()\n      UNWIND nodes(p) as node\n      DETACH DELETE node\n    `;\n\n    try {\n      await this.driver.executeQuery(clearMessagesCypherQuery, {\n        sessionId: this.sessionId,\n      });\n    } catch (e) {\n      const error = e as Error;\n      throw new Error(\n        `Ohno! Couldn't clear chat history.\\nCause: ${error.message}`\n      );\n    }\n  }\n\n  async close() {\n    await this.driver.close();\n  }\n}\n"],"mappings":";;;;;;;AAkBA,MAAM,gBAAgB;CACpB,kBAAkB;CAClB,kBAAkB;CAClB,YAAY;CACb;AAED,IAAa,0BAAb,MAAa,gCAAgCA,6BAAAA,2BAA2B;CACtE,eAAyB;EAAC;EAAa;EAAU;EAAW;EAAQ;CAEpE;CAEA;CAEA;CAEA;CAEA;CAEA,YAAY,EACV,aAAA,GAAA,KAAA,KAAoB,EACpB,mBAAmB,cAAc,kBACjC,mBAAmB,cAAc,kBACjC,KACA,UACA,UACA,aAAa,cAAc,cACU;AACrC,SAAO;AAEP,OAAK,YAAY;AACjB,OAAK,mBAAmB;AACxB,OAAK,mBAAmB;AACxB,OAAK,aAAa;AAElB,MAAI,OAAO,YAAY,SACrB,KAAI;AACF,QAAK,SAASC,aAAAA,QAAM,OAAO,KAAKC,aAAAA,KAAK,MAAM,UAAU,SAAS,CAAC;WACxD,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MACR,0FAA0F,MAAM,UACjG;;MAGH,OAAM,IAAI,MAAM,yCAAyC;;CAI7D,aAAa,WACX,OACkC;EAClC,MAAM,WAAW,IAAI,wBAAwB,MAAM;AAEnD,MAAI;AACF,SAAM,SAAS,oBAAoB;WAC5B,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MACR,8DAA8D,MAAM,UACrE;;AAGH,SAAO;;CAGT,MAAM,qBAA0C;AAE9C,SADqB,MAAM,KAAK,OAAO,eAAe;;CAIxD,MAAM,cAAsC;EAC1C,MAAM,yBAAyB;2BACR,KAAK,iBAAiB;;;yCAGR,KAAK,aAAa,IAAI,EAAE;;;;;;AAO7D,MAAI;GACF,MAAM,EAAE,YAAY,MAAM,KAAK,OAAO,aACpC,wBACA,EACE,WAAW,KAAK,WACjB,CACF;AAGD,WAAA,GAAA,yBAAA,iCAFgB,QAAQ,KAAK,WAAmB,OAAO,IAAI,SAAS,CAAC,CAEtB;WACxC,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MAAM,wCAAwC,MAAM,UAAU;;;CAI5E,MAAM,WAAW,SAAqC;EACpD,MAAM,wBAAwB;2BACP,KAAK,iBAAiB;;;8DAGa,KAAK,iBAAiB;;;;;;;AAQhF,MAAI;AACF,SAAM,KAAK,OAAO,aAAa,uBAAuB;IACpD,WAAW,KAAK;IAChB,MAAM,QAAQ,SAAS;IACvB,SAAS,QAAQ;IAClB,CAAC;WACK,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MAAM,uCAAuC,MAAM,UAAU;;;CAI3E,MAAM,QAAQ;EACZ,MAAM,2BAA2B;6BACR,KAAK,iBAAiB;;;;AAK/C,MAAI;AACF,SAAM,KAAK,OAAO,aAAa,0BAA0B,EACvD,WAAW,KAAK,WACjB,CAAC;WACK,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MACR,8CAA8C,MAAM,UACrD;;;CAIL,MAAM,QAAQ;AACZ,QAAM,KAAK,OAAO,OAAO"}

package/dist/stores/message/neo4j.d.cts

@@ -0,0 +1,40 @@
+import { ServerInfo } from "neo4j-driver";
+import { BaseListChatMessageHistory } from "@langchain/core/chat_history";
+import { BaseMessage } from "@langchain/core/messages";
+
+//#region src/stores/message/neo4j.d.ts
+type Neo4jChatMessageHistoryConfigInput = {
+  sessionId?: string | number;
+  sessionNodeLabel?: string;
+  messageNodeLabel?: string;
+  url: string;
+  username: string;
+  password: string;
+  windowSize?: number;
+};
+declare class Neo4jChatMessageHistory extends BaseListChatMessageHistory {
+  lc_namespace: string[];
+  sessionId: string | number;
+  sessionNodeLabel: string;
+  messageNodeLabel: string;
+  windowSize: number;
+  private driver;
+  constructor({
+    sessionId,
+    sessionNodeLabel,
+    messageNodeLabel,
+    url,
+    username,
+    password,
+    windowSize
+  }: Neo4jChatMessageHistoryConfigInput);
+  static initialize(props: Neo4jChatMessageHistoryConfigInput): Promise<Neo4jChatMessageHistory>;
+  verifyConnectivity(): Promise<ServerInfo>;
+  getMessages(): Promise<BaseMessage[]>;
+  addMessage(message: BaseMessage): Promise<void>;
+  clear(): Promise<void>;
+  close(): Promise<void>;
+}
+//#endregion
+export { Neo4jChatMessageHistory, Neo4jChatMessageHistoryConfigInput };
+//# sourceMappingURL=neo4j.d.cts.map

package/dist/stores/message/neo4j.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"neo4j.d.cts","names":[],"sources":["../../../src/stores/message/neo4j.ts"],"mappings":";;;;;KAQY,kCAAA;EACV,SAAA;EACA,gBAAA;EACA,gBAAA;EACA,GAAA;EACA,QAAA;EACA,QAAA;EACA,UAAA;AAAA;AAAA,cASW,uBAAA,SAAgC,0BAAA;EAC3C,YAAA;EAEA,SAAA;EAEA,gBAAA;EAEA,gBAAA;EAEA,UAAA;EAAA,QAEQ,MAAA;EAER,WAAA,CAAA;IACE,SAAA;IACA,gBAAA;IACA,gBAAA;IACA,GAAA;IACA,QAAA;IACA,QAAA;IACA;EAAA,GACC,kCAAA;EAAA,OAsBU,UAAA,CACX,KAAA,EAAO,kCAAA,GACN,OAAA,CAAQ,uBAAA;EAeL,kBAAA,CAAA,GAAsB,OAAA,CAAQ,UAAA;EAK9B,WAAA,CAAA,GAAe,OAAA,CAAQ,WAAA;EA4BvB,UAAA,CAAW,OAAA,EAAS,WAAA,GAAc,OAAA;EAyBlC,KAAA,CAAA,GAAK,OAAA;EAmBL,KAAA,CAAA,GAAK,OAAA;AAAA"}

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

@@ -0,0 +1,40 @@
+import { ServerInfo } from "neo4j-driver";
+import { BaseListChatMessageHistory } from "@langchain/core/chat_history";
+import { BaseMessage } from "@langchain/core/messages";
+
+//#region src/stores/message/neo4j.d.ts
+type Neo4jChatMessageHistoryConfigInput = {
+  sessionId?: string | number;
+  sessionNodeLabel?: string;
+  messageNodeLabel?: string;
+  url: string;
+  username: string;
+  password: string;
+  windowSize?: number;
+};
+declare class Neo4jChatMessageHistory extends BaseListChatMessageHistory {
+  lc_namespace: string[];
+  sessionId: string | number;
+  sessionNodeLabel: string;
+  messageNodeLabel: string;
+  windowSize: number;
+  private driver;
+  constructor({
+    sessionId,
+    sessionNodeLabel,
+    messageNodeLabel,
+    url,
+    username,
+    password,
+    windowSize
+  }: Neo4jChatMessageHistoryConfigInput);
+  static initialize(props: Neo4jChatMessageHistoryConfigInput): Promise<Neo4jChatMessageHistory>;
+  verifyConnectivity(): Promise<ServerInfo>;
+  getMessages(): Promise<BaseMessage[]>;
+  addMessage(message: BaseMessage): Promise<void>;
+  clear(): Promise<void>;
+  close(): Promise<void>;
+}
+//#endregion
+export { Neo4jChatMessageHistory, Neo4jChatMessageHistoryConfigInput };
+//# sourceMappingURL=neo4j.d.ts.map

package/dist/stores/message/neo4j.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"neo4j.d.ts","names":[],"sources":["../../../src/stores/message/neo4j.ts"],"mappings":";;;;;KAQY,kCAAA;EACV,SAAA;EACA,gBAAA;EACA,gBAAA;EACA,GAAA;EACA,QAAA;EACA,QAAA;EACA,UAAA;AAAA;AAAA,cASW,uBAAA,SAAgC,0BAAA;EAC3C,YAAA;EAEA,SAAA;EAEA,gBAAA;EAEA,gBAAA;EAEA,UAAA;EAAA,QAEQ,MAAA;EAER,WAAA,CAAA;IACE,SAAA;IACA,gBAAA;IACA,gBAAA;IACA,GAAA;IACA,QAAA;IACA,QAAA;IACA;EAAA,GACC,kCAAA;EAAA,OAsBU,UAAA,CACX,KAAA,EAAO,kCAAA,GACN,OAAA,CAAQ,uBAAA;EAeL,kBAAA,CAAA,GAAsB,OAAA,CAAQ,UAAA;EAK9B,WAAA,CAAA,GAAe,OAAA,CAAQ,WAAA;EA4BvB,UAAA,CAAW,OAAA,EAAS,WAAA,GAAc,OAAA;EAyBlC,KAAA,CAAA,GAAK,OAAA;EAmBL,KAAA,CAAA,GAAK,OAAA;AAAA"}

package/dist/stores/message/neo4j.js

@@ -0,0 +1,112 @@
+import neo4j, { auth } from "neo4j-driver";
+import { v4 } from "uuid";
+import { BaseListChatMessageHistory } from "@langchain/core/chat_history";
+import { mapStoredMessagesToChatMessages } from "@langchain/core/messages";
+//#region src/stores/message/neo4j.ts
+const defaultConfig = {
+	sessionNodeLabel: "ChatSession",
+	messageNodeLabel: "ChatMessage",
+	windowSize: 3
+};
+var Neo4jChatMessageHistory = class Neo4jChatMessageHistory extends BaseListChatMessageHistory {
+	lc_namespace = [
+		"langchain",
+		"stores",
+		"message",
+		"neo4j"
+	];
+	sessionId;
+	sessionNodeLabel;
+	messageNodeLabel;
+	windowSize;
+	driver;
+	constructor({ sessionId = v4(), sessionNodeLabel = defaultConfig.sessionNodeLabel, messageNodeLabel = defaultConfig.messageNodeLabel, url, username, password, windowSize = defaultConfig.windowSize }) {
+		super();
+		this.sessionId = sessionId;
+		this.sessionNodeLabel = sessionNodeLabel;
+		this.messageNodeLabel = messageNodeLabel;
+		this.windowSize = windowSize;
+		if (url && username && password) try {
+			this.driver = neo4j.driver(url, auth.basic(username, password));
+		} catch (e) {
+			const error = e;
+			throw new Error(`Could not create a Neo4j driver instance. Please check the connection details.\nCause: ${error.message}`);
+		}
+		else throw new Error("Neo4j connection details not provided.");
+	}
+	static async initialize(props) {
+		const instance = new Neo4jChatMessageHistory(props);
+		try {
+			await instance.verifyConnectivity();
+		} catch (e) {
+			const error = e;
+			throw new Error(`Could not verify connection to the Neo4j database.\nCause: ${error.message}`);
+		}
+		return instance;
+	}
+	async verifyConnectivity() {
+		return await this.driver.getServerInfo();
+	}
+	async getMessages() {
+		const getMessagesCypherQuery = `
+      MERGE (chatSession:${this.sessionNodeLabel} {id: $sessionId})
+      WITH chatSession
+      MATCH (chatSession)-[:LAST_MESSAGE]->(lastMessage)
+      MATCH p=(lastMessage)<-[:NEXT*0..${this.windowSize * 2 - 1}]-()
+      WITH p, length(p) AS length
+      ORDER BY length DESC LIMIT 1
+      UNWIND reverse(nodes(p)) AS node
+      RETURN {data:{content: node.content}, type:node.type} AS result
+    `;
+		try {
+			const { records } = await this.driver.executeQuery(getMessagesCypherQuery, { sessionId: this.sessionId });
+			return mapStoredMessagesToChatMessages(records.map((record) => record.get("result")));
+		} catch (e) {
+			const error = e;
+			throw new Error(`Ohno! Couldn't get messages.\nCause: ${error.message}`);
+		}
+	}
+	async addMessage(message) {
+		const addMessageCypherQuery = `
+      MERGE (chatSession:${this.sessionNodeLabel} {id: $sessionId})
+      WITH chatSession
+      OPTIONAL MATCH (chatSession)-[lastMessageRel:LAST_MESSAGE]->(lastMessage)
+      CREATE (chatSession)-[:LAST_MESSAGE]->(newLastMessage:${this.messageNodeLabel})
+      SET newLastMessage += {type:$type, content:$content}
+      WITH newLastMessage, lastMessageRel, lastMessage
+      WHERE lastMessage IS NOT NULL
+      CREATE (lastMessage)-[:NEXT]->(newLastMessage)
+      DELETE lastMessageRel
+    `;
+		try {
+			await this.driver.executeQuery(addMessageCypherQuery, {
+				sessionId: this.sessionId,
+				type: message.getType(),
+				content: message.content
+			});
+		} catch (e) {
+			const error = e;
+			throw new Error(`Ohno! Couldn't add message.\nCause: ${error.message}`);
+		}
+	}
+	async clear() {
+		const clearMessagesCypherQuery = `
+      MATCH p=(chatSession:${this.sessionNodeLabel} {id: $sessionId})-[:LAST_MESSAGE]->(lastMessage)<-[:NEXT*0..]-()
+      UNWIND nodes(p) as node
+      DETACH DELETE node
+    `;
+		try {
+			await this.driver.executeQuery(clearMessagesCypherQuery, { sessionId: this.sessionId });
+		} catch (e) {
+			const error = e;
+			throw new Error(`Ohno! Couldn't clear chat history.\nCause: ${error.message}`);
+		}
+	}
+	async close() {
+		await this.driver.close();
+	}
+};
+//#endregion
+export { Neo4jChatMessageHistory };
+
+//# sourceMappingURL=neo4j.js.map

package/dist/stores/message/neo4j.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"neo4j.js","names":["uuidv4"],"sources":["../../../src/stores/message/neo4j.ts"],"sourcesContent":["import neo4j, { Driver, Record, auth, type ServerInfo } from \"neo4j-driver\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport { BaseListChatMessageHistory } from \"@langchain/core/chat_history\";\nimport {\n  BaseMessage,\n  mapStoredMessagesToChatMessages,\n} from \"@langchain/core/messages\";\n\nexport type Neo4jChatMessageHistoryConfigInput = {\n  sessionId?: string | number;\n  sessionNodeLabel?: string;\n  messageNodeLabel?: string;\n  url: string;\n  username: string;\n  password: string;\n  windowSize?: number;\n};\n\nconst defaultConfig = {\n  sessionNodeLabel: \"ChatSession\",\n  messageNodeLabel: \"ChatMessage\",\n  windowSize: 3,\n};\n\nexport class Neo4jChatMessageHistory extends BaseListChatMessageHistory {\n  lc_namespace: string[] = [\"langchain\", \"stores\", \"message\", \"neo4j\"];\n\n  sessionId: string | number;\n\n  sessionNodeLabel: string;\n\n  messageNodeLabel: string;\n\n  windowSize: number;\n\n  private driver: Driver;\n\n  constructor({\n    sessionId = uuidv4(),\n    sessionNodeLabel = defaultConfig.sessionNodeLabel,\n    messageNodeLabel = defaultConfig.messageNodeLabel,\n    url,\n    username,\n    password,\n    windowSize = defaultConfig.windowSize,\n  }: Neo4jChatMessageHistoryConfigInput) {\n    super();\n\n    this.sessionId = sessionId;\n    this.sessionNodeLabel = sessionNodeLabel;\n    this.messageNodeLabel = messageNodeLabel;\n    this.windowSize = windowSize;\n\n    if (url && username && password) {\n      try {\n        this.driver = neo4j.driver(url, auth.basic(username, password));\n      } catch (e) {\n        const error = e as Error;\n        throw new Error(\n          `Could not create a Neo4j driver instance. Please check the connection details.\\nCause: ${error.message}`\n        );\n      }\n    } else {\n      throw new Error(\"Neo4j connection details not provided.\");\n    }\n  }\n\n  static async initialize(\n    props: Neo4jChatMessageHistoryConfigInput\n  ): Promise<Neo4jChatMessageHistory> {\n    const instance = new Neo4jChatMessageHistory(props);\n\n    try {\n      await instance.verifyConnectivity();\n    } catch (e) {\n      const error = e as Error;\n      throw new Error(\n        `Could not verify connection to the Neo4j database.\\nCause: ${error.message}`\n      );\n    }\n\n    return instance;\n  }\n\n  async verifyConnectivity(): Promise<ServerInfo> {\n    const connectivity = await this.driver.getServerInfo();\n    return connectivity;\n  }\n\n  async getMessages(): Promise<BaseMessage[]> {\n    const getMessagesCypherQuery = `\n      MERGE (chatSession:${this.sessionNodeLabel} {id: $sessionId})\n      WITH chatSession\n      MATCH (chatSession)-[:LAST_MESSAGE]->(lastMessage)\n      MATCH p=(lastMessage)<-[:NEXT*0..${this.windowSize * 2 - 1}]-()\n      WITH p, length(p) AS length\n      ORDER BY length DESC LIMIT 1\n      UNWIND reverse(nodes(p)) AS node\n      RETURN {data:{content: node.content}, type:node.type} AS result\n    `;\n\n    try {\n      const { records } = await this.driver.executeQuery(\n        getMessagesCypherQuery,\n        {\n          sessionId: this.sessionId,\n        }\n      );\n      const results = records.map((record: Record) => record.get(\"result\"));\n\n      return mapStoredMessagesToChatMessages(results);\n    } catch (e) {\n      const error = e as Error;\n      throw new Error(`Ohno! Couldn't get messages.\\nCause: ${error.message}`);\n    }\n  }\n\n  async addMessage(message: BaseMessage): Promise<void> {\n    const addMessageCypherQuery = `\n      MERGE (chatSession:${this.sessionNodeLabel} {id: $sessionId})\n      WITH chatSession\n      OPTIONAL MATCH (chatSession)-[lastMessageRel:LAST_MESSAGE]->(lastMessage)\n      CREATE (chatSession)-[:LAST_MESSAGE]->(newLastMessage:${this.messageNodeLabel})\n      SET newLastMessage += {type:$type, content:$content}\n      WITH newLastMessage, lastMessageRel, lastMessage\n      WHERE lastMessage IS NOT NULL\n      CREATE (lastMessage)-[:NEXT]->(newLastMessage)\n      DELETE lastMessageRel\n    `;\n\n    try {\n      await this.driver.executeQuery(addMessageCypherQuery, {\n        sessionId: this.sessionId,\n        type: message.getType(),\n        content: message.content,\n      });\n    } catch (e) {\n      const error = e as Error;\n      throw new Error(`Ohno! Couldn't add message.\\nCause: ${error.message}`);\n    }\n  }\n\n  async clear() {\n    const clearMessagesCypherQuery = `\n      MATCH p=(chatSession:${this.sessionNodeLabel} {id: $sessionId})-[:LAST_MESSAGE]->(lastMessage)<-[:NEXT*0..]-()\n      UNWIND nodes(p) as node\n      DETACH DELETE node\n    `;\n\n    try {\n      await this.driver.executeQuery(clearMessagesCypherQuery, {\n        sessionId: this.sessionId,\n      });\n    } catch (e) {\n      const error = e as Error;\n      throw new Error(\n        `Ohno! Couldn't clear chat history.\\nCause: ${error.message}`\n      );\n    }\n  }\n\n  async close() {\n    await this.driver.close();\n  }\n}\n"],"mappings":";;;;;AAkBA,MAAM,gBAAgB;CACpB,kBAAkB;CAClB,kBAAkB;CAClB,YAAY;CACb;AAED,IAAa,0BAAb,MAAa,gCAAgC,2BAA2B;CACtE,eAAyB;EAAC;EAAa;EAAU;EAAW;EAAQ;CAEpE;CAEA;CAEA;CAEA;CAEA;CAEA,YAAY,EACV,YAAYA,IAAQ,EACpB,mBAAmB,cAAc,kBACjC,mBAAmB,cAAc,kBACjC,KACA,UACA,UACA,aAAa,cAAc,cACU;AACrC,SAAO;AAEP,OAAK,YAAY;AACjB,OAAK,mBAAmB;AACxB,OAAK,mBAAmB;AACxB,OAAK,aAAa;AAElB,MAAI,OAAO,YAAY,SACrB,KAAI;AACF,QAAK,SAAS,MAAM,OAAO,KAAK,KAAK,MAAM,UAAU,SAAS,CAAC;WACxD,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MACR,0FAA0F,MAAM,UACjG;;MAGH,OAAM,IAAI,MAAM,yCAAyC;;CAI7D,aAAa,WACX,OACkC;EAClC,MAAM,WAAW,IAAI,wBAAwB,MAAM;AAEnD,MAAI;AACF,SAAM,SAAS,oBAAoB;WAC5B,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MACR,8DAA8D,MAAM,UACrE;;AAGH,SAAO;;CAGT,MAAM,qBAA0C;AAE9C,SADqB,MAAM,KAAK,OAAO,eAAe;;CAIxD,MAAM,cAAsC;EAC1C,MAAM,yBAAyB;2BACR,KAAK,iBAAiB;;;yCAGR,KAAK,aAAa,IAAI,EAAE;;;;;;AAO7D,MAAI;GACF,MAAM,EAAE,YAAY,MAAM,KAAK,OAAO,aACpC,wBACA,EACE,WAAW,KAAK,WACjB,CACF;AAGD,UAAO,gCAFS,QAAQ,KAAK,WAAmB,OAAO,IAAI,SAAS,CAAC,CAEtB;WACxC,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MAAM,wCAAwC,MAAM,UAAU;;;CAI5E,MAAM,WAAW,SAAqC;EACpD,MAAM,wBAAwB;2BACP,KAAK,iBAAiB;;;8DAGa,KAAK,iBAAiB;;;;;;;AAQhF,MAAI;AACF,SAAM,KAAK,OAAO,aAAa,uBAAuB;IACpD,WAAW,KAAK;IAChB,MAAM,QAAQ,SAAS;IACvB,SAAS,QAAQ;IAClB,CAAC;WACK,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MAAM,uCAAuC,MAAM,UAAU;;;CAI3E,MAAM,QAAQ;EACZ,MAAM,2BAA2B;6BACR,KAAK,iBAAiB;;;;AAK/C,MAAI;AACF,SAAM,KAAK,OAAO,aAAa,0BAA0B,EACvD,WAAW,KAAK,WACjB,CAAC;WACK,GAAG;GACV,MAAM,QAAQ;AACd,SAAM,IAAI,MACR,8CAA8C,MAAM,UACrD;;;CAIL,MAAM,QAAQ;AACZ,QAAM,KAAK,OAAO,OAAO"}

package/dist/vectorstores/index.cjs

@@ -0,0 +1 @@
+require("./neo4j_vector.cjs");

package/dist/vectorstores/index.d.ts

@@ -0,0 +1 @@
+import { DistanceStrategy, IndexType, Metadata, Neo4jVectorStore, Neo4jVectorStoreArgs, SearchType, constructMetadataFilter, isVersionLessThan, removeLuceneChars } from "./neo4j_vector.js";

package/dist/vectorstores/index.js

@@ -0,0 +1,2 @@
+import "./neo4j_vector.js";
+export {};