Adding Workflow API support for AI grading