Method remove_memory

nextpy/ai/memory/buffer_summary.py:134–209

Remove a memory from the store.

(self, prompt: str, llm: Any, threshold=1)

        return summarized_memory

    def remove_memory(self, prompt: str, llm: Any, threshold=1) -> None:
        """Remove a memory from the store."""
        # update messages
        for conversation in self.messages:
            if conversation["prompt"] == prompt:
                # updating messages
                self.messages.remove(conversation)

                # updating other variables
                self.current_summary = ""
                self.current_buffer = ""
                self.messages_in_buffer.clear()
                self.messages_in_summary.clear()

                # get new messages in memory
                new_messages = [
                    item
                    for item in self.messages
                    if item not in self.messages_in_summary
                    and item not in self.messages_in_buffer
                ]

                if len(new_messages) != 0:
                    # split conversations for buffer and summary according to threshold
                    buffer_messages = self.messages[-1 * threshold :]  # buffer
                    summary_messages = [
                        item for item in self.messages if item not in buffer_messages
                    ]  # summary

                    # get messages in the buffer format as text
                    self.current_buffer = ""
                    for conversation in buffer_messages:
                        self.current_buffer = (
                            self.current_buffer
                            + "Human: "
                            + conversation["prompt"]
                            + "\n"
                            + "AI: "
                            + conversation["llm_response"]
                            + "\n"
                        )
                    self.messages_in_buffer = buffer_messages

                    # get the new messages to be added in the summary
                    additions_to_summary = [
                        item
                        for item in summary_messages
                        if item not in self.messages_in_summary
                    ]

                    if len(additions_to_summary) != 0:
                        additions_to_summary_as_text = ""
                        print(additions_to_summary_as_text)

                        for conversation in additions_to_summary:
                            additions_to_summary_as_text = (
                                additions_to_summary_as_text
                                + "Human: "