add sentToLLM to ModelingStore

Author: robertbrignull

extensions/ql-vscode/src/common/interface-types.ts

@@ -543,6 +543,11 @@ interface SetInProgressMethodsMessage {
   methods: string[];
 }
 
+interface SetSentToLLMMethodsMessage {
+  t: "setSentToLLMMethods";
+  methods: string[];
+}
+
 interface SwitchModeMessage {
   t: "switchMode";
   mode: Mode;
@@ -626,6 +631,7 @@ export type ToModelEditorMessage =
   | SetModeledMethodsMessage
   | SetModifiedMethodsMessage
   | SetInProgressMethodsMessage
+  | SetSentToLLMMethodsMessage
   | RevealMethodMessage
   | SetAccessPathSuggestionsMessage;
 

extensions/ql-vscode/src/model-editor/auto-modeler.ts

@@ -159,6 +159,12 @@ export class AutoModeler {
             this.databaseItem,
             candidateSignatures,
           );
+
+          // Let the UI know which methods have been sent to the LLM
+          this.modelingStore.addSentToLLMMethods(
+            this.databaseItem,
+            candidateSignatures,
+          );
         }
       } finally {
         // Clear out in progress methods in case anything went wrong

extensions/ql-vscode/src/model-editor/model-editor-view.ts

@@ -835,6 +835,17 @@ export class ModelEditorView extends AbstractWebview<
       }),
     );
 
+    this.push(
+      this.modelingEvents.onSentToLLMMethodsChanged(async (event) => {
+        if (event.dbUri === this.databaseItem.databaseUri.toString()) {
+          await this.postMessage({
+            t: "setSentToLLMMethods",
+            methods: Array.from(event.methods),
+          });
+        }
+      }),
+    );
+
     this.push(
       this.modelingEvents.onRevealInModelEditor(async (event) => {
         if (event.dbUri === this.databaseItem.databaseUri.toString()) {

extensions/ql-vscode/src/model-editor/modeling-events.ts

@@ -42,13 +42,19 @@ interface SelectedMethodChangedEvent {
   readonly modeledMethods: readonly ModeledMethod[];
   readonly isModified: boolean;
   readonly isInProgress: boolean;
+  readonly hasBeenSentToLLM: boolean;
 }
 
 interface InProgressMethodsChangedEvent {
   readonly dbUri: string;
   readonly methods: ReadonlySet<string>;
 }
 
+interface SentToLLMMethodsChangedEvent {
+  readonly dbUri: string;
+  readonly methods: ReadonlySet<string>;
+}
+
 interface RevealInModelEditorEvent {
   dbUri: string;
   method: Method;
@@ -69,6 +75,7 @@ export class ModelingEvents extends DisposableObject {
   public readonly onModifiedMethodsChanged: AppEvent<ModifiedMethodsChangedEvent>;
   public readonly onSelectedMethodChanged: AppEvent<SelectedMethodChangedEvent>;
   public readonly onInProgressMethodsChanged: AppEvent<InProgressMethodsChangedEvent>;
+  public readonly onSentToLLMMethodsChanged: AppEvent<SentToLLMMethodsChangedEvent>;
   public readonly onRevealInModelEditor: AppEvent<RevealInModelEditorEvent>;
   public readonly onFocusModelEditor: AppEvent<FocusModelEditorEvent>;
 
@@ -82,6 +89,7 @@ export class ModelingEvents extends DisposableObject {
   private readonly onModifiedMethodsChangedEventEmitter: AppEventEmitter<ModifiedMethodsChangedEvent>;
   private readonly onSelectedMethodChangedEventEmitter: AppEventEmitter<SelectedMethodChangedEvent>;
   private readonly onInProgressMethodsChangedEventEmitter: AppEventEmitter<InProgressMethodsChangedEvent>;
+  private readonly onSentToLLMMethodsChangedEventEmitter: AppEventEmitter<SentToLLMMethodsChangedEvent>;
   private readonly onRevealInModelEditorEventEmitter: AppEventEmitter<RevealInModelEditorEvent>;
   private readonly onFocusModelEditorEventEmitter: AppEventEmitter<FocusModelEditorEvent>;
 
@@ -141,6 +149,12 @@ export class ModelingEvents extends DisposableObject {
     this.onInProgressMethodsChanged =
       this.onInProgressMethodsChangedEventEmitter.event;
 
+    this.onSentToLLMMethodsChangedEventEmitter = this.push(
+      app.createEventEmitter<SentToLLMMethodsChangedEvent>(),
+    );
+    this.onSentToLLMMethodsChanged =
+      this.onSentToLLMMethodsChangedEventEmitter.event;
+
     this.onRevealInModelEditorEventEmitter = this.push(
       app.createEventEmitter<RevealInModelEditorEvent>(),
     );
@@ -226,6 +240,7 @@ export class ModelingEvents extends DisposableObject {
     modeledMethods: ModeledMethod[],
     isModified: boolean,
     isInProgress: boolean,
+    hasBeenSentToLLM: boolean,
   ) {
     this.onSelectedMethodChangedEventEmitter.fire({
       databaseItem,
@@ -234,6 +249,7 @@ export class ModelingEvents extends DisposableObject {
       modeledMethods,
       isModified,
       isInProgress,
+      hasBeenSentToLLM,
     });
   }
 
@@ -247,6 +263,16 @@ export class ModelingEvents extends DisposableObject {
     });
   }
 
+  public fireSentToLLMMethodsChangedEvent(
+    dbUri: string,
+    methods: ReadonlySet<string>,
+  ) {
+    this.onSentToLLMMethodsChangedEventEmitter.fire({
+      dbUri,
+      methods,
+    });
+  }
+
   public fireRevealInModelEditorEvent(dbUri: string, method: Method) {
     this.onRevealInModelEditorEventEmitter.fire({
       dbUri,

extensions/ql-vscode/src/model-editor/modeling-store.ts

@@ -14,6 +14,7 @@ interface InternalDbModelingState {
   modeledMethods: Record<string, ModeledMethod[]>;
   modifiedMethodSignatures: Set<string>;
   inProgressMethods: Set<string>;
+  sentToLLMMethods: Set<string>;
   selectedMethod: Method | undefined;
   selectedUsage: Usage | undefined;
 }
@@ -26,6 +27,7 @@ interface DbModelingState {
   readonly modeledMethods: Readonly<Record<string, readonly ModeledMethod[]>>;
   readonly modifiedMethodSignatures: ReadonlySet<string>;
   readonly inProgressMethods: ReadonlySet<string>;
+  readonly sentToLLMMethods: ReadonlySet<string>;
   readonly selectedMethod: Method | undefined;
   readonly selectedUsage: Usage | undefined;
 }
@@ -37,6 +39,7 @@ interface SelectedMethodDetails {
   readonly modeledMethods: readonly ModeledMethod[];
   readonly isModified: boolean;
   readonly isInProgress: boolean;
+  readonly hasBeenSentToLLM: boolean;
 }
 
 export class ModelingStore extends DisposableObject {
@@ -59,6 +62,7 @@ export class ModelingStore extends DisposableObject {
       mode,
       modeledMethods: {},
       modifiedMethodSignatures: new Set(),
+      sentToLLMMethods: new Set(),
       selectedMethod: undefined,
       selectedUsage: undefined,
       inProgressMethods: new Set(),
@@ -301,13 +305,15 @@ export class ModelingStore extends DisposableObject {
     const modeledMethods = dbState.modeledMethods[method.signature] ?? [];
     const isModified = dbState.modifiedMethodSignatures.has(method.signature);
     const isInProgress = dbState.inProgressMethods.has(method.signature);
+    const hasBeenSentToLLM = dbState.sentToLLMMethods.has(method.signature);
     this.modelingEvents.fireSelectedMethodChangedEvent(
       dbItem,
       method,
       usage,
       modeledMethods,
       isModified,
       isInProgress,
+      hasBeenSentToLLM,
     );
   }
 
@@ -336,6 +342,15 @@ export class ModelingStore extends DisposableObject {
     });
   }
 
+  public addSentToLLMMethods(dbItem: DatabaseItem, sentToLLMMethods: string[]) {
+    this.changeSentToLLMMethods(dbItem, (state) => {
+      state.sentToLLMMethods = new Set([
+        ...state.sentToLLMMethods,
+        ...sentToLLMMethods,
+      ]);
+    });
+  }
+
   public getSelectedMethodDetails(): SelectedMethodDetails | undefined {
     const dbState = this.getInternalStateForActiveDb();
     if (!dbState) {
@@ -356,6 +371,7 @@ export class ModelingStore extends DisposableObject {
         selectedMethod.signature,
       ),
       isInProgress: dbState.inProgressMethods.has(selectedMethod.signature),
+      hasBeenSentToLLM: dbState.sentToLLMMethods.has(selectedMethod.signature),
     };
   }
 
@@ -412,4 +428,18 @@ export class ModelingStore extends DisposableObject {
       state.inProgressMethods,
     );
   }
+
+  private changeSentToLLMMethods(
+    dbItem: DatabaseItem,
+    updateState: (state: InternalDbModelingState) => void,
+  ) {
+    const state = this.getState(dbItem);
+
+    updateState(state);
+
+    this.modelingEvents.fireSentToLLMMethodsChangedEvent(
+      dbItem.databaseUri.toString(),
+      state.sentToLLMMethods,
+    );
+  }
 }

extensions/ql-vscode/src/view/model-editor/ModelEditor.tsx

@@ -148,6 +148,10 @@ export function ModelEditor({
             setInProgressMethods(new Set(msg.methods));
             break;
           }
+          case "setSentToLLMMethods": {
+            // TODO: set state
+            break;
+          }
           case "revealMethod":
             setRevealedMethodSignature(msg.methodSignature);
             break;