From f64522fbafe94bb8cb5fde4e8828dae45bd55f9e Mon Sep 17 00:00:00 2001
From: maks-operlejn-ds <142261444+maks-operlejn-ds@users.noreply.github.com>
Date: Mon, 9 Oct 2023 20:11:05 +0200
Subject: [PATCH] Reset deanonymizer mapping (#11559)

@hwchase17 @baskaryan
---
 .../langchain_experimental/data_anonymizer/base.py            | 4 ++++
 .../langchain_experimental/data_anonymizer/presidio.py        | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/libs/experimental/langchain_experimental/data_anonymizer/base.py b/libs/experimental/langchain_experimental/data_anonymizer/base.py
index 6cca949c0d6..6092ec8fb74 100644
--- a/libs/experimental/langchain_experimental/data_anonymizer/base.py
+++ b/libs/experimental/langchain_experimental/data_anonymizer/base.py
@@ -47,3 +47,7 @@ class ReversibleAnonymizerBase(AnonymizerBase):
         deanonymizer_matching_strategy: Callable[[str, MappingDataType], str],
     ) -> str:
         """Abstract method to deanonymize text"""
+
+    @abstractmethod
+    def reset_deanonymizer_mapping(self) -> None:
+        """Abstract method to reset deanonymizer mapping"""
diff --git a/libs/experimental/langchain_experimental/data_anonymizer/presidio.py b/libs/experimental/langchain_experimental/data_anonymizer/presidio.py
index 0b4012cdb72..b10841b4d81 100644
--- a/libs/experimental/langchain_experimental/data_anonymizer/presidio.py
+++ b/libs/experimental/langchain_experimental/data_anonymizer/presidio.py
@@ -313,6 +313,10 @@ class PresidioReversibleAnonymizer(PresidioAnonymizerBase, ReversibleAnonymizerB
 
         return text_to_deanonymize
 
+    def reset_deanonymizer_mapping(self) -> None:
+        """Reset the deanonymizer mapping"""
+        self._deanonymizer_mapping = DeanonymizerMapping()
+
     def save_deanonymizer_mapping(self, file_path: Union[Path, str]) -> None:
         """Save the deanonymizer mapping to a JSON or YAML file.