farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-29 11:01:18 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update MongoDBMemory.ts

Browse Source
This commit is contained in:
Anuj Verma
2024-01-28 21:33:09 +05:30
committed by GitHub
parent e8deeb25cf
commit e154461f1d
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/memory/MongoDBMemory/MongoDBMemory.ts
+11 -11
View File
@@ -1,19 +1,19 @@
import { MongoClient, Collection, Document } from 'mongodb'; import { MongoClient, Collection, Document } from 'mongodb'
import { MongoDBChatMessageHistory } from 'langchain/stores/message/mongodb'; import { MongoDBChatMessageHistory } from 'langchain/stores/message/mongodb'
import { BufferMemory, BufferMemoryInput } from 'langchain/memory'; import { BufferMemory, BufferMemoryInput } from 'langchain/memory'
import { mapStoredMessageToChatMessage, AIMessage, HumanMessage, BaseMessage } from 'langchain/schema'; import { mapStoredMessageToChatMessage, AIMessage, HumanMessage, BaseMessage } from 'langchain/schema'
import { convertBaseMessagetoIMessage, getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils'; import { convertBaseMessagetoIMessage, getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils'
import { FlowiseMemory, ICommonObject, IMessage, INode, INodeData, INodeParams, MemoryMethods, MessageType } from '../../../src/Interface'; import { FlowiseMemory, ICommonObject, IMessage, INode, INodeData, INodeParams, MemoryMethods, MessageType } from '../../../src/Interface'
let mongoClientSingleton = null; let mongoClientSingleton = null
const getMongoClient = async (mongoDBConnectUrl) => { const getMongoClient = async (mongoDBConnectUrl) => {
if (!mongoClientSingleton) { if (!mongoClientSingleton) {
mongoClientSingleton = new MongoClient(mongoDBConnectUrl, { useNewUrlParser: true, useUnifiedTopology: true }); mongoClientSingleton = new MongoClient(mongoDBConnectUrl, { useNewUrlParser: true, useUnifiedTopology: true })
await mongoClientSingleton.connect(); await mongoClientSingleton.connect()
} }
return mongoClientSingleton; return mongoClientSingleton
}; }
class MongoDB_Memory implements INode { class MongoDB_Memory implements INode {
label: string label: string