Skip to content

Commit 91a3219

Browse files
authored
reversed ordering of context analysis roles (#60)
Signed-off-by: Shonda-Adena-Witherspoon <[email protected]> Co-authored-by: Shonda-Adena-Witherspoon <[email protected]>
1 parent a862480 commit 91a3219

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

tests/generative_detectors/test_granite_guardian.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -485,10 +485,10 @@ def test_request_to_chat_completion_request_prompt_analysis(granite_guardian_det
485485
)
486486
assert type(chat_request) == ChatCompletionRequest
487487
assert len(chat_request.messages) == 2
488-
assert chat_request.messages[0]["role"] == "user"
489-
assert chat_request.messages[0]["content"] == CONTENT
490-
assert chat_request.messages[1]["role"] == "context"
491-
assert chat_request.messages[1]["content"] == "extra! " + CONTEXT_DOC
488+
assert chat_request.messages[0]["role"] == "context"
489+
assert chat_request.messages[0]["content"] == "extra! " + CONTEXT_DOC
490+
assert chat_request.messages[1]["role"] == "user"
491+
assert chat_request.messages[1]["content"] == CONTENT
492492
assert chat_request.model == MODEL_NAME
493493
# detector_paramas
494494
assert chat_request.n == 2

vllm_detector_adapter/generative_detectors/granite_guardian.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -282,8 +282,8 @@ def _request_to_chat_completion_request(
282282
elif risk_name in self.PROMPT_CONTEXT_ANALYSIS_RISKS:
283283
# Prompt analysis
284284
messages = [
285-
{"role": "user", "content": content},
286285
{"role": "context", "content": context_text},
286+
{"role": "user", "content": content},
287287
]
288288
else:
289289
# Return error if risk names are not expected ones

0 commit comments

Comments
 (0)