src/cuga/backend/memory/__init__.py

try:
    from cuga.backend.memory.memory import Memory as Memory
    from cuga.backend.memory.agentic_memory.client.exceptions import (
        APIRequestException as APIRequestException,
        FactNotFoundException as FactNotFoundException,
        MemoryClientException as MemoryClientException,
        NamespaceNotFoundException as NamespaceNotFoundException,
    )
    from cuga.backend.memory.agentic_memory.schema import (
        Fact as Fact,
        Message as Message,
        Namespace as Namespace,
        RecordedFact as RecordedFact,
        Run as Run,
    )

    __all__ = [
        "Memory",
        "APIRequestException",
        "FactNotFoundException",
        "MemoryClientException",
        "NamespaceNotFoundException",
        "Fact",
        "Message",
        "Namespace",
        "RecordedFact",
        "Run",
    ]
except ImportError:
    __all__ = []