@ivotoby/postgram-cli 1.2.0 → 1.4.0

package/README.md

@@ -56,6 +56,9 @@ pgm task complete <task-id> --version 1
 # Sync markdown directory
 pgm sync ./notes --repo my-notes
 
+# Check enrichment/extraction queue status
+pgm queue
+
 # JSON output (all commands)
 pgm store "hello" --json
 ```

package/dist/client.js

@@ -159,6 +159,9 @@ export function createPgmClient(options) {
             const query = qs.toString();
             return request(options, `/api/entities/${entityId}/edges${query ? `?${query}` : ''}`);
         },
+        getQueueStatus() {
+            return request(options, '/api/queue');
+        },
         expandGraph(entityId, params = {}) {
             const qs = new URLSearchParams();
             if (params.depth !== undefined)

package/dist/pgm.js

@@ -29,6 +29,13 @@ function formatSearchResults(results) {
         if (result.entity.content && result.entity.content !== result.chunk_content) {
             lines.push(`  entity: ${result.entity.content}`);
         }
+        if (result.related && result.related.length > 0) {
+            lines.push(`  related (${result.related.length}):`);
+            for (const rel of result.related) {
+                const arrow = rel.direction === 'outgoing' ? '->' : '<-';
+                lines.push(`    ${arrow} [${rel.relation}] ${rel.entity.type} ${shortId(rel.entity.id)}`);
+            }
+        }
     }
     return lines;
 }
@@ -237,6 +244,7 @@ program
     .option('--limit <limit>', 'result limit', '10')
     .option('--threshold <threshold>', 'similarity threshold', '0.35')
     .option('--recency-weight <recencyWeight>', 'recency weight', '0.1')
+    .option('--expand-graph', 'include graph-connected entities in results')
     .action(async (query, options, command) => {
     await runWithClient(command, async (client, json) => {
         const body = await client.searchEntities({
@@ -247,7 +255,8 @@ program
             owner: options.owner,
             limit: Number(options.limit),
             threshold: Number(options.threshold),
-            recency_weight: Number(options.recencyWeight)
+            recency_weight: Number(options.recencyWeight),
+            expand_graph: options.expandGraph === true ? true : undefined
         });
         return json ? body : formatSearchResults(body.results);
     });
@@ -476,6 +485,29 @@ taskCommand
             });
     });
 });
+program
+    .command('queue')
+    .description('Show enrichment and extraction queue status')
+    .action(async (_options, command) => {
+    await runWithClient(command, async (client, json) => {
+        const body = await client.getQueueStatus();
+        if (json)
+            return body;
+        const e = body.embedding;
+        const age = e.oldest_pending_secs !== null ? ` oldest_pending=${e.oldest_pending_secs}s` : '';
+        const lines = [
+            `embedding:  pending=${e.pending}  completed=${e.completed}  failed=${e.failed}  retry_eligible=${e.retry_eligible}${age}`
+        ];
+        if (body.extraction) {
+            const x = body.extraction;
+            lines.push(`extraction: pending=${x.pending}  completed=${x.completed}  failed=${x.failed}`);
+        }
+        else {
+            lines.push('extraction: disabled');
+        }
+        return lines;
+    });
+});
 program
     .command('backup')
     .description('Create a database backup')

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ivotoby/postgram-cli",
-  "version": "1.2.0",
+  "version": "1.4.0",
   "description": "Postgram CLI — store, search, and manage entities from the command line",
   "type": "module",
   "bin": {