diff --git a/libs/checkpoint-sqlite/langgraph/cache/sqlite/__init__.py b/libs/checkpoint-sqlite/langgraph/cache/sqlite/__init__.py index 780e88282..9942a1228 100644 --- a/libs/checkpoint-sqlite/langgraph/cache/sqlite/__init__.py +++ b/libs/checkpoint-sqlite/langgraph/cache/sqlite/__init__.py @@ -5,13 +5,12 @@ import datetime import sqlite3 import threading from collections.abc import Mapping, Sequence -from typing import Generic from langgraph.cache.base import BaseCache, FullKey, Namespace, ValueT from langgraph.checkpoint.serde.base import SerializerProtocol -class SqliteCache(BaseCache[ValueT], Generic[ValueT]): +class SqliteCache(BaseCache[ValueT]): """File-based cache using SQLite.""" def __init__( diff --git a/libs/checkpoint/langgraph/cache/memory/__init__.py b/libs/checkpoint/langgraph/cache/memory/__init__.py index c39d4b40f..a6d858e80 100644 --- a/libs/checkpoint/langgraph/cache/memory/__init__.py +++ b/libs/checkpoint/langgraph/cache/memory/__init__.py @@ -3,13 +3,12 @@ from __future__ import annotations import datetime import threading from collections.abc import Mapping, Sequence -from typing import Generic from langgraph.cache.base import BaseCache, FullKey, Namespace, ValueT from langgraph.checkpoint.serde.base import SerializerProtocol -class InMemoryCache(BaseCache[ValueT], Generic[ValueT]): +class InMemoryCache(BaseCache[ValueT]): def __init__(self, *, serde: SerializerProtocol | None = None): super().__init__(serde=serde) self._cache: dict[Namespace, dict[str, tuple[str, bytes, float | None]]] = {}