diff --git a/pyproject.toml b/pyproject.toml index 31c7be39..32eaa71c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api" [tool.poetry] name = "swarms" -version = "3.8.5" +version = "3.8.7" description = "Swarms - Pytorch" license = "MIT" authors = ["Kye Gomez "] diff --git a/swarms/memory/__init__.py b/swarms/memory/__init__.py index f6ab4750..c8330fa8 100644 --- a/swarms/memory/__init__.py +++ b/swarms/memory/__init__.py @@ -1,4 +1,5 @@ from swarms.memory.base_vectordb import AbstractVectorDatabase +from swarms.memory.base_db import AbstractDatabase from swarms.memory.short_term_memory import ShortTermMemory from swarms.memory.sqlite import SQLiteDB from swarms.memory.weaviate_db import WeaviateDB @@ -6,6 +7,7 @@ from swarms.memory.visual_memory import VisualShortTermMemory __all__ = [ "AbstractVectorDatabase", + "AbstractDatabase", "ShortTermMemory", "SQLiteDB", "WeaviateDB",