diff --git a/text_2_sql/autogen/src/autogen_text_2_sql/__init__.py b/text_2_sql/autogen/src/autogen_text_2_sql/__init__.py
index bf72f7d..03e6104 100644
--- a/text_2_sql/autogen/src/autogen_text_2_sql/__init__.py
+++ b/text_2_sql/autogen/src/autogen_text_2_sql/__init__.py
@@ -1,6 +1,19 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 from autogen_text_2_sql.autogen_text_2_sql import AutoGenText2Sql
-from text_2_sql_core.payloads.interaction_payloads import UserMessagePayload
+from text_2_sql_core.payloads.interaction_payloads import (
+    UserMessagePayload,
+    DismabiguationRequestsPayload,
+    AnswerWithSourcesPayload,
+    ProcessingUpdatePayload,
+    InteractionPayload,
+)
 
-__all__ = ["AutoGenText2Sql", "UserMessagePayload"]
+__all__ = [
+    "AutoGenText2Sql",
+    "UserMessagePayload",
+    "DismabiguationRequestsPayload",
+    "AnswerWithSourcesPayload",
+    "ProcessingUpdatePayload",
+    "InteractionPayload",
+]