update: new types

Author: maksutovic

Sources/CompilerSwiftAI/Model Calling/ChatHistory.swift

@@ -28,28 +28,28 @@ actor ChatHistory {
     }
 
     init(systemPrompt: String) {
-        self._messages = [Message(role: .system, content: systemPrompt)]
+        self._messages = [Message(role: .system, content: [.text(systemPrompt)])]
     }
 
     func notifyMessageUpdate() {
         continuation?.yield(_messages)
     }
 
     func addUserMessage(_ content: String) {
-        _messages.append(Message(role: .user, content: content))
+        _messages.append(Message(role: .user, content: [.text(content)]))
         notifyMessageUpdate()
     }
 
     func addAssistantMessage(_ content: String) {
-        _messages.append(Message(role: .assistant, content: content))
+        _messages.append(Message(role: .assistant, content: [.text(content)]))
         notifyMessageUpdate()
     }
 
     /// Start a new streaming response from the assistant
     @discardableResult
     func beginStreamingResponse() -> UUID {
         let id = UUID()
-        let msg = Message(id: id, role: .assistant, content: "", state: .streaming(""))
+        let msg = Message(id: id, role: .assistant, content: [.text("")], state: .streaming(""))
         _messages.append(msg)
         messageID = id
         notifyMessageUpdate()
@@ -66,7 +66,7 @@ actor ChatHistory {
         _messages[idx] = Message(
             id: old.id,
             role: old.role,
-            content: partial,
+            content: [.text(partial)],
             state: .streaming(partial)
         )
         notifyMessageUpdate()
@@ -81,7 +81,7 @@ actor ChatHistory {
         _messages[idx] = Message(
             id: id,
             role: .assistant,
-            content: finalContent,
+            content: [.text(finalContent)],
             state: .complete
         )
         messageID = nil