Did you set PYTHONPATH to find the psi4 modules? Run
psi4 --psiapi
to get the paths you need to export.