diff --git a/vizro-ai/CHANGELOG.md b/vizro-ai/CHANGELOG.md
index 4f1bc4fb9..d5bd0a439 100644
--- a/vizro-ai/CHANGELOG.md
+++ b/vizro-ai/CHANGELOG.md
@@ -11,6 +11,22 @@ See the fragment files in the [changelog.d directory](https://github.com/mckinse
 
 <!-- scriv-insert-here -->
 
+<a id='changelog-0.2.0'></a>
+
+# 0.2.0 — 2024-05-09
+
+## Removed
+
+- Removed `temperature` and `model_name` arguments of `VizroAI` class. For current configuration options, visit the [Vizro-AI docs](https://vizro.readthedocs.io/projects/vizro-ai/en/latest/pages/explanation/faq/#what-parameters-does-vizro-ai-support) ([#423](https://github.com/mckinsey/vizro/pull/423))
+
+## Added
+
+- Enable customization of LLM models provided to Vizro-AI class. ([#423](https://github.com/mckinsey/vizro/pull/423))
+
+## Changed
+
+- `VizroAI.plot` now generates and returns a `plotly.graph_objects.Figure` object. ([#411](https://github.com/mckinsey/vizro/pull/441)). This facilitates the integration of Vizro-AI charts into the `vizro` dashboard, visit [Use Vizro-AI dynamically to return a fig object](https://vizro.readthedocs.io/projects/vizro-ai/en/latest/pages/user-guides/add-generated-chart-usecase/#use-vizro-ais-generated-code) for details.
+
 <a id='changelog-0.1.2'></a>
 
 # 0.1.2 — 2024-03-13
diff --git a/vizro-ai/changelog.d/20240313_190013_maximilian_schulz_version_bump.md b/vizro-ai/changelog.d/20240313_190013_maximilian_schulz_version_bump.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240313_190013_maximilian_schulz_version_bump.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240325_170323_jo_stichbury_revise_vizro_ai_docs.md b/vizro-ai/changelog.d/20240325_170323_jo_stichbury_revise_vizro_ai_docs.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240325_170323_jo_stichbury_revise_vizro_ai_docs.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240328_174607_nadija_ratkusic_graca_add_genAI_use_case.md b/vizro-ai/changelog.d/20240328_174607_nadija_ratkusic_graca_add_genAI_use_case.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240328_174607_nadija_ratkusic_graca_add_genAI_use_case.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240402_180718_jo_stichbury_update_core_to_highlight_ai.md b/vizro-ai/changelog.d/20240402_180718_jo_stichbury_update_core_to_highlight_ai.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240402_180718_jo_stichbury_update_core_to_highlight_ai.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240404_112532_jo_stichbury_update_build_docs_settings.md b/vizro-ai/changelog.d/20240404_112532_jo_stichbury_update_build_docs_settings.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240404_112532_jo_stichbury_update_build_docs_settings.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240415_091003_huong_li_nguyen_0_1_15.md b/vizro-ai/changelog.d/20240415_091003_huong_li_nguyen_0_1_15.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240415_091003_huong_li_nguyen_0_1_15.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240415_145151_jo_stichbury_fix_merge_vale_added.md b/vizro-ai/changelog.d/20240415_145151_jo_stichbury_fix_merge_vale_added.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240415_145151_jo_stichbury_fix_merge_vale_added.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240415_223857_nadija_ratkusic_graca_refactor_vizroai_class.md b/vizro-ai/changelog.d/20240415_223857_nadija_ratkusic_graca_refactor_vizroai_class.md
deleted file mode 100644
index 8fba12247..000000000
--- a/vizro-ai/changelog.d/20240415_223857_nadija_ratkusic_graca_refactor_vizroai_class.md
+++ /dev/null
@@ -1,45 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-
-### Removed
-
-- Removed `temperature` and `model_name` arguments of `VizroAI` class. For current configuration options, visit the [Vizro-AI docs](https://vizro.readthedocs.io/projects/vizro-ai/en/latest/pages/explanation/faq/#what-parameters-does-vizro-ai-support) ([#423](https://github.com/mckinsey/vizro/pull/423))
-
-### Added
-
-- Enable customization of LLM models provided to Vizro-AI class. ([#423](https://github.com/mckinsey/vizro/pull/423))
-
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240418_131121_jo_stichbury_add_awesome_vizro.md b/vizro-ai/changelog.d/20240418_131121_jo_stichbury_add_awesome_vizro.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240418_131121_jo_stichbury_add_awesome_vizro.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240419_145915_jo_stichbury_fix_formatting_issue.md b/vizro-ai/changelog.d/20240419_145915_jo_stichbury_fix_formatting_issue.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240419_145915_jo_stichbury_fix_formatting_issue.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240419_165541_alexey_snigir_vizro_ai_tests_fix.md b/vizro-ai/changelog.d/20240419_165541_alexey_snigir_vizro_ai_tests_fix.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240419_165541_alexey_snigir_vizro_ai_tests_fix.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240424_120614_jo_stichbury_update_environment_text_vizroai.md b/vizro-ai/changelog.d/20240424_120614_jo_stichbury_update_environment_text_vizroai.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240424_120614_jo_stichbury_update_environment_text_vizroai.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240425_125608_nadija_ratkusic_graca_add_how_to_guide_vizroai.md b/vizro-ai/changelog.d/20240425_125608_nadija_ratkusic_graca_add_how_to_guide_vizroai.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240425_125608_nadija_ratkusic_graca_add_how_to_guide_vizroai.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240426_164549_nadija_ratkusic_graca_vizro_ai_return_fig_object.md b/vizro-ai/changelog.d/20240426_164549_nadija_ratkusic_graca_vizro_ai_return_fig_object.md
deleted file mode 100644
index cc959e0ec..000000000
--- a/vizro-ai/changelog.d/20240426_164549_nadija_ratkusic_graca_vizro_ai_return_fig_object.md
+++ /dev/null
@@ -1,47 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-
-### Changed
-
-- `VizroAI.plot` now returns `plotly.graph_objects` fig object. ([#411](https://github.com/mckinsey/vizro/pull/441))
-
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240426_172859_anna_xiong_add_azure_open_ai.md b/vizro-ai/changelog.d/20240426_172859_anna_xiong_add_azure_open_ai.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240426_172859_anna_xiong_add_azure_open_ai.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240430_095908_anna_xiong.md b/vizro-ai/changelog.d/20240430_095908_anna_xiong.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240430_095908_anna_xiong.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240501_111259_anna_xiong_langchian_verison_upperbound.md b/vizro-ai/changelog.d/20240501_111259_anna_xiong_langchian_verison_upperbound.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240501_111259_anna_xiong_langchian_verison_upperbound.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240502_112941_lingyi_zhang_default_model.md b/vizro-ai/changelog.d/20240502_112941_lingyi_zhang_default_model.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240502_112941_lingyi_zhang_default_model.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240503_150635_jo_stichbury_add_gif_to_index.md b/vizro-ai/changelog.d/20240503_150635_jo_stichbury_add_gif_to_index.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240503_150635_jo_stichbury_add_gif_to_index.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240507_133149_alexey_snigir_vizro_ai_tests_approval.md b/vizro-ai/changelog.d/20240507_133149_alexey_snigir_vizro_ai_tests_approval.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240507_133149_alexey_snigir_vizro_ai_tests_approval.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240508_181644_alexey_snigir_fix_vizro_ai_tests.md b/vizro-ai/changelog.d/20240508_181644_alexey_snigir_fix_vizro_ai_tests.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240508_181644_alexey_snigir_fix_vizro_ai_tests.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240508_210432_nadija_ratkusic_graca_vizro_ai_tidy_docs.md b/vizro-ai/changelog.d/20240508_210432_nadija_ratkusic_graca_vizro_ai_tidy_docs.md
deleted file mode 100644
index f1f65e73c..000000000
--- a/vizro-ai/changelog.d/20240508_210432_nadija_ratkusic_graca_vizro_ai_tidy_docs.md
+++ /dev/null
@@ -1,48 +0,0 @@
-<!--
-A new scriv changelog fragment.
-
-Uncomment the section that is right (remove the HTML comment wrapper).
--->
-
-<!--
-### Highlights ✨
-
-- A bullet item for the Highlights ✨ category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Removed
-
-- A bullet item for the Removed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Added
-
-- A bullet item for the Added category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Changed
-
-- A bullet item for the Changed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Deprecated
-
-- A bullet item for the Deprecated category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Fixed
-
-- A bullet item for the Fixed category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
-<!--
-### Security
-
-- A bullet item for the Security category with a link to the relevant PR at the end of your entry, e.g. Enable feature XXX ([#1](https://github.com/mckinsey/vizro/pull/1))
-
--->
diff --git a/vizro-ai/changelog.d/20240313_183650_maximilian_schulz_0_1_2.md b/vizro-ai/changelog.d/20240509_190629_nadija_ratkusic_graca_0_2_0.md
similarity index 100%
rename from vizro-ai/changelog.d/20240313_183650_maximilian_schulz_0_1_2.md
rename to vizro-ai/changelog.d/20240509_190629_nadija_ratkusic_graca_0_2_0.md
diff --git a/vizro-ai/src/vizro_ai/__init__.py b/vizro-ai/src/vizro_ai/__init__.py
index 8d971a4dd..96dcbf4c3 100644
--- a/vizro-ai/src/vizro_ai/__init__.py
+++ b/vizro-ai/src/vizro_ai/__init__.py
@@ -9,6 +9,6 @@
 
 __all__ = ["VizroAI"]
 
-__version__ = "0.2.0.dev0"
+__version__ = "0.2.0"
 
 logging.basicConfig(level=os.getenv("VIZRO_AI_LOG_LEVEL", "INFO"))