github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-29 01:48:57 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 0 deletions
Show Stats 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], deanonymizer_matching_strategy: Callable[[str, MappingDataType], str],
) -> str: ) -> str:
"""Abstract method to deanonymize text""" """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 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: def save_deanonymizer_mapping(self, file_path: Union[Path, str]) -> None:
"""Save the deanonymizer mapping to a JSON or YAML file. """Save the deanonymizer mapping to a JSON or YAML file.