from crewai import Agent, Task, Crew
from groundtruth.crewai import (
    GroundTruthExecuteTool,
    GroundTruthVerifyTool,
    GroundTruthAwaitApprovalTool,
)

# Tools
execute       = GroundTruthExecuteTool(agent_name="support-crew")
verify        = GroundTruthVerifyTool()
await_approval = GroundTruthAwaitApprovalTool()

# Agents
triage = Agent(
    role="Support Triage",
    goal="Categorize incoming tickets by type and urgency",
    backstory="You triage support tickets for a SaaS company.",
)

replier = Agent(
    role="Support Reply Agent",
    goal="Draft accurate, helpful replies to customer tickets",
    backstory=(
        "You draft replies to support tickets. Before sending ANY reply, "
        "you MUST use the GroundTruth Verify tool to check your draft, "
        "then use the GroundTruth Execute tool to send it. Never send "
        "a reply without going through GroundTruth first."
    ),
    tools=[execute, verify],
)

escalation = Agent(
    role="Escalation Handler",
    goal="Handle blocked or escalated actions and wait for human decisions",
    backstory=(
        "You handle cases where GroundTruth escalates an action. "
        "Use GroundTruth Await Approval with the execution ID to wait "
        "for the human reviewer's decision, then report the outcome."
    ),
    tools=[await_approval],
)

# Tasks
triage_task = Task(
    description="Categorize ticket #4521: 'I want to cancel my subscription and get a refund.'",
    expected_output="Category and urgency level.",
    agent=triage,
)

reply_task = Task(
    description=(
        "Draft and send a reply to ticket #4521 about the cancellation request. "
        "First verify your draft with GroundTruth Verify, then send it via "
        "GroundTruth Execute with action='reply_ticket'."
    ),
    expected_output="The GroundTruth execution result (approved, blocked, or escalated).",
    agent=replier,
    context=[triage_task],
)

await_task = Task(
    description=(
        "If the previous reply was ESCALATED, use GroundTruth Await Approval "
        "with the execution ID to wait for the human decision. "
        "If approved, confirm to the customer. If rejected, report the rejection."
    ),
    expected_output="The final approval decision and next steps.",
    agent=escalation,
    context=[reply_task],
)

crew = Crew(
    agents=[triage, replier, escalation],
    tasks=[triage_task, reply_task, await_task],
)
result = crew.kickoff()
print(result)

from crewai import Agent, Task, Crew
from groundtruth.crewai import (
    GroundTruthExecuteTool,
    GroundTruthVerifyTool,
    GroundTruthAwaitApprovalTool,
)

execute        = GroundTruthExecuteTool(agent_name="sales-crew", user_id="sales-team")
verify         = GroundTruthVerifyTool()
await_approval = GroundTruthAwaitApprovalTool()

researcher = Agent(
    role="Prospect Researcher",
    goal="Research prospects and identify relevant pain points",
    backstory="You research companies to find the best angle for outreach.",
)

copywriter = Agent(
    role="Sales Copywriter",
    goal="Write personalized, accurate outreach emails",
    backstory=(
        "You write cold outreach emails. Before sending ANY email, you MUST: "
        "1) Use GroundTruth Verify to check all claims about your product. "
        "2) Use GroundTruth Execute with action='send_email' to send it. "
        "NEVER mention competitor names. NEVER fabricate product features."
    ),
    tools=[execute, verify],
)

qa = Agent(
    role="QA Reviewer",
    goal="Handle escalated emails and wait for human approval",
    backstory=(
        "You handle emails that GroundTruth escalated for review. "
        "Use GroundTruth Await Approval with the execution ID to wait "
        "for the manager's decision, then report the outcome."
    ),
    tools=[verify, await_approval],
)

research_task = Task(
    description="Research Acme Corp — find their industry, size, and likely pain points.",
    expected_output="A brief prospect profile.",
    agent=researcher,
)

email_task = Task(
    description=(
        "Draft and send a personalized outreach email to sarah@acme.com. "
        "Reference our product's actual features from the knowledge base. "
        "Verify the draft first, then send via GroundTruth Execute."
    ),
    expected_output="The execution result from GroundTruth.",
    agent=copywriter,
    context=[research_task],
)

approval_task = Task(
    description=(
        "If the email was ESCALATED, use GroundTruth Await Approval with the "
        "execution ID to wait for the manager's decision. "
        "Report whether the email was approved and sent, or rejected."
    ),
    expected_output="The final approval decision.",
    agent=qa,
    context=[email_task],
)

crew = Crew(
    agents=[researcher, copywriter, qa],
    tasks=[research_task, email_task, approval_task],
)
result = crew.kickoff()

from crewai import Agent, Task, Crew
from groundtruth.crewai import GroundTruthExecuteTool, GroundTruthAwaitApprovalTool

execute = GroundTruthExecuteTool(agent_name="support-crew", session_id="sess_123")
await_approval = GroundTruthAwaitApprovalTool()

agent = Agent(
    role="Support Agent",
    goal="Handle tickets with human oversight for sensitive actions",
    backstory=(
        "You handle support tickets. Submit actions via GroundTruth Execute. "
        "If an action is ESCALATED, use GroundTruth Await Approval with the "
        "execution ID and wait for the human decision before proceeding."
    ),
    tools=[execute, await_approval],
)

submit_task = Task(
    description=(
        "Reply to ticket #7890 about the customer's refund request. "
        "Use GroundTruth Execute with action='reply_ticket'."
    ),
    expected_output="The execution result.",
    agent=agent,
)

followup_task = Task(
    description=(
        "If the previous action was ESCALATED, use GroundTruth Await Approval "
        "with the execution ID to wait for the human decision. "
        "If approved, confirm to the customer. If rejected, revise the reply."
    ),
    expected_output="Final status after human review.",
    agent=agent,
    context=[submit_task],
)

crew = Crew(agents=[agent], tasks=[submit_task, followup_task])
crew.kickoff()