Skip to content
This repository was archived by the owner on Jun 5, 2025. It is now read-only.

Commit 615a838

Browse files
authored
Add malicious package test for vllm (#912)
Signed-off-by: Radoslav Dimitrov <[email protected]>
1 parent a53187c commit 615a838

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

tests/integration/vllm/testcases.yaml

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,3 +63,26 @@ testcases:
6363
if __name__ == '__main__':
6464
test_call_api()
6565
66+
vllm_malicious_package_question:
67+
name: VLLM Malicious Package
68+
provider: vllm
69+
url: http://127.0.0.1:8989/vllm/chat/completions
70+
data: |
71+
{
72+
"max_tokens":4096,
73+
"messages":[
74+
{
75+
"content":"You are a coding assistant.",
76+
"role":"system"
77+
},
78+
{
79+
"content":"Is it safe to use invokehttp?",
80+
"role":"user"
81+
}
82+
],
83+
"model":"Qwen/Qwen2.5-Coder-0.5B-Instruct",
84+
"stream":true,
85+
"temperature":0
86+
}
87+
contains: |
88+
CodeGate detected one or more malicious, deprecated or archived packages.

0 commit comments

Comments
 (0)