Import NoMemory and add it as a memory_backend option in get_memory function

This commit is contained in:
Merwane Hamadi
2023-04-12 17:48:08 -07:00
parent a0f900f2da
commit 62bd93a4d2

View File

@@ -1,4 +1,5 @@
from memory.local import LocalCache
from memory.no_memory import NoMemory
# List of supported memory backends
# Add a backend to this list if the import attempt is successful
@@ -34,6 +35,8 @@ def get_memory(cfg, init=False):
" use Redis as a memory backend.")
else:
memory = RedisMemory(cfg)
elif cfg.memory_backend == "no_memory":
memory = NoMemory(cfg)
if memory is None:
memory = LocalCache(cfg)
@@ -50,4 +53,5 @@ __all__ = [
"LocalCache",
"RedisMemory",
"PineconeMemory",
"NoMemory"
]