From 9e2a3d898895ef238269ab331fa06c2445e096ef Mon Sep 17 00:00:00 2001 From: di-sukharev Date: Tue, 7 Mar 2023 16:16:12 +0800 Subject: [PATCH] getMessagesPromisesByLines --- src/generateCommitMessageFromGitDiff.ts | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/src/generateCommitMessageFromGitDiff.ts b/src/generateCommitMessageFromGitDiff.ts index c3fb638..87ca579 100644 --- a/src/generateCommitMessageFromGitDiff.ts +++ b/src/generateCommitMessageFromGitDiff.ts @@ -122,3 +122,20 @@ export const generateCommitMessageWithChatCompletion = async ( return { error: GenerateCommitMessageErrorEnum.internalError }; } }; + +function getMessagesPromisesByLines(fileDiff: string, separator: string) { + const [fileHeader, ...fileDiffByLines] = fileDiff.split('\n@@'); + const lineDiffsWithHeader = fileDiffByLines.map( + (d) => fileHeader + '\n@@' + d + ); + + const mergedLines = mergeStrings(lineDiffsWithHeader, MAX_REQ_TOKENS); + + const commitMsgsFromFileLineDiffs = mergedLines.map((d) => { + const messages = generateCommitMessageChatCompletionPrompt(separator + d); + + return api.generateCommitMessage(messages); + }); + + return commitMsgsFromFileLineDiffs; +}