From 73a0a34e392d0ce6a5de8f935bbeb023a89a99a4 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Tue, 8 Oct 2024 17:47:03 -0700
Subject: [PATCH] Kill non-llama guard shields

---
 .../templates/docker/llamastack-local-cpu/run.yaml   | 12 ------------
 .../templates/docker/llamastack-local-gpu/run.yaml   | 12 ------------
 tests/examples/local-run.yaml                        | 12 ------------
 3 files changed, 36 deletions(-)

diff --git a/llama_stack/distribution/templates/docker/llamastack-local-cpu/run.yaml b/llama_stack/distribution/templates/docker/llamastack-local-cpu/run.yaml
index 1efa28cdc..3a7514147 100644
--- a/llama_stack/distribution/templates/docker/llamastack-local-cpu/run.yaml
+++ b/llama_stack/distribution/templates/docker/llamastack-local-cpu/run.yaml
@@ -49,18 +49,6 @@ shields:
   type: llama_guard
   provider_id: meta-reference
   params: {}
-- identifier: code_scanner_guard
-  type: llama_guard
-  provider_id: meta-reference
-  params: {}
-- identifier: injection_shield
-  type: llama_guard
-  provider_id: meta-reference
-  params: {}
-- identifier: jailbreak_shield
-  type: llama_guard
-  provider_id: meta-reference
-  params: {}
 memory_banks:
 - identifier: vector
   provider_id: meta-reference
diff --git a/llama_stack/distribution/templates/docker/llamastack-local-gpu/run.yaml b/llama_stack/distribution/templates/docker/llamastack-local-gpu/run.yaml
index 949e78eda..3b9cd6866 100644
--- a/llama_stack/distribution/templates/docker/llamastack-local-gpu/run.yaml
+++ b/llama_stack/distribution/templates/docker/llamastack-local-gpu/run.yaml
@@ -52,18 +52,6 @@ shields:
   type: llama_guard
   provider_id: meta-reference
   params: {}
-- identifier: code_scanner_guard
-  type: llama_guard
-  provider_id: meta-reference
-  params: {}
-- identifier: injection_shield
-  type: llama_guard
-  provider_id: meta-reference
-  params: {}
-- identifier: jailbreak_shield
-  type: llama_guard
-  provider_id: meta-reference
-  params: {}
 memory_banks:
 - identifier: vector
   provider_id: meta-reference
diff --git a/tests/examples/local-run.yaml b/tests/examples/local-run.yaml
index 365cbb7c6..06006bcb5 100644
--- a/tests/examples/local-run.yaml
+++ b/tests/examples/local-run.yaml
@@ -57,18 +57,6 @@ shields:
   type: llama_guard
   provider_id: meta-reference
   params: {}
-- identifier: code_scanner_guard
-  type: llama_guard
-  provider_id: meta-reference
-  params: {}
-- identifier: injection_shield
-  type: llama_guard
-  provider_id: meta-reference
-  params: {}
-- identifier: jailbreak_shield
-  type: llama_guard
-  provider_id: meta-reference
-  params: {}
 memory_banks:
 - identifier: vector
   provider_id: meta-reference