Files
Auto-GPT/scripts/factory.py

11 lines
304 B
Python

from providers.pinecone import PineconeMemory
from providers.weaviate import WeaviateMemory
class MemoryFactory:
@staticmethod
def get_memory(mem_type):
if mem_type == 'pinecone':
return PineconeMemory()
if mem_type == 'weaviate':
return WeaviateMemory()