From eef61e2b26ca29f8bfdb7c2ac0feff27671d3b98 Mon Sep 17 00:00:00 2001 From: Mustafa Elbehery Date: Tue, 12 Aug 2025 09:17:37 +0200 Subject: [PATCH] fix: instantiate abstract PromptGuardSafetyImpl class Signed-off-by: Mustafa Elbehery --- .../providers/inline/safety/prompt_guard/prompt_guard.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/llama_stack/providers/inline/safety/prompt_guard/prompt_guard.py b/llama_stack/providers/inline/safety/prompt_guard/prompt_guard.py index e11ec5cf5..801500dee 100644 --- a/llama_stack/providers/inline/safety/prompt_guard/prompt_guard.py +++ b/llama_stack/providers/inline/safety/prompt_guard/prompt_guard.py @@ -64,6 +64,9 @@ class PromptGuardSafetyImpl(Safety, ShieldsProtocolPrivate): return await self.shield.run(messages) + async def run_moderation(self, input: str | list[str], model: str): + raise NotImplementedError("run_moderation not implemented for PromptGuard") + class PromptGuardShield: def __init__(