fix config issues for shields

Author: radofuchs

run.yaml

@@ -134,6 +134,7 @@ server:
 shields: 
   - shield_id: llama-guard-shield
     provider_id: llama-guard
+    provider_shield_id: "gpt-3.5-turbo"  # Model to use for safety checks
 vector_dbs:
   - vector_db_id: my_knowledge_base
     embedding_model: sentence-transformers/all-mpnet-base-v2

tests/e2e/configs/run-azure.yaml

@@ -123,6 +123,7 @@ server:
 shields: 
   - shield_id: llama-guard-shield
     provider_id: llama-guard
+    provider_shield_id: "gpt-4o-mini"
 models:
   - model_id: gpt-4o-mini
     model_type: llm

tests/e2e/configs/run-ci.yaml

@@ -131,9 +131,10 @@ server:
   tls_cafile: null
   tls_certfile: null
   tls_keyfile: null
-shields:
+shields: 
   - shield_id: llama-guard-shield
     provider_id: llama-guard
+    provider_shield_id: "gpt-4-turbo"
 vector_dbs:
   - vector_db_id: my_knowledge_base
     embedding_model: sentence-transformers/all-mpnet-base-v2

tests/e2e/configs/run-rhaiis.yaml

@@ -126,6 +126,7 @@ server:
 shields: 
   - shield_id: llama-guard-shield
     provider_id: llama-guard
+    provider_shield_id: "meta-llama/Llama-3.1-8B-Instruct"
 models:
   - metadata:
       embedding_dimension: 768 # Depends on chosen model