Skip to content

Commit ce73741

Browse files
hntrlDag7
andauthored
chore(redis/v1): patch changes (#9190)
Co-authored-by: Dag7 <Dag7@users.noreply.github.com>
1 parent 499ca4d commit ce73741

File tree

3 files changed

+954
-18
lines changed

3 files changed

+954
-18
lines changed

libs/providers/langchain-redis/src/tests/vectorstores.int.test.ts

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,4 +73,33 @@ describe("RedisVectorStore", () => {
7373
new Document({ metadata: { foo: uuid }, pageContent }),
7474
]);
7575
});
76+
77+
test("delete documents by ids", async () => {
78+
const documentIds = ["doc1", "doc2"];
79+
const documentKeys = documentIds.map((id) => `test:${id}`);
80+
const pageContent = faker.lorem.sentence(5);
81+
82+
const documents = documentKeys.map((key) => ({
83+
pageContent,
84+
metadata: {
85+
id: key,
86+
},
87+
}));
88+
89+
await vectorStore.addDocuments(documents, {
90+
keys: documentKeys,
91+
});
92+
93+
const results = await vectorStore.similaritySearch(pageContent, 2);
94+
expect(results).toHaveLength(2);
95+
expect(results.map((result) => result.metadata.id)).toEqual(documentKeys);
96+
97+
await vectorStore.delete({ ids: [documentIds[0]] });
98+
99+
const results2 = await vectorStore.similaritySearch(pageContent, 2);
100+
expect(results2).toHaveLength(1);
101+
expect(results2.map((result) => result.metadata.id)).toEqual(
102+
documentKeys.slice(1)
103+
);
104+
});
76105
});

0 commit comments

Comments
 (0)