diff --git a/.github/workflows/shared-test-trtllm-nightly-image.yml b/.github/workflows/shared-test-trtllm-nightly-image.yml
index 7409afc5f7bb..0382a42d7ee6 100644
--- a/.github/workflows/shared-test-trtllm-nightly-image.yml
+++ b/.github/workflows/shared-test-trtllm-nightly-image.yml
@@ -27,7 +27,9 @@ jobs:
steps:
- uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
- name: Test nightly selection and media harness
- run: python3 -m unittest discover -s lib/sidecar/trtllm/ci/tests -v
+ run: |
+ python3 -m pip install 'PyYAML==6.0.3'
+ python3 -m unittest discover -s lib/sidecar/trtllm/ci/tests -v
- name: Resolve latest upstream nightly
id: config
run: >-
@@ -61,7 +63,7 @@ jobs:
strategy:
fail-fast: false
matrix:
- media: [image, video]
+ media: [image, video, audio]
needs: build
runs-on: [gpu-1, amd64, v2, us-west-2]
timeout-minutes: 30
diff --git a/lib/sidecar/testkit/tests/support/trtllm.rs b/lib/sidecar/testkit/tests/support/trtllm.rs
index 00f030d9d297..4fcb88c8189d 100644
--- a/lib/sidecar/testkit/tests/support/trtllm.rs
+++ b/lib/sidecar/testkit/tests/support/trtllm.rs
@@ -42,9 +42,7 @@ pub struct Discovery {
pub calls: AtomicUsize,
pub unavailable: AtomicBool,
pub hold: AtomicBool,
- pub portable_multimodal: AtomicBool,
- pub advertises_image: AtomicBool,
- pub advertises_video: AtomicBool,
+ pub model_info: Mutex