format the codebase

Author: olasunkanmi-SE

.gitignore

@@ -12,3 +12,4 @@ webviewUi/dist
 samples
 patterns
 .codebuddy
+scripts

src/services/code-indexing.ts

@@ -58,7 +58,9 @@ export class CodeIndexingService {
       if (!mappedCode) {
         throw new Error("Failed to build codebase map");
       }
-      const ats = Object.values(mappedCode).flatMap((repo) => Object.values(repo.modules));
+      const ats = Object.values(mappedCode).flatMap((repo) =>
+        Object.values(repo.modules),
+      );
       const mapper = new CodeStructureMapper(ats);
       return mapper.normalizeData();
     } catch (error) {
@@ -73,7 +75,8 @@ export class CodeIndexingService {
    */
   async generateFunctionDescription(): Promise<IFunctionData[]> {
     try {
-      const functions = (await this.buildFunctionStructureMap()) as IFunctionData[];
+      const functions =
+        (await this.buildFunctionStructureMap()) as IFunctionData[];
       if (!functions?.length) {
         throw new Error("failed to generate ATS");
       }
@@ -95,7 +98,10 @@ export class CodeIndexingService {
         item.compositeText = `Description: ${item.description} Function: ${item.name} ${item.returnType} Dependencies: ${(item.dependencies ?? []).join(", ")}`;
       }
     });
-    const functionWithEmbeddings = await this.embeddingService.processFunctions(functionsWithDescription, true);
+    const functionWithEmbeddings = await this.embeddingService.processFunctions(
+      functionsWithDescription,
+      true,
+    );
     return functionWithEmbeddings;
   }
 

src/services/context-retriever.ts

@@ -1,6 +1,9 @@
 import * as vscode from "vscode";
 import { Orchestrator } from "../agents/orchestrator";
-import { IFileToolConfig, IFileToolResponse } from "../application/interfaces/agent.interface";
+import {
+  IFileToolConfig,
+  IFileToolResponse,
+} from "../application/interfaces/agent.interface";
 import { Logger } from "../infrastructure/logger/logger";
 import { getAPIKeyAndModel, getGenerativeAiModel } from "./../utils/utils";
 import { EmbeddingService } from "./embedding";
@@ -51,7 +54,9 @@ export class ContextRetriever {
   //   }
   // }
 
-  async readFiles(fileConfigs: IFileToolConfig[]): Promise<IFileToolResponse[]> {
+  async readFiles(
+    fileConfigs: IFileToolConfig[],
+  ): Promise<IFileToolResponse[]> {
     const files = fileConfigs.flatMap((file) => file);
     const promises = files.map(async (file) => {
       try {
@@ -65,7 +70,9 @@ export class ContextRetriever {
       }
     });
     const results = await Promise.all(promises);
-    return results.filter((result): result is IFileToolResponse => result !== undefined);
+    return results.filter(
+      (result): result is IFileToolResponse => result !== undefined,
+    );
   }
 
   async readFileContent(filePath: string): Promise<string> {

src/services/embedding.ts

@@ -1,4 +1,8 @@
-import { EmbedContentResponse, GenerativeModel, GoogleGenerativeAI } from "@google/generative-ai";
+import {
+  EmbedContentResponse,
+  GenerativeModel,
+  GoogleGenerativeAI,
+} from "@google/generative-ai";
 import { EmbeddingsConfig } from "../application/constant";
 import { IFunctionData } from "../application/interfaces";
 import { Logger, LogLevel } from "../infrastructure/logger/logger";
@@ -24,7 +28,8 @@ interface BatchProcessResult {
  * @class EmbeddingService
  */
 export class EmbeddingService {
-  private static readonly DEFAULT_OPTIONS: Required<EmbeddingServiceOptions> = EmbeddingsConfig;
+  private static readonly DEFAULT_OPTIONS: Required<EmbeddingServiceOptions> =
+    EmbeddingsConfig;
 
   private readonly options: Required<EmbeddingServiceOptions>;
   private readonly requestInterval: number;
@@ -56,7 +61,10 @@ export class EmbeddingService {
     try {
       const vscode = require("vscode");
       const config = vscode.workspace?.getConfiguration?.();
-      return (config?.get("codebuddy.embeddingModel") as string) || "gemini-2.0-flash";
+      return (
+        (config?.get("codebuddy.embeddingModel") as string) ||
+        "gemini-2.0-flash"
+      );
     } catch {
       // Fallback if vscode module is not available (e.g., in tests)
       return "gemini-2.0-flash";
@@ -171,7 +179,9 @@ export class EmbeddingService {
    * @returns {Promise<IFunctionData>} The function data with the generated embeddings.
    * @memberof EmbeddingService
    */
-  private async generateFunctionEmbeddings(item: IFunctionData): Promise<IFunctionData> {
+  private async generateFunctionEmbeddings(
+    item: IFunctionData,
+  ): Promise<IFunctionData> {
     try {
       const embedding = await this.generateEmbedding(item.compositeText);
       return {
@@ -202,7 +212,7 @@ export class EmbeddingService {
   private async processBatchWithRetry(
     batch: IFunctionData[],
     lastRequestTime: number,
-    forEmbedding: boolean
+    forEmbedding: boolean,
   ): Promise<BatchProcessResult> {
     let retries = 0;
     const generateComments: IFunctionData[] = [];
@@ -216,17 +226,23 @@ export class EmbeddingService {
         }
 
         if (forEmbedding) {
-          const embeddings = await Promise.all(batch.map((item) => this.generateFunctionEmbeddings(item)));
+          const embeddings = await Promise.all(
+            batch.map((item) => this.generateFunctionEmbeddings(item)),
+          );
           generateEmbeddings.push(...embeddings.filter(Boolean));
         } else {
           const comments = await Promise.all(
             batch.map((item) => {
               if (!item.description) {
                 return item.content ? this.generateText(item) : null;
               }
-            })
+            }),
+          );
+          generateComments.push(
+            ...comments.filter(
+              (comment): comment is IFunctionData => comment !== null,
+            ),
           );
-          generateComments.push(...comments.filter((comment): comment is IFunctionData => comment !== null));
         }
 
         return {
@@ -262,7 +278,10 @@ export class EmbeddingService {
    * @returns {Promise<IFunctionData[]>} The processed function data, including any generated embeddings or text.
    * @memberof EmbeddingService
    */
-  public async processFunctions(data: IFunctionData[], forEmbedding = false): Promise<IFunctionData[]> {
+  public async processFunctions(
+    data: IFunctionData[],
+    forEmbedding = false,
+  ): Promise<IFunctionData[]> {
     try {
       const result = await this.processWithRateLimit(data, forEmbedding);
 
@@ -290,7 +309,7 @@ export class EmbeddingService {
    */
   private async processWithRateLimit(
     data: IFunctionData[],
-    forEmbedding: boolean
+    forEmbedding: boolean,
   ): Promise<{
     successful: IFunctionData[];
     failed: IFunctionData[];
@@ -304,9 +323,17 @@ export class EmbeddingService {
 
       try {
         await this.delay(60000);
-        const result = await this.processBatchWithRetry(batch, lastRequestTime, forEmbedding);
+        const result = await this.processBatchWithRetry(
+          batch,
+          lastRequestTime,
+          forEmbedding,
+        );
 
-        successful.push(...(forEmbedding ? result.generateEmbeddings : result.generateComments));
+        successful.push(
+          ...(forEmbedding
+            ? result.generateEmbeddings
+            : result.generateComments),
+        );
         lastRequestTime = Date.now();
 
         this.logger.info(`Batch processed`, { startIndex: i });
@@ -315,7 +342,9 @@ export class EmbeddingService {
           startIndex: i,
           error,
         });
-        failed.push(...batch.map((item) => ({ ...item, error: error as Error })));
+        failed.push(
+          ...batch.map((item) => ({ ...item, error: error as Error })),
+        );
       }
     }
 

src/services/url-reranker.ts

@@ -23,8 +23,12 @@ interface IRerankingConfig {
 export class UrlReranker {
   private static readonly CONFIG: IRerankingConfig = URL_RERANKING_CONFIG;
   private static readonly KEYWORDS: string[] = RELEVANT_CODING_KEY_WORDS;
-  private static readonly WEBSITE_REPUTATIONS: Record<string, number> = REPUTATION_RANK_MAP;
-  private static readonly CATEGORIES: Record<string, { type: string; score: number }> = URL_CATEGORIES;
+  private static readonly WEBSITE_REPUTATIONS: Record<string, number> =
+    REPUTATION_RANK_MAP;
+  private static readonly CATEGORIES: Record<
+    string,
+    { type: string; score: number }
+  > = URL_CATEGORIES;
   private readonly groqLLM: GroqLLM;
   private static instance: UrlReranker;
   query: string;
@@ -36,7 +40,8 @@ export class UrlReranker {
     const { apiKey, model } = getAPIKeyAndModel(currentModel);
 
     // For URL reranking, prefer Groq if available, otherwise fallback to current selection
-    const rerankerModel = currentModel.toLowerCase() === "groq" ? currentModel : "Groq";
+    const rerankerModel =
+      currentModel.toLowerCase() === "groq" ? currentModel : "Groq";
     const { apiKey: rerankerApiKey } = getAPIKeyAndModel(rerankerModel);
 
     this.groqLLM = GroqLLM.getInstance({
@@ -69,7 +74,9 @@ export class UrlReranker {
   private calculateTitleRelevanceScore(title: string): number {
     if (!title) return 0;
 
-    const matches = UrlReranker.KEYWORDS.filter((keyword) => title.toLowerCase().includes(keyword.toLowerCase()));
+    const matches = UrlReranker.KEYWORDS.filter((keyword) =>
+      title.toLowerCase().includes(keyword.toLowerCase()),
+    );
 
     return matches.length / UrlReranker.KEYWORDS.length;
   }
@@ -83,7 +90,10 @@ export class UrlReranker {
     const hostname = this.extractHostname(metadata);
     const domain = hostname.split(".").slice(-2).join(".");
 
-    return UrlReranker.WEBSITE_REPUTATIONS[domain] || UrlReranker.WEBSITE_REPUTATIONS.default;
+    return (
+      UrlReranker.WEBSITE_REPUTATIONS[domain] ||
+      UrlReranker.WEBSITE_REPUTATIONS.default
+    );
   }
 
   /**
@@ -95,7 +105,9 @@ export class UrlReranker {
     if (!metadata.content) return 0;
 
     const codeBlockCount = this.countCodeBlocks(metadata.content);
-    const hasAdequateExplanation = this.hasAdequateExplanation(metadata.content);
+    const hasAdequateExplanation = this.hasAdequateExplanation(
+      metadata.content,
+    );
 
     return codeBlockCount + (hasAdequateExplanation ? 1 : 0);
   }
@@ -126,7 +138,9 @@ export class UrlReranker {
    * @returns Final score for reranking
    */
   calculateFinalScore(metadata: IPageMetada): number {
-    const titleRelevanceScore = this.calculateTitleRelevanceScore(metadata.title ?? "");
+    const titleRelevanceScore = this.calculateTitleRelevanceScore(
+      metadata.title ?? "",
+    );
     const reputationScore = this.calculateSourceReputationScore(metadata);
     const contentQualityScore = this.calculateContentQualityScore(metadata);
     const diversityScore = this.calculateDiversityScore(metadata);