add document content to support pdf

Author: fumito-ito

Sources/AnthropicSwiftSDK/Entity/Content/DocumentContent.swift

@@ -0,0 +1,42 @@
+//
+//  DocumentContent.swift
+//  AnthropicSwiftSDK
+//
+//  Created by 伊藤史 on 2024/11/05.
+//
+import Foundation
+
+public struct DocumentContent {
+    public enum DocumentContentType: String, Codable {
+        case base64
+    }
+
+    public enum DocumentContentMediaType: String, Codable {
+        case pdf = "application/pdf"
+    }
+
+    public let type: DocumentContentType
+    public let mediaType: DocumentContentMediaType
+    public let data: Data
+
+    public init(type: DocumentContentType, mediaType: DocumentContentMediaType, data: Data) {
+        self.type = type
+        self.mediaType = mediaType
+        self.data = data
+    }
+}
+
+extension DocumentContent: Codable {
+    enum CodingKeys: String, CodingKey {
+        case type
+        case mediaType = "media_type"
+        case data
+    }
+
+    public init(from decoder: any Decoder) throws {
+        let container = try decoder.container(keyedBy: CodingKeys.self)
+        self.type = try container.decode(DocumentContent.DocumentContentType.self, forKey: .type)
+        self.mediaType = try container.decode(DocumentContent.DocumentContentMediaType.self, forKey: .mediaType)
+        self.data = try container.decode(Data.self, forKey: .data)
+    }
+}

Tests/AnthropicSwiftSDKTests/Entity/ContentTests.swift

@@ -40,6 +40,26 @@ final class ContentTests: XCTestCase {
         )
     }
 
+    func testEncodeContentDocument() throws {
+        let expect = Content.document(
+            .init(
+                type: .base64,
+                mediaType: .pdf,
+                data: "data".data(using: .utf8)!
+            )
+        )
+
+        let dictionary = try XCTUnwrap(expect.toDictionary(encoder))
+        XCTAssertEqual(dictionary["type"] as! String, "document")
+        let source = try XCTUnwrap(dictionary["source"] as? [String: Any])
+        XCTAssertEqual(source["type"] as! String, "base64")
+        XCTAssertEqual(source["media_type"] as! String, "application/pdf")
+        XCTAssertEqual(
+            source["data"] as! String,
+            "data".data(using: .utf8)!.base64EncodedString()
+        )
+    }
+
     func testEncodeContentToolResult() throws {
         let expect = Content.toolResult(
             .init(