Source code for langgraph_agent_toolkit.core.memory.base

from abc import ABC, abstractmethod
from contextlib import AbstractAsyncContextManager
from typing import Any, TypeVar


T = TypeVar("T", bound=Any)




class BaseMemoryBackend(ABC):
    """Base class for memory backends."""



    @abstractmethod
    def validate_config(self) -> bool:
        """Validate that all necessary configuration is set.

        Returns:
            True if configuration is valid

        Raises:
            ValueError: If required configuration is missing

        """
        pass




    @abstractmethod
    def get_checkpoint_saver(self) -> AbstractAsyncContextManager[T]:
        """Get the checkpoint saver for the memory backend.

        Returns:
            A configured checkpoint saver

        """
        pass




    @abstractmethod
    def get_memory_store(self) -> AbstractAsyncContextManager[T]:
        """Get the memory store for the memory backend.

        Returns:
            A configured memory store

        """
        pass