fix safety using inference (#99)

llama_stack/providers/impls/meta_reference/safety/shields/llama_guard.py

@@ -18,6 +18,7 @@ from transformers import (
 )
 
 from .base import CANNED_RESPONSE_TEXT, OnViolationAction, ShieldBase, ShieldResponse
+from llama_models.llama3.api.datatypes import Message, Role
 
 
 SAFE_RESPONSE = "safe"

llama_stack/providers/registry/safety.py

@@ -28,6 +28,9 @@ def available_providers() -> List[ProviderSpec]:
             ],
             module="llama_stack.providers.impls.meta_reference.safety",
             config_class="llama_stack.providers.impls.meta_reference.safety.SafetyConfig",
+            api_dependencies=[
+                Api.inference,
+            ],
         ),
         remote_provider_spec(
             api=Api.safety,