Skip to content

Commit 6c552cb

Browse files
authored
ci(langchain): update apm test to use classic langchain for in-memory vectorstore (#6721)
1 parent b259523 commit 6c552cb

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

packages/datadog-plugin-langchain/test/index.spec.js

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,8 @@ const { withVersions } = require('../../dd-trace/test/setup/mocha')
1010

1111
const isDdTrace = iastFilter.isDdTrace
1212

13+
const semifies = require('semifies')
14+
1315
describe('Plugin', () => {
1416
let langchainOpenai
1517
let langchainAnthropic
@@ -114,9 +116,15 @@ describe('Plugin', () => {
114116
langchainTools = require(`../../../versions/@langchain/core@${version}`)
115117
.get('@langchain/core/tools')
116118

117-
MemoryVectorStore = require(`../../../versions/@langchain/core@${version}`)
118-
.get('langchain/vectorstores/memory')
119-
.MemoryVectorStore
119+
if (semifies(realVersion, '>=1.0')) {
120+
MemoryVectorStore = require('../../../versions/@langchain/classic@>=1.0')
121+
.get('@langchain/classic/vectorstores/memory')
122+
.MemoryVectorStore
123+
} else {
124+
MemoryVectorStore = require(`../../../versions/langchain@${version}`)
125+
.get('langchain/vectorstores/memory')
126+
.MemoryVectorStore
127+
}
120128
})
121129

122130
describe('llm', () => {

0 commit comments

Comments
 (0)