Consider using asynchronous alternative function は completionHandler じゃなくて、asyncメソッドとして呼んだら消えないですかね

let error = await self.chat.predictTokens(for: prompt, tokenHandler: { token in continuation.yield(token) }) continuation.finish(throwing: error)

(edited)