User Simulator Agent

import pytest
import scenario
 
class TechnicalSupportAgent(scenario.AgentAdapter):
    async def call(self, input: scenario.AgentInput) -> scenario.AgentReturnTypes:
        # Your technical support agent implementation
        user_message = input.last_new_user_message_str()
        return await my_support_bot.process(user_message)
 
@pytest.mark.asyncio
async def test_frustrated_customer_handling():
    result = await scenario.run(
        name="frustrated customer with internet issues",
        description="""
            User is a non-technical person experiencing slow internet for 3 days.
            They've already tried calling support twice with no resolution.
            They're frustrated and tired of technical jargon. They just want
            their internet to work and are losing patience with troubleshooting steps.
        """,
        agents=[
            TechnicalSupportAgent(),
            scenario.UserSimulatorAgent(
                model="openai/gpt-4o",
                temperature=0.3  # Some variability for realistic frustration
            ),
            scenario.JudgeAgent(criteria=[
                "Agent acknowledges the customer's frustration empathetically",
                "Agent avoids excessive technical jargon",
                "Agent provides simple, clear instructions",
                "Agent offers escalation if troubleshooting doesn't work",
                "Agent remains professional despite customer frustration"
            ])
        ],
        max_turns=10
    )
    
    assert result.success
    print(f"Test completed with {len(result.messages)} messages")