[FEATURE] Preserving schemas when saving as avro

[goodwanghan]

Describe the solution you'd like Currently when saving avro, we lose the original types. This can be done if we implement something like this.