From 3944f29addc1a2ea908e7ff8a78e36f21bd5c9db Mon Sep 17 00:00:00 2001
From: Eesa Hamza <eesa.hamza@gmail.com>
Date: Sun, 16 Apr 2023 21:40:09 +0300
Subject: [PATCH] Fixed new backends not being added to supported memory

---
 README.md                  | 2 +-
 autogpt/memory/__init__.py | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 21f3ccf2..bfcd395c 100644
--- a/README.md
+++ b/README.md
@@ -195,7 +195,7 @@ python -m autogpt --help
 ```bash
 python -m autogpt --ai-settings <filename>
 ```
-* Specify one of 3 memory backends: `local`, `redis`, `pinecone` or `no_memory`
+* Specify a memory backend
 ```bash
 python -m autogpt --use-memory  <memory-backend>
 ```
diff --git a/autogpt/memory/__init__.py b/autogpt/memory/__init__.py
index e2ee44a4..f5afb8c9 100644
--- a/autogpt/memory/__init__.py
+++ b/autogpt/memory/__init__.py
@@ -23,12 +23,16 @@ except ImportError:
 
 try:
     from autogpt.memory.weaviate import WeaviateMemory
+
+    supported_memory.append("weaviate")
 except ImportError:
     # print("Weaviate not installed. Skipping import.")
     WeaviateMemory = None
 
 try:
     from autogpt.memory.milvus import MilvusMemory
+
+    supported_memory.append("milvus")
 except ImportError:
     # print("pymilvus not installed. Skipping import.")
     MilvusMemory = None