mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-11 07:25:29 +00:00
AstraDB: use includeSimilarity option instead of $similarity (#13512)
- **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
This commit is contained in:
parent
7100d586ef
commit
8362bd729b
@ -480,12 +480,11 @@ class AstraDB(VectorStore):
|
|||||||
self.collection.paginated_find(
|
self.collection.paginated_find(
|
||||||
filter=metadata_parameter,
|
filter=metadata_parameter,
|
||||||
sort={"$vector": embedding},
|
sort={"$vector": embedding},
|
||||||
options={"limit": k},
|
options={"limit": k, "includeSimilarity": True},
|
||||||
projection={
|
projection={
|
||||||
"_id": 1,
|
"_id": 1,
|
||||||
"content": 1,
|
"content": 1,
|
||||||
"metadata": 1,
|
"metadata": 1,
|
||||||
"$similarity": 1,
|
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
@ -609,12 +608,11 @@ class AstraDB(VectorStore):
|
|||||||
self.collection.paginated_find(
|
self.collection.paginated_find(
|
||||||
filter=metadata_parameter,
|
filter=metadata_parameter,
|
||||||
sort={"$vector": embedding},
|
sort={"$vector": embedding},
|
||||||
options={"limit": fetch_k},
|
options={"limit": fetch_k, "includeSimilarity": True},
|
||||||
projection={
|
projection={
|
||||||
"_id": 1,
|
"_id": 1,
|
||||||
"content": 1,
|
"content": 1,
|
||||||
"metadata": 1,
|
"metadata": 1,
|
||||||
"$similarity": 1,
|
|
||||||
"$vector": 1,
|
"$vector": 1,
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user