test: Add granite-embedding-125m model to integration tests (#31)

nathan-weinberg · web-flow · commit db3c99b08d88 · 2025-09-15T08:57:56.000-04:00
Enabling granite-embedding-125m will allow the embedding tests to run.

&lt;!-- This is an auto-generated comment: release notes by coderabbit.ai
--&gt;
## Summary by CodeRabbit

* **Tests**
* Updated integration test script to include an additional
embedding-model parameter alongside the existing text-model parameter,
ensuring integration tests exercise embedding behavior as well as text
handling.
&lt;!-- end of auto-generated comment: release notes by coderabbit.ai --&gt;
diff --git a/tests/run_integration_tests.sh b/tests/run_integration_tests.sh
@@ -53,6 +53,7 @@ function run_integration_tests() {
     uv run pytest -s -v tests/integration/inference/ \
         --stack-config=server:"$STACK_CONFIG_PATH" \
         --text-model=vllm-inference/"$INFERENCE_MODEL" \
+        --embedding-model=granite-embedding-125m \
         -k "not ($SKIP_TESTS)"
 }
 

Original file line number	Diff line number	Diff line change
`@@ -53,6 +53,7 @@ function run_integration_tests() {`
`53`	`53`	`uv run pytest -s -v tests/integration/inference/ \`
`54`	`54`	`--stack-config=server:"$STACK_CONFIG_PATH" \`
`55`	`55`	`--text-model=vllm-inference/"$INFERENCE_MODEL" \`
	`56`	`+ --embedding-model=granite-embedding-125m \`
`56`	`57`	`-k "not ($SKIP_TESTS)"`
`57`	`58`	`}`
`58`	`59`