Added /purgeDB endpoint

Author: preeesha

ai-assistant/src/RocketChatterApp.ts

@@ -26,6 +26,7 @@ import { SuggestCommand } from "./commands/SuggestCommand";
 import { TranslateCommand } from "./commands/TranslateCommand";
 import { WhyUsedCommand } from "./commands/WhyUsedCommand";
 import { IngestEndpoint } from "./endpoints/ingest";
+import { PurgeDBEndpoint } from "./endpoints/purgeDB";
 import { handleModalViewSubmit } from "./utils/handleModalViewSubmit";
 
 export class RocketChatterApp extends App {
@@ -65,7 +66,11 @@ export class RocketChatterApp extends App {
         await configuration.api.provideApi({
             visibility: ApiVisibility.PUBLIC,
             security: ApiSecurity.UNSECURE,
-            endpoints: [new IngestEndpoint(this)],
+            endpoints: [
+                new IngestEndpoint(this),
+                new PurgeDBEndpoint(this),
+            ],
         });
+
     }
 }

ai-assistant/src/endpoints/purgeDB.ts

@@ -0,0 +1,66 @@
+import {
+    IHttp,
+    IModify,
+    IPersistence,
+    IRead,
+} from "@rocket.chat/apps-engine/definition/accessors";
+import {
+    ApiEndpoint,
+    IApiEndpointInfo,
+    IApiRequest,
+    IApiResponse,
+} from "@rocket.chat/apps-engine/definition/api";
+import { IDB } from "../core/db/db.types";
+import { Neo4j } from "../core/db/neo4j";
+
+export class PurgeDBEndpoint extends ApiEndpoint {
+    public path = "purgeDB";
+
+    async emptyDB(db: IDB) {
+        const query = `MATCH (n) DETACH DELETE n`;
+        await db.run(query);
+        console.error("Failed to empty DB");
+    }
+
+    async setupIndices(db: IDB) {
+        const query = [
+            // Drop existing indices
+            "DROP INDEX `nameEmbeddings`;",
+            "DROP INDEX `codeEmbeddings`;",
+
+            // Create indices for name embeddings
+            "CREATE VECTOR INDEX `nameEmbeddings`",
+            "FOR (n: Node) ON (n.nameEmbeddings)",
+            "OPTIONS {indexConfig: {",
+            "   `vector.dimensions`: 768,",
+            "   `vector.similarity_function`: 'COSINE'",
+            "}};",
+
+            // Create indices for code embeddings
+            "CREATE VECTOR INDEX `codeEmbeddings`",
+            "FOR (n: Node) ON (n.codeEmbeddings)",
+            "OPTIONS {indexConfig: {",
+            "   `vector.dimensions`: 768,",
+            "   `vector.similarity_function`: 'COSINE'",
+            "}};",
+
+        ].join("\n");
+        await db.run(query);
+    }
+
+    public async post(
+        request: IApiRequest,
+        endpoint: IApiEndpointInfo,
+        read: IRead,
+        modify: IModify,
+        http: IHttp,
+        persis: IPersistence
+    ): Promise<IApiResponse> {
+        const db = new Neo4j(http);
+
+        await this.emptyDB(db)
+        await this.setupIndices(db)
+
+        return this.success();
+    }
+}