From d89bbbe4b34369c564d9b31967ad83f6e5a73fc3 Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 30 Aug 2023 10:56:59 -0400 Subject: [PATCH] swamrs.agents.memory -> swarms.memory --- swarms/memory/base.py | 2 +- swarms/memory/utils.py | 2 +- swarms/orchestrator/orchestrate.py | 2 +- tests/agents/memory/main.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/swarms/memory/base.py b/swarms/memory/base.py index 6af5015b..ff51e0da 100644 --- a/swarms/memory/base.py +++ b/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 diff --git a/swarms/memory/utils.py b/swarms/memory/utils.py index 5c4792e6..43f0622e 100644 --- a/swarms/memory/utils.py +++ b/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 diff --git a/swarms/orchestrator/orchestrate.py b/swarms/orchestrator/orchestrate.py index 902b5f4b..4e697598 100644 --- a/swarms/orchestrator/orchestrate.py +++ b/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): diff --git a/tests/agents/memory/main.py b/tests/agents/memory/main.py index f7ae0ec4..c156ab3d 100644 --- a/tests/agents/memory/main.py +++ b/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