I feel Allah (SWT) wants to get rid of me/doesn’t want me amongst his servants anymore.

I have fear of dying as an atheist, non-shia. What if one day Allah (SWT) just leaves me?

Allah doesn’t leave you if you don’t leave Him. He is merciful and generous. Try to fulfill the obligations he has asked you to fulfill, believe in his oneness, believe in the Day of Judgment, increase your knowledge and if you sin repent. You’ll be ok.