aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-19 06:54:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Merwane Hamadi
2023-04-12 17:48:08 -07:00
parent a0f900f2da
commit 62bd93a4d2
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/memory/__init__.py
View File

@@ -1,4 +1,5 @@
from memory.local import LocalCache from memory.local import LocalCache
from memory.no_memory import NoMemory
# List of supported memory backends # List of supported memory backends
# Add a backend to this list if the import attempt is successful # 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.") " use Redis as a memory backend.")
else: else:
memory = RedisMemory(cfg) memory = RedisMemory(cfg)
elif cfg.memory_backend == "no_memory":
memory = NoMemory(cfg)
if memory is None: if memory is None:
memory = LocalCache(cfg) memory = LocalCache(cfg)
@@ -50,4 +53,5 @@ __all__ = [
"LocalCache", "LocalCache",
"RedisMemory", "RedisMemory",
"PineconeMemory", "PineconeMemory",
"NoMemory"
] ]