diff --git a/snowflake/cortex/__init__.py b/snowflake/cortex/__init__.py
index 92ab345..f153b1d 100644
--- a/snowflake/cortex/__init__.py
+++ b/snowflake/cortex/__init__.py
@@ -1,5 +1,5 @@
 from snowflake.cortex._classify_text import ClassifyText
-from snowflake.cortex._complete import Complete, CompleteOptions
+from snowflake.cortex._complete import Complete, CompleteOptions, ConversationMessage
 from snowflake.cortex._embed_text_768 import EmbedText768
 from snowflake.cortex._embed_text_1024 import EmbedText1024
 from snowflake.cortex._extract_answer import ExtractAnswer
@@ -11,6 +11,7 @@
     "ClassifyText",
     "Complete",
     "CompleteOptions",
+    "ConversationMessage",
     "EmbedText768",
     "EmbedText1024",
     "ExtractAnswer",