diff --git a/scripts/memory/no_memory.py b/scripts/memory/no_memory.py index 45dbd734..830982f9 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 fa21124b..20a905b3 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