@prajwolkc/stk 0.3.0 → 0.4.0

package/dist/mcp/server.js

@@ -1008,6 +1008,202 @@ server.tool("stk_cost", "Track costs across your stack: Stripe fees, Vercel usag
         content: [{ type: "text", text: JSON.stringify({ costs }, null, 2) }],
     };
 });
+// ──────────────────────────────────────────
+// Brain: Supabase Knowledge Base Client
+// ──────────────────────────────────────────
+function getBrainClient() {
+    const url = process.env.SUPABASE_URL;
+    const key = process.env.SUPABASE_SERVICE_KEY;
+    if (!url || !key)
+        return null;
+    return {
+        async query(table, params = {}) {
+            const searchParams = new URLSearchParams(params);
+            const res = await fetch(`${url}/rest/v1/${table}?${searchParams}`, {
+                headers: {
+                    apikey: key,
+                    Authorization: `Bearer ${key}`,
+                    "Content-Type": "application/json",
+                    Prefer: "count=exact",
+                },
+            });
+            const count = res.headers.get("content-range")?.split("/")[1] ?? null;
+            const data = await res.json();
+            return { data, count, ok: res.ok };
+        },
+        async insert(table, row) {
+            const res = await fetch(`${url}/rest/v1/${table}`, {
+                method: "POST",
+                headers: {
+                    apikey: key,
+                    Authorization: `Bearer ${key}`,
+                    "Content-Type": "application/json",
+                    Prefer: "return=representation",
+                },
+                body: JSON.stringify(row),
+            });
+            const data = await res.json();
+            return { data, ok: res.ok };
+        },
+    };
+}
+// ──────────────────────────────────────────
+// Tool: stk_brain_search
+// ──────────────────────────────────────────
+server.tool("stk_brain_search", "Search the knowledge base for SaaS patterns, best practices, and architecture examples from top open-source projects (LangChain, Ollama, Transformers, llama.cpp, vLLM, AutoGen, OpenAI Cookbook). Use this when you need to know how successful projects solve specific problems.", {
+    query: z.string().describe("What to search for (e.g., 'authentication', 'real-time updates', 'payment integration')"),
+    category: z.string().optional().describe("Filter: architecture, auth, payments, database, api, deployment, testing, performance, security, ml, realtime, general"),
+}, async ({ query, category }) => {
+    const brain = getBrainClient();
+    if (!brain)
+        return { content: [{ type: "text", text: JSON.stringify({ error: "SUPABASE_URL or SUPABASE_SERVICE_KEY not set" }) }] };
+    // Try ilike search on content and title
+    const words = query.split(" ").filter(w => w.length > 2);
+    const searchWord = words[0] ?? query;
+    const params = {
+        or: `(title.ilike.%${searchWord}%,content.ilike.%${searchWord}%)`,
+        limit: "10",
+        order: "source",
+    };
+    if (category)
+        params["category"] = `eq.${category}`;
+    const { data, ok } = await brain.query("knowledge", params);
+    if (!ok)
+        return { content: [{ type: "text", text: JSON.stringify({ error: "Query failed", data }) }] };
+    return {
+        content: [{
+                type: "text",
+                text: JSON.stringify({
+                    query,
+                    results: (data ?? []).map((r) => ({ source: r.source, category: r.category, title: r.title, content: r.content, tags: r.tags })),
+                    total: data?.length ?? 0,
+                }, null, 2),
+            }],
+    };
+});
+// ──────────────────────────────────────────
+// Tool: stk_brain_patterns
+// ──────────────────────────────────────────
+server.tool("stk_brain_patterns", "Get best practice patterns for a specific feature. Returns how top projects implement auth, payments, real-time, caching, APIs, etc.", {
+    feature: z.string().describe("The feature or pattern (e.g., 'authentication', 'webhooks', 'caching', 'model serving', 'fine-tuning')"),
+}, async ({ feature }) => {
+    const brain = getBrainClient();
+    if (!brain)
+        return { content: [{ type: "text", text: JSON.stringify({ error: "SUPABASE_URL or SUPABASE_SERVICE_KEY not set" }) }] };
+    const { data, ok } = await brain.query("knowledge", {
+        or: `(title.ilike.%${feature}%,content.ilike.%${feature}%)`,
+        limit: "15",
+        order: "source",
+    });
+    if (!ok)
+        return { content: [{ type: "text", text: JSON.stringify({ error: "Query failed" }) }] };
+    // Group by source
+    const grouped = {};
+    for (const item of data ?? []) {
+        if (!grouped[item.source])
+            grouped[item.source] = [];
+        grouped[item.source].push({ title: item.title, category: item.category, content: item.content });
+    }
+    return {
+        content: [{
+                type: "text",
+                text: JSON.stringify({
+                    feature,
+                    patterns: grouped,
+                    totalSources: Object.keys(grouped).length,
+                    totalPatterns: data?.length ?? 0,
+                }, null, 2),
+            }],
+    };
+});
+// ──────────────────────────────────────────
+// Tool: stk_brain_stack
+// ──────────────────────────────────────────
+server.tool("stk_brain_stack", "Get recommendations specific to YOUR stack (Supabase + Vercel + Stripe + Node.js). Filters knowledge for patterns matching your technology choices.", {
+    question: z.string().describe("What you want to build or solve (e.g., 'add user auth', 'implement webhooks', 'optimize queries')"),
+}, async ({ question }) => {
+    const brain = getBrainClient();
+    if (!brain)
+        return { content: [{ type: "text", text: JSON.stringify({ error: "SUPABASE_URL or SUPABASE_SERVICE_KEY not set" }) }] };
+    const words = question.split(" ").filter(w => w.length > 3);
+    const searchWord = words[0] ?? question;
+    const { data, ok } = await brain.query("knowledge", {
+        or: `(content.ilike.%${searchWord}%,title.ilike.%${searchWord}%)`,
+        limit: "10",
+    });
+    if (!ok)
+        return { content: [{ type: "text", text: JSON.stringify({ error: "Query failed" }) }] };
+    return {
+        content: [{
+                type: "text",
+                text: JSON.stringify({
+                    question,
+                    stack: ["Supabase", "Vercel", "Stripe", "Node.js/Express"],
+                    recommendations: (data ?? []).map((r) => ({ source: r.source, title: r.title, content: r.content, relevance: r.category })),
+                    total: data?.length ?? 0,
+                }, null, 2),
+            }],
+    };
+});
+// ──────────────────────────────────────────
+// Tool: stk_brain_learn
+// ──────────────────────────────────────────
+server.tool("stk_brain_learn", "Save new knowledge to the brain. Use this to remember patterns, solutions, or learnings for future reference across all projects.", {
+    title: z.string().describe("Short title"),
+    content: z.string().describe("The knowledge — pattern, solution, or best practice"),
+    source: z.string().optional().describe("Where this came from (e.g., 'project:worldchat', 'github:vercel/next.js')"),
+    category: z.string().optional().describe("Category: architecture, auth, payments, database, api, deployment, testing, performance, security, ml, general"),
+    tags: z.array(z.string()).optional().describe("Tags for searchability"),
+}, async ({ title, content, source, category, tags }) => {
+    const brain = getBrainClient();
+    if (!brain)
+        return { content: [{ type: "text", text: JSON.stringify({ error: "SUPABASE_URL or SUPABASE_SERVICE_KEY not set" }) }] };
+    const { data, ok } = await brain.insert("knowledge", {
+        title,
+        content,
+        source: source ?? "manual",
+        category: category ?? "general",
+        tags: tags ?? [],
+        created_at: new Date().toISOString(),
+    });
+    if (!ok)
+        return { content: [{ type: "text", text: JSON.stringify({ error: "Insert failed", data }) }] };
+    return {
+        content: [{
+                type: "text",
+                text: JSON.stringify({ learned: true, title, message: "Knowledge saved. I can recall this in future conversations." }, null, 2),
+            }],
+    };
+});
+// ──────────────────────────────────────────
+// Tool: stk_brain_stats
+// ──────────────────────────────────────────
+server.tool("stk_brain_stats", "Check what the brain knows — total knowledge entries, categories, sources, and coverage.", {}, async () => {
+    const brain = getBrainClient();
+    if (!brain)
+        return { content: [{ type: "text", text: JSON.stringify({ error: "SUPABASE_URL or SUPABASE_SERVICE_KEY not set" }) }] };
+    const { data, count } = await brain.query("knowledge", { select: "category,source", limit: "1000" });
+    const categories = {};
+    const sources = {};
+    for (const row of data ?? []) {
+        categories[row.category] = (categories[row.category] || 0) + 1;
+        sources[row.source] = (sources[row.source] || 0) + 1;
+    }
+    return {
+        content: [{
+                type: "text",
+                text: JSON.stringify({
+                    totalKnowledge: count ?? data?.length ?? 0,
+                    categories,
+                    sources,
+                    topSources: Object.entries(sources)
+                        .sort(([, a], [, b]) => b - a)
+                        .slice(0, 10)
+                        .map(([name, count]) => ({ name, count })),
+                }, null, 2),
+            }],
+    };
+});
 // Helper
 function detectGitHubRepo() {
     try {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@prajwolkc/stk",
-  "version": "0.3.0",
-  "description": "One CLI to deploy, monitor, and debug your entire stack. Health checks, deploy watching, env sync, logs, and GitHub issues — all from one command.",
+  "version": "0.4.0",
+  "description": "One CLI to deploy, monitor, debug, and learn about your entire stack. Infrastructure monitoring, knowledge base brain, deploy watching, and GitHub issues — all from one command.",
   "type": "module",
   "license": "MIT",
   "author": "prajwolkc",