diff --git a/scripts/memory/no_memory.py b/scripts/memory/no_memory.py index 45dbd7349b..830982f9c4 100644 --- a/scripts/memory/no_memory.py +++ b/scripts/memory/no_memory.py @@ -2,6 +2,7 @@ from typing import Optional, List, Any from memory.base import MemoryProviderSingleton + class NoMemory(MemoryProviderSingleton): def __init__(self, cfg): """ diff --git a/scripts/memory/pinecone.py b/scripts/memory/pinecone.py index fa21124b5c..20a905b32c 100644 --- a/scripts/memory/pinecone.py +++ b/scripts/memory/pinecone.py @@ -5,6 +5,7 @@ from memory.base import MemoryProviderSingleton, get_ada_embedding from logger import logger from colorama import Fore, Style + class PineconeMemory(MemoryProviderSingleton): def __init__(self, cfg): pinecone_api_key = cfg.pinecone_api_key