swamrs.agents.memory -> swarms.memory

swarms/memory/base.py

@@ -26,7 +26,7 @@ from langchain.callbacks.manager import (
     CallbackManagerForRetrieverRun,
 )
 
-from swarms.agents.memory.document import Document
+from swarms.memory.document import Document
 from swarms.utils.embeddings.base import Embeddings
 
 from langchain.schema import BaseRetriever

swarms/memory/utils.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, List
 
-from swarms.agents.memory.base import get_buffer_string
+from swarms.memory.base import get_buffer_string
 
 
 

swarms/orchestrator/orchestrate.py

@@ -3,7 +3,7 @@ import threading
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List
 
-from swarms.agents.memory.ocean import OceanDB
+from swarms.memory.ocean import OceanDB
 
 
 class Orchestrator(ABC):

tests/agents/memory/main.py

@@ -1,6 +1,6 @@
 import pytest
 from unittest.mock import Mock
-from swarms.agents.memory.oceandb import OceanDB
+from swarms.memory.oceandb import OceanDB
 
 
 @pytest.fixture