diff --git a/langgraph/checkpoint/aiosqlite.py b/langgraph/checkpoint/aiosqlite.py index a374886cd..81a5a0943 100644 --- a/langgraph/checkpoint/aiosqlite.py +++ b/langgraph/checkpoint/aiosqlite.py @@ -28,10 +28,10 @@ class AsyncSqliteSaver(BaseCheckpointSaver, AbstractAsyncContextManager): async def __aexit__( self, - __exc_type: type[BaseException] | None, - __exc_value: BaseException | None, - __traceback: TracebackType | None, - ) -> bool | None: + __exc_type: Optional[type[BaseException]], + __exc_value: Optional[BaseException], + __traceback: Optional[TracebackType], + ) -> Optional[bool]: return await self.conn.close() async def setup(self) -> None: diff --git a/langgraph/checkpoint/sqlite.py b/langgraph/checkpoint/sqlite.py index d2b801f19..b7dbc04c8 100644 --- a/langgraph/checkpoint/sqlite.py +++ b/langgraph/checkpoint/sqlite.py @@ -28,10 +28,10 @@ class SqliteSaver(BaseCheckpointSaver, AbstractContextManager): def __exit__( self, - __exc_type: type[BaseException] | None, - __exc_value: BaseException | None, - __traceback: TracebackType | None, - ) -> bool | None: + __exc_type: Optional[type[BaseException]], + __exc_value: Optional[BaseException], + __traceback: Optional[TracebackType], + ) -> Optional[bool]: return self.conn.close() def setup(self) -> None: