It could be the fuel pump or possibly the fuel pressure regulator. If you could check what you are getting fuel pressure at idle it might help point you in the right direction. I had the same sort of problem with my truck, 2.3t swap, been fixing problems on it ever since I bought the thing, lol...