Skip to content

Commit bea4b35

Browse files
committed
Update expectation for MLX image processing test
1 parent e90669b commit bea4b35

1 file changed

Lines changed: 11 additions & 2 deletions

File tree

Tests/AnyLanguageModelTests/MLXLanguageModelTests.swift

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import Testing
44
@testable import AnyLanguageModel
55

66
#if MLX
7+
import MLXLMCommon
78
private let shouldRunMLXTests = {
89
// Enable when explicitly requested via environment variable
910
if ProcessInfo.processInfo.environment["ENABLE_MLX_TESTS"] != nil {
@@ -154,7 +155,11 @@ import Testing
154155
)
155156
])
156157
let session = LanguageModelSession(model: visionModel, transcript: transcript)
157-
let response = try await session.respond(to: "")
158+
var options = GenerationOptions()
159+
var mlxOptions = MLXLanguageModel.CustomGenerationOptions.default
160+
mlxOptions.userInputProcessing = .resize(to: CGSize(width: 512, height: 512))
161+
options[custom: MLXLanguageModel.self] = mlxOptions
162+
let response = try await session.respond(to: "", options: options)
158163
#expect(!response.content.isEmpty)
159164
}
160165

@@ -168,7 +173,11 @@ import Testing
168173
)
169174
])
170175
let session = LanguageModelSession(model: visionModel, transcript: transcript)
171-
let response = try await session.respond(to: "")
176+
var options = GenerationOptions()
177+
var mlxOptions = MLXLanguageModel.CustomGenerationOptions.default
178+
mlxOptions.userInputProcessing = .resize(to: CGSize(width: 512, height: 512))
179+
options[custom: MLXLanguageModel.self] = mlxOptions
180+
let response = try await session.respond(to: "", options: options)
172181
#expect(!response.content.isEmpty)
173182
}
174183

0 commit comments

Comments
 (0)