The short story you're thinking of is titled 'Trurl’s Machine'. It's the second story in the collection 'The Cyberiad'.

... a very good book, by the way.