Skip to content

Commit 9b267c0

Browse files
BrandonBuiBrandon Bui
andauthored
Switch orchestrators for executors (#8)
* Change deployment_name to model_name and generalize chat target initialization * swapped Prompt Sending Orchestrator with Prompt Sending Attack --------- Co-authored-by: Brandon Bui <brandonbui@Brandons-MacBook-Pro.local>
1 parent 25eed7c commit 9b267c0

1 file changed

Lines changed: 10 additions & 8 deletions

File tree

pyritship/app.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,11 @@
44
import os
55
import inspect
66
import importlib
7-
from pyrit.common import default_values, initialize_pyrit, IN_MEMORY
7+
from pyrit.common import default_values
8+
from pyrit.setup import initialize_pyrit_async, IN_MEMORY
89
from pyrit.prompt_converter import PromptConverter
910
from pyrit.prompt_target import OpenAIChatTarget
10-
from pyrit.orchestrator import PromptSendingOrchestrator
11+
from pyrit.executor.attack import PromptSendingAttack, MarkdownAttackResultPrinter
1112
from pyrit.score import SelfAskTrueFalseScorer
1213
from dotenv import load_dotenv
1314

@@ -52,22 +53,23 @@ def convert(converter_name:str):
5253

5354
@app.route('/prompt/generate', methods=['POST'])
5455
def generate_prompt():
55-
# Initialize AOAI GPT-4o target
56+
# Initialize chat target
5657
global chat_target
5758
if (chat_target is None):
5859
chat_target = initialize_chat_target()
5960

60-
promptSendingOrchestrator = PromptSendingOrchestrator(objective_target=chat_target)
61+
prompt_sending_attack = PromptSendingAttack(objective_target=chat_target)
6162
# Extract input data from json payload
6263
data = request.get_json()
6364
prompt_goal = data['prompt_goal']
6465

65-
generated_prompt = asyncio.run(promptSendingOrchestrator.send_prompts_async(prompt_list=[prompt_goal]))[0].request_pieces[0].converted_value
66-
return jsonify({"prompt": generated_prompt})
66+
result = asyncio.run(prompt_sending_attack.execute_async(objective=prompt_goal)).last_response.converted_value
67+
68+
return jsonify({"prompt": result})
6769

6870
@app.route('/prompt/score/SelfAskTrueFalseScorer', methods=['POST'])
6971
def score():
70-
# Initialize AOAI GPT-4o target
72+
# Initialize chat target
7173
global chat_target
7274
if (chat_target is None):
7375
chat_target = initialize_chat_target()
@@ -97,7 +99,7 @@ def score():
9799
)
98100

99101
def initialize_chat_target():
100-
initialize_pyrit(memory_db_type=IN_MEMORY)
102+
asyncio.run(initialize_pyrit_async(memory_db_type=IN_MEMORY))
101103

102104
chat_target = OpenAIChatTarget(
103105
model_name=os.environ.get("OPENAI_CHAT_MODEL_NAME"),

0 commit comments

Comments
 (0)