Source code for autogen_core._cache_store
from typing import Dict, Generic, Optional, Protocol, TypeVar
T = TypeVar("T")
[docs]
class CacheStore(Protocol, Generic[T]):
"""
This protocol defines the basic interface for store/cache operations.
Sub-classes should handle the lifecycle of underlying storage.
"""
[docs]
def get(self, key: str, default: Optional[T] = None) -> Optional[T]:
"""
Retrieve an item from the store.
Args:
key: The key identifying the item in the store.
default (optional): The default value to return if the key is not found.
Defaults to None.
Returns:
The value associated with the key if found, else the default value.
"""
...
[docs]
def set(self, key: str, value: T) -> None:
"""
Set an item in the store.
Args:
key: The key under which the item is to be stored.
value: The value to be stored in the store.
"""
...
[docs]
class InMemoryStore(CacheStore[T]):
def __init__(self) -> None:
self.store: Dict[str, T] = {}
[docs]
def get(self, key: str, default: Optional[T] = None) -> Optional[T]:
return self.store.get(key, default)
[docs]
def set(self, key: str, value: T) -> None:
self.store[key] = value