diff --git a/mas/memory/mas_memory/GMemory.py b/mas/memory/mas_memory/GMemory.py
index cf0619a..33da5e5 100644
--- a/mas/memory/mas_memory/GMemory.py
+++ b/mas/memory/mas_memory/GMemory.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass, replace
 from langchain_chroma import Chroma
-from langchain.docstore.document import Document
+from langchain_core.documents import Document
 import os
 import copy
 import re
@@ -890,4 +890,4 @@ def _retrieve_rule_index(self, operation_rule_text: str) -> int:
         for idx, insight in enumerate(self.insights_memory):
             if insight['rule'] in operation_rule_text:
                 return idx
-        return -1
\ No newline at end of file
+        return -1
diff --git a/mas/memory/mas_memory/generative.py b/mas/memory/mas_memory/generative.py
index ddded2f..7e8a404 100644
--- a/mas/memory/mas_memory/generative.py
+++ b/mas/memory/mas_memory/generative.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from langchain_chroma import Chroma
-from langchain.docstore.document import Document
+from langchain_core.documents import Document
 import re
 
 from .memory_base import MASMemoryBase
diff --git a/mas/memory/mas_memory/memorybank.py b/mas/memory/mas_memory/memorybank.py
index 679c6b1..fec741e 100644
--- a/mas/memory/mas_memory/memorybank.py
+++ b/mas/memory/mas_memory/memorybank.py
@@ -2,7 +2,7 @@
 import copy
 from dataclasses import dataclass, field
 from langchain_chroma import Chroma
-from langchain.docstore.document import Document
+from langchain_core.documents import Document
 
 from .memory_base import MASMemoryBase
 from .prompt import MEMORYBANK
diff --git a/mas/memory/mas_memory/metagpt.py b/mas/memory/mas_memory/metagpt.py
index fac20ab..7b44228 100644
--- a/mas/memory/mas_memory/metagpt.py
+++ b/mas/memory/mas_memory/metagpt.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from langchain_chroma import Chroma
-from langchain.docstore.document import Document
+from langchain_core.documents import Document
 
 from ..common import MASMessage
 from .memory_base import MASMemoryBase
@@ -65,4 +65,4 @@ def retrieve_memory(
             false_task_messages.append(mas_message)
 
         return true_task_messages, false_task_messages, []
-    
\ No newline at end of file
+    
diff --git a/mas/memory/mas_memory/voyager.py b/mas/memory/mas_memory/voyager.py
index fed3cc4..c6f660c 100644
--- a/mas/memory/mas_memory/voyager.py
+++ b/mas/memory/mas_memory/voyager.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from langchain_chroma import Chroma
-from langchain.docstore.document import Document
+from langchain_core.documents import Document
 
 from .memory_base import MASMemoryBase
 from .prompt import VOYAGER
@@ -96,4 +96,4 @@ def retrieve_memory(
             mas_message: MASMessage = MASMessage.from_dict(meta_data)
             false_task_messages.append(mas_message)
         
-        return true_task_messages, false_task_messages, []
\ No newline at end of file
+        return true_task_messages, false_task_messages, []