from memory.local import LocalCache
try:
    from memory.redismem import RedisMemory
except ImportError:
    print("Redis not installed. Skipping import.")
    RedisMemory = None

try:
    from memory.pinecone import PineconeMemory
except ImportError:
    print("Pinecone not installed. Skipping import.")
    PineconeMemory = None

try:
    from memory.weaviate import WeaviateMemory
except ImportError:
    print("Weaviate not installed. Skipping import.")
    WeaviateMemory = None

def get_memory(cfg, init=False):
    memory = None
    if cfg.memory_backend == "pinecone":
        if not PineconeMemory:
            print("Error: Pinecone is not installed. Please install pinecone"
                  " to use Pinecone as a memory backend.")
        else:
            memory = PineconeMemory(cfg)
            if init:
                memory.clear()
    elif cfg.memory_backend == "redis":
        if not RedisMemory:
            print("Error: Redis is not installed. Please install redis-py to"
                  " use Redis as a memory backend.")
        else:
            memory = RedisMemory(cfg)
    elif cfg.memory_backend == "weaviate":
        if not WeaviateMemory:
            print("Error: Weaviate is not installed. Please install weaviate-client to"
                  " use Weaviate as a memory backend.")
        else:
            memory = WeaviateMemory(cfg)
            
    if memory is None:
        memory = LocalCache(cfg)
        if init:
            memory.clear()
    return memory


__all__ = [
    "get_memory",
    "LocalCache",
    "RedisMemory",
    "PineconeMemory",
    "WeaviateMemory"
]