How should I make Dokka code samples working correctly in IDEA?

Put them in test source set for module you documenting

THat's what I did and in my library I have the code sampel working but when I use my library it is not working anymore. Should they be added to the source JAR ?

Yes, you should add them to sources JAR, but unsure will it work or not. If not is is bug anyway