langchain

1 files changed, 15 insertions, 0 deletions

diff --git a/stdout_filterer.py b/stdout_filterer.py
new file mode 100644
index 0000000..c4cf845
--- /dev/null
+++ b/stdout_filterer.py
@@ -0,0 +1,15 @@
+import re
+
+
+class RedactPhoneNumbers:
+    def __init__(self, stream):
+        self.stream = stream
+
+    def write(self, text):
+        # Regular expression to match phone numbers
+        phone_regex = r"(\+\d{1,2}\s?)?1?\-?\.?\s?\(?\d{3}\)?[\s.-]?\d{3}[\s.-]?\d{4}"
+        redacted_text = re.sub(phone_regex, "****", text)
+        self.stream.write(redacted_text)
+
+    def flush(self):
+        self.stream.flush()