chore: use env vars for mongo namespace

Author: himanshusinghs

.github/workflows/accuracy-tests.yml

@@ -25,6 +25,8 @@ jobs:
       MDB_AZURE_OPEN_AI_API_KEY: ${{ secrets.ACCURACY_AZURE_OPEN_AI_API_KEY }}
       MDB_AZURE_OPEN_AI_API_URL: ${{ vars.ACCURACY_AZURE_OPEN_AI_API_URL }}
       MDB_ACCURACY_MDB_URL: ${{ secrets.ACCURACY_MDB_CONNECTION_STRING }}
+      MDB_ACCURACY_MDB_DB: ${{ vars.ACCURACY_MDB_DB }}
+      MDB_ACCURACY_MDB_COLLECTION: ${{ vars.ACCURACY_MDB_COLLECTION }}
       MDB_ACCURACY_BASELINE_COMMIT: ${{ github.event.pull_request.base.sha || '' }}
     steps:
       - uses: GitHubSecurityLab/actions-permissions/monitor@v1

scripts/accuracy/run-accuracy-tests.sh

@@ -10,6 +10,8 @@ export MDB_ACCURACY_RUN_ID=$(npx uuid v4)
 
 # For providing a mongodb based storage to store accuracy result
 # export MDB_ACCURACY_MDB_URL=""
+# export MDB_ACCURACY_MDB_DB=""
+# export MDB_ACCURACY_MDB_COLLECTION=""
 
 # By default we run all the tests under tests/accuracy folder unless a path is
 # specified in the command line. Such as:

tests/accuracy/sdk/accuracy-result-storage/get-accuracy-result-storage.ts

@@ -3,8 +3,9 @@ import { MongoDBBasedResultStorage } from "./mongodb-storage.js";
 import { AccuracyResultStorage } from "./result-storage.js";
 
 export function getAccuracyResultStorage(): AccuracyResultStorage {
-    if (process.env.MDB_ACCURACY_MDB_URL) {
-        return new MongoDBBasedResultStorage(process.env.MDB_ACCURACY_MDB_URL);
+    const { MDB_ACCURACY_MDB_URL, MDB_ACCURACY_MDB_DB, MDB_ACCURACY_MDB_COLLECTION } = process.env;
+    if (MDB_ACCURACY_MDB_URL && MDB_ACCURACY_MDB_DB && MDB_ACCURACY_MDB_COLLECTION) {
+        return new MongoDBBasedResultStorage(MDB_ACCURACY_MDB_URL, MDB_ACCURACY_MDB_DB, MDB_ACCURACY_MDB_COLLECTION);
     }
     return new DiskBasedResultStorage();
 }

tests/accuracy/sdk/accuracy-result-storage/mongodb-storage.ts

@@ -13,11 +13,13 @@ export class MongoDBBasedResultStorage implements AccuracyResultStorage {
 
     constructor(
         connectionString: string,
+        database: string,
+        collection: string,
         // Omitting these as they might contain large chunk of texts
         private readonly omittedModelResponseFields: (keyof ModelResponse)[] = ["messages", "text"]
     ) {
         this.client = new MongoClient(connectionString);
-        this.resultCollection = this.client.db("mongodb-mcp-server").collection<AccuracyResult>("accuracy-results");
+        this.resultCollection = this.client.db(database).collection<AccuracyResult>(collection);
     }
 
     async getAccuracyResult(commitSHA: string, runId?: string): Promise<AccuracyResult | null> {