diff --git a/requirements.txt b/requirements.txt
index 708a2fbc..90bee338 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -130,4 +130,5 @@ beautifulsoup4
 celery
 redis
 google-search-results==2.4.2
-Pillow
\ No newline at end of file
+Pillow
+fiass-cpu
\ No newline at end of file
diff --git a/swarms/agents/base.py b/swarms/agents/base.py
index 30d1d2dc..e81e9e39 100644
--- a/swarms/agents/base.py
+++ b/swarms/agents/base.py
@@ -6,11 +6,13 @@ from langchain.chains.llm import LLMChain
 from pydantic import ValidationError
 
 from swarms.agents.memory.base import VectorStoreRetriever
-from swarms.agents.memory.base_memory import BaseChatMessageHistory
-from swarms.agents.memory.chat_message_history import ChatMessageHistory
+from swarms.agents.memory.base_memory import ChatMessageHistory, BaseChatMessageHistory
+
+
 from swarms.agents.memory.document import Document
 from swarms.agents.models.base import AbstractModel
 from swarms.agents.models.prompts.agent_output_parser import AgentOutputParser
+
 from swarms.agents.models.prompts.agent_prompt import (
     AIMessage,
     HumanMessage,