Skip to content

Commit 9cf39e7

Browse files
committed
Do not require images and videos in init(messages:)
1 parent 11fb312 commit 9cf39e7

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

Libraries/MLXLMCommon/UserInput.swift

+7-3
Original file line numberDiff line numberDiff line change
@@ -151,13 +151,17 @@ public struct UserInput: Sendable {
151151
}
152152

153153
public init(
154-
messages: [Chat.Message], images: [Image] = [Image](), videos: [Video] = [Video](),
154+
messages: [Chat.Message],
155155
tools: [ToolSpec]? = nil,
156156
additionalContext: [String: Any]? = nil
157157
) {
158158
self.prompt = .chat(messages)
159-
self.images = images
160-
self.videos = videos
159+
self.images = messages.reduce(into: []) { result, message in
160+
result.append(contentsOf: message.images)
161+
}
162+
self.videos = messages.reduce(into: []) { result, message in
163+
result.append(contentsOf: message.videos)
164+
}
161165
self.tools = tools
162166
self.additionalContext = additionalContext
163167
}

0 commit comments

Comments
 (0)