From a54ef5ab589c1ae027a1ae49d93229a47619c535 Mon Sep 17 00:00:00 2001 From: Alex Lee Date: Sat, 15 Mar 2025 18:53:12 -0700 Subject: [PATCH] await async calls --- libs/community/langchain_community/storage/sql.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/community/langchain_community/storage/sql.py b/libs/community/langchain_community/storage/sql.py index 6709a24a8dc..5c9f95a0b2c 100644 --- a/libs/community/langchain_community/storage/sql.py +++ b/libs/community/langchain_community/storage/sql.py @@ -184,7 +184,7 @@ class SQLStore(BaseStore[str, bytes]): LangchainKeyValueStores.namespace == self.namespace, ) ) - results = session.execute(stmt).all() + results = await session.execute(stmt).all() return [result[0].value for result in results] def mget(self, keys: Sequence[str]) -> List[Optional[bytes]]: @@ -201,7 +201,7 @@ class SQLStore(BaseStore[str, bytes]): async def amset(self, key_value_pairs: Sequence[Tuple[str, bytes]]) -> None: async with self._make_async_session() as session: await self._amdelete([key for key, _ in key_value_pairs], session) - session.add_all( + await session.add_all( [ LangchainKeyValueStores( namespace=self.namespace,