File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed
tests/generative_detectors
vllm_detector_adapter/generative_detectors Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments