await async calls

This commit is contained in:
Alex Lee 2025-03-15 18:53:12 -07:00
parent 0077286176
commit a54ef5ab58

View File

@ -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,