Skip to content

Local conversation memory driver

LocalConversationMemoryDriver

Bases: BaseConversationMemoryDriver

Source code in griptape/griptape/drivers/memory/conversation/local_conversation_memory_driver.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
@define
class LocalConversationMemoryDriver(BaseConversationMemoryDriver):
    file_path: str = field(default="griptape_memory.json", kw_only=True)

    def store(self, memory: ConversationMemory) -> None:
        with open(self.file_path, "w") as file:
            file.write(memory.to_json())

    def load(self) -> Optional[ConversationMemory]:
        if not os.path.exists(self.file_path):
            return None
        with open(self.file_path, "r") as file:
            memory = ConversationMemory.from_json(file.read())

            memory.driver = self

            return memory

file_path: str = field(default='griptape_memory.json', kw_only=True) class-attribute instance-attribute

load()

Source code in griptape/griptape/drivers/memory/conversation/local_conversation_memory_driver.py 
16
17
18
19
20
21
22
23
24
def load(self) -> Optional[ConversationMemory]:
    if not os.path.exists(self.file_path):
        return None
    with open(self.file_path, "r") as file:
        memory = ConversationMemory.from_json(file.read())

        memory.driver = self

        return memory

store(memory)

Source code in griptape/griptape/drivers/memory/conversation/local_conversation_memory_driver.py 
12
13
14
def store(self, memory: ConversationMemory) -> None:
    with open(self.file_path, "w") as file:
        file.write(memory.to_json())