diff --git a/packages/modules/qdrant/src/qdrant-container.test.ts b/packages/modules/qdrant/src/qdrant-container.test.ts index 1af9220ae..4f6257653 100644 --- a/packages/modules/qdrant/src/qdrant-container.test.ts +++ b/packages/modules/qdrant/src/qdrant-container.test.ts @@ -8,7 +8,7 @@ describe("QdrantContainer", () => { // connectQdrantSimple { it("should connect to the client", async () => { - const container = await new QdrantContainer().start(); + const container = await new QdrantContainer("qdrant/qdrant:v1.13.4").start(); const client = new QdrantClient({ url: `http://${container.getRestHostAddress()}` }); @@ -22,7 +22,7 @@ describe("QdrantContainer", () => { it("should work with valid API keys", async () => { const apiKey = crypto.randomUUID(); - const container = await new QdrantContainer().withApiKey(apiKey).start(); + const container = await new QdrantContainer("qdrant/qdrant:v1.13.4").withApiKey(apiKey).start(); const client = new QdrantClient({ url: `http://${container.getRestHostAddress()}`, apiKey }); @@ -35,21 +35,23 @@ describe("QdrantContainer", () => { it("should fail for invalid API keys", async () => { const apiKey = crypto.randomUUID(); - const container = await new QdrantContainer().withApiKey(apiKey).start(); + const container = await new QdrantContainer("qdrant/qdrant:v1.13.4").withApiKey(apiKey).start(); const client = new QdrantClient({ url: `http://${container.getRestHostAddress()}`, apiKey: "INVALID_KEY_" + crypto.randomUUID(), }); - expect(client.getCollections()).rejects.toThrow("Forbidden"); + expect(client.getCollections()).rejects.toThrow("Unauthorized"); await container.stop(); }); // connectQdrantWithConfig { it("should work with config files - valid API key", async () => { - const container = await new QdrantContainer().withConfigFile(path.resolve(__dirname, "test_config.yaml")).start(); + const container = await new QdrantContainer("qdrant/qdrant:v1.13.4") + .withConfigFile(path.resolve(__dirname, "test_config.yaml")) + .start(); const client = new QdrantClient({ url: `http://${container.getRestHostAddress()}`, apiKey: "SOME_TEST_KEY" }); @@ -60,14 +62,16 @@ describe("QdrantContainer", () => { // } it("should work with config files - invalid API key", async () => { - const container = await new QdrantContainer().withConfigFile(path.resolve(__dirname, "test_config.yaml")).start(); + const container = await new QdrantContainer("qdrant/qdrant:v1.13.4") + .withConfigFile(path.resolve(__dirname, "test_config.yaml")) + .start(); const client = new QdrantClient({ url: `http://${container.getRestHostAddress()}`, apiKey: "INVALID_KEY_" + crypto.randomUUID(), }); - expect(client.getCollections()).rejects.toThrow("Forbidden"); + expect(client.getCollections()).rejects.toThrow("Unauthorized"); await container.stop(); }); diff --git a/packages/modules/qdrant/src/qdrant-container.ts b/packages/modules/qdrant/src/qdrant-container.ts index d4f32a293..df13a37ba 100644 --- a/packages/modules/qdrant/src/qdrant-container.ts +++ b/packages/modules/qdrant/src/qdrant-container.ts @@ -8,7 +8,7 @@ export class QdrantContainer extends GenericContainer { private apiKey: string | undefined; private configFilePath: string | undefined; - constructor(image = "qdrant/qdrant:v1.8.1") { + constructor(image = "qdrant/qdrant:v1.13.4") { super(image); this.withExposedPorts(QDRANT_REST_PORT, QDRANT_GRPC_PORT); this.withWaitStrategy(