feat: add driverInfo to mongodb component (#2779)
* feat: add driverInfo to mongodb component NODE-6240 * chore: add a getVersion utility function
This commit is contained in:
parent
363d1bfc44
commit
9e88c45051
|
|
@ -4,7 +4,7 @@ import { MongoDBAtlasVectorSearch } from '@langchain/mongodb'
|
||||||
import { Embeddings } from '@langchain/core/embeddings'
|
import { Embeddings } from '@langchain/core/embeddings'
|
||||||
import { Document } from '@langchain/core/documents'
|
import { Document } from '@langchain/core/documents'
|
||||||
import { ICommonObject, INode, INodeData, INodeOutputsValue, INodeParams, IndexingResult } from '../../../src/Interface'
|
import { ICommonObject, INode, INodeData, INodeOutputsValue, INodeParams, IndexingResult } from '../../../src/Interface'
|
||||||
import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils'
|
import { getBaseClasses, getCredentialData, getCredentialParam, getVersion } from '../../../src/utils'
|
||||||
import { addMMRInputParams, resolveVectorStoreOrRetriever } from '../VectorStoreUtils'
|
import { addMMRInputParams, resolveVectorStoreOrRetriever } from '../VectorStoreUtils'
|
||||||
|
|
||||||
class MongoDBAtlas_VectorStores implements INode {
|
class MongoDBAtlas_VectorStores implements INode {
|
||||||
|
|
@ -191,15 +191,17 @@ let mongoClientSingleton: MongoClient
|
||||||
let mongoUrl: string
|
let mongoUrl: string
|
||||||
|
|
||||||
const getMongoClient = async (newMongoUrl: string) => {
|
const getMongoClient = async (newMongoUrl: string) => {
|
||||||
|
const driverInfo = { name: 'Flowise', version: (await getVersion()).version }
|
||||||
|
|
||||||
if (!mongoClientSingleton) {
|
if (!mongoClientSingleton) {
|
||||||
// if client does not exist
|
// if client does not exist
|
||||||
mongoClientSingleton = new MongoClient(newMongoUrl)
|
mongoClientSingleton = new MongoClient(newMongoUrl, { driverInfo })
|
||||||
mongoUrl = newMongoUrl
|
mongoUrl = newMongoUrl
|
||||||
return mongoClientSingleton
|
return mongoClientSingleton
|
||||||
} else if (mongoClientSingleton && newMongoUrl !== mongoUrl) {
|
} else if (mongoClientSingleton && newMongoUrl !== mongoUrl) {
|
||||||
// if client exists but url changed
|
// if client exists but url changed
|
||||||
mongoClientSingleton.close()
|
mongoClientSingleton.close()
|
||||||
mongoClientSingleton = new MongoClient(newMongoUrl)
|
mongoClientSingleton = new MongoClient(newMongoUrl, { driverInfo })
|
||||||
mongoUrl = newMongoUrl
|
mongoUrl = newMongoUrl
|
||||||
return mongoClientSingleton
|
return mongoClientSingleton
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -772,3 +772,28 @@ export const prepareSandboxVars = (variables: IVariable[]) => {
|
||||||
}
|
}
|
||||||
return vars
|
return vars
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let version: string
|
||||||
|
export const getVersion: () => Promise<{ version: string }> = async () => {
|
||||||
|
if (version != null) return { version }
|
||||||
|
|
||||||
|
const checkPaths = [
|
||||||
|
path.join(__dirname, '..', 'package.json'),
|
||||||
|
path.join(__dirname, '..', '..', 'package.json'),
|
||||||
|
path.join(__dirname, '..', '..', '..', 'package.json'),
|
||||||
|
path.join(__dirname, '..', '..', '..', '..', 'package.json'),
|
||||||
|
path.join(__dirname, '..', '..', '..', '..', '..', 'package.json')
|
||||||
|
]
|
||||||
|
for (const checkPath of checkPaths) {
|
||||||
|
try {
|
||||||
|
const content = await fs.promises.readFile(checkPath, 'utf8')
|
||||||
|
const parsedContent = JSON.parse(content)
|
||||||
|
version = parsedContent.version
|
||||||
|
return { version }
|
||||||
|
} catch {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
throw new Error('None of the package.json paths could be parsed')
|
||||||
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue