mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-19 14:04:29 +01:00
13 lines
357 B
Python
13 lines
357 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()
|
|
|
|
raise ValueError('Unknown memory provider') |