They were simply totally original screenplays written for the films. No book or anything. Well apart from the young Vito parts in the Godfather novel were masterfully included into Part II.

I'm sorry I don't think I've heard of anything on Puzo already having ideas for a sequel but I wouldn't put it past him. I bet he made notes somewhere. And so would FFC after seeing the success of GFI.