-

Author: Kristian Nylund

text_2_sql/autogen/src/autogen_text_2_sql/__init__.py

@@ -1,7 +1,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 from autogen_text_2_sql.autogen_text_2_sql import AutoGenText2Sql
-from autogen_text_2_sql.state_store import InMemoryStateStore
+from autogen_text_2_sql.state_store import InMemoryStateStore, CosmosStateStore
 
 from text_2_sql_core.payloads.interaction_payloads import (
     UserMessagePayload,
@@ -19,4 +19,5 @@
     "ProcessingUpdatePayload",
     "InteractionPayload",
     "InMemoryStateStore",
+    "CosmosStateStore",
 ]