ai framework changes

made some framework changes