Function processNextEditData

core/nextEdit/context/processNextEditData.ts:43–151

({
  filePath,
  beforeContent,
  afterContent,
  cursorPosBeforeEdit,
  cursorPosAfterPrevEdit,
  ide,
  configHandler,
  getDefinitionsFromLsp,
  recentlyEditedRanges,
  recentlyVisitedRanges,
  workspaceDir,
  modelNameOrInstance,
}: ProcessNextEditDataParams)

}

export const processNextEditData = async ({
  filePath,
  beforeContent,
  afterContent,
  cursorPosBeforeEdit,
  cursorPosAfterPrevEdit,
  ide,
  configHandler,
  getDefinitionsFromLsp,
  recentlyEditedRanges,
  recentlyVisitedRanges,
  workspaceDir,
  modelNameOrInstance,
}: ProcessNextEditDataParams) => {
  // To switch to the user's autocomplete model, uncomment the following lines
  // const { config } = await configHandler.loadConfig();
  // const autocompleteModel =
  //   (modelNameOrInstance || config?.selectedModelByRole.autocomplete) ??
  //   undefined;

  const modelName = "Codestral";
  const modelProvider = "mistral";
  const maxPromptTokens = randomNumberBetween(500, 12000);

  const autocompleteContext = await getAutocompleteContext(
    filePath,
    cursorPosBeforeEdit,
    ide,
    configHandler,
    getDefinitionsFromLsp,
    recentlyEditedRanges,
    recentlyVisitedRanges,
    maxPromptTokens,
    beforeContent,
    modelName,
  );

  NextEditProvider.getInstance().addAutocompleteContext(autocompleteContext);

  // console.log(
  //   createDiff(beforeContent, afterContent, filePath, DiffFormatType.Unified),
  // );

  let filenamesAndDiffs: filenameAndDiff[] = [];

  const timestamp = Date.now();
  let prevEdits: prevEdit[] = getPrevEditsDescending(); // edits from most to least recent
  if (prevEdits.length > 0) {
    // if last edit was 10+ minutes ago or the workspace changed, forget previous edits
    if (
      timestamp - prevEdits[0].timestamp >= 1000 * 60 * 10 ||
      workspaceDir !== prevEdits[0].workspaceUri
    ) {
      prevEditLruCache.clear();
      prevEdits = [];
    }

    // extract filenames and diffs for logging