From 8362bd729bda9d5562508ba1bb5f9019dd9ae451 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Boschi?= Date: Mon, 20 Nov 2023 02:54:35 +0100 Subject: [PATCH] AstraDB: use includeSimilarity option instead of $similarity (#13512) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - **Description:** AstraDB is going to deprecate the `$similarity` projection property in favor of the ´includeSimilarity´ option flag. I moved all the queries to the new format. - **Tag maintainer:** @hemidactylus - **Twitter handle:** nicoloboschi --- libs/langchain/langchain/vectorstores/astradb.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/libs/langchain/langchain/vectorstores/astradb.py b/libs/langchain/langchain/vectorstores/astradb.py index ed972a479c9..d1c25e02f46 100644 --- a/libs/langchain/langchain/vectorstores/astradb.py +++ b/libs/langchain/langchain/vectorstores/astradb.py @@ -480,12 +480,11 @@ class AstraDB(VectorStore): self.collection.paginated_find( filter=metadata_parameter, sort={"$vector": embedding}, - options={"limit": k}, + options={"limit": k, "includeSimilarity": True}, projection={ "_id": 1, "content": 1, "metadata": 1, - "$similarity": 1, }, ) ) @@ -609,12 +608,11 @@ class AstraDB(VectorStore): self.collection.paginated_find( filter=metadata_parameter, sort={"$vector": embedding}, - options={"limit": fetch_k}, + options={"limit": fetch_k, "includeSimilarity": True}, projection={ "_id": 1, "content": 1, "metadata": 1, - "$similarity": 1, "$vector": 1, }, )