github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-02 03:15:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

astradb[patch]: Use astrapy's upsert_one method in AstraDBStore (#18063)

Browse Source 
As `upsert` is deprecated
This commit is contained in:
Christophe Bornet 2024-02-26 04:04:18 +01:00 committed by GitHub
parent e42110f720
commit 4d6cd5b46a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/partners/astradb/langchain_astradb/storage.py
View File

@ -60,12 +60,12 @@ class AstraDBBaseStore(Generic[V], BaseStore[str, V], ABC):
def mset(self, key_value_pairs: Sequence[Tuple[str, V]]) -> None: def mset(self, key_value_pairs: Sequence[Tuple[str, V]]) -> None:
self.astra_env.ensure_db_setup() self.astra_env.ensure_db_setup()
for k, v in key_value_pairs: 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: async def amset(self, key_value_pairs: Sequence[Tuple[str, V]]) -> None:
await self.astra_env.aensure_db_setup() await self.astra_env.aensure_db_setup()
for k, v in key_value_pairs: 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)} {"_id": k, "value": self.encode_value(v)}
) )