@virsanghavi/axis-server 1.5.0 → 1.6.0

package/dist/mcp-server.mjs

@@ -1166,6 +1166,7 @@ import fs4 from "fs";
 
 // ../../src/local/local-search.ts
 import fs3 from "fs/promises";
+import fsSync2 from "fs";
 import path3 from "path";
 var SKIP_DIRS = /* @__PURE__ */ new Set([
   "node_modules",
@@ -1321,34 +1322,61 @@ var STOP_WORDS = /* @__PURE__ */ new Set([
   "most",
   "some",
   "any",
-  "find",
-  "show",
-  "get",
-  "look",
-  "search",
-  "locate",
-  "check",
-  "file",
-  "files",
-  "code",
-  "function",
-  "class",
-  "method",
   "there",
   "here",
   "just",
   "also",
   "very",
   "really",
-  "quite"
+  "quite",
+  "show",
+  "look",
+  "locate",
+  "using",
+  "used",
+  "need",
+  "want"
 ]);
 var MAX_FILE_SIZE = 256 * 1024;
 var MAX_RESULTS = 20;
 var CONTEXT_LINES = 2;
 var MAX_MATCHES_PER_FILE = 6;
 function extractKeywords(query) {
-  const raw = query.toLowerCase().replace(/[^\w\s\-_.]/g, " ").split(/\s+/).filter((w) => w.length >= 2 && !STOP_WORDS.has(w));
-  return [...new Set(raw)];
+  const words = query.toLowerCase().replace(/[^\w\s\-_.]/g, " ").split(/\s+/).filter((w) => w.length >= 2);
+  const filtered = words.filter((w) => !STOP_WORDS.has(w));
+  const result = filtered.length > 0 ? filtered : words.filter((w) => w.length >= 3);
+  return [...new Set(result)];
+}
+var PROJECT_ROOT_MARKERS = [
+  ".git",
+  ".axis",
+  "package.json",
+  "Cargo.toml",
+  "go.mod",
+  "pyproject.toml",
+  "setup.py",
+  "Gemfile",
+  "pom.xml",
+  "tsconfig.json",
+  ".cursorrules",
+  "AGENTS.md"
+];
+function detectProjectRoot(startDir) {
+  let current = startDir;
+  const root = path3.parse(current).root;
+  while (current !== root) {
+    for (const marker of PROJECT_ROOT_MARKERS) {
+      try {
+        fsSync2.accessSync(path3.join(current, marker));
+        return current;
+      } catch {
+      }
+    }
+    const parent = path3.dirname(current);
+    if (parent === current) break;
+    current = parent;
+  }
+  return startDir;
 }
 async function walkDir(dir, maxDepth = 12) {
   const results = [];
@@ -1428,8 +1456,12 @@ async function searchFile(filePath, rootDir, keywords) {
   return { filePath, relativePath, score, matchedKeywords, regions };
 }
 async function localSearch(query, rootDir) {
-  const cwd = rootDir || process.cwd();
+  const rawCwd = rootDir || process.cwd();
+  const cwd = detectProjectRoot(rawCwd);
   const keywords = extractKeywords(query);
+  if (cwd !== rawCwd) {
+    logger.info(`[localSearch] Detected project root: ${cwd} (CWD was: ${rawCwd})`);
+  }
   if (keywords.length === 0) {
     return "Could not extract meaningful search terms from the query. Try being more specific (e.g. 'authentication middleware' instead of 'how does it work').";
   }
@@ -1557,8 +1589,20 @@ var subscription = {
 };
 async function verifySubscription() {
   if (!apiSecret) {
-    subscription = { checked: true, valid: true, plan: "local", reason: "No API key configured \u2014 local mode", checkedAt: Date.now() };
-    logger.info("[subscription] No API key \u2014 running in local/dev mode, skipping verification");
+    const hasDirectSupabase = !useRemoteApiOnly && !!process.env.NEXT_PUBLIC_SUPABASE_URL && !!process.env.SUPABASE_SERVICE_ROLE_KEY;
+    if (hasDirectSupabase) {
+      subscription = { checked: true, valid: true, plan: "developer", reason: "Direct Supabase mode \u2014 no API key needed", checkedAt: Date.now() };
+      logger.info("[subscription] Direct Supabase credentials found \u2014 developer mode, skipping verification");
+      return subscription;
+    }
+    subscription = {
+      checked: true,
+      valid: false,
+      plan: "none",
+      reason: "no_api_key",
+      checkedAt: Date.now()
+    };
+    logger.error("[subscription] No API key configured. Axis requires an API key from https://useaxis.dev/dashboard");
     return subscription;
   }
   const verifyUrl = apiUrl.endsWith("/v1") ? `${apiUrl}/verify` : `${apiUrl}/v1/verify`;
@@ -1616,6 +1660,22 @@ function isSubscriptionStale() {
   return Date.now() - subscription.checkedAt > RECHECK_INTERVAL_MS;
 }
 function getSubscriptionBlockMessage() {
+  if (subscription.reason === "no_api_key") {
+    return [
+      "\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550",
+      "  Axis API key required",
+      "",
+      "  No API key found. Axis requires an active subscription",
+      "  and a valid API key to operate.",
+      "",
+      "  1. Sign up or log in at https://useaxis.dev",
+      "  2. Subscribe to Axis Pro",
+      "  3. Generate an API key from the dashboard",
+      "  4. Add AXIS_API_KEY to your mcp.json configuration",
+      "  5. Restart your IDE",
+      "\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550"
+    ].join("\n");
+  }
   return [
     "\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550",
     "  Axis Pro subscription required",
@@ -1644,7 +1704,7 @@ async function ensureFileSystem() {
   try {
     const fs5 = await import("fs/promises");
     const path5 = await import("path");
-    const fsSync2 = await import("fs");
+    const fsSync3 = await import("fs");
     const cwd = process.cwd();
     logger.info(`Server CWD: ${cwd}`);
     const historyDir = path5.join(cwd, "history");
@@ -1653,7 +1713,7 @@ async function ensureFileSystem() {
     const axisDir = path5.join(cwd, ".axis");
     const axisInstructions = path5.join(axisDir, "instructions");
     const legacyInstructions = path5.join(cwd, "agent-instructions");
-    if (fsSync2.existsSync(legacyInstructions) && !fsSync2.existsSync(axisDir)) {
+    if (fsSync3.existsSync(legacyInstructions) && !fsSync3.existsSync(axisDir)) {
       logger.info("Using legacy agent-instructions directory");
     } else {
       await fs5.mkdir(axisInstructions, { recursive: true }).catch(() => {
@@ -2036,41 +2096,51 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
   }
   if (name === SEARCH_CONTEXT_TOOL) {
     const query = String(args?.query);
-    let ragResults = null;
-    try {
-      const remote = await manager.searchContext(query, nerveCenter.currentProjectName);
-      if (remote && !remote.includes("No results found") && remote.trim().length > 20) {
-        ragResults = remote;
-      }
-    } catch {
-    }
-    if (!ragResults && ragEngine) {
-      try {
-        const localRag = await ragEngine.search(query);
-        if (localRag.length > 0) {
-          ragResults = localRag.join("\n---\n");
-        }
-      } catch {
-      }
-    }
-    let localResults = null;
+    logger.info(`[search_codebase] Query: "${query}"`);
+    let localResults = "";
     try {
       localResults = await localSearch(query);
+      logger.info(`[search_codebase] Local search completed: ${localResults.length} chars`);
     } catch (e) {
       logger.warn(`[search_codebase] Local search error: ${e}`);
+      localResults = "";
     }
-    const parts = [];
-    if (ragResults) {
-      parts.push("## Indexed Results (RAG)\n\n" + ragResults);
-    }
-    if (localResults && !localResults.startsWith("No matches found") && !localResults.startsWith("Could not extract")) {
-      parts.push("## Local Codebase Search\n\n" + localResults);
-    } else if (!ragResults) {
-      return { content: [{ type: "text", text: localResults || "No results found." }] };
+    let ragResults = null;
+    const RAG_TIMEOUT_MS = 3e3;
+    try {
+      const ragPromise = (async () => {
+        try {
+          const remote = await manager.searchContext(query, nerveCenter.currentProjectName);
+          if (remote && !remote.includes("No results found") && remote.trim().length > 20) {
+            return remote;
+          }
+        } catch {
+        }
+        if (ragEngine) {
+          try {
+            const results = await ragEngine.search(query);
+            if (results.length > 0) return results.join("\n---\n");
+          } catch {
+          }
+        }
+        return null;
+      })();
+      ragResults = await Promise.race([
+        ragPromise,
+        new Promise((resolve) => setTimeout(() => resolve(null), RAG_TIMEOUT_MS))
+      ]);
+      if (ragResults) {
+        logger.info(`[search_codebase] RAG returned results (${ragResults.length} chars)`);
+      }
+    } catch {
     }
-    if (parts.length === 0) {
-      return { content: [{ type: "text", text: "No results found for this query." }] };
+    const hasLocal = localResults && !localResults.startsWith("No matches found") && !localResults.startsWith("Could not extract");
+    if (!hasLocal && !ragResults) {
+      return { content: [{ type: "text", text: localResults || "No results found for this query." }] };
     }
+    const parts = [];
+    if (hasLocal) parts.push(localResults);
+    if (ragResults) parts.push("## Indexed Results (RAG)\n\n" + ragResults);
     return { content: [{ type: "text", text: parts.join("\n\n---\n\n") }] };
   }
   if (name === "get_subscription_status") {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@virsanghavi/axis-server",
-    "version": "1.5.0",
+    "version": "1.6.0",
     "description": "Axis MCP Server CLI",
     "main": "dist/index.js",
     "bin": {