From 7d9ea152cb164f371ac6d468fdfa31f60a4e137a Mon Sep 17 00:00:00 2001 From: hupe1980 Date: Mon, 29 Apr 2024 22:57:25 +0200 Subject: [PATCH] Update docs --- docs/scanner.ipynb | 53 +++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 52 insertions(+), 1 deletion(-) diff --git a/docs/scanner.ipynb b/docs/scanner.ipynb index c1dd246..76460d8 100644 --- a/docs/scanner.ipynb +++ b/docs/scanner.ipynb @@ -42,7 +42,58 @@ "cell_type": "markdown", "metadata": {}, "source": [ - "### Extensibility with Plugins\n", + "### ImageMarkdownInjectionPlugin" + ] + }, + { + "cell_type": "markdown", + "metadata": {}, + "source": [ + "### ManyShotPlugin" + ] + }, + { + "cell_type": "markdown", + "metadata": {}, + "source": [ + "### PromptInjectionPlugin" + ] + }, + { + "cell_type": "markdown", + "metadata": {}, + "source": [ + "### RepeatedTokenPlugin" + ] + }, + { + "cell_type": "markdown", + "metadata": {}, + "source": [ + "### SelfSimilarityPlugin" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "metadata": {}, + "outputs": [], + "source": [ + "from aisploit.scanner.plugins import SelfSimilarityPlugin\n", + "\n", + "plugin = SelfSimilarityPlugin(\n", + " questions=[\n", + " \"What's the name of the Voodoo Lady in the Monkey Island games?\"\n", + " ],\n", + " num_samples=1,\n", + ")" + ] + }, + { + "cell_type": "markdown", + "metadata": {}, + "source": [ + "### Extensibility with custom Plugins\n", "\n", "The scanner is extensible with plugins, which allow users to add custom scanning capabilities tailored to their requirements. Below is an example of how to create a custom plugin:" ]