refactor: marks images as optional (#32)

kevinhermawan · Jul 30, 2024 · 85a7c55 · 85a7c55
1 parent 38f738d
commit 85a7c55
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Sources/OllamaKit/RequestData/OKChatRequestData.swift b/Sources/OllamaKit/RequestData/OKChatRequestData.swift
@@ -39,9 +39,9 @@ public struct OKChatRequestData: Encodable {
         public let content: String
 
         /// An optional array of base64-encoded images.
-        public let images: [String]
+        public let images: [String]?
 
-        public init(role: Role, content: String, images: [String] = []) {
+        public init(role: Role, content: String, images: [String]? = nil) {
             self.role = role
             self.content = content
             self.images = images

diff --git a/Sources/OllamaKit/RequestData/OKGenerateRequestData.swift b/Sources/OllamaKit/RequestData/OKGenerateRequestData.swift
@@ -18,7 +18,7 @@ public struct OKGenerateRequestData: Encodable {
     public let prompt: String
 
     /// /// An optional array of base64-encoded images.
-    public let images: [String]
+    public let images: [String]?
 
     /// An optional string specifying the system message.
     public var system: String?
@@ -29,7 +29,7 @@ public struct OKGenerateRequestData: Encodable {
     /// Optional ``OKCompletionOptions`` providing additional configuration for the generation request.
     public var options: OKCompletionOptions?
 
-    public init(model: String, prompt: String, images: [String] = []) {
+    public init(model: String, prompt: String, images: [String]? = nil) {
         self.stream = true
         self.model = model
         self.prompt = prompt