From 9615a5b428f5fb99fea3ab1781da2a593ca009b5 Mon Sep 17 00:00:00 2001 From: Alex Lee Date: Mon, 17 Mar 2025 17:02:04 -0700 Subject: [PATCH] use Union --- libs/community/langchain_community/storage/sql.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/community/langchain_community/storage/sql.py b/libs/community/langchain_community/storage/sql.py index d697db4fc6b..d2d3a0e3eef 100644 --- a/libs/community/langchain_community/storage/sql.py +++ b/libs/community/langchain_community/storage/sql.py @@ -213,7 +213,9 @@ class SQLStore(BaseStore[str, bytes]): ) await session.commit() - def mset(self, key_value_pairs: Sequence[Tuple[str, bytes | Document]]) -> None: + def mset( + self, key_value_pairs: Sequence[Tuple[str, Union[bytes, Document]]] + ) -> None: values = dict(key_value_pairs) with self._make_sync_session() as session: self._mdelete(list(values.keys()), session)