From 00e902455f2c53a1085d810faf3da327932b4c07 Mon Sep 17 00:00:00 2001 From: Ben Constable Date: Mon, 13 Jan 2025 12:36:09 +0000 Subject: [PATCH] Update payloads --- .../autogen/src/autogen_text_2_sql/__init__.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) 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", +]