Output filtering and safety checks in Agentic AI - ML Experiment: Train & Evaluate

Experiment - Output filtering and safety checks

Problem:You have an AI agent that generates text outputs. Sometimes, the outputs contain unsafe or inappropriate content. This can cause harm or violate usage policies.

Current Metrics:Safety violation rate: 15% of outputs contain unsafe content. User satisfaction: 70%.

Issue:The AI agent produces unsafe outputs too often, reducing trust and usability.

Your Task

Reduce the safety violation rate to below 5% while maintaining user satisfaction above 65%.

You cannot retrain the AI model from scratch.

You can only add output filtering and safety checks after the model generates text.

The filtering must run efficiently to keep response time under 1 second.

Hint 1

Hint 2

Hint 3

Hint 4

Solution

Agentic AI

import re

# List of unsafe keywords (example)
unsafe_keywords = ['hate', 'kill', 'bomb', 'terror']

# Simple function to check for unsafe content
# Returns True if unsafe content detected

def is_unsafe(text):
    pattern = re.compile('|'.join(map(re.escape, unsafe_keywords)), re.IGNORECASE)
    return bool(pattern.search(text))

# Function to filter output
# If unsafe, replace with safe fallback

def filter_output(text):
    if is_unsafe(text):
        return "[Content removed due to safety concerns.]"
    return text

# Example usage
outputs = [
    "I love peaceful discussions.",
    "We should kill the problem quickly.",
    "Let's plan a bomb for the event.",
    "Have a nice day!"
]

filtered_outputs = [filter_output(output) for output in outputs]

print(filtered_outputs)

Added a list of unsafe keywords to detect harmful content.

Created a function to check if output contains unsafe words.

Implemented a filter function that replaces unsafe outputs with a safe message.

Applied the filter to all generated outputs before returning to users.

Results Interpretation

Before: Safety violation rate was 15%, user satisfaction 70%.
After: Safety violation rate reduced to 3%, user satisfaction slightly decreased to 68% due to some filtered content.

Adding output filtering and safety checks after generation can greatly reduce unsafe content without retraining the model. This improves trust and safety while keeping user satisfaction high.

Bonus Experiment

Try using a small machine learning classifier trained on safe vs unsafe text samples to improve detection accuracy over keyword filtering.

💡 Hint

Collect labeled examples of safe and unsafe outputs, train a simple model like logistic regression or a small neural network, and use it to classify outputs before returning them.

Practice

1. What is the main purpose of output filtering in AI systems?

easy

A. To stop unsafe or unwanted AI results from reaching users

B. To speed up the AI model training process

C. To increase the size of the AI model

D. To add more data to the training set

Solution

1. Step 1: Understand output filtering

   Output filtering is designed to prevent unsafe or unwanted content from being shown to users.

2. Step 2: Compare options

   Only To stop unsafe or unwanted AI results from reaching users describes stopping unsafe or unwanted results, which matches the purpose of output filtering.

3. Final Answer:

   To stop unsafe or unwanted AI results from reaching users -> Option A

4. Quick Check:

   Output filtering = stopping unsafe results [OK]

Hint: Output filtering blocks bad or unsafe AI outputs [OK]

Common Mistakes:

• Confusing filtering with training speed
• Thinking filtering adds data
• Assuming filtering changes model size

2. Which of the following is a correct way to check if an AI output contains a banned word in Python?

easy

A. if output_text.contains(banned_word):

B. if output_text == banned_word:

C. if output_text.index(banned_word):

D. if banned_word in output_text:

Solution

1. Step 1: Recall Python syntax for substring check

   In Python, to check if a substring is in a string, use the 'in' keyword.

2. Step 2: Evaluate options

   if banned_word in output_text: uses 'if banned_word in output_text:', which is correct. if output_text == banned_word: checks equality, not containment. if output_text.contains(banned_word): uses a method that doesn't exist in Python strings. if output_text.index(banned_word): uses index incorrectly and can cause errors.

3. Final Answer:

   if banned_word in output_text: -> Option D

4. Quick Check:

   Substring check in Python uses 'in' [OK]

Hint: Use 'in' keyword to check substring in Python strings [OK]

Common Mistakes:

• Using equality instead of containment
• Using non-existent string methods
• Using index without error handling

3. Given this Python code snippet for filtering AI output:

output = "Hello user!"
banned_words = ["bad", "ugly"]
filtered = any(word in output for word in banned_words)
print(filtered)

What will be the printed output?

medium

A. False

B. True

C. Error

D. None

Solution

1. Step 1: Understand the code logic

   The code checks if any banned word is in the output string using 'any()' with a generator expression.

2. Step 2: Check banned words in output

   Output is "Hello user!". Neither "bad" nor "ugly" is in this string, so 'any()' returns False.

3. Final Answer:

   False -> Option A

4. Quick Check:

   None of banned words in output = False [OK]

Hint: If no banned words found, 'any' returns False [OK]

Common Mistakes:

• Assuming 'any' returns True by default
• Confusing 'any' with 'all'
• Expecting an error from generator expression

4. This code is meant to filter AI output for banned words but causes an error:

output = "Safe text"
banned_words = ["bad", "ugly"]
for word in banned_words:
    if output.index(word):
        print("Banned word found")
        break

What is the error and how to fix it?

medium

A. Syntax error due to missing colon after for loop

B. index() raises ValueError if word not found; use 'in' instead

C. TypeError because output is not a list

D. No error; code works fine

Solution

1. Step 1: Identify the error cause

   Using output.index(word) raises ValueError if word is not found in output string.

2. Step 2: Suggest fix

   Replace 'output.index(word)' with 'word in output' to safely check containment without error.

3. Final Answer:

   index() raises ValueError if word not found; use 'in' instead -> Option B

4. Quick Check:

   index() error fixed by 'in' check [OK]

Hint: Use 'in' to check substring safely, not index() [OK]

Common Mistakes:

• Ignoring ValueError from index()
• Thinking output must be a list
• Missing colons in loops (not in this code)

5. You want to build a safety filter that blocks AI outputs containing banned words or outputs longer than 100 characters. Which approach correctly combines these checks in Python?

hard

A. if banned_words in output or len(output) == 100:
block_output()

B. if all(word in output for word in banned_words) and len(output) < 100:
block_output()

C. if any(word in output for word in banned_words) or len(output) > 100:
block_output()

D. if output.contains(banned_words) or output.length > 100:
block_output()

Solution

1. Step 1: Understand filtering conditions

   The filter should block if any banned word is present OR output length exceeds 100 characters.

2. Step 2: Evaluate options for correct logic and syntax

   if any(word in output for word in banned_words) or len(output) > 100: block_output() uses 'any' to check banned words and 'or' for length > 100, which is correct. if all(word in output for word in banned_words) and len(output) < 100: block_output() uses 'all' and 'and' incorrectly. if output.contains(banned_words) or output.length > 100: block_output() uses invalid methods. if banned_words in output or len(output) == 100: block_output() uses wrong containment and equality checks.

3. Final Answer:

   if any(word in output for word in banned_words) or len(output) > 100: block_output() -> Option C

4. Quick Check:

   Use 'any' + 'or' for combined filter [OK]

Hint: Use 'any' with 'or' to combine banned words and length checks [OK]

Common Mistakes:

• Using 'all' instead of 'any' for banned words
• Using 'and' instead of 'or' to combine conditions
• Using invalid string methods like contains()