FEAT: Integrate XPIATestOrchestrator with the AI Recruiter

build_scripts/check_links.py

@@ -26,7 +26,7 @@
 
 
 def extract_urls(file_path):
-    with open(file_path, "r") as file:
+    with open(file_path, "r", encoding="utf-8") as file:
         content = file.read()
     matches = URL_PATTERN.findall(content)
     # Flatten the list of tuples and filter out empty strings

doc/code/converters/pdf_converter.ipynb

@@ -30,16 +30,7 @@
    "execution_count": null,
    "id": "1",
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{'__type__': 'TextTarget', '__module__': 'pyrit.prompt_target.text_target'}: user: D:\\git\\PyRIT-internal\\PyRIT\\dbdata\\prompt-memory-entries\\urls\\1738382955429053.pdf\n",
-      "\u001b[1m\u001b[34muser: D:\\git\\PyRIT-internal\\PyRIT\\dbdata\\prompt-memory-entries\\urls\\1738382955429053.pdf\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "import pathlib\n",
     "\n",
@@ -77,7 +68,7 @@
     "# Initialize the PDFConverter\n",
     "pdf_converter = PDFConverter(\n",
     "    prompt_template=prompt_template,\n",
-    "    font_type=\"Arial\",\n",
+    "    font_type=\"Helvetica\",\n",
     "    font_size=12,\n",
     "    page_width=210,\n",
     "    page_height=297,\n",
@@ -110,22 +101,7 @@
    "execution_count": null,
    "id": "3",
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{'__type__': 'TextTarget', '__module__': 'pyrit.prompt_target.text_target'}: user: D:\\git\\PyRIT-internal\\PyRIT\\dbdata\\prompt-memory-entries\\urls\\1738380368779812.pdf\n"
-     ]
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "\u001b[1m\u001b[34muser: D:\\git\\PyRIT-internal\\PyRIT\\dbdata\\prompt-memory-entries\\urls\\1738380368779812.pdf\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "# Define a simple string prompt (no templates)\n",
     "prompt = \"This is a simple test string for PDF generation. No templates here!\"\n",
@@ -136,7 +112,7 @@
     "# Initialize the PDFConverter without a template\n",
     "pdf_converter = PDFConverter(\n",
     "    prompt_template=None,  # No template provided\n",
-    "    font_type=\"Arial\",\n",
+    "    font_type=\"Helvetica\",\n",
     "    font_size=12,\n",
     "    page_width=210,\n",
     "    page_height=297,\n",
@@ -171,19 +147,7 @@
    "execution_count": null,
    "id": "5",
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[09:10:28][26][ai-red-team][INFO][Processing page 0 with 2 injection items.]\n",
-      "[09:10:28][31][ai-red-team][INFO][Processing page 1 with 2 injection items.]\n",
-      "[09:10:28][33][ai-red-team][INFO][Processing page 2 with 2 injection items.]\n",
-      "{'__type__': 'TextTarget', '__module__': 'pyrit.prompt_target.text_target'}: user: D:\\git\\PyRIT-internal\\PyRIT\\dbdata\\prompt-memory-entries\\urls\\1738429828035527.pdf\n",
-      "\u001b[1m\u001b[34muser: D:\\git\\PyRIT-internal\\PyRIT\\dbdata\\prompt-memory-entries\\urls\\1738429828035527.pdf\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "import tempfile\n",
     "from pathlib import Path\n",
@@ -208,8 +172,24 @@
     "\n",
     "# Define injection items\n",
     "injection_items = [\n",
-    "    {\"page\": 0, \"x\": 50, \"y\": 700, \"text\": \"Injected Text\", \"font_size\": 12, \"font\": \"Helvetica\", \"font_color\": (255, 0, 0)},  # Red text\n",
-    "    {\"page\": 1, \"x\": 100, \"y\": 600, \"text\": \"Confidential\", \"font_size\": 10, \"font\": \"Helvetica\", \"font_color\": (0, 0, 255)}  # Blue text\n",
+    "    {\n",
+    "        \"page\": 0,\n",
+    "        \"x\": 50,\n",
+    "        \"y\": 700,\n",
+    "        \"text\": \"Injected Text\",\n",
+    "        \"font_size\": 12,\n",
+    "        \"font\": \"Helvetica\",\n",
+    "        \"font_color\": (255, 0, 0),\n",
+    "    },  # Red text\n",
+    "    {\n",
+    "        \"page\": 1,\n",
+    "        \"x\": 100,\n",
+    "        \"y\": 600,\n",
+    "        \"text\": \"Confidential\",\n",
+    "        \"font_size\": 10,\n",
+    "        \"font\": \"Helvetica\",\n",
+    "        \"font_color\": (0, 0, 255),\n",
+    "    },  # Blue text\n",
     "]\n",
     "\n",
     "# Define a simple string prompt (no templates)\n",
@@ -221,7 +201,7 @@
     "# Initialize the PDFConverter with the existing PDF and injection items\n",
     "pdf_converter = PDFConverter(\n",
     "    prompt_template=None,  # No template provided\n",
-    "    font_type=\"Arial\",\n",
+    "    font_type=\"Helvetica\",\n",
     "    font_size=12,\n",
     "    page_width=210,\n",
     "    page_height=297,\n",
@@ -263,16 +243,8 @@
    "cell_metadata_filter": "-all"
   },
   "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
    "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.12.8"
+   "version": "3.11.9"
   }
  },
  "nbformat": 4,

doc/code/converters/pdf_converter.py

@@ -121,11 +121,10 @@
 await orchestrator.send_prompts_async(prompt_list=prompts)  # type: ignore
 await orchestrator.print_conversations_async()  # type: ignore
 
-import tempfile
-
 # %% [markdown]
 # # Modify Existing PDF with Injection Items
 # %%
+import tempfile
 from pathlib import Path
 
 import requests