AI's aren't going to replace lawyers in the near term, particularly in discovery - doc production work, but for a different reason.
Why bring in an expensive machine when you can get a swarm of peons for really cheap, and throw them away when you're done?
In most large cities in the country, there are way too many lawyers (yeah, insert favorite joke here), with more being hatched every year. When a big case comes up, or any legal matter requiring a lot of gruntwork, such as going through tons of documents, a lot of law firms hire throw-away lawyers for some times as little as $20 an hour -- and for most of these, they have more folks looking for work than they need. Why would you go with an AI in a situation like that? They're too expensive (at this time), and if the docs are in printed form (which is how the other side will present them to the other side's life as difficult as possible), the docs have to be handled, bates stamped, scanned, and then analyzed. Why not hire a roomful of out of work lawyers $20 an hour to do that, with a few more at a higher rate (say, $30) to do spot check and general QA, eventually feeding to that high priced ($125 an hour) law firm paralegal. And dump them when the task is done.
(Disclosure: I passed the California bar in 1990 and have been through this ratshit. People that save every email they've ever received or sent make a lot of money for law firms handling discovery. Please, don't save all that shit unless it's really needed and useful?)