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