guardrails-ai / competitor_check

Guardrails AI: Competitor Check - Validates that LLM-generated text is not naming any competitors from a given list
Apache License 2.0
1 stars 3 forks source link

Overview

| Developed by | Guardrails AI | | Date of development | Feb 15, 2024 | | Validator type | Brand risk, QA, chatbots | | Blog | | | License | Apache 2 | | Input/Output | Output |

Description

Intended Use

This validator ensures that no competitors for an organization are being named. In order to use this validator, you need to provide a list of competitors that you don’t want to name.

Requirements

Installation

guardrails hub install hub://guardrails/competitor_check

Usage Examples

Validating string output via Python

In this example, we apply the validator to a string output generated by an LLM.

# Import Guard and Validator
from guardrails import Guard
from guardrails.hub import CompetitorCheck

# Setup Guard
guard = Guard().use(
    CompetitorCheck, ["Apple", "Samsung"], "exception"
)

response = guard.validate(
    "The apple doesn't fall far from the tree."
)  # Validator passes

try:
    response = guard.validate("Apple just released a new iPhone.")  # Validator fails
except Exception as e:
    print(e)

Output:

Validation failed for field with errors: Found the following competitors: [['Apple']]. Please avoid naming those competitors next time

Validating JSON output via Python

In this example, we apply the validator to a string that is a field within a Pydantic object.

# Import Guard and Validator
from pydantic import BaseModel, Field
from guardrails.hub import CompetitorCheck
from guardrails import Guard

# Initialize Validator
val = CompetitorCheck(competitors=["Apple", "Samsung"], on_fail="exception")

# Create Pydantic BaseModel
class MarketingCopy(BaseModel):
    product_name: str
    product_description: str = Field(
        description="Description about the product", validators=[val]
    )

# Create a Guard to check for valid Pydantic output
guard = Guard.from_pydantic(output_class=MarketingCopy)

# Run LLM output generating JSON through guard
try:
    guard.parse(
        """
        {
            "product_name": "Galaxy S23+",
            "product_description": "Samsung's latest flagship phone with 5G capabilities"
        }
        """
    )
except Exception as e:
    print(e)

Output:

Validation failed for field with errors: Found the following competitors: [['Samsung']]. Please avoid naming those competitors next time

API Reference

__init__(self, competitors, on_fail="noop")


validate(self, value, metadata={}) -> ValidationResult