Add ByteStream and ChatMessage to Haystack init

deepset-ai · Dec 11, 2023 · 128f079 · 128f079
1 parent 820d9c3
commit 128f079
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/haystack/__init__.py b/haystack/__init__.py
@@ -2,7 +2,7 @@
 from haystack.core.serialization import default_from_dict, default_to_dict
 from haystack.core.errors import DeserializationError, ComponentError
 from haystack.pipeline import Pipeline
-from haystack.dataclasses import Document, Answer, GeneratedAnswer, ExtractedAnswer
+from haystack.dataclasses import Answer, ByteStream, ChatMessage, Document, ExtractedAnswer, GeneratedAnswer
 
 
 __all__ = [
@@ -12,8 +12,10 @@
     "DeserializationError",
     "ComponentError",
     "Pipeline",
-    "Document",
     "Answer",
-    "GeneratedAnswer",
+    "ByteStream",
+    "ChatMessage",
+    "Document",
     "ExtractedAnswer",
+    "GeneratedAnswer",
 ]