memax-cli 0.0.1 → 0.1.0-alpha.10

package/src/commands/recall.ts

@@ -0,0 +1,163 @@
+import chalk from "chalk";
+import { apiPost } from "../lib/api.js";
+
+interface RecallResult {
+  notes: Array<{
+    id: string;
+    title: string;
+    chunk_content: string;
+    heading_chain: string;
+    relevance_score: number;
+    category: string;
+    source: string;
+    age: string;
+  }>;
+  query_metadata: {
+    intent: string;
+    total_candidates: number;
+    latency_ms: number;
+  };
+}
+
+interface RecallOptions {
+  category?: string;
+  tags?: string;
+  limit?: string;
+  format?: string;
+  includeArchived?: boolean;
+  hook?: boolean;
+  maxTokens?: string;
+}
+
+export async function recallCommand(
+  query: string | undefined,
+  options: RecallOptions,
+): Promise<void> {
+  if (!query) {
+    // Read from stdin if no query argument
+    if (!process.stdin.isTTY) {
+      const chunks: Buffer[] = [];
+      for await (const chunk of process.stdin) {
+        chunks.push(chunk);
+      }
+      query = Buffer.concat(chunks).toString("utf-8").trim();
+    }
+    if (!query) {
+      console.error(chalk.red('Provide a query: memax recall "your question"'));
+      process.exit(1);
+    }
+  }
+
+  const limit = parseInt(options.limit ?? "5", 10);
+
+  try {
+    const result = await apiPost<RecallResult>("/v1/recall", {
+      query,
+      category: options.category ?? "",
+      limit,
+      include_archived: options.includeArchived ?? false,
+      source: options.hook ? "hook" : "cli",
+      working_dir: process.cwd(),
+    });
+
+    const notes = result.notes ?? [];
+
+    if (options.format === "json") {
+      console.log(JSON.stringify({ ...result, notes }, null, 2));
+      return;
+    }
+
+    // Hook mode: output clean context for agent injection
+    if (options.hook) {
+      if (notes.length === 0) return;
+      let output = "<memax-context>\n";
+      output += "## Relevant Context (from Memax)\n\n";
+      for (const note of notes) {
+        const heading = note.heading_chain ? ` — ${note.heading_chain}` : "";
+        output += `### ${note.title} [${note.category}, ${note.age}]${heading}\n`;
+        output += note.chunk_content + "\n\n";
+      }
+      output += "</memax-context>";
+
+      // Truncate to fit within token budget (1 token ≈ 4 characters)
+      const maxTokens = options.maxTokens
+        ? parseInt(options.maxTokens, 10)
+        : undefined;
+      if (maxTokens) {
+        const maxChars = maxTokens * 4;
+        if (output.length > maxChars) {
+          output =
+            output.substring(0, maxChars - 50) +
+            "\n\n[truncated to fit token budget]\n</memax-context>";
+        }
+      }
+
+      console.log(output);
+      return;
+    }
+
+    // Pipe-friendly plain text output when stdout is not a TTY
+    if (!process.stdout.isTTY) {
+      if (notes.length === 0) {
+        return;
+      }
+      for (const note of notes) {
+        const score = (note.relevance_score * 100).toFixed(0);
+        console.log(
+          `${note.title} [${note.category}] ${score}% · ${note.age} (${note.id})`,
+        );
+        if (note.heading_chain) {
+          console.log(`  ${note.heading_chain}`);
+        }
+        const lines = note.chunk_content.split("\n").slice(0, 3);
+        for (const line of lines) {
+          console.log(`  ${line}`);
+        }
+        console.log();
+      }
+      return;
+    }
+
+    // Human-readable output
+    if (notes.length === 0) {
+      console.log(chalk.yellow("No results found."));
+      console.log(
+        chalk.gray(
+          `  Searched ${result.query_metadata.total_candidates} chunks in ${result.query_metadata.latency_ms}ms`,
+        ),
+      );
+      return;
+    }
+
+    console.log(
+      chalk.blue(`${notes.length} result${notes.length > 1 ? "s" : ""}`),
+      chalk.gray(
+        `(${result.query_metadata.total_candidates} chunks searched, ${result.query_metadata.latency_ms}ms)`,
+      ),
+    );
+    console.log();
+
+    for (const note of notes) {
+      const score = (note.relevance_score * 100).toFixed(0);
+      console.log(
+        chalk.bold(note.title),
+        chalk.gray(`[${note.category}]`),
+        chalk.cyan(`${score}%`),
+        chalk.gray(`· ${note.age}`),
+      );
+      console.log(chalk.gray(`  id: ${note.id}`));
+      if (note.heading_chain) {
+        console.log(chalk.gray(`  ${note.heading_chain}`));
+      }
+      // Show first 3 lines of chunk content
+      const lines = note.chunk_content.split("\n").slice(0, 3);
+      for (const line of lines) {
+        console.log(chalk.white(`  ${line}`));
+      }
+      console.log();
+    }
+  } catch (err) {
+    console.error(chalk.red(`Recall failed: ${(err as Error).message}`));
+    process.exit(1);
+  }
+}