You also can't have a human in the next room feeding you lines... even if that person is an attorney.
However, a party / attorney certainly can bring in notes / a casebook / etc. You can usually bring in a laptop, with search functions, pdfs, etc. An AI that quickly presents relevant information, documents, caselaw, etc. would 100% be allowed. However, if they're sworn in to testify, these would usually be taken away, because testimony is supposed to be from personal knowledge / memory.