Skip to content

Local conversation memory driver

LocalConversationMemoryDriver

Bases: BaseConversationMemoryDriver

Source code in 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, metadata={"serializable": True})

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

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

            memory.driver = self

            return memory

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

load()

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

        memory.driver = self

        return memory

store(memory)

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