Added @discardableResult to most of the streaming message methods.

btfranklin · Jun 17, 2023 · 028a2ad · 028a2ad
1 parent 7ca0a00
commit 028a2ad
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/Sources/CleverBird/chat/streaming/StreamableChatThread.swift b/Sources/CleverBird/chat/streaming/StreamableChatThread.swift
@@ -9,23 +9,27 @@ public class StreamableChatThread {
         self.chatThread = chatThread
     }
 
+    @discardableResult
     public func addSystemMessage(_ content: String) -> Self {
-        _ = self.chatThread.addSystemMessage(content)
+        self.chatThread.addSystemMessage(content)
         return self
     }
 
+    @discardableResult
     public func addUserMessage(_ content: String) -> Self {
-        _ = self.chatThread.addUserMessage(content)
+        self.chatThread.addUserMessage(content)
         return self
     }
 
+    @discardableResult
     public func addAssistantMessage(_ content: String) -> Self {
-        _ = self.chatThread.addAssistantMessage(content)
+        self.chatThread.addAssistantMessage(content)
         return self
     }
 
+    @discardableResult
     public func addMessage(_ message: ChatMessage) -> Self {
-        _ = self.chatThread.addMessage(message)
+        self.chatThread.addMessage(message)
         return self
     }