add modelProvider to PromptLog

Author: Patrick-Erichsen

core/config/types.ts

@@ -387,6 +387,7 @@ declare global {
   
   export interface PromptLog {
     modelTitle: string;
+    modelProvider: string;
     completionOptions: CompletionOptions;
     prompt: string;
     completion: string;

core/index.d.ts

@@ -455,6 +455,7 @@ export type FileSymbolMap = Record<string, SymbolWithRange[]>;
 
 export interface PromptLog {
   modelTitle: string;
+  modelProvider: string;
   completionOptions: CompletionOptions;
   prompt: string;
   completion: string;

core/llm/index.ts

@@ -779,6 +779,7 @@ export abstract class BaseLLM implements ILLM {
 
     return {
       modelTitle: this.title ?? completionOptions.model,
+      modelProvider: this.underlyingProviderName,
       prompt,
       completion,
       completionOptions,
@@ -1094,6 +1095,7 @@ export abstract class BaseLLM implements ILLM {
 
     return {
       modelTitle: this.title ?? completionOptions.model,
+      modelProvider: this.underlyingProviderName,
       prompt,
       completion,
       completionOptions,

core/llm/llm-pre-fetch.vitest.ts

@@ -23,6 +23,7 @@ async function dudLLMCall(llm: ILLM, messages: ChatMessage[]) {
         model: "",
       },
       modelTitle: "",
+      modelProvider: "",
       prompt: "",
     });
     abortController.abort();

core/llm/streamChat.ts

@@ -40,6 +40,7 @@ export async function* llmStreamChat(
   // Log to return in case of error
   const errorPromptLog = {
     modelTitle: model?.title ?? model?.model,
+    modelProvider: model?.underlyingProviderName ?? "unknown",
     completion: "",
     prompt: "",
     completionOptions: {

core/nextEdit/context/processNextEditData.ts

@@ -127,6 +127,7 @@ export const processNextEditData = async ({
         context: autocompleteContext,
         modelProvider,
         modelName,
+        modelTitle: modelName,
       },
     });
   }

extensions/vscode/src/diff/vertical/manager.ts

@@ -537,6 +537,7 @@ export class VerticalDiffManager {
       timestamp: new Date().toISOString(),
       modelProvider: model.underlyingProviderName,
       modelName: model.title ?? "",
+      modelTitle: model.title ?? "",
       prompt: stripImages(prompt),
       completion: newCode,
       previousCode,

extensions/vscode/src/extension/EditOutcomeTracker.ts

@@ -7,6 +7,7 @@ export interface PendingEditData {
   timestamp: string;
   modelProvider: string;
   modelName: string;
+  modelTitle: string;
   prompt: string;
   completion: string;
   previousCode: string;
@@ -57,6 +58,7 @@ class EditOutcomeTracker {
       data: {
         modelProvider: pendingEdit.modelProvider,
         modelName: pendingEdit.modelName,
+        modelTitle: pendingEdit.modelName,
         prompt: pendingEdit.prompt,
         completion: pendingEdit.completion,
         previousCode: pendingEdit.previousCode,

extensions/vscode/src/extension/VsCodeMessenger.ts

@@ -30,6 +30,7 @@ import { showTutorial } from "../util/tutorial";
 import { getExtensionUri } from "../util/vscode";
 import { VsCodeIde } from "../VsCodeIde";
 import { VsCodeWebviewProtocol } from "../webviewProtocol";
+
 import { VsCodeExtension } from "./VsCodeExtension";
 
 type ToIdeOrWebviewFromCoreProtocol = ToIdeFromCoreProtocol &
@@ -241,6 +242,7 @@ export class VsCodeMessenger {
           completion: fileAfterEdit ?? "",
           modelProvider: model.underlyingProviderName,
           modelName: model.title ?? "",
+          modelTitle: model.title ?? "",
           filepath: msg.data.range.filepath,
         },
       });

extensions/vscode/src/quickEdit/QuickEditQuickPick.ts

@@ -238,15 +238,15 @@ export class QuickEdit {
     prompt: string,
     path: string | undefined,
   ) => {
-    const model = await this.getCurModel();
-    if (!model) {
+    const modelTitle = await this.getCurModel();
+    if (!modelTitle) {
       throw new Error("No model selected");
     }
 
     const { config } = await this.configHandler.loadConfig();
     const rules = config?.rules ?? []; // no need to error - getCurModel above will handle
 
-    await this._streamEditWithInputAndContext(prompt, model, rules);
+    await this._streamEditWithInputAndContext(prompt, modelTitle, rules);
     this.openAcceptRejectMenu(prompt, path);
   };