From 1152356a3abcac73cd8a7a88dc0f2c6c774be080 Mon Sep 17 00:00:00 2001 From: Kaushik Date: Mon, 17 Feb 2025 13:45:48 +0530 Subject: [PATCH] add comment describing the fiddlecube adapter --- .../providers/remote/safety/fiddlecube/fiddlecube.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/llama_stack/providers/remote/safety/fiddlecube/fiddlecube.py b/llama_stack/providers/remote/safety/fiddlecube/fiddlecube.py index 7a4faea2e..821e66d63 100644 --- a/llama_stack/providers/remote/safety/fiddlecube/fiddlecube.py +++ b/llama_stack/providers/remote/safety/fiddlecube/fiddlecube.py @@ -27,6 +27,14 @@ logger = logging.getLogger(__name__) class FiddlecubeSafetyAdapter(Safety, ShieldsProtocolPrivate): + """ + Implementation of the Fiddlecube Safety API. + + Accepts a list of messages for content moderation. + Optionally, a list of `excluded_categories` can be provided to exclude certain categories from moderation. + Category name strings map to the llama-guard risk taxonomy. + """ + def __init__(self, config: FiddlecubeSafetyConfig) -> None: self.config = config self.registered_shields = []