@virsanghavi/axis-server 1.0.4 → 1.0.6

package/dist/mcp-server.mjs

@@ -110,17 +110,18 @@ var ContextManager = class {
       return `Updated ${filename}`;
     });
   }
-  async searchContext(query) {
+  async searchContext(query, projectName = "default") {
     if (!this.apiUrl) {
       throw new Error("SHARED_CONTEXT_API_URL not configured.");
     }
-    const response = await fetch(`${this.apiUrl}/search`, {
+    const endpoint = this.apiUrl.endsWith("/v1") ? `${this.apiUrl}/search` : `${this.apiUrl}/v1/search`;
+    const response = await fetch(endpoint, {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
         "Authorization": `Bearer ${this.apiSecret || ""}`
       },
-      body: JSON.stringify({ query })
+      body: JSON.stringify({ query, projectName })
     });
     if (!response.ok) {
       const text = await response.text();
@@ -134,18 +135,19 @@ var ContextManager = class {
     }
     throw new Error("No results format recognized.");
   }
-  async embedContent(items) {
+  async embedContent(items, projectName = "default") {
     if (!this.apiUrl) {
       console.warn("Skipping RAG embedding: SHARED_CONTEXT_API_URL not configured.");
       return;
     }
-    const response = await fetch(`${this.apiUrl}/embed`, {
+    const endpoint = this.apiUrl.endsWith("/v1") ? `${this.apiUrl}/embed` : `${this.apiUrl}/v1/embed`;
+    const response = await fetch(endpoint, {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
         "Authorization": `Bearer ${this.apiSecret || ""}`
       },
-      body: JSON.stringify({ items })
+      body: JSON.stringify({ items, projectName })
     });
     if (!response.ok) {
       const text = await response.text();
@@ -238,6 +240,9 @@ var NerveCenter = class {
   get projectId() {
     return this._projectId;
   }
+  get currentProjectName() {
+    return this.projectName;
+  }
   async init() {
     await this.loadState();
     await this.detectProjectName();
@@ -758,43 +763,49 @@ var nerveCenter = new NerveCenter(manager, {
   supabaseServiceRoleKey: process.env.SUPABASE_SERVICE_ROLE_KEY,
   projectName: process.env.PROJECT_NAME || "default"
 });
-var ragEngine = new RagEngine(
-  process.env.NEXT_PUBLIC_SUPABASE_URL,
-  process.env.SUPABASE_SERVICE_ROLE_KEY,
-  process.env.OPENAI_API_KEY || ""
-  // Project ID is loaded async by NerveCenter... tricky dependency.
-  // We'll let NerveCenter expose it or pass it later.
-);
+var ragEngine;
+if (process.env.NEXT_PUBLIC_SUPABASE_URL && process.env.SUPABASE_SERVICE_ROLE_KEY) {
+  ragEngine = new RagEngine(
+    process.env.NEXT_PUBLIC_SUPABASE_URL,
+    process.env.SUPABASE_SERVICE_ROLE_KEY,
+    process.env.OPENAI_API_KEY || ""
+  );
+}
 async function ensureFileSystem() {
-  const fs3 = await import("fs/promises");
-  const path3 = await import("path");
-  const fsSync = await import("fs");
-  const cwd = process.cwd();
-  const historyDir = path3.join(cwd, "history");
-  await fs3.mkdir(historyDir, { recursive: true }).catch(() => {
-  });
-  const axisDir = path3.join(cwd, ".axis");
-  const axisInstructions = path3.join(axisDir, "instructions");
-  const legacyInstructions = path3.join(cwd, "agent-instructions");
-  if (fsSync.existsSync(legacyInstructions) && !fsSync.existsSync(axisDir)) {
-    logger.info("Using legacy agent-instructions directory");
-  } else {
-    await fs3.mkdir(axisInstructions, { recursive: true }).catch(() => {
+  try {
+    const fs3 = await import("fs/promises");
+    const path3 = await import("path");
+    const fsSync = await import("fs");
+    const cwd = process.cwd();
+    logger.info(`Server CWD: ${cwd}`);
+    const historyDir = path3.join(cwd, "history");
+    await fs3.mkdir(historyDir, { recursive: true }).catch(() => {
     });
-    const defaults = [
-      ["context.md", "# Project Context\n\n"],
-      ["conventions.md", "# Coding Conventions\n\n"],
-      ["activity.md", "# Activity Log\n\n"]
-    ];
-    for (const [file, content] of defaults) {
-      const p = path3.join(axisInstructions, file);
-      try {
-        await fs3.access(p);
-      } catch {
-        await fs3.writeFile(p, content);
-        logger.info(`Created default context file: ${file}`);
+    const axisDir = path3.join(cwd, ".axis");
+    const axisInstructions = path3.join(axisDir, "instructions");
+    const legacyInstructions = path3.join(cwd, "agent-instructions");
+    if (fsSync.existsSync(legacyInstructions) && !fsSync.existsSync(axisDir)) {
+      logger.info("Using legacy agent-instructions directory");
+    } else {
+      await fs3.mkdir(axisInstructions, { recursive: true }).catch(() => {
+      });
+      const defaults = [
+        ["context.md", "# Project Context\n\n"],
+        ["conventions.md", "# Coding Conventions\n\n"],
+        ["activity.md", "# Activity Log\n\n"]
+      ];
+      for (const [file, content] of defaults) {
+        const p = path3.join(axisInstructions, file);
+        try {
+          await fs3.access(p);
+        } catch {
+          await fs3.writeFile(p, content);
+          logger.info(`Created default context file: ${file}`);
+        }
       }
     }
+  } catch (error) {
+    logger.warn("Could not initialize local file system. Persistence features (context.md) may be disabled.", { error: String(error) });
   }
 }
 var server = new Server(
@@ -1084,13 +1095,29 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
   if (name === "index_file") {
     const filePath = String(args?.filePath);
     const content = String(args?.content);
-    const success = await ragEngine.indexContent(filePath, content);
-    return { content: [{ type: "text", text: success ? "Indexed." : "Failed." }] };
+    try {
+      await manager.embedContent([{ content, metadata: { filePath } }], nerveCenter.currentProjectName);
+      return { content: [{ type: "text", text: "Indexed via Remote API." }] };
+    } catch (e) {
+      if (ragEngine) {
+        const success = await ragEngine.indexContent(filePath, content);
+        return { content: [{ type: "text", text: success ? "Indexed locally." : "Local index failed." }] };
+      }
+      return { content: [{ type: "text", text: `Indexing failed: ${e}` }], isError: true };
+    }
   }
   if (name === SEARCH_CONTEXT_TOOL) {
     const query = String(args?.query);
-    const results = await ragEngine.search(query);
-    return { content: [{ type: "text", text: results.join("\n---\n") }] };
+    try {
+      const results = await manager.searchContext(query, nerveCenter.currentProjectName);
+      return { content: [{ type: "text", text: results }] };
+    } catch (e) {
+      if (ragEngine) {
+        const results = await ragEngine.search(query);
+        return { content: [{ type: "text", text: results.join("\n---\n") }] };
+      }
+      return { content: [{ type: "text", text: `Search failed: ${e}` }], isError: true };
+    }
   }
   if (name === "get_subscription_status") {
     const email = String(args?.email);
@@ -1162,9 +1189,9 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
 async function main() {
   await ensureFileSystem();
   await nerveCenter.init();
-  if (nerveCenter.projectId) {
+  if (nerveCenter.projectId && ragEngine) {
     ragEngine.setProjectId(nerveCenter.projectId);
-    logger.info(`RAG Engine linked to Project ID: ${nerveCenter.projectId}`);
+    logger.info(`Local RAG Engine linked to Project ID: ${nerveCenter.projectId}`);
   }
   const transport = new StdioServerTransport();
   await server.connect(transport);

package/package.json

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