Don’t store originalDocument

src/commands.ts

@@ -58,10 +58,7 @@ export function acceptSuggestionCommand(view: EditorView) {
 }
 
 /**
- * Rejecting a suggestion: this looks at the currently-shown suggestion
- * and reverses it, clears the suggestion, and makes sure
- * that we don't add that clearing transaction to history and we don't
- * trigger a new suggestion because of it.
+ * Cycle through suggested AI completed code.
  */
 export const nextSuggestionCommand: Command = (view: EditorView) => {
   const { state } = view;

src/completionDecoration.ts

@@ -26,7 +26,7 @@ export const completionDecoration = StateField.define<CompletionState>({
   update(state: CompletionState, transaction: Transaction) {
     for (const effect of transaction.effects) {
       if (effect.is(addSuggestions)) {
-        const { changeSpecs, index, originalDocument } = effect.value;
+        const { changeSpecs, index } = effect.value;
 
         // NOTE: here we're adjusting the decoration range
         // to refer to locations in the document _after_ we've
@@ -45,7 +45,6 @@ export const completionDecoration = StateField.define<CompletionState>({
           index,
           decorations,
           changeSpecs,
-          originalDocument,
           reverseChangeSet: effect.value.reverseChangeSet,
         };
       }

src/plugin.ts

@@ -101,6 +101,7 @@ export {
   copilotIgnore,
   codeiumConfig,
   codeiumOtherDocumentsConfig,
+  nextSuggestionCommand,
   type CodeiumOtherDocumentsConfig,
   type CodeiumConfig,
 };

src/types.ts

@@ -1,4 +1,4 @@
-import type { ChangeSet, Text } from "@codemirror/state";
+import type { ChangeSet } from "@codemirror/state";
 import type { DecorationSet } from "@codemirror/view";
 
 /**
@@ -7,7 +7,6 @@ import type { DecorationSet } from "@codemirror/view";
  */
 export type CompletionState = null | {
   index: number;
-  originalDocument: Text;
   reverseChangeSet: ChangeSet;
   changeSpecs: SimpleChangeSpec[][];
   decorations: DecorationSet;