Skip to content

Commit 3565c1e

Browse files
authored
Use convertTranscriptToMLXChat in MLXLanguageModel streamResponse (#67)
1 parent 2db5fd1 commit 3565c1e

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

Sources/AnyLanguageModel/Models/MLXLanguageModel.swift

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -189,14 +189,8 @@ import Foundation
189189

190190
let generateParameters = toGenerateParameters(options)
191191

192-
var chat: [MLXLMCommon.Chat.Message] = []
193-
194-
if let instructionSegments = extractInstructionSegments(from: session) {
195-
chat.append(convertSegmentsToMLXSystemMessage(instructionSegments))
196-
}
197-
198-
let userSegments = extractPromptSegments(from: session, fallbackText: prompt.description)
199-
chat.append(convertSegmentsToMLXMessage(userSegments))
192+
// Build chat history from full transcript
193+
let chat = convertTranscriptToMLXChat(session: session, fallbackPrompt: prompt.description)
200194

201195
let userInput = MLXLMCommon.UserInput(
202196
chat: chat,

0 commit comments

Comments
 (0)