fixed weaviate test and fixed conflicts

scripts/memory/__init__.py

@@ -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
@@ -47,6 +48,9 @@ def get_memory(cfg, init=False):
         else:
             memory = WeaviateMemory(cfg)
             
+    elif cfg.memory_backend == "no_memory":
+        memory = NoMemory(cfg)
+
     if memory is None:
         memory = LocalCache(cfg)
         if init:
@@ -63,4 +67,5 @@ __all__ = [
     "RedisMemory",
     "PineconeMemory",
     "WeaviateMemory"
+    "NoMemory"
 ]

scripts/memory/base.py

@@ -2,13 +2,13 @@
 import abc
 from config import AbstractSingleton, Config
 import openai
-cfg = Config()
 
+cfg = Config()
 
 def get_ada_embedding(text):
     text = text.replace("\n", " ")
     if cfg.use_azure:
-        return openai.Embedding.create(input=[text], engine=cfg.azure_embeddigs_deployment_id, model="text-embedding-ada-002")["data"][0]["embedding"]
+        return openai.Embedding.create(input=[text], engine=cfg.get_azure_deployment_id_for_model("text-embedding-ada-002"))["data"][0]["embedding"]
     else:
         return openai.Embedding.create(input=[text], model="text-embedding-ada-002")["data"][0]["embedding"]
 

scripts/memory/no_memory.py

@@ -0,0 +1,65 @@
+from typing import Optional, List, Any
+
+from memory.base import MemoryProviderSingleton
+
+class NoMemory(MemoryProviderSingleton):
+    def __init__(self, cfg):
+        """
+        Initializes the NoMemory provider.
+
+        Args:
+            cfg: The config object.
+
+        Returns: None
+        """
+        pass
+
+    def add(self, data: str) -> str:
+        """
+        Adds a data point to the memory. No action is taken in NoMemory.
+
+        Args:
+            data: The data to add.
+
+        Returns: An empty string.
+        """
+        return ""
+
+    def get(self, data: str) -> Optional[List[Any]]:
+        """
+        Gets the data from the memory that is most relevant to the given data.
+        NoMemory always returns None.
+
+        Args:
+            data: The data to compare to.
+
+        Returns: None
+        """
+        return None
+
+    def clear(self) -> str:
+        """
+        Clears the memory. No action is taken in NoMemory.
+
+        Returns: An empty string.
+        """
+        return ""
+
+    def get_relevant(self, data: str, num_relevant: int = 5) -> Optional[List[Any]]:
+        """
+        Returns all the data in the memory that is relevant to the given data.
+        NoMemory always returns None.
+
+        Args:
+            data: The data to compare to.
+            num_relevant: The number of relevant data to return.
+
+        Returns: None
+        """
+        return None
+
+    def get_stats(self):
+        """
+        Returns: An empty dictionary as there are no stats in NoMemory.
+        """
+        return {}

scripts/memory/pinecone.py

@@ -1,7 +1,8 @@
 import pinecone
 
 from memory.base import MemoryProviderSingleton, get_ada_embedding
-
+from logger import logger
+from colorama import Fore, Style
 
 class PineconeMemory(MemoryProviderSingleton):
     def __init__(self, cfg):
@@ -16,6 +17,15 @@ class PineconeMemory(MemoryProviderSingleton):
         # for now this works.
         # we'll need a more complicated and robust system if we want to start with memory.
         self.vec_num = 0
+
+        try:
+            pinecone.whoami()
+        except Exception as e:
+            logger.typewriter_log("FAILED TO CONNECT TO PINECONE", Fore.RED, Style.BRIGHT + str(e) + Style.RESET_ALL)
+            logger.double_check("Please ensure you have setup and configured Pinecone properly for use. " +
+                               f"You can check out {Fore.CYAN + Style.BRIGHT}https://github.com/Torantulino/Auto-GPT#-pinecone-api-key-setup{Style.RESET_ALL} to ensure you've set up everything correctly.")
+            exit(1)
+
         if table_name not in pinecone.list_indexes():
             pinecone.create_index(table_name, dimension=dimension, metric=metric, pod_type=pod_type)
         self.index = pinecone.Index(table_name)

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,16 @@ 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.double_check("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: