update tests

Author: mrlynn

bin/commands/config/clear-config.js

@@ -3,12 +3,15 @@ import fs from 'fs';
 import chalk from 'chalk';
 
 export async function clearConfig(configPath) {
-  if (fs.existsSync(configPath)) {
-    fs.unlinkSync(configPath);
-    console.log(chalk.green("🗑️ Configuration deleted successfully."));
-    return true;
-  } else {
+  try {
+    if (fs.existsSync(configPath)) {
+      fs.unlinkSync(configPath);
+      return true;
+    }
     console.warn(chalk.yellow("⚠️ No configuration file found."));
+    return true; // Changed to return true even when file doesn't exist
+  } catch (error) {
+    console.error(chalk.red(`❌ Error clearing config: ${error.message}`));
     return false;
   }
 }

bin/commands/index/create-index.js

@@ -25,44 +25,56 @@ export async function createIndex(config) {
       console.log(chalk.blue("🔍 Existing indexes:"), existingIndexes);
 
       await client.close();
-      throw error; 
+      throw new Error("createSearchIndexes() is not available"); 
     }
 
     if (!config || !config.embedding || !config.embedding.dimensions) {
       console.error(chalk.red("❌ MongoDB Error: Missing embedding dimensions in config."));
-      throw new Error("Missing embedding dimensions in config."); // ✅ Instead of process.exit(1)
+      throw new Error("Missing embedding dimensions in config.");
     }
 
     console.log(chalk.blue(`📌 Creating Vector Search Index: ${config.indexName}...`));
 
-    const indexDefinition = {
+    const indexConfig = {
       name: config.indexName || "vector_index",
       definition: {
         mappings: {
-          dynamic: false,  // ✅ Fix: Set dynamic to false to explicitly define the index
+          dynamic: false,  // More restrictive than true, better for vector search
           fields: {
             [config.embedding.path || "embedding"]: {
-              type: "vector",
-              numDimensions: config.embedding.dimensions || 1536,  // ✅ Ensure dimensions are set
+              type: "knnVector",
+              dimensions: config.embedding.dimensions,
               similarity: config.embedding.similarity || "cosine"
             }
           }
         }
       }
     };
-    
 
-    console.log(chalk.blue(`🔍 Debug: Index definition: `, JSON.stringify(indexDefinition, null, 2)));
+    console.log(chalk.blue(`🔍 Debug: Index definition: `), JSON.stringify(indexConfig, null, 2));
 
-    const indexResult = await collection.createSearchIndexes([indexDefinition]);
+    try {
+      const indexResult = await collection.createSearchIndex(indexConfig);
+      console.log(chalk.green(`✅ Vector Search Index "${indexConfig.name}" created successfully!`));
+      console.log(chalk.blue(`🔍 Index creation result:`), indexResult);
+      
+      await client.close();
+      console.log(chalk.blue("🔍 MongoDB connection closed."));
+      
+      return {
+        success: true,
+        message: "Vector search index created successfully",
+        indexName: indexConfig.name,
+        result: indexResult
+      };
+    } catch (error) {
+      console.error(chalk.red(`❌ Error creating search index: ${error.message}`));
+      await client.close();
+      throw error;
+    }
 
-    console.log(chalk.green(`✅ Vector Search Index "${config.indexName}" created successfully!`));
-    console.log(chalk.blue(`🔍 Index creation result:`), indexResult);
-    return indexResult;
-    await client.close();
-    console.log(chalk.blue("🔍 MongoDB connection closed."));
   } catch (error) {
-    console.error(chalk.red(`❌ MongoDB Error: ${error.message}`));
-    throw error; 
+    console.error(chalk.red(`❌ Error: ${error.message}`));
+    throw error;
   }
-}
+}

bin/utils/error-handling.js

@@ -1,7 +1,7 @@
 // bin/utils/error-handling.js
 import chalk from 'chalk';
 
-export function wrapCommand(command) {
+export function withErrorHandling(command) {
   return async (...args) => {
     try {
       return await command(...args);
@@ -10,7 +10,11 @@ export function wrapCommand(command) {
       if (process.env.DEBUG) {
         console.error(chalk.gray("\nDebug Stack Trace:"), error.stack);
       }
-      process.exit(1);
+      // Don't exit if we're in a test environment
+      if (process.env.NODE_ENV !== 'test') {
+        process.exit(1);
+      }
+      throw error; // Re-throw for test environments
     }
   };
 }

bin/utils/formatting.js

@@ -91,13 +91,18 @@ function displayResult(result, index) {
 }
 
 export function formatDocument(doc) {
-    const formatted = {};
-    for (const key in doc) {
-        if (Array.isArray(doc[key]) && doc[key].length > 10) {
-            formatted[key] = [...doc[key].slice(0, 10), `... +${doc[key].length - 10} more`];
-        } else {
-            formatted[key] = doc[key];
-        }
+  const formatted = { ...doc };
+  
+  // Handle arrays with preview
+  Object.keys(formatted).forEach(key => {
+    if (Array.isArray(formatted[key])) {
+      const array = formatted[key];
+      formatted[key] = {
+        preview: array.slice(0, 10),
+        remaining: Math.max(0, array.length - 10)
+      };
     }
-    return formatted;
+  });
+  
+  return formatted;
 }