def generate_agent(**kwargs) -> schema.StateAgent:
    args = {
        "state_code": _STATE_CODE,
        "agent_type": StateAgentType.JUDGE.value,
    }
    args.update(kwargs)
    return schema.StateAgent(**args)
def generate_test_assessment_agent() -> state_schema.StateAgent:
    instance = state_schema.StateAgent(
        agent_id=1010,
        external_id='ASSAGENT1234',
        agent_type=StateAgentType.SUPERVISION_OFFICER.value,
        state_code='us_ca',
        full_name='JOHN SMITH',
    )
    return instance