CrewAI

from benchspan import BenchGuard, InjectionDetectedError
from crewai import Agent, Crew, Task

guard = BenchGuard(api_key="ag_live_...", agent="research-crew", mode="block")

researcher = Agent(
    role="Senior Researcher",
    goal="Find accurate information on the topic",
    tools=[web_search_tool, document_reader],
)

writer = Agent(
    role="Technical Writer",
    goal="Write clear summaries",
)

tasks = [
    Task(description="Research {topic}", agent=researcher),
    Task(description="Summarize findings", agent=writer),
]

crew = Crew(
    agents=[researcher, writer],
    tasks=tasks,
    callbacks=[guard],  # every tool output scanned before the LLM sees it
)

try:
    result = crew.kickoff(inputs={"topic": "indirect prompt injection"})
except InjectionDetectedError as e:
    print(f"Crew blocked an injection: score={e.result.score:.4f} id={e.result.id}")