Law firm restricts AI after 'significant' staff use