Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update MongoDBMemory.ts

This commit is contained in:
Anuj Verma 2024-01-28 21:33:09 +05:30 committed by GitHub
parent e8deeb25cf
commit e154461f1d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
packages/components/nodes/memory/MongoDBMemory/MongoDBMemory.ts
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