fix langchainv2 test

Arize-ai · Oct 24, 2024 · a68d976 · a68d976
1 parent 3df6725
commit a68d976
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 5 deletions.
diff --git a/js/package.json b/js/package.json
@@ -43,8 +43,13 @@
   ],
   "pnpm": {
     "overrides": {
+      "@langchain/core@^0.2>langchain": "^0.2.17",
+      "@langchain/core@^0.1>langchain": "^0.1.37",
+      "langchain@^0.2>@langchain/core": "^0.2.36",
       "@langchain/openai@^0.2>@langchain/core": "^0.2.36",
-      "@langchain/openai@^0.1>@langchain/core": "^0.1.63"
+      "@langchain/openai@^0.1>@langchain/core": "^0.1.63",
+      "@langchain/[email protected]>langchain": "^0.1.37",
+      "@langchain/[email protected]>langchain": "^0.2.17"
     }
   }
 }
diff --git a/js/packages/openinference-instrumentation-langchain/test/langchainV2.test.ts b/js/packages/openinference-instrumentation-langchain/test/langchainV2.test.ts
@@ -325,7 +325,13 @@ describe("LangChainInstrumentation", () => {
       `${LLM_OUTPUT_MESSAGES}.0.${MESSAGE_ROLE}`
     ];
 
-    expect(span.attributes).toStrictEqual(expectedStreamingAttributes);
+    const actualAttributes = { ...span.attributes };
+    const output = JSON.parse(String(actualAttributes[OUTPUT_VALUE]));
+    delete output.generations[0][0].message.kwargs.id;
+    const newOutputValue = JSON.stringify(output);
+    actualAttributes[OUTPUT_VALUE] = newOutputValue;
+
+    expect(actualAttributes).toStrictEqual(expectedStreamingAttributes);
   });
 
   it("should add documents to retriever spans", async () => {

diff --git a/js/pnpm-lock.yaml b/js/pnpm-lock.yaml