Consolidating Safety tests from various places under client-sdk

llama_stack/providers/inline/safety/code_scanner/__init__.py

@@ -4,10 +4,10 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
-from .config import CodeShieldConfig
+from .config import CodeScannerConfig
 
 
-async def get_provider_impl(config: CodeShieldConfig, deps):
+async def get_provider_impl(config: CodeScannerConfig, deps):
     from .code_scanner import MetaReferenceCodeScannerSafetyImpl
 
     impl = MetaReferenceCodeScannerSafetyImpl(config, deps)

llama_stack/templates/ollama/run-with-safety.yaml

@@ -32,6 +32,9 @@ providers:
   - provider_id: llama-guard
     provider_type: inline::llama-guard
     config: {}
+  - provider_id: code-scanner
+    provider_type: inline::code-scanner
+    config: {}
   agents:
   - provider_id: meta-reference
     provider_type: inline::meta-reference
@@ -93,7 +96,11 @@ models:
 shields:
 - params: null
   shield_id: ${env.SAFETY_MODEL}
-  provider_id: null
+  provider_id: llama-guard
+  provider_shield_id: null
+- params: null
+  shield_id: CodeScanner
+  provider_id: code-scanner
   provider_shield_id: null
 memory_banks: []
 datasets: []

llama_stack/templates/together/run-with-safety.yaml

@@ -0,0 +1,146 @@
+version: '2'
+image_name: together
+docker_image: null
+conda_env: together
+apis:
+- agents
+- datasetio
+- eval
+- inference
+- memory
+- safety
+- scoring
+- telemetry
+providers:
+  inference:
+  - provider_id: together
+    provider_type: remote::together
+    config:
+      url: https://api.together.xyz/v1
+      api_key: ${env.TOGETHER_API_KEY}
+  - provider_id: sentence-transformers
+    provider_type: inline::sentence-transformers
+    config: {}
+  memory:
+  - provider_id: faiss
+    provider_type: inline::faiss
+    config:
+      kvstore:
+        type: sqlite
+        namespace: null
+        db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/together}/faiss_store.db
+  safety:
+  - provider_id: llama-guard
+    provider_type: inline::llama-guard
+    config: {}
+  - provider_id: llama-guard-vision
+    provider_type: inline::llama-guard
+    config: {}
+  - provider_id: code-scanner
+    provider_type: inline::code-scanner
+    config: {}
+  agents:
+  - provider_id: meta-reference
+    provider_type: inline::meta-reference
+    config:
+      persistence_store:
+        type: sqlite
+        namespace: null
+        db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/together}/agents_store.db
+  telemetry:
+  - provider_id: meta-reference
+    provider_type: inline::meta-reference
+    config:
+      service_name: ${env.OTEL_SERVICE_NAME:llama-stack}
+      sinks: ${env.TELEMETRY_SINKS:console,sqlite}
+      sqlite_db_path: ${env.SQLITE_DB_PATH:~/.llama/distributions/together/trace_store.db}
+  eval:
+  - provider_id: meta-reference
+    provider_type: inline::meta-reference
+    config: {}
+  datasetio:
+  - provider_id: huggingface
+    provider_type: remote::huggingface
+    config: {}
+  - provider_id: localfs
+    provider_type: inline::localfs
+    config: {}
+  scoring:
+  - provider_id: basic
+    provider_type: inline::basic
+    config: {}
+  - provider_id: llm-as-judge
+    provider_type: inline::llm-as-judge
+    config: {}
+  - provider_id: braintrust
+    provider_type: inline::braintrust
+    config:
+      openai_api_key: ${env.OPENAI_API_KEY:}
+metadata_store:
+  namespace: null
+  type: sqlite
+  db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/together}/registry.db
+models:
+- metadata: {}
+  model_id: meta-llama/Llama-3.1-8B-Instruct
+  provider_id: together
+  provider_model_id: meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo
+  model_type: llm
+- metadata: {}
+  model_id: meta-llama/Llama-3.1-70B-Instruct
+  provider_id: together
+  provider_model_id: meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo
+  model_type: llm
+- metadata: {}
+  model_id: meta-llama/Llama-3.1-405B-Instruct-FP8
+  provider_id: together
+  provider_model_id: meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo
+  model_type: llm
+- metadata: {}
+  model_id: meta-llama/Llama-3.2-3B-Instruct
+  provider_id: together
+  provider_model_id: meta-llama/Llama-3.2-3B-Instruct-Turbo
+  model_type: llm
+- metadata: {}
+  model_id: meta-llama/Llama-3.2-11B-Vision-Instruct
+  provider_id: together
+  provider_model_id: meta-llama/Llama-3.2-11B-Vision-Instruct-Turbo
+  model_type: llm
+- metadata: {}
+  model_id: meta-llama/Llama-3.2-90B-Vision-Instruct
+  provider_id: together
+  provider_model_id: meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo
+  model_type: llm
+- metadata: {}
+  model_id: meta-llama/Llama-Guard-3-8B
+  provider_id: together
+  provider_model_id: meta-llama/Meta-Llama-Guard-3-8B
+  model_type: llm
+- metadata: {}
+  model_id: meta-llama/Llama-Guard-3-11B-Vision
+  provider_id: together
+  provider_model_id: meta-llama/Llama-Guard-3-11B-Vision-Turbo
+  model_type: llm
+- metadata:
+    embedding_dimension: 384
+  model_id: all-MiniLM-L6-v2
+  provider_id: sentence-transformers
+  provider_model_id: null
+  model_type: embedding
+shields:
+- params: null
+  shield_id: meta-llama/Llama-Guard-3-8B
+  provider_id: llama-guard
+  provider_shield_id: null
+- params: null
+  shield_id: meta-llama/Llama-Guard-3-11B-Vision
+  provider_id: llama-guard-vision
+  provider_shield_id: null
+- params: null
+  shield_id: CodeScanner
+  provider_id: code-scanner
+  provider_shield_id: null
+memory_banks: []
+datasets: []
+scoring_fns: []
+eval_tasks: []

tests/client-sdk/safety/run_tests.md

@@ -0,0 +1,33 @@
+# Test with Llama Stack as Library using Ollama
+=
+```
+LLAMA_STACK_CONFIG=llama_stack/templates/ollama/run-with-safety.yaml pytest tests/client-sdk/safety -v
+```
+
+# Test with Llama Stack as Library using Together
+=
+```
+export TOGETHER_API_KEY={your_api_key}
+LLAMA_STACK_CONFIG=llama_stack/templates/together/run-with-safety.yaml pytest tests/client-sdk/safety -v
+```
+
+# Test against a local Llama Stack server instance
+```
+# Export Llama Stack naming vars
+export SAFETY_MODEL=meta-llama/Llama-Guard-3-1B
+export INFERENCE_MODEL="meta-llama/Llama-3.2-3B-Instruct"
+
+# Export Ollama naming vars
+export OLLAMA_INFERENCE_MODEL="llama3.2:3b-instruct-fp16"
+export OLLAMA_SAFETY_MODEL="llama-guard3:1b"
+
+# Start Ollama instance
+ollama run $OLLAMA_INFERENCE_MODEL --keepalive 60m
+ollama run $OLLAMA_SAFETY_MODEL --keepalive 60m
+
+# Start the Llama Stack server
+llama stack run ./llama_stack/templates/ollama/run-with-safety.yaml
+
+# Run the tests
+LLAMA_STACK_BASE_URL=http://localhost:5000 pytest tests/client-sdk/safety -v
+```