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

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