@takara-ai/miru-code 1.8.2 → 1.10.0

package/src/spinner.ts

@@ -8,11 +8,14 @@ export class Spinner {
   private frame = 0;
   private belowLines = 0;
 
-  constructor(private readonly message: string) {}
+  constructor(
+    private readonly message: string,
+    private readonly stderr: SpinnerOutput = process.stderr,
+  ) {}
 
   start(): void {
-    if (!process.stderr.isTTY) {
-      process.stderr.write(`${this.message}...\n`);
+    if (!this.stderr.isTTY) {
+      this.stderr.write(`${this.message}...\n`);
       return;
     }
     this.draw();
@@ -20,17 +23,17 @@ export class Spinner {
   }
 
   follow(line: string): void {
-    if (!process.stderr.isTTY) {
-      process.stderr.write(`${line}\n`);
+    if (!this.stderr.isTTY) {
+      this.stderr.write(`${line}\n`);
       return;
     }
-    const cols = process.stderr.columns || 80;
+    const cols = this.stderr.columns || 80;
     const rows = line.split("\n").reduce((count, part) => {
       return count + Math.max(1, Math.ceil(displayWidth(part) / cols));
     }, 0);
-    process.stderr.write(`\n${line}`);
+    this.stderr.write(`\n${line}`);
     this.belowLines += rows;
-    process.stderr.write(`\x1b[${this.belowLines}A`);
+    this.stderr.write(`\x1b[${this.belowLines}A`);
   }
 
   stop(finalMessage?: string): void {
@@ -38,19 +41,19 @@ export class Spinner {
       clearInterval(this.timer);
       this.timer = null;
     }
-    if (process.stderr.isTTY) {
-      process.stderr.write("\r\x1b[K");
+    if (this.stderr.isTTY) {
+      this.stderr.write("\r\x1b[K");
       if (finalMessage) {
-        process.stderr.write(finalMessage);
+        this.stderr.write(finalMessage);
       }
       if (this.belowLines > 0) {
-        process.stderr.write(`\x1b[${this.belowLines}B`);
+        this.stderr.write(`\x1b[${this.belowLines}B`);
       }
       if (finalMessage || this.belowLines > 0) {
-        process.stderr.write("\n");
+        this.stderr.write("\n");
       }
     } else if (finalMessage) {
-      process.stderr.write(`${finalMessage}\n`);
+      this.stderr.write(`${finalMessage}\n`);
     }
   }
 
@@ -68,16 +71,23 @@ export class Spinner {
 
   private draw(): void {
     const glyph = FRAMES[this.frame++ % FRAMES.length];
-    process.stderr.write(`\r${dim(glyph)} ${this.message}`);
+    this.stderr.write(`\r${dim(glyph)} ${this.message}`);
   }
 }
 
+interface SpinnerOutput {
+  isTTY?: boolean;
+  columns?: number;
+  write(text: string): unknown;
+}
+
 export async function withSpinner<T>(
   message: string,
   fn: () => Promise<T>,
   options?: { successMessage?: string; failMessage?: string },
+  output: SpinnerOutput = process.stderr,
 ): Promise<T> {
-  const spinner = new Spinner(message);
+  const spinner = new Spinner(message, output);
   spinner.start();
   try {
     const result = await fn();

package/src/tokenizer/bert-wordpiece.ts

@@ -99,9 +99,6 @@ function whitespaceTokenize(text: string): string[] {
 
 function splitOnPunctuation(text: string): string[] {
   const chars = [...text];
-  if (chars.length === 0) {
-    return [];
-  }
 
   const output: string[][] = [];
   let startNewWord = true;
@@ -138,9 +135,6 @@ function wordpieceTokenize(
   prefix: string,
   maxInputCharsPerWord: number,
 ): string[] {
-  if (token.length === 0) {
-    return [];
-  }
   if (token.length > maxInputCharsPerWord) {
     return [unkToken];
   }

package/src/utils.ts

@@ -100,22 +100,12 @@ export type ExpandResults = {
   chunks: Record<string, unknown>[];
 };
 
-export const DEFAULT_MCP_TOP_K = 3;
-export const MAX_MCP_TOP_K = 10;
+export const SEARCH_RESULT_COUNT = 5;
 
 /** Extra chunks before/after the expand anchor — MCP `expand` tool defaults. */
 export const DEFAULT_EXPAND_BEFORE = 1;
 export const DEFAULT_EXPAND_AFTER = 1;
 
-/** Clamp MCP top_k to a sane range; omit for the default. */
-export function clampMcpTopK(topK?: number): number {
-  const value = topK ?? DEFAULT_MCP_TOP_K;
-  if (!Number.isFinite(value) || value < 1) {
-    return DEFAULT_MCP_TOP_K;
-  }
-  return Math.min(Math.floor(value), MAX_MCP_TOP_K);
-}
-
 function chunkToResponseDict(chunk: Chunk, repoRoot?: string | null): Record<string, unknown> {
   const dict = chunkToDict(chunk);
   if (!repoRoot) {
@@ -175,10 +165,6 @@ export function expandChunksAtLine(
   const anchorIndex = fileChunks.findIndex(
     (chunk) => chunk.start_line === anchor.start_line && chunk.end_line === anchor.end_line,
   );
-  if (anchorIndex < 0) {
-    return { anchor, chunks: [anchor] };
-  }
-
   const start = Math.max(0, anchorIndex - before);
   const end = Math.min(fileChunks.length, anchorIndex + after + 1);
   return { anchor, chunks: fileChunks.slice(start, end) };

package/src/version.ts

@@ -26,8 +26,8 @@ export function miruVersion(): string {
  * - 0.x releases: `0.{minor}` (chunking may break on minor bumps pre-1.0).
  * - 1.x+: major version only.
  */
-export function indexCacheEpoch(): string {
-  const parts = packageJson.version.split(".");
+export function indexCacheEpoch(version: string = packageJson.version): string {
+  const parts = version.split(".");
   const major = parts[0] ?? "0";
   const minor = parts[1] ?? "0";
   if (major === "0") {
@@ -118,7 +118,7 @@ export async function fetchLatestPublishedVersion(fetchImpl: Fetcher = fetch): P
 }
 
 /** Print update hint to stderr when a newer package is on npm (at most once per day). */
-export async function maybeNotifyUpdate(): Promise<void> {
+export async function maybeNotifyUpdate(fetchImpl: Fetcher = fetch): Promise<void> {
   if (process.env.MIRU_NO_UPDATE_CHECK === "1") {
     return;
   }
@@ -133,7 +133,7 @@ export async function maybeNotifyUpdate(): Promise<void> {
   }
 
   try {
-    const latest = await fetchLatestPublishedVersion();
+    const latest = await fetchLatestPublishedVersion(fetchImpl);
     await writeUpdateCheckCache(latest);
     if (isVersionNewer(latest, current)) {
       writeUpdateNotice(latest, current);