Refactor

README.md

@@ -23,7 +23,7 @@ from typing import Any
 import textwrap
 from aisploit.core import BaseCallbackHandler, BasePromptValue, Score, Response
 from aisploit.model import ChatOpenAI
-from aisploit.redteam import RedTeamJob, RedTeamClassifierTask
+from aisploit.red_team import RedTeamJob, RedTeamClassifierTask
 from aisploit.target import target
 from aisploit.demo import GandalfBot, GandalfLevel, GandalfScorer
 

aisploit/target/langchain.py

@@ -19,4 +19,4 @@ def send_prompt(self, prompt: BasePromptValue) -> Response:
             parser = StrOutputParser()
             return Response(content=parser.invoke(response))
 
-        raise ValueError(f"Unsupported cache value {type(response)}")
+        raise ValueError(f"Unsupported response value {type(response)}")

examples/gandalf.ipynb

@@ -22,7 +22,7 @@
     "from dotenv import load_dotenv\n",
     "from aisploit.core import BaseCallbackHandler, BasePromptValue, Score, Response\n",
     "from aisploit.model import ChatOpenAI\n",
-    "from aisploit.redteam import RedTeamJob, RedTeamClassifierTask\n",
+    "from aisploit.red_team import RedTeamJob, RedTeamClassifierTask\n",
     "from aisploit.target import target\n",
     "from aisploit.demo import GandalfBot, GandalfLevel, GandalfScorer\n",
     "\n",

tests/redteam/test_report.py → tests/red_team/test_report.py

@@ -1,7 +1,7 @@
 import pytest
 
 from aisploit.core import BaseReport, Score
-from aisploit.redteam import RedTeamReport, RedTeamReportEntry
+from aisploit.red_team import RedTeamReport, RedTeamReportEntry
 
 
 @pytest.fixture