diff --git a/libs/providers/langchain-mongodb/package.json b/libs/providers/langchain-mongodb/package.json index 0d7da99f527c..249c407342a2 100644 --- a/libs/providers/langchain-mongodb/package.json +++ b/libs/providers/langchain-mongodb/package.json @@ -21,7 +21,7 @@ "lint:fix": "pnpm lint:eslint --fix && pnpm lint:dpdm", "clean": "rm -rf .turbo dist/", "test:watch": "vitest", - "test:int": "vitest run --mode int --testTimeout 100000 --maxWorkers=50%", + "test:int": "vitest run --mode int --testTimeout 10000 --maxWorkers=50%", "format": "prettier --config .prettierrc --write \"src\"", "format:check": "prettier --config .prettierrc --check \"src\"" }, diff --git a/libs/providers/langchain-mongodb/src/tests/storage.int.test.ts b/libs/providers/langchain-mongodb/src/tests/storage.int.test.ts index 540c824c1d0b..be14a05311ae 100644 --- a/libs/providers/langchain-mongodb/src/tests/storage.int.test.ts +++ b/libs/providers/langchain-mongodb/src/tests/storage.int.test.ts @@ -1,4 +1,4 @@ -import { describe, test, expect, beforeAll, afterAll } from "vitest"; +import { describe, test, expect, beforeAll, afterAll, afterEach } from "vitest"; import { v4 as uuidv4 } from "uuid"; import { Collection, MongoClient, ServerApiVersion } from "mongodb"; import { MongoDBStore } from "../storage.js"; diff --git a/libs/providers/langchain-mongodb/src/tests/vectorstores.int.test.ts b/libs/providers/langchain-mongodb/src/tests/vectorstores.int.test.ts index 11819abc9a02..5cd706d9a737 100644 --- a/libs/providers/langchain-mongodb/src/tests/vectorstores.int.test.ts +++ b/libs/providers/langchain-mongodb/src/tests/vectorstores.int.test.ts @@ -1,4 +1,4 @@ -import { beforeAll, expect, vi, test, type MockInstance } from "vitest"; +import { afterAll, beforeAll, beforeEach, describe, expect, vi, test, type MockInstance } from "vitest"; import { Collection, MongoClient } from "mongodb"; import { setTimeout } from "timers/promises"; import { OpenAIEmbeddings, AzureOpenAIEmbeddings } from "@langchain/openai"; @@ -33,7 +33,7 @@ beforeAll(async () => { client = new MongoClient(uri(), { monitorCommands: true }); await client.connect(); - const namespace = "langchain.test"; + const namespace = "langchain_test_db.langchain_test"; const [dbName, collectionName] = namespace.split("."); collection = await client.db(dbName).createCollection(collectionName); @@ -108,13 +108,13 @@ function getEmbeddings() { } test("MongoDBStore sets client metadata", () => { - const spy = jest.spyOn(client, "appendMetadata"); + const spy = vi.spyOn(client, "appendMetadata"); // eslint-disable-next-line no-new new PatchedVectorStore(getEmbeddings(), { collection, }); expect(spy).toHaveBeenCalledWith({ name: "langchainjs_vector" }); - jest.clearAllMocks(); + vi.clearAllMocks(); }); test("MongoDBAtlasVectorSearch with external ids", async () => { @@ -199,7 +199,7 @@ test("MongoDBAtlasVectorSearch with Maximal Marginal Relevance", async () => { const standardRetriever = await vectorStore.asRetriever(); - const standardRetrieverOutput = await standardRetriever.getRelevantDocuments( + const standardRetrieverOutput = await standardRetriever._getRelevantDocuments( "foo" ); expect(output).toHaveLength(texts.length); @@ -218,7 +218,7 @@ test("MongoDBAtlasVectorSearch with Maximal Marginal Relevance", async () => { }, }); - const retrieverOutput = await retriever.getRelevantDocuments("foo"); + const retrieverOutput = await retriever._getRelevantDocuments("foo"); expect(output).toHaveLength(texts.length); const retrieverActual = retrieverOutput.map((doc) => doc.pageContent);