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

resolve sqldocstore postgres compatibility issue

Browse Source
This commit is contained in:
Alex Lee 2025-03-15 16:46:54 -07:00
parent 91b0ea385c
commit 40979a0063
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -184,9 +184,8 @@ class SQLStore(BaseStore[str, bytes]):
LangchainKeyValueStores.namespace == self.namespace,
)
)
for v in await session.scalars(stmt):
result[v.key] = v.value
return [result.get(key) for key in keys]
results = session.execute(stmt).all()
return [result[0].value for result in results]
def mget(self, keys: Sequence[str]) -> List[Optional[bytes]]:
result = {}
@ -198,28 +197,27 @@ class SQLStore(BaseStore[str, bytes]):
LangchainKeyValueStores.namespace == self.namespace,
)
)
for v in session.scalars(stmt):
result[v.key] = v.value
return [result.get(key) for key in keys]
results = session.execute(stmt).all()
return [result[0].value for result in results]
async def amset(self, key_value_pairs: Sequence[Tuple[str, bytes]]) -> None:
async def amset(self, key_value_pairs: Sequence[Tuple[str, bytes]], encoding: str = 'utf8') -> None:
async with self._make_async_session() as session:
await self._amdelete([key for key, _ in key_value_pairs], session)
session.add_all(
[
LangchainKeyValueStores(namespace=self.namespace, key=k, value=v)
LangchainKeyValueStores(namespace=self.namespace, key=k, value=bytearray(v, encoding))
for k, v in key_value_pairs
]
)
await session.commit()
def mset(self, key_value_pairs: Sequence[Tuple[str, bytes]]) -> None:
def mset(self, key_value_pairs: Sequence[Tuple[str, bytes]], , encoding: str = 'utf8') -> None:
values: Dict[str, bytes] = dict(key_value_pairs)
with self._make_sync_session() as session:
self._mdelete(list(values.keys()), session)
session.add_all(
[
LangchainKeyValueStores(namespace=self.namespace, key=k, value=v)
LangchainKeyValueStores(namespace=self.namespace, key=k, value=bytearray(v, encoding))
for k, v in values.items()
]
)