Integrate with goai lib