github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-14 15:16:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

more linting fixes

Browse Source
This commit is contained in:
Alex Lee 2025-03-17 16:57:33 -07:00
parent 579688202f
commit 1085926d45
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libs/community/langchain_community/storage/sql.py
View File

@ -201,7 +201,7 @@ class SQLStore(BaseStore[str, bytes]):
async def amset(self, key_value_pairs: Sequence[Tuple[str, bytes]]) -> None: async def amset(self, key_value_pairs: Sequence[Tuple[str, bytes]]) -> None:
async with self._make_async_session() as session: async with self._make_async_session() as session:
await self._amdelete([key for key, _ in key_value_pairs], session) await self._amdelete([key for key, _ in key_value_pairs], session)
await session.add_all( session.add_all(
[ [
LangchainKeyValueStores( LangchainKeyValueStores(
namespace=self.namespace, namespace=self.namespace,
@ -213,8 +213,8 @@ class SQLStore(BaseStore[str, bytes]):
) )
await session.commit() await session.commit()
def mset(self, key_value_pairs: Sequence[Tuple[str, bytes]]) -> None: def mset(self, key_value_pairs: Sequence[Tuple[str, bytes | Document]]) -> None:
values: Dict[str, bytes] = dict(key_value_pairs) values = dict(key_value_pairs)
with self._make_sync_session() as session: with self._make_sync_session() as session:
self._mdelete(list(values.keys()), session) self._mdelete(list(values.keys()), session)
session.add_all( session.add_all(
@ -229,13 +229,12 @@ class SQLStore(BaseStore[str, bytes]):
) )
session.commit() session.commit()
def _bytes_or_document(self, v: Union[bytes, str, Document]) -> Union[bytes, bytearray]: def _bytes_or_document(self, v: Union[bytes, Document]) -> Union[bytes, bytearray]:
if type(v) is bytes: if type(v) is bytes:
return v return v
elif type(v) is Document: elif type(v) is Document:
return bytearray(v.page_content, "utf8") return bytearray(v.page_content, "utf8")
else: raise ValueError("Expecting bytes or langchain Document")
return bytes(v, "utf8")
def _mdelete(self, keys: Sequence[str], session: Session) -> None: def _mdelete(self, keys: Sequence[str], session: Session) -> None:
stmt = delete(LangchainKeyValueStores).filter( stmt = delete(LangchainKeyValueStores).filter(