Skip to content

Commit f98c531

Browse files
Rename to processedByAutoModel
1 parent 4c4820f commit f98c531

File tree

6 files changed

+49
-40
lines changed

6 files changed

+49
-40
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -543,8 +543,8 @@ interface SetInProgressMethodsMessage {
543543
methods: string[];
544544
}
545545

546-
interface SetSentToLLMMethodsMessage {
547-
t: "setSentToLLMMethods";
546+
interface SetProcessedByAutoModelMethodsMessage {
547+
t: "setProcessedByAutoModelMethods";
548548
methods: string[];
549549
}
550550

@@ -631,7 +631,7 @@ export type ToModelEditorMessage =
631631
| SetModeledMethodsMessage
632632
| SetModifiedMethodsMessage
633633
| SetInProgressMethodsMessage
634-
| SetSentToLLMMethodsMessage
634+
| SetProcessedByAutoModelMethodsMessage
635635
| RevealMethodMessage
636636
| SetAccessPathSuggestionsMessage;
637637

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ export class AutoModeler {
161161
);
162162

163163
// Let the UI know which methods have been sent to the LLM
164-
this.modelingStore.addSentToLLMMethods(
164+
this.modelingStore.addProcessedByAutoModelMethods(
165165
this.databaseItem,
166166
candidateSignatures,
167167
);

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

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -836,14 +836,16 @@ export class ModelEditorView extends AbstractWebview<
836836
);
837837

838838
this.push(
839-
this.modelingEvents.onSentToLLMMethodsChanged(async (event) => {
840-
if (event.dbUri === this.databaseItem.databaseUri.toString()) {
841-
await this.postMessage({
842-
t: "setSentToLLMMethods",
843-
methods: Array.from(event.methods),
844-
});
845-
}
846-
}),
839+
this.modelingEvents.onProcessedByAutoModelMethodsChanged(
840+
async (event) => {
841+
if (event.dbUri === this.databaseItem.databaseUri.toString()) {
842+
await this.postMessage({
843+
t: "setProcessedByAutoModelMethods",
844+
methods: Array.from(event.methods),
845+
});
846+
}
847+
},
848+
),
847849
);
848850

849851
this.push(

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

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -42,15 +42,15 @@ interface SelectedMethodChangedEvent {
4242
readonly modeledMethods: readonly ModeledMethod[];
4343
readonly isModified: boolean;
4444
readonly isInProgress: boolean;
45-
readonly hasBeenSentToLLM: boolean;
45+
readonly processedByAutoModel: boolean;
4646
}
4747

4848
interface InProgressMethodsChangedEvent {
4949
readonly dbUri: string;
5050
readonly methods: ReadonlySet<string>;
5151
}
5252

53-
interface SentToLLMMethodsChangedEvent {
53+
interface ProcessedByAutoModelMethodsChangedEvent {
5454
readonly dbUri: string;
5555
readonly methods: ReadonlySet<string>;
5656
}
@@ -75,7 +75,7 @@ export class ModelingEvents extends DisposableObject {
7575
public readonly onModifiedMethodsChanged: AppEvent<ModifiedMethodsChangedEvent>;
7676
public readonly onSelectedMethodChanged: AppEvent<SelectedMethodChangedEvent>;
7777
public readonly onInProgressMethodsChanged: AppEvent<InProgressMethodsChangedEvent>;
78-
public readonly onSentToLLMMethodsChanged: AppEvent<SentToLLMMethodsChangedEvent>;
78+
public readonly onProcessedByAutoModelMethodsChanged: AppEvent<ProcessedByAutoModelMethodsChangedEvent>;
7979
public readonly onRevealInModelEditor: AppEvent<RevealInModelEditorEvent>;
8080
public readonly onFocusModelEditor: AppEvent<FocusModelEditorEvent>;
8181

@@ -89,7 +89,7 @@ export class ModelingEvents extends DisposableObject {
8989
private readonly onModifiedMethodsChangedEventEmitter: AppEventEmitter<ModifiedMethodsChangedEvent>;
9090
private readonly onSelectedMethodChangedEventEmitter: AppEventEmitter<SelectedMethodChangedEvent>;
9191
private readonly onInProgressMethodsChangedEventEmitter: AppEventEmitter<InProgressMethodsChangedEvent>;
92-
private readonly onSentToLLMMethodsChangedEventEmitter: AppEventEmitter<SentToLLMMethodsChangedEvent>;
92+
private readonly onProcessedByAutoModelMethodsChangedEventEmitter: AppEventEmitter<ProcessedByAutoModelMethodsChangedEvent>;
9393
private readonly onRevealInModelEditorEventEmitter: AppEventEmitter<RevealInModelEditorEvent>;
9494
private readonly onFocusModelEditorEventEmitter: AppEventEmitter<FocusModelEditorEvent>;
9595

@@ -149,11 +149,11 @@ export class ModelingEvents extends DisposableObject {
149149
this.onInProgressMethodsChanged =
150150
this.onInProgressMethodsChangedEventEmitter.event;
151151

152-
this.onSentToLLMMethodsChangedEventEmitter = this.push(
153-
app.createEventEmitter<SentToLLMMethodsChangedEvent>(),
152+
this.onProcessedByAutoModelMethodsChangedEventEmitter = this.push(
153+
app.createEventEmitter<ProcessedByAutoModelMethodsChangedEvent>(),
154154
);
155-
this.onSentToLLMMethodsChanged =
156-
this.onSentToLLMMethodsChangedEventEmitter.event;
155+
this.onProcessedByAutoModelMethodsChanged =
156+
this.onProcessedByAutoModelMethodsChangedEventEmitter.event;
157157

158158
this.onRevealInModelEditorEventEmitter = this.push(
159159
app.createEventEmitter<RevealInModelEditorEvent>(),
@@ -240,7 +240,7 @@ export class ModelingEvents extends DisposableObject {
240240
modeledMethods: ModeledMethod[],
241241
isModified: boolean,
242242
isInProgress: boolean,
243-
hasBeenSentToLLM: boolean,
243+
processedByAutoModel: boolean,
244244
) {
245245
this.onSelectedMethodChangedEventEmitter.fire({
246246
databaseItem,
@@ -249,7 +249,7 @@ export class ModelingEvents extends DisposableObject {
249249
modeledMethods,
250250
isModified,
251251
isInProgress,
252-
hasBeenSentToLLM,
252+
processedByAutoModel,
253253
});
254254
}
255255

@@ -263,11 +263,11 @@ export class ModelingEvents extends DisposableObject {
263263
});
264264
}
265265

266-
public fireSentToLLMMethodsChangedEvent(
266+
public fireProcessedByAutoModelMethodsChangedEvent(
267267
dbUri: string,
268268
methods: ReadonlySet<string>,
269269
) {
270-
this.onSentToLLMMethodsChangedEventEmitter.fire({
270+
this.onProcessedByAutoModelMethodsChangedEventEmitter.fire({
271271
dbUri,
272272
methods,
273273
});

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

Lines changed: 22 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ interface InternalDbModelingState {
1414
modeledMethods: Record<string, ModeledMethod[]>;
1515
modifiedMethodSignatures: Set<string>;
1616
inProgressMethods: Set<string>;
17-
sentToLLMMethods: Set<string>;
17+
processedByAutoModelMethods: Set<string>;
1818
selectedMethod: Method | undefined;
1919
selectedUsage: Usage | undefined;
2020
}
@@ -27,7 +27,7 @@ interface DbModelingState {
2727
readonly modeledMethods: Readonly<Record<string, readonly ModeledMethod[]>>;
2828
readonly modifiedMethodSignatures: ReadonlySet<string>;
2929
readonly inProgressMethods: ReadonlySet<string>;
30-
readonly sentToLLMMethods: ReadonlySet<string>;
30+
readonly processedByAutoModelMethods: ReadonlySet<string>;
3131
readonly selectedMethod: Method | undefined;
3232
readonly selectedUsage: Usage | undefined;
3333
}
@@ -39,7 +39,7 @@ interface SelectedMethodDetails {
3939
readonly modeledMethods: readonly ModeledMethod[];
4040
readonly isModified: boolean;
4141
readonly isInProgress: boolean;
42-
readonly hasBeenSentToLLM: boolean;
42+
readonly processedByAutoModel: boolean;
4343
}
4444

4545
export class ModelingStore extends DisposableObject {
@@ -62,7 +62,7 @@ export class ModelingStore extends DisposableObject {
6262
mode,
6363
modeledMethods: {},
6464
modifiedMethodSignatures: new Set(),
65-
sentToLLMMethods: new Set(),
65+
processedByAutoModelMethods: new Set(),
6666
selectedMethod: undefined,
6767
selectedUsage: undefined,
6868
inProgressMethods: new Set(),
@@ -305,15 +305,17 @@ export class ModelingStore extends DisposableObject {
305305
const modeledMethods = dbState.modeledMethods[method.signature] ?? [];
306306
const isModified = dbState.modifiedMethodSignatures.has(method.signature);
307307
const isInProgress = dbState.inProgressMethods.has(method.signature);
308-
const hasBeenSentToLLM = dbState.sentToLLMMethods.has(method.signature);
308+
const processedByAutoModel = dbState.processedByAutoModelMethods.has(
309+
method.signature,
310+
);
309311
this.modelingEvents.fireSelectedMethodChangedEvent(
310312
dbItem,
311313
method,
312314
usage,
313315
modeledMethods,
314316
isModified,
315317
isInProgress,
316-
hasBeenSentToLLM,
318+
processedByAutoModel,
317319
);
318320
}
319321

@@ -342,11 +344,14 @@ export class ModelingStore extends DisposableObject {
342344
});
343345
}
344346

345-
public addSentToLLMMethods(dbItem: DatabaseItem, sentToLLMMethods: string[]) {
346-
this.changeSentToLLMMethods(dbItem, (state) => {
347-
state.sentToLLMMethods = new Set([
348-
...state.sentToLLMMethods,
349-
...sentToLLMMethods,
347+
public addProcessedByAutoModelMethods(
348+
dbItem: DatabaseItem,
349+
processedByAutoModelMethods: string[],
350+
) {
351+
this.changeProcessedByAutoModelMethods(dbItem, (state) => {
352+
state.processedByAutoModelMethods = new Set([
353+
...state.processedByAutoModelMethods,
354+
...processedByAutoModelMethods,
350355
]);
351356
});
352357
}
@@ -371,7 +376,9 @@ export class ModelingStore extends DisposableObject {
371376
selectedMethod.signature,
372377
),
373378
isInProgress: dbState.inProgressMethods.has(selectedMethod.signature),
374-
hasBeenSentToLLM: dbState.sentToLLMMethods.has(selectedMethod.signature),
379+
processedByAutoModel: dbState.processedByAutoModelMethods.has(
380+
selectedMethod.signature,
381+
),
375382
};
376383
}
377384

@@ -429,17 +436,17 @@ export class ModelingStore extends DisposableObject {
429436
);
430437
}
431438

432-
private changeSentToLLMMethods(
439+
private changeProcessedByAutoModelMethods(
433440
dbItem: DatabaseItem,
434441
updateState: (state: InternalDbModelingState) => void,
435442
) {
436443
const state = this.getState(dbItem);
437444

438445
updateState(state);
439446

440-
this.modelingEvents.fireSentToLLMMethodsChangedEvent(
447+
this.modelingEvents.fireProcessedByAutoModelMethodsChangedEvent(
441448
dbItem.databaseUri.toString(),
442-
state.sentToLLMMethods,
449+
state.processedByAutoModelMethods,
443450
);
444451
}
445452
}

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ export function ModelEditor({
148148
setInProgressMethods(new Set(msg.methods));
149149
break;
150150
}
151-
case "setSentToLLMMethods": {
151+
case "setProcessedByAutoModelMethods": {
152152
// TODO: set state
153153
break;
154154
}

0 commit comments

Comments
 (0)