Skip to content

Commit 4c6693a

Browse files
committed
ci: Switch vllm config to qwen3
It preforms better in tool calling and structured tests Signed-off-by: Derek Higgins <[email protected]>
1 parent e2e92d2 commit 4c6693a

File tree

3 files changed

+6
-10
lines changed

3 files changed

+6
-10
lines changed

.github/actions/setup-vllm/action.yml

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,14 @@ runs:
1111
--name vllm \
1212
-p 8000:8000 \
1313
--privileged=true \
14-
quay.io/higginsd/vllm-cpu:65393ee064 \
14+
quay.io/higginsd/vllm-cpu:65393ee064-qwen3 \
1515
--host 0.0.0.0 \
1616
--port 8000 \
1717
--enable-auto-tool-choice \
18-
--tool-call-parser llama3_json \
19-
--model /root/.cache/Llama-3.2-1B-Instruct \
20-
--served-model-name meta-llama/Llama-3.2-1B-Instruct
18+
--tool-call-parser hermes \
19+
--model /root/.cache/Qwen3-0.6B \
20+
--served-model-name Qwen/Qwen3-0.6B \
21+
--max-model-len 8192
2122
2223
# Wait for vllm to be ready
2324
echo "Waiting for vllm to be ready..."

scripts/integration-tests.sh

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -212,11 +212,6 @@ fi
212212
echo "=== Running Integration Tests ==="
213213
EXCLUDE_TESTS="builtin_tool or safety_with_image or code_interpreter or test_rag"
214214

215-
# Additional exclusions for vllm setup
216-
if [[ "$TEST_SETUP" == "vllm" ]]; then
217-
EXCLUDE_TESTS="${EXCLUDE_TESTS} or test_inference_store_tool_calls"
218-
fi
219-
220215
PYTEST_PATTERN="not( $EXCLUDE_TESTS )"
221216
if [[ -n "$TEST_PATTERN" ]]; then
222217
PYTEST_PATTERN="${PYTEST_PATTERN} and $TEST_PATTERN"

tests/integration/suites.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ class Setup(BaseModel):
7878
"VLLM_URL": "http://localhost:8000/v1",
7979
},
8080
defaults={
81-
"text_model": "vllm/meta-llama/Llama-3.2-1B-Instruct",
81+
"text_model": "vllm/Qwen/Qwen3-0.6B",
8282
"embedding_model": "sentence-transformers/all-MiniLM-L6-v2",
8383
},
8484
),

0 commit comments

Comments
 (0)