Update index.ts

Author: PV-Bhat

src/index.ts

@@ -1,14 +1,11 @@
-// src/index.ts --------------------------------------------------------------
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-
 import {
   InitializeRequestSchema,
   ListToolsRequestSchema,
   CallToolRequestSchema,
   CallToolRequest
 } from "@modelcontextprotocol/sdk/types.js";
-
 import { z } from "zod";
 
 import {
@@ -21,21 +18,23 @@ import {
   vibeLearnTool,   VibeLearnInput,   VibeLearnOutput
 } from "./tools/vibeLearn.js";
 
-/* --------------------------------------------------------------------- */
-/*  0.  optional Gemini startup                                          */
-/* --------------------------------------------------------------------- */
+// ────────────────────────────────
+// 0. optional Gemini startup
+// ────────────────────────────────
 if (process.env.GEMINI_API_KEY) {
   try {
     // initializeGemini(process.env.GEMINI_API_KEY);
     console.error("[LOG] Gemini initialised");
   } catch (e) {
     console.error("[ERR] Gemini init failed:", e);
   }
+} else {
+  console.error("[WARN] GEMINI_API_KEY not set – Gemini tools will fail");
 }
 
-/* --------------------------------------------------------------------- */
-/*  1.  Zod schemas (single source of truth)                             */
-/* --------------------------------------------------------------------- */
+// ────────────────────────────────
+// 1. Zod schemas
+// ────────────────────────────────
 const vibeCheckSchema = z.object({
   plan:        z.string(),
   userRequest: z.string(),
@@ -52,71 +51,78 @@ const vibeLearnSchema  = z.object({
   solution: z.string()
 });
 
-/* --------------------------------------------------------------------- */
-/*  2.  create Server with “tools” capability                            */
-/* --------------------------------------------------------------------- */
+// ────────────────────────────────
+// 2. Server with tools capability
+// ────────────────────────────────
 const server = new Server({
   name:    "vibe-check-mcp",
   version: "0.2.0",
   capabilities: { tools: {} }
 });
 
-/* --------------------------------------------------------------------- */
-/*  3.  initialize handler – MUST reply within 30 s                      */
-/* --------------------------------------------------------------------- */
+// ────────────────────────────────
+// 3. initialize handler
+// ────────────────────────────────
 server.setRequestHandler(InitializeRequestSchema, async req => ({
   protocolVersion: req.params.protocolVersion,
   serverInfo:      { name: "vibe-check-mcp", version: "0.2.0" },
-  capabilities:    server.capabilities          // echoes { tools: {} }
+  capabilities:    { tools: {} }     // ← fixed line
 }));
 
-/* --------------------------------------------------------------------- */
-/*  4.  tools/list handler – static descriptors                          */
-/* --------------------------------------------------------------------- */
+// ────────────────────────────────
+// 4. tools/list handler
+// ────────────────────────────────
 const toolList = [
   { name: "vibe_check",  description: "Metacognitive check …", inputSchema: vibeCheckSchema },
   { name: "vibe_distill",description: "Distils a plan …",      inputSchema: vibeDistillSchema },
   { name: "vibe_learn",  description: "Logs mistake patterns", inputSchema: vibeLearnSchema  }
 ];
 server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: toolList }));
 
-/* --------------------------------------------------------------------- */
-/*  5.  tools/call dispatcher                                            */
-/* --------------------------------------------------------------------- */
+// ────────────────────────────────
+// 5. tools/call dispatcher
+// ────────────────────────────────
 server.setRequestHandler(CallToolRequestSchema, async (req: CallToolRequest) => {
   const { name, arguments: raw = {} } = req.params;
 
   switch (name) {
     case "vibe_check": {
       const args: VibeCheckInput = vibeCheckSchema.parse(raw);
       const out:  VibeCheckOutput = await vibeCheckTool(args);
-      return { content:[{ type:"text",
-        text: out.questions + (out.patternAlert ? `\n\n**Pattern Alert:** ${out.patternAlert}` : "")
-      }]};
+      return {
+        content:[{ type:"text",
+          text: out.questions +
+               (out.patternAlert ? `\n\n**Pattern Alert:** ${out.patternAlert}` : "")
+        }]
+      };
     }
     case "vibe_distill": {
       const args: VibeDistillInput = vibeDistillSchema.parse(raw);
       const out:  VibeDistillOutput = await vibeDistillTool(args);
-      return { content:[{ type:"markdown",
-        markdown: `${out.distilledPlan}\n\n**Rationale:** ${out.rationale}`
-      }]};
+      return {
+        content:[{ type:"markdown",
+          markdown: `${out.distilledPlan}\n\n**Rationale:** ${out.rationale}`
+        }]
+      };
     }
     case "vibe_learn": {
       const args: VibeLearnInput = vibeLearnSchema.parse(raw);
       const out:  VibeLearnOutput = await vibeLearnTool(args);
       const summary = out.topCategories.map(c => `- ${c.category} (${c.count})`).join("\n");
-      return { content:[{ type:"text",
-        text:`✅ Logged. Current tally: ${out.currentTally}\n\n${summary}`
-      }]};
+      return {
+        content:[{ type:"text",
+          text:`✅ Logged. Current tally: ${out.currentTally}\n\n${summary}`
+        }]
+      };
     }
     default:
       throw new Error(`Unknown tool "${name}"`);
   }
 });
 
-/* --------------------------------------------------------------------- */
-/*  6.  connect via STDIO                                                */
-/* --------------------------------------------------------------------- */
+// ────────────────────────────────
+// 6. STDIO transport
+// ────────────────────────────────
 const transport = new StdioServerTransport();
 await server.connect(transport);
 console.error("[OK] vibe‑check‑mcp ready (stdio)");