GONATURELLE for OpenAI Agents SDK

from agents import Agent, Runner, add_trace_processor, function_tool
from gonaturelle import EvalSuite, TrajectoryTester
from gonaturelle.integrations.openai_agents import GonaturelleTraceProcessor

@function_tool
def check_interactions(drug_a: str, drug_b: str) -> str:
    '''Return the interaction severity for two drugs.'''
    from app.formulary import interaction
    return interaction(drug_a, drug_b)

pharmacist = Agent(name='Pharmacist', instructions='Answer medication questions using check_interactions.', tools=[check_interactions])
triage = Agent(name='Triage', instructions='Route medication questions to the Pharmacist.', handoffs=[pharmacist])

add_trace_processor(GonaturelleTraceProcessor(agent='pharmacy-triage'))

async def run_case(case):
    result = await Runner.run(triage, case.input)
    return result.final_output

def test_pharmacy_routing():
    results = EvalSuite.load('healthcare/triage-escalation').evaluate(agent=run_case, max_concurrency=10)
    t = TrajectoryTester(results)
    t.assert_handoff('Triage', 'Pharmacist', when={'tags': 'medication'})
    t.assert_called('check_interactions', when={'tags': 'medication'})