Skip to content

Commit b6f4e6c

Browse files
makeshnPouyanpi
andauthored
feat: add reasoning guardrail connector (NVIDIA-NeMo#1565)
* add reasoning guardrail connector Signed-off-by: Makesh Sreedhar <makeshn@nvidia.com> * update reasoning guardrail prompt Signed-off-by: Makesh Sreedhar <makeshn@nvidia.com> * feat(content_safety): add reasoning_enabled to prompt ctx Include the `reasoning_enabled` flag from the content safety config in the context passed to both input and output content safety check prompts. This enables prompt templates to conditionally adjust behavior based on the reasoning setting. Removes redundant config exposure in LLMTaskManager's default context. * revert streamers.py change * test(content_safety): add Nemotron reasoning parser tests * Revert "revert streamers.py change" This reverts commit e916dd4. * fix: add type ignore for streamer import * minor updates to docs, example Signed-off-by: Makesh Sreedhar <makeshn@nvidia.com> --------- Signed-off-by: Makesh Sreedhar <makeshn@nvidia.com> Co-authored-by: Pouyanpi <13303554+Pouyanpi@users.noreply.github.com>
1 parent 8e8c053 commit b6f4e6c

13 files changed

Lines changed: 1172 additions & 3 deletions

File tree

docs/getting-started/tutorials/index.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,13 @@ This section contains tutorials that help you get started with the NeMo Guardrai
2525
Check text inputs and outputs for harmful content using Nemotron Content Safety NIM.
2626
:::
2727

28+
:::{grid-item-card} Enforce Custom Safety and Dialogue Policies
29+
:link: nemotron-content-safety-reasoning-deployment
30+
:link-type: doc
31+
32+
Enforce customizable content safety and dialogue policies for your use-case by using our reasoning guard model Nemotron-Content-Safety-Reasoning-4B.
33+
:::
34+
2835
:::{grid-item-card} Restrict Topics
2936
:link: nemoguard-topiccontrol-deployment
3037
:link-type: doc
@@ -53,6 +60,7 @@ Add safety checks to images and text using a vision model as LLM-as-a-Judge.
5360
:maxdepth: 2
5461
5562
Check Harmful Content <nemotron-safety-guard-deployment>
63+
Content Safety Reasoning <nemotron-content-safety-reasoning-deployment>
5664
Restrict Topics <nemoguard-topiccontrol-deployment>
5765
Detect Jailbreak Attempts <nemoguard-jailbreakdetect-deployment>
5866
Add Multimodal Content Safety <multimodal>

0 commit comments

Comments
 (0)