From aca5287d9146b1fc32b56b366b2c0de8f9021b9a Mon Sep 17 00:00:00 2001
From: Sam Brenner <sam.brenner@datadoghq.com>
Date: Wed, 22 Oct 2025 09:34:07 -0400
Subject: [PATCH] fix other langchain test

---
 .../datadog-plugin-langchain/test/index.spec.js    | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/packages/datadog-plugin-langchain/test/index.spec.js b/packages/datadog-plugin-langchain/test/index.spec.js
index af30a270b91..a8a95188acb 100644
--- a/packages/datadog-plugin-langchain/test/index.spec.js
+++ b/packages/datadog-plugin-langchain/test/index.spec.js
@@ -10,6 +10,8 @@ const { withVersions } = require('../../dd-trace/test/setup/mocha')
 
 const isDdTrace = iastFilter.isDdTrace
 
+const semifies = require('semifies')
+
 describe('Plugin', () => {
   let langchainOpenai
   let langchainAnthropic
@@ -114,9 +116,15 @@ describe('Plugin', () => {
         langchainTools = require(`../../../versions/@langchain/core@${version}`)
           .get('@langchain/core/tools')
 
-        MemoryVectorStore = require(`../../../versions/@langchain/core@${version}`)
-          .get('langchain/vectorstores/memory')
-          .MemoryVectorStore
+        if (semifies(realVersion, '>=1.0')) {
+          MemoryVectorStore = require('../../../versions/@langchain/classic@>=1.0')
+            .get('@langchain/classic/vectorstores/memory')
+            .MemoryVectorStore
+        } else {
+          MemoryVectorStore = require(`../../../versions/langchain@${version}`)
+            .get('langchain/vectorstores/memory')
+            .MemoryVectorStore
+        }
       })
 
       describe('llm', () => {