Chore
* chore(release): v2.0.1 [skip ci] ([`2af2f2b`](https://github.com/AI21Labs/ai21-python/commit/2af2f2b3153a486a8df33b4266f54cf28f0da049))
Documentation
* docs: updated migration (54)
* docs: updated migration
* fix: example ([`bca0424`](https://github.com/AI21Labs/ai21-python/commit/bca0424f44ffb2c5e5a3b9f292eb08649a844f53))
Fix
* fix: Added badges (55) ([`96d716d`](https://github.com/AI21Labs/ai21-python/commit/96d716de7a49868c20461bc2dc34c142da36070b))