From e346714632908a734e2c4e5cd9bc4c1a4e8c799f Mon Sep 17 00:00:00 2001 From: Christophe Bornet Date: Sat, 24 Feb 2024 12:01:40 +0100 Subject: [PATCH] Use astrapy's upsert_one method in AstraDBStore --- libs/partners/astradb/langchain_astradb/storage.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/partners/astradb/langchain_astradb/storage.py b/libs/partners/astradb/langchain_astradb/storage.py index 119e82c2cf4965..da4cc58593d4e2 100644 --- a/libs/partners/astradb/langchain_astradb/storage.py +++ b/libs/partners/astradb/langchain_astradb/storage.py @@ -60,12 +60,12 @@ async def amget(self, keys: Sequence[str]) -> List[Optional[V]]: def mset(self, key_value_pairs: Sequence[Tuple[str, V]]) -> None: self.astra_env.ensure_db_setup() for k, v in key_value_pairs: - self.collection.upsert({"_id": k, "value": self.encode_value(v)}) + self.collection.upsert_one({"_id": k, "value": self.encode_value(v)}) async def amset(self, key_value_pairs: Sequence[Tuple[str, V]]) -> None: await self.astra_env.aensure_db_setup() for k, v in key_value_pairs: - await self.async_collection.upsert( + await self.async_collection.upsert_one( {"_id": k, "value": self.encode_value(v)} )