langchain.docstore no longer exists

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
+        return -1

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

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

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, []
-
+

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, []
+        return true_task_messages, false_task_messages, []