github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-22 11:00:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reset deanonymizer mapping (#11559)

Browse Source 
@hwchase17 @baskaryan
This commit is contained in:
maks-operlejn-ds
2023-10-09 20:11:05 +02:00
committed by GitHub
parent b14b65d62a
commit f64522fbaf
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/experimental/langchain_experimental/data_anonymizer/base.py
View File

@@ -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"""

4
libs/experimental/langchain_experimental/data_anonymizer/presidio.py
View File

@@ -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.