format

Author: olasunkanmiraymond

docs/infra.dot

@@ -10,7 +10,7 @@ export class Orchestrator extends EventEmitter implements vscode.Disposable {
     super();
     this.disposables.push(
       this.onStatusChange(this.handleStatus.bind(this)),
-      this.onPromptGenerated(this.handlePromptGeneratedEvent.bind(this))
+      this.onPromptGenerated(this.handlePromptGeneratedEvent.bind(this)),
       // this.onError(this.handleError.bind(this)),
     );
   }

docs/infra.svg

@@ -14,10 +14,10 @@ export class EventEmitter extends BaseEmitter<Record<string, IEventPayload>> {
   onBootstrap: vscode.Event<IEventPayload> = this.createEvent("onBootstrap");
   onFileUpload: vscode.Event<IEventPayload> = this.createEvent("onFileUpload");
   onFileProcessSuccess: vscode.Event<IEventPayload> = this.createEvent(
-    "onFileProcessSuccess"
+    "onFileProcessSuccess",
   );
   onActiveworkspaceUpdate: vscode.Event<IEventPayload> = this.createEvent(
-    "onActiveworkspaceUpdate"
+    "onActiveworkspaceUpdate",
   );
   onFilesRetrieved: vscode.Event<IEventPayload> =
     this.createEvent("onFilesRetrieved");

src/agents/orchestrator.ts

@@ -116,44 +116,44 @@ export async function activate(context: vscode.ExtensionContext) {
     } = OLA_ACTIONS;
     const getComment = new Comments(
       `${USER_MESSAGE} generates the code comments...`,
-      context
+      context,
     );
     const getInLineChat = new InLineChat(
       `${USER_MESSAGE} generates a response...`,
-      context
+      context,
     );
     const generateOptimizeCode = new OptimizeCode(
       `${USER_MESSAGE} optimizes the code...`,
-      context
+      context,
     );
     const generateRefactoredCode = new RefactorCode(
       `${USER_MESSAGE} refactors the code...`,
-      context
+      context,
     );
     const explainCode = new ExplainCode(
       `${USER_MESSAGE} explains the code...`,
-      context
+      context,
     );
     const generateReview = new ReviewCode(
       `${USER_MESSAGE} reviews the code...`,
-      context
+      context,
     );
     const codeChartGenerator = new CodeChartGenerator(
       `${USER_MESSAGE} creates the code chart...`,
-      context
+      context,
     );
     const generateCommitMessage = new GenerateCommitMessage(
       `${USER_MESSAGE} generates a commit message...`,
-      context
+      context,
     );
     const generateInterviewQuestions = new InterviewMe(
       `${USER_MESSAGE} generates interview questions...`,
-      context
+      context,
     );
 
     const generateUnitTests = new GenerateUnitTest(
       `${USER_MESSAGE} generates unit tests...`,
-      context
+      context,
     );
 
     const actionMap = {
@@ -167,7 +167,7 @@ export async function activate(context: vscode.ExtensionContext) {
         new FixError(
           `${USER_MESSAGE} finds a solution to the error...`,
           context,
-          errorMessage
+          errorMessage,
         ).execute(errorMessage),
       [explain]: async () => await explainCode.execute(),
       [commitMessage]: async () =>
@@ -177,15 +177,15 @@ export async function activate(context: vscode.ExtensionContext) {
     };
 
     const subscriptions: vscode.Disposable[] = Object.entries(actionMap).map(
-      ([action, handler]) => vscode.commands.registerCommand(action, handler)
+      ([action, handler]) => vscode.commands.registerCommand(action, handler),
     );
 
     const selectedGenerativeAiModel = getConfigValue("generativeAi.option");
 
     const quickFix = new CodeActionsProvider();
     quickFixCodeAction = vscode.languages.registerCodeActionsProvider(
       { scheme: "file", language: "*" },
-      quickFix
+      quickFix,
     );
 
     agentEventEmmitter = new EventEmitter();
@@ -233,13 +233,13 @@ export async function activate(context: vscode.ExtensionContext) {
         webviewProviderClass,
         subscriptions,
         quickFixCodeAction,
-        agentEventEmmitter
+        agentEventEmmitter,
       );
     }
   } catch (error) {
     Memory.clear();
     vscode.window.showErrorMessage(
-      "An Error occured while setting up generative AI model"
+      "An Error occured while setting up generative AI model",
     );
     console.log(error);
   }

src/emitter/publisher.ts

@@ -49,8 +49,8 @@ export class DeepseekLLM
   private initializeDisposable(): void {
     this.disposables.push(
       vscode.workspace.onDidChangeConfiguration(() =>
-        this.handleConfigurationChange()
-      )
+        this.handleConfigurationChange(),
+      ),
     );
   }
 
@@ -87,7 +87,7 @@ export class DeepseekLLM
 
   public async generateText(
     prompt: string,
-    instruction?: string
+    instruction?: string,
   ): Promise<string> {
     try {
       const messages = [
@@ -145,7 +145,7 @@ export class DeepseekLLM
         undefined,
         undefined,
         undefined,
-        true
+        true,
       );
       // Note this prompt should be for system instruction only.
       const prompt = createPrompt(userInput);
@@ -203,7 +203,7 @@ export class DeepseekLLM
                 {
                   name: response.choices[0].message.function_call.name,
                   args: JSON.parse(
-                    response.choices[0].message.function_call.arguments
+                    response.choices[0].message.function_call.arguments,
                   ),
                 },
               ];
@@ -246,8 +246,8 @@ export class DeepseekLLM
         const timeoutPromise = new Promise((_, reject) =>
           setTimeout(
             () => reject(new Error("Timeout Exceeded")),
-            this.timeOutMs
-          )
+            this.timeOutMs,
+          ),
         );
 
         const responsePromise = this.generateContentWithTools(userQuery);
@@ -276,17 +276,17 @@ export class DeepseekLLM
           ) {
             this.logger.warn(
               "Detecting no progress: same function calls repeated",
-              ""
+              "",
             );
 
             const regeneratedQuery = await this.generateText(
               userQuery,
-              "Rewrite the user query to more clearly and effectively express the user's underlying intent. The goal is to enable the system to retrieve and utilize the available tools more accurately. Identify the core information need and rephrase the query to highlight it. Consider what information the tools need to function optimally and ensure the query provides it."
+              "Rewrite the user query to more clearly and effectively express the user's underlying intent. The goal is to enable the system to retrieve and utilize the available tools more accurately. Identify the core information need and rephrase the query to highlight it. Consider what information the tools need to function optimally and ensure the query provides it.",
             );
 
             this.orchestrator.publish(
               "onQuery",
-              JSON.stringify(regeneratedQuery)
+              JSON.stringify(regeneratedQuery),
             );
 
             let answer = await this.processUserQuery(regeneratedQuery);
@@ -300,13 +300,13 @@ export class DeepseekLLM
           this.lastFunctionCalls.add(currentCallSignatures);
           if (this.lastFunctionCalls.size > 10) {
             this.lastFunctionCalls = new Set(
-              [...this.lastFunctionCalls].slice(-10)
+              [...this.lastFunctionCalls].slice(-10),
             );
           }
 
           if (toolCalls && toolCalls.length > 0) {
             this.logger.info(
-              `Function calls detected: ${JSON.stringify(toolCalls)}`
+              `Function calls detected: ${JSON.stringify(toolCalls)}`,
             );
 
             for (const functionCall of toolCalls) {
@@ -320,7 +320,7 @@ export class DeepseekLLM
                   functionCall,
                   functionResult,
                   undefined,
-                  false
+                  false,
                 );
 
                 const snapShot = this.createSnapShot({
@@ -336,7 +336,7 @@ export class DeepseekLLM
                 const retry = await vscode.window.showErrorMessage(
                   `Function call failed: ${error.message}. Retry or abort?`,
                   "Retry",
-                  "Abort"
+                  "Abort",
                 );
 
                 if (retry === "Retry") {
@@ -369,28 +369,28 @@ export class DeepseekLLM
       if (snapshot?.length > 0) {
         Memory.removeItems(
           "DEEPSEEK_SNAPSHOT",
-          Memory.get("DEEPSEEK_SNAPSHOT").length
+          Memory.get("DEEPSEEK_SNAPSHOT").length,
         );
       }
 
       return finalResult;
     } catch (error: any) {
       this.orchestrator.publish(
         "onError",
-        "Model not responding at this time, please try again"
+        "Model not responding at this time, please try again",
       );
       vscode.window.showErrorMessage("Error processing user query");
       this.logger.error(
         "Error generating queries, thoughts from user query",
-        error
+        error,
       );
       throw error;
     }
   }
 
   private async handleSingleFunctionCall(
     functionCall: any,
-    attempt: number = 0
+    attempt: number = 0,
   ): Promise<any> {
     const MAX_RETRIES = 3;
     const args = functionCall.args as Record<string, any>;
@@ -418,7 +418,7 @@ export class DeepseekLLM
       if (attempt < MAX_RETRIES) {
         this.logger.warn(
           `Retry attempt ${attempt + 1} for function ${name}`,
-          JSON.stringify({ error: error.message, args })
+          JSON.stringify({ error: error.message, args }),
         );
         return this.handleSingleFunctionCall(functionCall, attempt + 1);
       }
@@ -441,7 +441,7 @@ export class DeepseekLLM
     functionCall?: any,
     functionResponse?: any,
     chat?: any,
-    isInitialQuery: boolean = false
+    isInitialQuery: boolean = false,
   ): Promise<any[]> {
     let chatHistory: any = Memory.get(COMMON.DEEPSEEK_CHAT_HISTORY) || [];
     Memory.removeItems(COMMON.DEEPSEEK_CHAT_HISTORY);
@@ -471,15 +471,15 @@ export class DeepseekLLM
               },
             },
           ],
-        })
+        }),
       );
 
       // Add function result as user message
       chatHistory.push(
         Message.of({
           role: "user",
           content: `Tool result: ${JSON.stringify(functionResponse)}`,
-        })
+        }),
       );
     }