guardrails hub install hub://numbersstation/sql_column_presence

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

# Setup Guard
guard = Guard().use(SqlColumnPresence, ["name", "breed", "weight"], on_fail="exception")

guard.validate(
    "SELECT name, AVG(weight) FROM animals GROUP BY name"
)  # Validator passes

try:
    guard.validate(
        "SELECT name, color, AVG(weight) FROM animals GROUP BY name, color"
    )  # Validator fails
except Exception as e:
    print(e)

Validation failed for field with errors: Columns [color] not in [weight, name, breed]

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

# Initialize Validator
val = SqlColumnPresence(["name", "breed", "weight"])

# Create Pydantic BaseModel
class Report(BaseModel):
		name: str
		query: str = Field(validators=[val])

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

# Run LLM output generating JSON through guard
guard.parse("""
{
	"name": "Canine Lookup",
	"query": "SELECT name, AVG(weight) FROM animals GROUP BY name"
}
""")