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

lint fixes

Browse Source
This commit is contained in:
Alex Lee 2025-03-17 16:28:37 -07:00
parent 7285defea0
commit 579688202f
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -184,8 +184,8 @@ class SQLStore(BaseStore[str, bytes]):
LangchainKeyValueStores.namespace == self.namespace, LangchainKeyValueStores.namespace == self.namespace,
) )
) )
results = await session.execute(stmt).all() results = await session.execute(stmt)
return [result[0].value for result in results] return [result[0].value for result in results.all()]
def mget(self, keys: Sequence[str]) -> List[Optional[bytes]]: def mget(self, keys: Sequence[str]) -> List[Optional[bytes]]:
with self._make_sync_session() as session: with self._make_sync_session() as session:
@ -206,7 +206,7 @@ class SQLStore(BaseStore[str, bytes]):
LangchainKeyValueStores( LangchainKeyValueStores(
namespace=self.namespace, namespace=self.namespace,
key=k, key=k,
value=self._bytes_or_document(v) value=self._bytes_or_document(v),
) )
for k, v in key_value_pairs for k, v in key_value_pairs
] ]
@ -222,18 +222,20 @@ class SQLStore(BaseStore[str, bytes]):
LangchainKeyValueStores( LangchainKeyValueStores(
namespace=self.namespace, namespace=self.namespace,
key=k, key=k,
value=self._bytes_or_document(v) value=self._bytes_or_document(v),
) )
for k, v in values.items() for k, v in values.items()
] ]
) )
session.commit() session.commit()
def _bytes_or_document(self, v: Union[bytes, Document]) -> Union[bytes, bytearray]: def _bytes_or_document(self, v: Union[bytes, str, 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:
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(