Add RemovePunctuationConverter

Author: hupe1980

aisploit/converter/__init__.py

@@ -7,6 +7,7 @@
     KEYBOARD_NEIGHBORS_QWERTZ,
 )
 from .no_op import NoOpConverter
+from .remove_punctuation import RemovePunctuationConverter
 from .sequence import SequenceConverter
 
 __all__ = [
@@ -17,5 +18,6 @@
     "KEYBOARD_NEIGHBORS_QWERTY",
     "KEYBOARD_NEIGHBORS_QWERTZ",
     "NoOpConverter",
+    "RemovePunctuationConverter",
     "SequenceConverter",
 ]

aisploit/converter/remove_punctuation.py

@@ -0,0 +1,8 @@
+import string
+from ..core import BaseConverter
+
+
+class RemovePunctuationConverter(BaseConverter):
+    def _convert(self, prompt: str) -> str:
+        translator = str.maketrans('', '', string.punctuation)
+        return prompt.translate(translator)

examples/converter.ipynb

@@ -28,7 +28,13 @@
     "import os\n",
     "import textwrap\n",
     "from dotenv import load_dotenv\n",
-    "from aisploit.converter import Base64Converter, KeyboardTypoConverter, JoinConverter, GenderConverter\n",
+    "from aisploit.converter import (\n",
+    "    Base64Converter,\n",
+    "    KeyboardTypoConverter,\n",
+    "    JoinConverter,\n",
+    "    GenderConverter,\n",
+    "    RemovePunctuationConverter,\n",
+    ")\n",
     "from aisploit.models import ChatOpenAI\n",
     "\n",
     "load_dotenv()"
@@ -177,6 +183,38 @@
     "\n",
     "display(Markdown(converted_prompt.to_string()))"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## RemovePunctuationConverter"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "Hello world How are you"
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "converter = RemovePunctuationConverter()\n",
+    "converted_prompt = converter.convert(\"Hello, world! How are you?\")\n",
+    "\n",
+    "display(Markdown(converted_prompt.to_string()))"
+   ]
   }
  ],
  "metadata": {