From a850c27dd553cf6da5b29ac025d68dcce24781dc Mon Sep 17 00:00:00 2001
From: Eesa Hamza <eesa.hamza@gmail.com>
Date: Wed, 12 Apr 2023 23:13:34 +0300
Subject: [PATCH] Improved logging on connection fail to redis

---
 scripts/memory/redismem.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/scripts/memory/redismem.py b/scripts/memory/redismem.py
index 2082fe58..1267c4de 100644
--- a/scripts/memory/redismem.py
+++ b/scripts/memory/redismem.py
@@ -7,6 +7,8 @@ from redis.commands.search.indexDefinition import IndexDefinition, IndexType
 import numpy as np
 
 from memory.base import MemoryProviderSingleton, get_ada_embedding
+from logger import logger
+from colorama import Fore, Style 
 
 
 SCHEMA = [
@@ -44,6 +46,18 @@ class RedisMemory(MemoryProviderSingleton):
             db=0  # Cannot be changed
         )
         self.cfg = cfg
+
+        # Check redis connection
+        try:
+            self.redis.ping()
+        except redis.ConnectionError as e:
+            logger.typewriter_log("FAILED TO CONNECT TO REDIS", Fore.RED, Style.BRIGHT + str(e) + Style.RESET_ALL)
+            logger.typewriter_log("DOUBLE CHECK CONFIGURATION", 
+                                  Fore.YELLOW, 
+                                    "Please ensure you have setup and configured redis properly for use. " +
+                                   f"You can check out {Fore.CYAN + Style.BRIGHT}https://github.com/Torantulino/Auto-GPT#redis-setup{Style.RESET_ALL} to ensure you've set up everything correctly.")
+            exit(1)
+
         if cfg.wipe_redis_on_start:
             self.redis.flushall()
         try: