File tree Expand file tree Collapse file tree 1 file changed +11
-3
lines changed
packages/datadog-plugin-langchain/test Expand file tree Collapse file tree 1 file changed +11
-3
lines changed Original file line number Diff line number Diff line change @@ -10,6 +10,8 @@ const { withVersions } = require('../../dd-trace/test/setup/mocha')
1010
1111const isDdTrace = iastFilter . isDdTrace
1212
13+ const semifies = require ( 'semifies' )
14+
1315describe ( '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' , ( ) => {
You can’t perform that action at this time.
0 commit comments