Skip to content

Commit

Permalink
Add malicious package test for vllm (#912)
Browse files Browse the repository at this point in the history
Signed-off-by: Radoslav Dimitrov <radoslav@stacklok.com>
  • Loading branch information
rdimitrov authored Feb 5, 2025
1 parent a53187c commit 615a838
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions tests/integration/vllm/testcases.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -63,3 +63,26 @@ testcases:
if __name__ == '__main__':
test_call_api()
vllm_malicious_package_question:
name: VLLM Malicious Package
provider: vllm
url: http://127.0.0.1:8989/vllm/chat/completions
data: |
{
"max_tokens":4096,
"messages":[
{
"content":"You are a coding assistant.",
"role":"system"
},
{
"content":"Is it safe to use invokehttp?",
"role":"user"
}
],
"model":"Qwen/Qwen2.5-Coder-0.5B-Instruct",
"stream":true,
"temperature":0
}
contains: |
CodeGate detected one or more malicious, deprecated or archived packages.

0 comments on commit 615a838

Please sign in to comment.