Guardrails 是一种用于约束生成式 AI 输出的工具。它允许开发人员定义规则和约束,确保生成的内容符合预期。
以下是一个简单的 Guardrails 示例:
from guardrails import Guardrails
# 定义规则
rules = {
"max_length": 100,
"no_profanity": True
}
# 初始化 Guardrails
rails = Guardrails(rules)
# 应用规则
output = rails.apply("生成的文本内容")
print(output)
集成到 LangChain
Guardrails 可以与 LangChain 集成,以确保代理生成的输出符合规则。
from langchain.chains import LLMChain
from guardrails import Guardrails
# 定义规则
rules = {
"max_length": 100,
"no_profanity": True
}
# 初始化 Guardrails
rails = Guardrails(rules)
# 创建 LangChain
chain = LLMChain(llm=llm, guardrails=rails)
# 运行链
result = chain.run("输入内容")
print(result)
使用 Guardrails 可以显著提高生成式 AI 的可靠性和安全性。