git push --set-upstream origin chat_history_handling

Author: Olasunkanmi Oyinlola

src/commands/handler.ts

@@ -3,7 +3,11 @@ import Anthropic from "@anthropic-ai/sdk";
 import { GenerativeModel, GoogleGenerativeAI } from "@google/generative-ai";
 import Groq from "groq-sdk";
 import * as vscode from "vscode";
-import { APP_CONFIG, COMMON, generativeAiModels } from "../application/constant";
+import {
+  APP_CONFIG,
+  COMMON,
+  generativeAiModels,
+} from "../application/constant";
 import { AnthropicWebViewProvider } from "../webview-providers/anthropic";
 import { GeminiWebViewProvider } from "../webview-providers/gemini";
 import { GroqWebViewProvider } from "../webview-providers/groq";
@@ -38,7 +42,7 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
   constructor(
     private readonly action: string,
     _context: vscode.ExtensionContext,
-    errorMessage?: string
+    errorMessage?: string,
   ) {
     this.context = _context;
     this.error = errorMessage;
@@ -69,21 +73,23 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
     return getConfigValue(configKey);
   }
 
-  protected createModel(): { generativeAi: string; model: any; modelName: string } | undefined {
+  protected createModel():
+    | { generativeAi: string; model: any; modelName: string }
+    | undefined {
     try {
       let model;
       let modelName = "";
       if (!this.generativeAi) {
         vscodeErrorMessage(
-          "Configuration not found. Go to settings, search for Your coding buddy. Fill up the model and model name"
+          "Configuration not found. Go to settings, search for Your coding buddy. Fill up the model and model name",
         );
       }
       if (this.generativeAi === generativeAiModels.GROQ) {
         const apiKey = this.groqApiKey;
         modelName = this.groqModel;
         if (!apiKey || !modelName) {
           vscodeErrorMessage(
-            "Configuration not found. Go to settings, search for Your coding buddy. Fill up the model and model name"
+            "Configuration not found. Go to settings, search for Your coding buddy. Fill up the model and model name",
           );
         }
         model = this.createGroqModel(apiKey);
@@ -109,7 +115,9 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
       return { generativeAi: this.generativeAi, model, modelName };
     } catch (error) {
       console.error("Error creating model:", error);
-      vscode.window.showErrorMessage("An error occurred while creating the model. Please try again.");
+      vscode.window.showErrorMessage(
+        "An error occurred while creating the model. Please try again.",
+      );
     }
   }
 
@@ -142,7 +150,9 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
     return new Groq({ apiKey });
   }
 
-  protected async generateModelResponse(text: string): Promise<string | Anthropic.Messages.Message | undefined> {
+  protected async generateModelResponse(
+    text: string,
+  ): Promise<string | Anthropic.Messages.Message | undefined> {
     try {
       const activeModel = this.createModel();
       if (!activeModel) {
@@ -179,7 +189,7 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
 
       if (!response) {
         throw new Error(
-          "Could not generate response. Check your settings, ensure the API keys and Model Name is added properly."
+          "Could not generate response. Check your settings, ensure the API keys and Model Name is added properly.",
         );
       }
       if (this.action.includes("chart")) {
@@ -188,7 +198,9 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
       return response;
     } catch (error) {
       this.logger.error("Error generating response:", error);
-      vscode.window.showErrorMessage("An error occurred while generating the response. Please try again.");
+      vscode.window.showErrorMessage(
+        "An error occurred while generating the response. Please try again.",
+      );
     }
   }
 
@@ -199,12 +211,19 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
     return inputString;
   }
 
-  async generateGeminiResponse(model: any, text: string): Promise<string | undefined> {
+  async generateGeminiResponse(
+    model: any,
+    text: string,
+  ): Promise<string | undefined> {
     const result = await model.generateContent(text);
     return result ? await result.response.text() : undefined;
   }
 
-  private async anthropicResponse(model: Anthropic, generativeAiModel: string, userPrompt: string) {
+  private async anthropicResponse(
+    model: Anthropic,
+    generativeAiModel: string,
+    userPrompt: string,
+  ) {
     try {
       const response = await model.messages.create({
         model: generativeAiModel,
@@ -218,9 +237,15 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
     }
   }
 
-  private async groqResponse(model: Groq, prompt: string, generativeAiModel: string): Promise<string | undefined> {
+  private async groqResponse(
+    model: Groq,
+    prompt: string,
+    generativeAiModel: string,
+  ): Promise<string | undefined> {
     try {
-      const chatHistory = Memory.has(COMMON.ANTHROPIC_CHAT_HISTORY) ? Memory.get(COMMON.GROQ_CHAT_HISTORY) : [];
+      const chatHistory = Memory.has(COMMON.ANTHROPIC_CHAT_HISTORY)
+        ? Memory.get(COMMON.GROQ_CHAT_HISTORY)
+        : [];
       const params = {
         messages: [
           ...chatHistory,
@@ -232,7 +257,8 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
         model: generativeAiModel,
       };
 
-      const completion: Groq.Chat.ChatCompletion = await model.chat.completions.create(params);
+      const completion: Groq.Chat.ChatCompletion =
+        await model.chat.completions.create(params);
       return completion.choices[0]?.message?.content ?? undefined;
     } catch (error) {
       this.logger.error("Error generating response:", error);
@@ -243,7 +269,9 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
 
   abstract createPrompt(text?: string): any;
 
-  async generateResponse(message?: string): Promise<string | Anthropic.Messages.Message | undefined> {
+  async generateResponse(
+    message?: string,
+  ): Promise<string | Anthropic.Messages.Message | undefined> {
     this.logger.info(this.action);
     let prompt;
     const selectedCode = this.getSelectedWindowArea();
@@ -255,7 +283,9 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
     if (message && selectedCode) {
       prompt = await this.createPrompt(`${message} \n ${selectedCode}`);
     } else {
-      message ? (prompt = await this.createPrompt(message)) : (prompt = await this.createPrompt(selectedCode));
+      message
+        ? (prompt = await this.createPrompt(message))
+        : (prompt = await this.createPrompt(selectedCode));
     }
 
     if (!prompt) {
@@ -338,7 +368,9 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
         placeHolder: "Enter instructions for CodeBuddy",
         ignoreFocusOut: true,
         validateInput: (text) => {
-          return text === "" ? "Enter instructions for CodeBuddy or press Escape to close chat box" : null;
+          return text === ""
+            ? "Enter instructions for CodeBuddy or press Escape to close chat box"
+            : null;
         },
       });
       return userPrompt;
@@ -350,7 +382,9 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
   async execute(message?: string): Promise<void> {
     try {
       let prompt: string | undefined;
-      const response = (await this.generateResponse(prompt ? prompt : message)) as string;
+      const response = (await this.generateResponse(
+        prompt ? prompt : message,
+      )) as string;
       if (!response) {
         vscode.window.showErrorMessage("model not reponding, try again later");
         return;
@@ -383,7 +417,10 @@ export abstract class CodeCommandHandler implements ICodeCommandHandler {
           break;
       }
     } catch (error) {
-      this.logger.error("Error while passing model response to the webview", error);
+      this.logger.error(
+        "Error while passing model response to the webview",
+        error,
+      );
     }
   }
 }

src/emitter/publisher.ts

@@ -9,16 +9,27 @@ export class EventEmitter extends BaseEmitter<Record<string, IEventPayload>> {
   onPromptGenerated: vscode.Event<IEventPayload> = this.createEvent("onQuery");
   onThinking: vscode.Event<IEventPayload> = this.createEvent("onThinking");
   onResponse: vscode.Event<IEventPayload> = this.createEvent("onResponse");
-  onSecretChange: vscode.Event<IEventPayload> = this.createEvent("onSecretChange");
+  onSecretChange: vscode.Event<IEventPayload> =
+    this.createEvent("onSecretChange");
   onBootstrap: vscode.Event<IEventPayload> = this.createEvent("onBootstrap");
   onFileUpload: vscode.Event<IEventPayload> = this.createEvent("onFileUpload");
-  onFileProcessSuccess: vscode.Event<IEventPayload> = this.createEvent("onFileProcessSuccess");
-  onActiveworkspaceUpdate: vscode.Event<IEventPayload> = this.createEvent("onActiveworkspaceUpdate");
-  onFilesRetrieved: vscode.Event<IEventPayload> = this.createEvent("onFilesRetrieved");
-  onStrategizing: vscode.Event<IEventPayload> = this.createEvent("onStrategizing");
-  onModelChange: vscode.Event<IEventPayload> = this.createEvent("onModelChange");
-  onModelChangeSuccess: vscode.Event<IEventPayload> = this.createEvent("onModelChangeSuccess");
-  onHistoryUpdated: vscode.Event<IEventPayload> = this.createEvent("onHistoryUpdated");
+  onFileProcessSuccess: vscode.Event<IEventPayload> = this.createEvent(
+    "onFileProcessSuccess",
+  );
+  onActiveworkspaceUpdate: vscode.Event<IEventPayload> = this.createEvent(
+    "onActiveworkspaceUpdate",
+  );
+  onFilesRetrieved: vscode.Event<IEventPayload> =
+    this.createEvent("onFilesRetrieved");
+  onStrategizing: vscode.Event<IEventPayload> =
+    this.createEvent("onStrategizing");
+  onModelChange: vscode.Event<IEventPayload> =
+    this.createEvent("onModelChange");
+  onModelChangeSuccess: vscode.Event<IEventPayload> = this.createEvent(
+    "onModelChangeSuccess",
+  );
+  onHistoryUpdated: vscode.Event<IEventPayload> =
+    this.createEvent("onHistoryUpdated");
 
   /**
    * Emits a generic event with specified status, message, and optional data.

src/extension.ts

@@ -1,7 +1,12 @@
 import * as fs from "fs";
 import * as path from "path";
 import * as vscode from "vscode";
-import { APP_CONFIG, generativeAiModels, OLA_ACTIONS, USER_MESSAGE } from "./application/constant";
+import {
+  APP_CONFIG,
+  generativeAiModels,
+  OLA_ACTIONS,
+  USER_MESSAGE,
+} from "./application/constant";
 import { Comments } from "./commands/comment";
 import { ExplainCode } from "./commands/explain";
 import { FixError } from "./commands/fixError";
@@ -86,7 +91,8 @@ export async function activate(context: vscode.ExtensionContext) {
     const { apiKey, model } = getAPIKeyAndModel("gemini");
     FileUploadService.initialize(apiKey);
     await credentials.initialize(context);
-    const session: vscode.AuthenticationSession | undefined = await credentials.getSession();
+    const session: vscode.AuthenticationSession | undefined =
+      await credentials.getSession();
     logger.info(`Logged into GitHub as ${session?.account.label}`);
     Memory.getInstance();
 
@@ -108,17 +114,47 @@ export async function activate(context: vscode.ExtensionContext) {
       generateCodeChart,
       inlineChat,
     } = OLA_ACTIONS;
-    const getComment = new Comments(`${USER_MESSAGE} generates the code comments...`, context);
-    const getInLineChat = new InLineChat(`${USER_MESSAGE} generates a response...`, context);
-    const generateOptimizeCode = new OptimizeCode(`${USER_MESSAGE} optimizes the code...`, context);
-    const generateRefactoredCode = new RefactorCode(`${USER_MESSAGE} refactors the code...`, context);
-    const explainCode = new ExplainCode(`${USER_MESSAGE} explains the code...`, context);
-    const generateReview = new ReviewCode(`${USER_MESSAGE} reviews the code...`, context);
-    const codeChartGenerator = new CodeChartGenerator(`${USER_MESSAGE} creates the code chart...`, context);
-    const generateCommitMessage = new GenerateCommitMessage(`${USER_MESSAGE} generates a commit message...`, context);
-    const generateInterviewQuestions = new InterviewMe(`${USER_MESSAGE} generates interview questions...`, context);
-
-    const generateUnitTests = new GenerateUnitTest(`${USER_MESSAGE} generates unit tests...`, context);
+    const getComment = new Comments(
+      `${USER_MESSAGE} generates the code comments...`,
+      context,
+    );
+    const getInLineChat = new InLineChat(
+      `${USER_MESSAGE} generates a response...`,
+      context,
+    );
+    const generateOptimizeCode = new OptimizeCode(
+      `${USER_MESSAGE} optimizes the code...`,
+      context,
+    );
+    const generateRefactoredCode = new RefactorCode(
+      `${USER_MESSAGE} refactors the code...`,
+      context,
+    );
+    const explainCode = new ExplainCode(
+      `${USER_MESSAGE} explains the code...`,
+      context,
+    );
+    const generateReview = new ReviewCode(
+      `${USER_MESSAGE} reviews the code...`,
+      context,
+    );
+    const codeChartGenerator = new CodeChartGenerator(
+      `${USER_MESSAGE} creates the code chart...`,
+      context,
+    );
+    const generateCommitMessage = new GenerateCommitMessage(
+      `${USER_MESSAGE} generates a commit message...`,
+      context,
+    );
+    const generateInterviewQuestions = new InterviewMe(
+      `${USER_MESSAGE} generates interview questions...`,
+      context,
+    );
+
+    const generateUnitTests = new GenerateUnitTest(
+      `${USER_MESSAGE} generates unit tests...`,
+      context,
+    );
 
     const actionMap = {
       [comment]: async () => await getComment.execute(),
@@ -128,21 +164,29 @@ export async function activate(context: vscode.ExtensionContext) {
       [interviewMe]: async () => await generateInterviewQuestions.execute(),
       [generateUnitTest]: async () => await generateUnitTests.execute(),
       [fix]: (errorMessage: string) =>
-        new FixError(`${USER_MESSAGE} finds a solution to the error...`, context, errorMessage).execute(errorMessage),
+        new FixError(
+          `${USER_MESSAGE} finds a solution to the error...`,
+          context,
+          errorMessage,
+        ).execute(errorMessage),
       [explain]: async () => await explainCode.execute(),
-      [commitMessage]: async () => await generateCommitMessage.execute("commitMessage"),
+      [commitMessage]: async () =>
+        await generateCommitMessage.execute("commitMessage"),
       [generateCodeChart]: async () => await codeChartGenerator.execute(),
       [inlineChat]: async () => await getInLineChat.execute(),
     };
 
-    let subscriptions: vscode.Disposable[] = Object.entries(actionMap).map(([action, handler]) =>
-      vscode.commands.registerCommand(action, handler)
+    let subscriptions: vscode.Disposable[] = Object.entries(actionMap).map(
+      ([action, handler]) => vscode.commands.registerCommand(action, handler),
     );
 
     const selectedGenerativeAiModel = getConfigValue("generativeAi.option");
 
     const quickFix = new CodeActionsProvider();
-    quickFixCodeAction = vscode.languages.registerCodeActionsProvider({ scheme: "file", language: "*" }, quickFix);
+    quickFixCodeAction = vscode.languages.registerCodeActionsProvider(
+      { scheme: "file", language: "*" },
+      quickFix,
+    );
 
     agentEventEmmitter = new EventEmitter();
 
@@ -186,12 +230,23 @@ export async function activate(context: vscode.ExtensionContext) {
       const modelConfig = modelConfigurations[selectedGenerativeAiModel];
       const apiKey = getConfigValue(modelConfig.key);
       const apiModel = getConfigValue(modelConfig.model);
-      providerManager.initializeProvider(selectedGenerativeAiModel, apiKey, apiModel, true);
+      providerManager.initializeProvider(
+        selectedGenerativeAiModel,
+        apiKey,
+        apiModel,
+        true,
+      );
     }
-    context.subscriptions.push(...subscriptions, quickFixCodeAction, agentEventEmmitter);
+    context.subscriptions.push(
+      ...subscriptions,
+      quickFixCodeAction,
+      agentEventEmmitter,
+    );
   } catch (error) {
     Memory.clear();
-    vscode.window.showErrorMessage("An Error occured while setting up generative AI model");
+    vscode.window.showErrorMessage(
+      "An Error occured while setting up generative AI model",
+    );
     console.log(error);
   }
 }