diff --git a/libs/core/langchain_core/stores.py b/libs/core/langchain_core/stores.py index 7f51b4c4b5a..939be003748 100644 --- a/libs/core/langchain_core/stores.py +++ b/libs/core/langchain_core/stores.py @@ -56,22 +56,22 @@ class BaseStore(ABC, Generic[K, V]): class MyInMemoryStore(BaseStore[str, int]): - def __init__(self): - self.store = {} + def __init__(self) -> None: + self.store: dict[str, int] = {} - def mget(self, keys): + def mget(self, keys: Sequence[str]) -> list[Optional[int]]: return [self.store.get(key) for key in keys] - def mset(self, key_value_pairs): + def mset(self, key_value_pairs: Sequence[tuple[str, int]]) -> None: for key, value in key_value_pairs: self.store[key] = value - def mdelete(self, keys): + def mdelete(self, keys: Sequence[str]) -> None: for key in keys: if key in self.store: del self.store[key] - def yield_keys(self, prefix=None): + def yield_keys(self, prefix: Optional[str] = None) -> Iterator[str]: if prefix is None: yield from self.store.keys() else: