Function getCellContentTextRequiredForBigCode

src/utility.js:266–304

(currctJupyterModel)

function getCellContentTextRequiredForBigCode(currctJupyterModel) {
    const context = getActiveContext(currctJupyterModel)

    if(context.length == 0){
        return [null, null]
    }


    let prompt = "<start_jupyter>"
    let cellCodeText = ""
    let isLastLine = true

    for (let index = 0; index < context.length; index++) {
        const lineInformation = context[index]

        if (lineInformation.isCursor) {
            prompt += bigcodeFormattPrefixMap[lineInformation.type] + cellCodeText + lineInformation.content
            isLastLine = judgeIsTheLastLine(context, index)
            break
        }

        if (index == context.length - 1) {
            prompt += bigcodeFormattPrefixMap[lineInformation.type] + cellCodeText + lineInformation.content
            break
        }

        const nextLineInformation = context[index + 1]

        if (lineInformation.cellIndex != nextLineInformation.cellIndex || lineInformation.type != nextLineInformation.type) {
            prompt += bigcodeFormattPrefixMap[lineInformation.type] + cellCodeText + lineInformation.content
            cellCodeText = ""
        } else {
            cellCodeText += lineInformation.content + "\n"
        }

    }

    return [prompt, isLastLine]
}