from piqrypt_crewai import AuditedAgent, AuditedCrew, stamp_task

# Drop-in replacement for Agent
researcher = AuditedAgent(
    role="Researcher",
    goal="Find accurate information on AI compliance",
    backstory="Expert in regulatory frameworks",
    identity_file="researcher.json"   # piqrypt identity
)

writer = AuditedAgent(
    role="Writer",
    goal="Write clear compliance reports",
    backstory="Technical writer",
    identity_file="writer.json"
)

# Drop-in replacement for Crew
crew = AuditedCrew(
    agents=[researcher, writer],
    tasks=[research_task, write_task],
    identity_file="crew-coordinator.json"
)

result = crew.kickoff(inputs={"topic": "EU AI Act compliance"})

# Export tamper-proof audit trail
crew.export_audit("eu-ai-act-audit.json")
# $ piqrypt verify eu-ai-act-audit.json

from piqrypt_crewai import stamp_task

@stamp_task("research", identity_file="my-agent.json")
def research(topic: str) -> str:
    return your_research_logic(topic)

@stamp_task("write_report", identity_file="my-agent.json")
def write_report(research: str) -> str:
    return your_writing_logic(research)