Message improvements

Author: pblazej

VoiceAgent/Agent/AgentSession.swift

@@ -143,14 +143,17 @@ final class AgentSession: ObservableObject {
         error = nil
     }
 
-    func send(message: SentMessage) async {
+    @discardableResult
+    func send(text: String) async -> SentMessage {
+        let message = SentMessage(id: UUID().uuidString, timestamp: Date(), content: .userText(text))
         do {
             for sender in senders {
                 try await sender.send(message)
             }
         } catch {
             self.error = .failedToSend(error)
         }
+        return message
     }
 
     func getMessageHistory() -> [ReceivedMessage] {

VoiceAgent/Chat/Message.swift

@@ -9,6 +9,7 @@ struct ReceivedMessage: Identifiable, Equatable, Codable, Sendable {
     enum Content: Equatable, Codable, Sendable {
         case agentTranscript(String)
         case userTranscript(String)
+        case userText(String)
     }
 }
 

VoiceAgent/Chat/Send/TextMessageSender.swift

@@ -26,7 +26,7 @@ actor TextMessageSender: MessageSender, MessageReceiver {
         let loopbackMessage = ReceivedMessage(
             id: message.id,
             timestamp: message.timestamp,
-            content: .userTranscript(text)
+            content: .userText(text)
         )
 
         messageContinuation?.yield(loopbackMessage)

VoiceAgent/Chat/View/ChatTextInputView.swift

@@ -74,6 +74,6 @@ struct ChatTextInputView: View {
             messageText = ""
             keyboardFocus = false
         }
-        await session.send(message: SentMessage(id: UUID().uuidString, timestamp: Date(), content: .userText(messageText)))
+        await session.send(text: messageText)
     }
 }

VoiceAgent/Chat/View/ChatView.swift

@@ -10,7 +10,7 @@ struct ChatView: View {
     private func message(_ message: ReceivedMessage) -> some View {
         ZStack {
             switch message.content {
-            case let .userTranscript(text):
+            case let .userTranscript(text), let .userText(text):
                 userTranscript(text)
             case let .agentTranscript(text):
                 agentTranscript(text)