mikoshi 0.1.0 → 0.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mikoshi",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Private local code search + MCP",
   "type": "module",
   "bin": {

package/src/mikoshi/indexing/indexer.js

@@ -57,7 +57,7 @@ function reuseChunks(prevChunks, prevEmbeddings) {
   return { chunksByFile, embeddingById };
 }
 
-export async function indexRepo(repoPath, configOverride = null) {
+export async function indexRepo(repoPath, configOverride = null, providerOverride = null) {
   const config = configOverride || loadConfig();
   const repoRoot = path.resolve(repoPath);
 
@@ -136,10 +136,10 @@ export async function indexRepo(repoPath, configOverride = null) {
     }
   }
 
-  let provider = null;
+  let provider = providerOverride;
   let pendingEmbeddings = [];
   if (pendingTexts.length) {
-    provider = getEmbeddingsProvider(config);
+    if (!provider) provider = getEmbeddingsProvider(config);
     pendingEmbeddings = await provider.embedTexts(pendingTexts);
   }
 

package/src/mikoshi/mcp_server/server.js

@@ -5,6 +5,7 @@ import { searchRepo } from "../retrieval/hybrid.js";
 import { loadConfig } from "../config.js";
 import { IndexStore } from "../indexing/index_store.js";
 import path from "node:path";
+import { getEmbeddingsProvider } from "../retrieval/semantic.js";
 
 const TOOL_NAME = "codebase-retrieval";
 const TOOL_SCHEMA = {
@@ -54,12 +55,13 @@ async function handleToolCall(params) {
   const config = loadConfig();
   const repoRoot = path.resolve(repoPath);
   const store = new IndexStore(repoRoot, config.index_root);
+  const provider = getEmbeddingsProvider(config);
   if (!store.exists()) {
-    await indexRepo(repoRoot, config);
+    await indexRepo(repoRoot, config, provider);
   } else {
-    await indexRepo(repoRoot, config);
+    await indexRepo(repoRoot, config, provider);
   }
-  const results = await searchRepo(repoRoot, query, k);
+  const results = await searchRepo(repoRoot, query, k, provider);
   return {
     isError: false,
     content: [{ type: "text", text: JSON.stringify(results) }],
@@ -67,6 +69,17 @@ async function handleToolCall(params) {
   };
 }
 
+const args = new Set(process.argv.slice(2));
+const quiet = args.has("--quiet");
+const shouldAnnounce =
+  !quiet && (process.env.MIKOSHI_MCP_BANNER === "1" || process.stdin.isTTY);
+
+if (shouldAnnounce) {
+  console.error("Mikoshi MCP running (stdio). Waiting for client...");
+}
+
+process.env.MIKOSHI_SILENT = "1";
+
 const rl = readline.createInterface({ input: process.stdin, crlfDelay: Infinity });
 
 rl.on("line", async (line) => {
@@ -88,7 +101,7 @@ rl.on("line", async (line) => {
         result: {
           protocolVersion: params?.protocolVersion || "2024-11-05",
           capabilities: { tools: {} },
-          serverInfo: { name: "mikoshi", version: "0.1.0" },
+          serverInfo: { name: "mikoshi", version: "0.1.2" },
         },
       });
       return;

package/src/mikoshi/retrieval/hybrid.js

@@ -54,7 +54,7 @@ export async function hybridSearch(query, chunks, lexical, semantic, k = 8, alph
   return reranked.slice(0, k);
 }
 
-export async function searchRepo(repoPath, query, k = 8) {
+export async function searchRepo(repoPath, query, k = 8, providerOverride = null) {
   const config = loadConfig();
   const repoRoot = path.resolve(repoPath);
   const store = new IndexStore(repoRoot, config.index_root);
@@ -71,7 +71,7 @@ export async function searchRepo(repoPath, query, k = 8) {
   if (!chunks.length) return [];
 
   const lexical = new LexicalIndex(chunks.map((chunk) => chunk.text));
-  const provider = getEmbeddingsProvider(config);
+  const provider = providerOverride || getEmbeddingsProvider(config);
   const semantic = new SemanticSearcher(embeddingsData.embeddings, embeddingsData.dim, provider);
   const hits = await hybridSearch(query, chunks, lexical, semantic, k);
 

package/src/mikoshi/retrieval/semantic.js

@@ -103,15 +103,16 @@ class TransformersEmbeddingsProvider {
 
       const offline = isOffline();
       const timeoutMs = downloadTimeoutMs();
+      const silent = isSilent();
 
-      console.error("🧠 Loading embeddings model…");
+      if (!silent) console.error("🧠 Loading embeddings model…");
 
       env.allowRemoteModels = false;
       try {
         const extractor = await pipeline("feature-extraction", this.config.embeddings.model, {
           quantized: false,
         });
-        console.error("✅ Embeddings model loaded");
+        if (!silent) console.error("✅ Embeddings model loaded");
         return extractor;
       } catch (err) {
         if (offline) {
@@ -121,14 +122,16 @@ class TransformersEmbeddingsProvider {
         }
       }
 
-      console.error("⬇️ Downloading local model (one-time)…");
+      if (!silent) console.error("⬇️ Downloading local model (one-time)…");
       env.allowRemoteModels = true;
       const extractor = await withTimeout(
         pipeline("feature-extraction", this.config.embeddings.model, { quantized: false }),
         timeoutMs
       );
-      console.error("✅ Model ready");
-      console.error("✅ Embeddings model loaded");
+      if (!silent) {
+        console.error("✅ Model ready");
+        console.error("✅ Embeddings model loaded");
+      }
       return extractor;
     })();
     return this._pipelinePromise;
@@ -153,6 +156,11 @@ function isOffline() {
   return ["1", "true", "yes", "on"].includes(raw.toLowerCase());
 }
 
+function isSilent() {
+  const raw = process.env.MIKOSHI_SILENT || "";
+  return ["1", "true", "yes", "on"].includes(raw.toLowerCase());
+}
+
 function downloadTimeoutMs() {
   const raw = process.env.MIKOSHI_MODEL_DOWNLOAD_TIMEOUT_SEC;
   const seconds = raw ? Number.parseInt(raw, 10) : 300;