fixes

scripts/integration-tests.sh

@@ -141,6 +141,10 @@ fi
 
 # Apply setup-specific environment variables (needed for server startup and tests)
 echo "=== Applying Setup Environment Variables ==="
+
+# the server needs this
+export LLAMA_STACK_INFERENCE_MODE=$INFERENCE_MODE
+
 SETUP_ENV=$(PYTHONPATH=$THIS_DIR/.. python "$THIS_DIR/get_setup_env.py" --suite "$TEST_SUITE" --setup "$TEST_SETUP" --format bash)
 echo "Setting up environment variables:"
 echo "$SETUP_ENV"

tests/integration/suites.py

@@ -51,6 +51,7 @@ SETUP_DEFINITIONS: dict[str, Setup] = {
         description="Local Ollama provider with text + safety models",
         env={
             "OLLAMA_URL": "http://0.0.0.0:11434",
+            "SAFETY_MODEL": "ollama/llama-guard3:1b",
         },
         defaults={
             "text_model": "ollama/llama3.2:3b-instruct-fp16",