When I use it I often give a final prompt like "Now combine the above answers together into a function that accept the following arguments...". This has worked well for my use cases.