Evaluating Agent Quality: Task Completion Rate

# evaluation/task_completion.py
from openai import AsyncAzureOpenAI
from pydantic import BaseModel

class TaskEvalResult(BaseModel):
    task_id: str
    goal: str
    agent_answer: str
    reference_answer: str
    completed: bool
    score: float  # 0.0-1.0
    reason: str

async def evaluate_task_completion(
    task_id: str,
    goal: str,
    agent_answer: str,
    reference_answer: str,
    client: AsyncAzureOpenAI,
) -> TaskEvalResult:
    """LLM-as-judge evaluation of task completion."""

    judge_prompt = f"""You are evaluating an AI agent's answer.

Task: {goal}

Reference answer (what a correct response should cover):
{reference_answer}

Agent's answer:
{agent_answer}

Rate the agent's answer on a scale of 0-10:
- 10: Fully correct, complete, covers all key points
- 7-9: Mostly correct with minor gaps
- 4-6: Partially correct, missing key information
- 1-3: Mostly wrong but contains some relevant information
- 0: Completely wrong or refused to answer

Respond as JSON: {{"score": <0-10>, "completed": <true/false>, "reason": "<one sentence>"}}"""

    response = await client.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": judge_prompt}],
        response_format={"type": "json_object"},
        temperature=0,
    )

    data = eval(response.choices[0].message.content)
    return TaskEvalResult(
        task_id=task_id,
        goal=goal,
        agent_answer=agent_answer,
        reference_answer=reference_answer,
        completed=data["score"] >= 7,
        score=data["score"] / 10.0,
        reason=data["reason"],
    )

# evaluation/step_efficiency.py
from dataclasses import dataclass

@dataclass
class StepMetrics:
    task_id: str
    steps_used: int
    optimal_steps: int
    efficiency: float  # 0.0-1.0, higher is better
    tool_calls: list[str]

def compute_step_efficiency(
    steps_used: int,
    optimal_steps: int,
) -> float:
    """Efficiency = optimal / actual. 1.0 = perfect, lower = less efficient."""
    if steps_used == 0:
        return 0.0
    return min(1.0, optimal_steps / steps_used)

# Golden dataset: each task has an expected step count
GOLDEN_TASKS = [
    {
        "task_id": "drug-info-ibuprofen",
        "goal": "What is ibuprofen used for?",
        "optimal_steps": 1,  # Simple lookup: one tool call
    },
    {
        "task_id": "interaction-warfarin-ibuprofen",
        "goal": "Check interaction between warfarin and ibuprofen",
        "optimal_steps": 2,  # Lookup both drugs, then check interaction
    },
    {
        "task_id": "dosage-pediatric-amoxicillin",
        "goal": "What is the pediatric dosage for amoxicillin by weight?",
        "optimal_steps": 3,  # Lookup drug, find pediatric section, compute by weight
    },
]

def analyze_efficiency(run_log: dict, golden: dict) -> StepMetrics:
    return StepMetrics(
        task_id=golden["task_id"],
        steps_used=len(run_log["tool_calls"]),
        optimal_steps=golden["optimal_steps"],
        efficiency=compute_step_efficiency(
            len(run_log["tool_calls"]),
            golden["optimal_steps"],
        ),
        tool_calls=[tc["tool_name"] for tc in run_log["tool_calls"]],
    )

# evaluation/trajectory.py

REFERENCE_TRAJECTORIES = {
    "interaction-warfarin-ibuprofen": {
        "expected_tools": ["search_drug_database", "check_interaction"],
        "expected_tool_order": True,  # Must be in this order
        "required_content": ["bleeding risk", "NSAID", "anticoagulant"],
    }
}

def evaluate_trajectory(
    task_id: str,
    actual_tool_calls: list[str],
    final_answer: str,
) -> dict:
    if task_id not in REFERENCE_TRAJECTORIES:
        return {"evaluated": False}

    ref = REFERENCE_TRAJECTORIES[task_id]

    # Check all expected tools were called
    expected = set(ref["expected_tools"])
    actual_set = set(actual_tool_calls)
    missing_tools = expected - actual_set
    extra_tools = actual_set - expected

    # Check required content in final answer
    answer_lower = final_answer.lower()
    missing_content = [
        phrase for phrase in ref["required_content"]
        if phrase.lower() not in answer_lower
    ]

    return {
        "task_id": task_id,
        "missing_tools": list(missing_tools),
        "extra_tools": list(extra_tools),
        "missing_content": missing_content,
        "passed": len(missing_tools) == 0 and len(missing_content) == 0,
    }

# evaluation/sampling.py
import random
import structlog

log = structlog.get_logger()

def should_sample_for_review(run_id: str, rate: float = 0.05) -> bool:
    """Randomly flag 5% of runs for human review."""
    return random.random() < rate

def flag_for_review(run_id: str, reason: str, priority: str = "normal"):
    """Add run to human review queue."""
    log.info(
        "flagged_for_review",
        run_id=run_id,
        reason=reason,
        priority=priority,
    )
    # In production: add to review queue (DB table, Slack message, etc.)

# Also flag specific cases automatically:
# - Agent hit max iterations
# - Output safety check triggered
# - Step efficiency below 0.3
# - User rated the response poorly (thumbs down)

async def post_run_review_check(run_result: dict):
    if run_result.get("hit_max_iterations"):
        flag_for_review(run_result["run_id"], "hit_max_iterations", priority="high")

    if run_result.get("output_blocked"):
        flag_for_review(run_result["run_id"], "output_blocked", priority="high")

    if run_result.get("step_efficiency", 1.0) < 0.3:
        flag_for_review(run_result["run_id"], "low_efficiency", priority="normal")

    if should_sample_for_review(run_result["run_id"]):
        flag_for_review(run_result["run_id"], "random_sample", priority="low")

# evaluation/run_eval.py
import asyncio
from evaluation.task_completion import evaluate_task_completion
from evaluation.trajectory import evaluate_trajectory

EVAL_TASKS = [
    # (task_id, goal, reference_answer)
    ("drug-info-ibuprofen", "What is ibuprofen?", "NSAID, pain reliever, COX inhibitor"),
    ("interaction-check", "Is it safe to combine warfarin and ibuprofen?", "Major interaction, increased bleeding risk, consult pharmacist"),
    # ... 50 more tasks
]

async def run_evaluation_suite(agent, client):
    results = []
    for task_id, goal, reference in EVAL_TASKS:
        # Run the agent
        run_log = await agent.run_with_logging(goal)

        # Evaluate task completion
        completion = await evaluate_task_completion(
            task_id, goal, run_log["answer"], reference, client
        )

        # Evaluate trajectory
        trajectory = evaluate_trajectory(
            task_id, run_log["tool_calls"], run_log["answer"]
        )

        results.append({
            "task_id": task_id,
            "completed": completion.completed,
            "score": completion.score,
            "trajectory_passed": trajectory.get("passed", True),
        })

    # Compute aggregate metrics
    completion_rate = sum(r["completed"] for r in results) / len(results)
    avg_score = sum(r["score"] for r in results) / len(results)

    print(f"Task completion rate: {completion_rate:.1%}")
    print(f"Average score: {avg_score:.2f}")

    # Fail CI if quality dropped
    if completion_rate < 0.80:
        raise AssertionError(f"Completion rate {completion_rate:.1%} below 80% threshold")

    return results

if __name__ == "__main__":
    asyncio.run(run_evaluation_suite(agent=..., client=...))