| 5 | |
| 6 | |
| 7 | const cleanUpBigcodeOutput = (suggestion, isLastLine) => { |
| 8 | suggestion = suggestion.replace("# -*- coding: utf-8 -*-\n\n", "") |
| 9 | let outPutIndex = suggestion.indexOf('<jupyter_output>') |
| 10 | |
| 11 | if (outPutIndex != -1){ |
| 12 | suggestion = suggestion.slice(0, outPutIndex) |
| 13 | } |
| 14 | |
| 15 | const unnecessaryTag = '<|endoftext|>' |
| 16 | const unnecessaryTagIndex = suggestion.indexOf(unnecessaryTag) |
| 17 | |
| 18 | |
| 19 | suggestion = unnecessaryTagIndex == -1 ? suggestion : suggestion.slice(0, unnecessaryTagIndex) |
| 20 | |
| 21 | return isLastLine ? suggestion : suggestion.replace("\n","") |
| 22 | } |
| 23 | |
| 24 | |
| 25 | const cleanUpOpenaiOutput = (suggestion, isLastLine) => { |