Fix san report

hupe1980 · Apr 10, 2024 · d0c52b3 · d0c52b3
1 parent 92137c7
commit d0c52b3
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/aisploit/scanner/templates/report.md b/aisploit/scanner/templates/report.md
@@ -13,7 +13,7 @@ No issues!
 | Prompt | Converter | Response | RTT (seconds) |
 |--------|-----------|----------|---------------|
 {% for issue in issues -%}
-|{{ issue.send_report_entry.prompt }}|{{ issue.send_report_entry.converter }}|{{ issue.send_report_entry.response }}| {{ issue.send_report_entry.round_trip_time }} |
+|{{ issue.send_report_entry.prompt }}|{{ issue.send_report_entry.converter }}|{{ issue.send_report_entry.response|replace('\n', '<br>') }}| {{ issue.send_report_entry.round_trip_time }} |
 {% endfor %}
 {% endfor %}
 {% endif %}
diff --git a/aisploit/sender/job.py b/aisploit/sender/job.py
@@ -53,7 +53,7 @@ def _send_prompt(
         self, prompt: BasePromptValue, converter: Optional[BaseConverter] = None
     ) -> SendReportEntry:
         start_time = datetime.now()
-        response = self._target.send_prompt(prompt)
+        response = self._target.send_prompt(prompt).strip()
         end_time = datetime.now()
 
         return SendReportEntry(