fix ai class function arguments bug