Merge pull request #13 from Compiler-Inc/feat/add-tool-call-content-type

feat: add tool call content type

Author: AtharvaVaidya

Sources/CompilerSwiftAI/Model Calling/Domain Models/ChatCompletionChunk.swift

@@ -66,7 +66,7 @@ public struct ChatDelta: Decodable, Sendable {
 }
 
 /// Represents a delta for a tool call in a streamed response
-public struct ToolCallDelta: Decodable, Sendable {
+public struct ToolCallDelta: Codable, Sendable, Equatable {
     /// The index of this tool call in the array
     public let index: Int
     /// The ID of the tool call (only present in first delta)
@@ -78,7 +78,7 @@ public struct ToolCallDelta: Decodable, Sendable {
 }
 
 /// Represents a delta for a function call in a streamed response
-public struct FunctionCallDelta: Decodable, Sendable {
+public struct FunctionCallDelta: Codable, Sendable, Equatable {
     /// The name of the function (only present in first delta)
     public let name: String?
     /// The arguments being built up

Sources/CompilerSwiftAI/Model Calling/Domain Models/Message.swift

@@ -18,18 +18,24 @@ public struct Message: Sendable, Equatable {
     public enum Content: Sendable, Equatable {
         case text(String)
         case image(String)
+        case toolCall(ToolCallDelta)
+        case toolCallResult(String)
 
         var type: ContentType {
             switch self {
             case .text: return .text
             case .image: return .imageUrl
+            case .toolCall: return .text
+            case .toolCallResult: return .text
             }
         }
 
         var value: String {
             switch self {
             case .text(let text): return text
             case .image(let url): return url
+            case .toolCall(let toolCall): return "Function call: \(toolCall.function?.name ?? "")"
+            case .toolCallResult(let result): return result
             }
         }
     }

Sources/CompilerSwiftAI/Network Models/MessageDTO.swift

@@ -1,22 +1,26 @@
-//  Copyright © 2025 Compiler, Inc. All rights reserved.
+//  Copyright 2025 Compiler, Inc. All rights reserved.
 
 import Foundation
 
 public struct MessageDTO: Codable, Sendable {
     public enum ContentType: String, Codable, Sendable {
         case text = "text"
         case imageUrl = "image_url"
+        case toolCall = "tool_call"
+        case toolCallResult = "tool_call_result"
     }
 
     public struct Content: Codable, Sendable {
         let type: ContentType
         let text: String?
         let imageUrl: ImageUrl?
+        let toolCall: ToolCallDelta?
 
         private enum CodingKeys: String, CodingKey {
             case type
             case text
             case imageUrl = "image_url"
+            case toolCall = "tool_call"
         }
     }
 
@@ -34,9 +38,13 @@ public struct MessageDTO: Codable, Sendable {
         self.content = message.content.map { content in
             switch content {
             case .text(let text):
-                return Content(type: .text, text: text, imageUrl: nil)
+                return Content(type: .text, text: text, imageUrl: nil, toolCall: nil)
             case .image(let url):
-                return Content(type: .imageUrl, text: nil, imageUrl: ImageUrl(url: url))
+                return Content(type: .imageUrl, text: nil, imageUrl: ImageUrl(url: url), toolCall: nil)
+            case .toolCall(let delta):
+                return Content(type: .toolCall, text: nil, imageUrl: nil, toolCall: delta)
+            case .toolCallResult(let result):
+                return Content(type: .toolCallResult, text: result, imageUrl: nil, toolCall: nil)
             }
         }
     }
@@ -46,12 +54,16 @@ public struct MessageDTO: Codable, Sendable {
             id: id,
             role: .init(rawValue: role) ?? .user,
             content: content.compactMap { content in
-                if let text = content.text {
-                    return .text(text)
-                } else if let imageUrl = content.imageUrl {
-                    return .image(imageUrl.url)
+                switch content.type {
+                case .text:
+                    return content.text.map { .text($0) }
+                case .imageUrl:
+                    return content.imageUrl.map { .image($0.url) }
+                case .toolCall:
+                    return content.toolCall.map { .toolCall($0) }
+                case .toolCallResult:
+                    return content.text.map { .toolCallResult($0) }
                 }
-                return nil
             }
         )
     }