Skip to content

Commit 1005ca9

Browse files
committed
Fix default plugins
1 parent 85322dc commit 1005ca9

File tree

2 files changed

+17
-32
lines changed

2 files changed

+17
-32
lines changed

aisploit/scanner/job.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2,22 +2,15 @@
22
from typing import List, Optional, Sequence
33

44
from .plugin import Plugin
5-
from .plugins import ImageMarkdownInjectionPlugin, PromptInjectionPlugin
5+
from .plugins import PromptInjectionPlugin
66
from .report import Issue, ScanReport
77
from ..core import BaseJob, BaseTarget, CallbackManager, Callbacks
88

99

1010
@dataclass
1111
class ScannerJob(BaseJob):
1212
target: BaseTarget
13-
plugins: Sequence[Plugin] = field(
14-
default_factory=lambda: [
15-
PromptInjectionPlugin(name="prompt_injection"),
16-
ImageMarkdownInjectionPlugin(
17-
name="image_markdown_injection", domain="cxd47vgx2z2qyzr637trlgzogfm6ayyn.oastify.com"
18-
),
19-
]
20-
)
13+
plugins: Sequence[Plugin] = field(default_factory=lambda: [PromptInjectionPlugin(name="prompt_injection")])
2114
callbacks: Callbacks = field(default_factory=list)
2215

2316
def execute(self, *, run_id: Optional[str] = None, tags: Optional[Sequence[str]] = None) -> ScanReport:

0 commit comments

Comments
 (0)