@coreviz/cli 1.0.6 → 1.0.8

package/README.md

@@ -66,6 +66,9 @@ Search local images using natural language:
 npx @coreviz/cli search "a person wearing a red t-shirt"
 ```
 
+![Screenshot of CoreViz CLI visually searching through a folder using AI.](./screenshots/search.png)
+
+
 This will index the images in your current directory (creating a `.index.db` file) and return the top matches for your query.
 
 ## Development

package/bin/cli.js

@@ -296,9 +296,12 @@ program.command('describe <image-path>')
 
 program.command('search <query>')
     .description('Search for images in the current directory using AI')
-    .action(async (query) => {
+    .option('-m, --mode <mode>', 'The mode to use for embedding. Defaults to "local".', 'local')
+    .action(async (query, options) => {
         intro(chalk.bgHex('#663399').white('CoreViz'));
 
+        const mode = options.mode || 'local';
+
         const session = config.get('session');
         if (!session || !session.access_token) {
             cancel('You are not logged in. Please run `coreviz login` first.');
@@ -345,6 +348,14 @@ program.command('search <query>')
             }
         }
 
+        if (mode === 'local') {
+            // You're using the local model, it might take a few minutes for the model to load on the first run.
+            spinner.text = "Loading local model, this might take a few minutes to load the first time...";
+            spinner.start();
+            await coreviz.embedLocal('text', { type: 'text', mode: mode });
+            spinner.stop();
+        }
+
         for (const file of files) {
             const filePath = path.join(process.cwd(), file);
             const stats = fs.statSync(filePath);
@@ -361,7 +372,7 @@ program.command('search <query>')
 
             try {
                 const base64Image = readImageAsBase64(filePath);
-                const { embedding } = await coreviz.embed(base64Image, { type: 'image', mode: 'local' });
+                const { embedding } = await coreviz.embed(base64Image, { type: 'image', mode: mode });
 
                 upsertFile.run(file, mtime, JSON.stringify(embedding));
             } catch (error) {
@@ -373,7 +384,7 @@ program.command('search <query>')
         spinner.text = "Processing search query...";
 
         try {
-            const { embedding: queryEmbedding } = await coreviz.embed(query, { type: 'text', mode: 'local' });
+            const { embedding: queryEmbedding } = await coreviz.embed(query, { type: 'text', mode: mode });
 
             const rows = db.prepare('SELECT path, embedding FROM images').all();
             const results = [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@coreviz/cli",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "type": "module",
   "description": "CoreViz CLI tool",
   "main": "index.js",
@@ -27,7 +27,7 @@
   "homepage": "https://github.com/CoreViz/cli#readme",
   "dependencies": {
     "@clack/prompts": "^0.11.0",
-    "@coreviz/sdk": "^1.0.7",
+    "@coreviz/sdk": "^1.0.9",
     "better-auth": "^1.4.2",
     "better-sqlite3": "^12.4.6",
     "chalk": "^5.6.2",