ChatThread now automatically appends a completion message to itself.

btfranklin · May 6, 2023 · 8767527 · 8767527
1 parent 0b27339
commit 8767527
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Sources/CleverBird/chat/ChatThread+complete.swift b/Sources/CleverBird/chat/ChatThread+complete.swift
@@ -18,8 +18,14 @@ extension ChatThread {
             let request = try await self.connection.createRequest(for: requestBody)
             let response = try await self.connection.client.send(request)
             let completion = response.value
+            let firstChoiceMessage = completion.choices.first?.message
 
-            return completion.choices.first?.message
+            // Append the response message to the thread
+            if let firstChoiceMessage {
+                _ = addMessage(firstChoiceMessage)
+            }
+
+            return firstChoiceMessage
         } catch {
             logger("Error executing request: \(error.localizedDescription)")
             return nil