Fixed debugging

craigloewen-msft · Apr 9, 2024 · d4c44bc · d4c44bc
1 parent 6811e88
commit d4c44bc
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 266 deletions.
diff --git a/backendsrc/embeddingsHandler.js b/backendsrc/embeddingsHandler.js
@@ -20,6 +20,8 @@ class embeddingsHandler {
         this.maxConcurrentRequests = 3;
         this.pineconeSemaphore = new Semaphore(this.maxConcurrentRequests);
         this.azureSemaphore = new Semaphore(this.maxConcurrentRequests);
+
+        this.debugDisableEmbeddings = inConfigObject.debugDisableEmbeddings;
     }
 
     async addEmbedding(inputIssue) {
@@ -29,6 +31,10 @@ class embeddingsHandler {
 
             let embeddingObject = null;
 
+            if (this.debugDisableEmbeddings) {
+                return;
+            }
+
             await this.azureSemaphore.runExclusive(async () => {
                 embeddingObject = await this.azureClient.getEmbeddings("issue-body-embeddings-model", description);
             });
@@ -65,6 +71,10 @@ class embeddingsHandler {
         // Create title + body description
         const description = [issueDescription];
 
+        if (this.debugDisableEmbeddings) {
+            return [];
+        }
+
         // Query azure for embeddings
         const inputVector = await this.azureClient.getEmbeddings("issue-body-embeddings-model", description);
 

diff --git a/defaultconfig.js b/defaultconfig.js
@@ -8,4 +8,6 @@ module.exports = {
     'ghToken': null,
     'azureOpenAIAPIKey': "key",
     'azureEndpointURL' : "url",
+    'pineconeAPIKey' : 'key',
+    'debugDisableEmbeddings': true,
 };