Removed problems with nodes note getting fetched due to bad queries

Author: preeesha

ai-assistant/src/commands/DocumentCommand.ts

@@ -62,8 +62,6 @@ export class DocumentCommand implements ISlashCommand {
         );
         if (!result) return null;
 
-        console.log(result);
-
         const answer = result
             .split("<ANSWER_START>")[1]
             .split("<ANSWER_END>")[0]

ai-assistant/src/commands/ImportanceCommand.ts

@@ -31,7 +31,7 @@ export class ImportanceCommand implements ISlashCommand {
 				LIMIT 1
 			`
         );
-        if (!maxOutDegreeQuery) return 0;
+        if (!maxOutDegreeQuery.length) return 0;
         const maxOutDegree = maxOutDegreeQuery[0].get("outDegree").toNumber();
 
         const outDegree = await db.run(
@@ -41,7 +41,7 @@ export class ImportanceCommand implements ISlashCommand {
 			`,
             { id: node.id }
         );
-        if (!outDegree) return 0;
+        if (!outDegree.length) return 0;
         const centrality = outDegree[0].get("outDegree").toNumber();
 
         const relativeCentrality = centrality / maxOutDegree;
@@ -58,7 +58,7 @@ export class ImportanceCommand implements ISlashCommand {
 				LIMIT 1
 			`
         );
-        if (!maxInDegreeQuery) return 0;
+        if (!maxInDegreeQuery.length) return 0;
         const maxInDegree = maxInDegreeQuery[0].get("inDegree").toNumber();
 
         const inDegree = await db.run(
@@ -68,7 +68,7 @@ export class ImportanceCommand implements ISlashCommand {
 			`,
             { id: node.id }
         );
-        if (!inDegree) return 0;
+        if (!inDegree.length) return 0;
         const criticality = inDegree[0].get("inDegree").toNumber();
 
         const relativeCriticality = criticality / maxInDegree;

ai-assistant/src/core/query.ts

@@ -15,15 +15,15 @@ export namespace Query {
             `
 				CALL db.index.vector.queryNodes("${indexName}", 2, $vector)
 				YIELD node, score 
-				WHERE score >= ${threshold}
-				WITH node, score
-				MATCH (node)-[r]->(relatedNode)
-				RETURN node, COLLECT(relatedNode) AS relatedNodes, score
-				ORDER BY score DESC
+                WHERE score >= ${threshold}
+                WITH node, score
+                OPTIONAL MATCH (node)-[r]->(relatedNode)
+                RETURN node, COLLECT(relatedNode) AS relatedNodes, score
+                ORDER BY score DESC
 			`,
             { vector }
         );
-        if (!result) return [];
+        if (!result.length) return [];
 
         const nodes: DBNode[] = [];
         for (const record of result) {
@@ -42,11 +42,12 @@ export namespace Query {
         for (const keyword of keywords) {
             const queryVector = await embeddingModel.generate(keyword);
             if (!queryVector) continue;
+
             const result = await getDBNodesFromVectorQuery(
                 db,
                 "nameEmbeddings",
                 queryVector,
-                0.7
+                0.85
             );
             results.push(...result);
         }

ai-assistant/src/endpoints/purgeDB.ts

@@ -18,10 +18,7 @@ export class PurgeDBEndpoint extends ApiEndpoint {
 
     async emptyDB(db: IDB) {
         const query = `MATCH (n) DETACH DELETE n`;
-        const res = await db.run(query);
-        if (!res) {
-            throw new Error(JSON.stringify(res));
-        }
+        await db.run(query);
     }
 
     async setupIndices(db: IDB) {

ai-assistant/src/modals/styleguideModal.ts

@@ -54,7 +54,7 @@ async function process(http: IHttp, query: string): Promise<string | null> {
      * ---------------------------------------------------------------------------------------------
      */
     const dbResults = await db.run(`MATCH (n:Styleguide) RETURN n`);
-    if (!dbResults) return null;
+    if (!dbResults.length) return null;
 
     const styleGuides = dbResults.map((record) => record.get("n").properties);
     if (!styleGuides.length) return null;