diff --git a/scripts/memory/pinecone.py b/scripts/memory/pinecone.py index 8e1eaa570..eeafd9c9b 100644 --- a/scripts/memory/pinecone.py +++ b/scripts/memory/pinecone.py @@ -2,7 +2,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,7 +17,18 @@ class PineconeMemory(MemoryProviderSingleton): # this assumes we don't start with memory. # for now this works. # we'll need a more complicated and robust system if we want to start with memory. - self.vec_num = 0 + 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.typewriter_log("DOUBLE CHECK CONFIGURATION", + Fore.YELLOW, + "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) diff --git a/scripts/memory/redismem.py b/scripts/memory/redismem.py index 1267c4de2..b2d5326f8 100644 --- a/scripts/memory/redismem.py +++ b/scripts/memory/redismem.py @@ -54,7 +54,7 @@ class RedisMemory(MemoryProviderSingleton): 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. " + + "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)