Make ChatMessage Sendable

Author: intitni

Tool/Sources/ChatBasic/ChatMessage.swift

@@ -1,12 +1,12 @@
-import CodableWrappers
+@preconcurrency import CodableWrappers
 import Foundation
 
 /// A chat message that can be sent or received.
-public struct ChatMessage: Equatable, Codable {
+public struct ChatMessage: Equatable, Codable, Sendable {
     public typealias ID = String
 
     /// The role of a message.
-    public enum Role: String, Codable, Equatable {
+    public enum Role: String, Codable, Equatable, Sendable {
         case system
         case user
         case assistant
@@ -15,7 +15,7 @@ public struct ChatMessage: Equatable, Codable {
     }
 
     /// A function call that can be made by the bot.
-    public struct FunctionCall: Codable, Equatable {
+    public struct FunctionCall: Codable, Equatable, Sendable {
         /// The name of the function.
         public var name: String
         /// Arguments in the format of a JSON string.
@@ -27,7 +27,7 @@ public struct ChatMessage: Equatable, Codable {
     }
 
     /// A tool call that can be made by the bot.
-    public struct ToolCall: Codable, Equatable, Identifiable {
+    public struct ToolCall: Codable, Equatable, Identifiable, Sendable {
         public var id: String
         /// The type of tool call.
         public var type: String
@@ -49,7 +49,7 @@ public struct ChatMessage: Equatable, Codable {
     }
 
     /// The response of a tool call
-    public struct ToolCallResponse: Codable, Equatable {
+    public struct ToolCallResponse: Codable, Equatable, Sendable {
         /// The content of the response.
         public var content: String
         /// The summary of the response to display in UI.
@@ -61,10 +61,10 @@ public struct ChatMessage: Equatable, Codable {
     }
 
     /// A reference to include in a chat message.
-    public struct Reference: Codable, Equatable, Identifiable {
+    public struct Reference: Codable, Equatable, Identifiable, Sendable {
         /// The kind of reference.
-        public enum Kind: Codable, Equatable {
-            public enum Symbol: String, Codable {
+        public enum Kind: Codable, Equatable, Sendable {
+            public enum Symbol: String, Codable, Sendable {
                 case `class`
                 case `struct`
                 case `enum`
@@ -77,6 +77,7 @@ public struct ChatMessage: Equatable, Codable {
                 case function
                 case method
             }
+
             /// Code symbol.
             case symbol(Symbol, uri: String, startLine: Int?, endLine: Int?)
             /// Some text.
@@ -138,10 +139,10 @@ public struct ChatMessage: Equatable, Codable {
 
     /// The id of the message.
     public var id: ID
-    
+
     /// The id of the sender of the message.
     public var senderId: String?
-    
+
     /// The id of the message that this message is a response to.
     public var responseTo: ID?
 
@@ -151,7 +152,7 @@ public struct ChatMessage: Equatable, Codable {
     /// The references of this message.
     @FallbackDecoding<EmptyArray<Reference>>
     public var references: [Reference]
-    
+
     /// Cache the message in the prompt if possible.
     public var cacheIfPossible: Bool
 
@@ -190,15 +191,15 @@ public struct ChatMessage: Equatable, Codable {
     }
 }
 
-public struct ReferenceKindFallback: FallbackValueProvider {
+public struct ReferenceKindFallback: FallbackValueProvider, Sendable {
     public static var defaultValue: ChatMessage.Reference.Kind { .other(kind: "Unknown") }
 }
 
-public struct ReferenceIDFallback: FallbackValueProvider {
+public struct ReferenceIDFallback: FallbackValueProvider, Sendable {
     public static var defaultValue: String { UUID().uuidString }
 }
 
-public struct ChatMessageRoleFallback: FallbackValueProvider {
+public struct ChatMessageRoleFallback: FallbackValueProvider, Sendable {
     public static var defaultValue: ChatMessage.Role { .user }
 }