wyrm-mcp 7.3.2 → 7.3.3

package/dist/handlers/search.js

@@ -1,15 +1,338 @@
-import{TOOL_ANNOTATIONS as A}from"../tool-annotations.js";import{renderResult as I,withGlyph as q,cacheKeyFor as C}from"../render.js";import{sanitizeFtsQuery as Q}from"../security.js";import{constellationQuery as E}from"../constellation.js";const B=[{name:"wyrm_search",description:"Use to find anything by keyword or meaning across all projects - find past sessions that mention a topic, look up everything we have about a service, or ask have I solved this problem before in any other project (cross-project constellation). FTS/vector/hybrid; where a function or symbol is defined across repos lives here too (wyrm_symbol_search).",inputSchema:{type:"object",properties:{query:{type:"string"},type:{type:"string",enum:["all","sessions","quests","data","artifacts","entities"]},mode:{type:"string",enum:["lexical","semantic","hybrid"],description:"Default hybrid, else lexical"},projectPath:{type:"string",description:"Limit to one project"}},required:["query"]},outputSchema:{type:"object",properties:{query:{type:"string"},mode:{type:"string",enum:["lexical","semantic","hybrid"]},sections:{type:"array",items:{type:"object",properties:{label:{type:"string"},count:{type:"integer"},items:{type:"array",items:{type:"string"}}},required:["label","count","items"]}},skills:{type:"array"},entities:{type:"array"}},required:["query","mode","sections","skills","entities"]},annotations:A.wyrm_search,aliases:[],handler:async(j,{store:l,graph:$,vectors:b,cache:w,decrypt:S})=>{const{query:d,type:_,mode:k,projectPath:m}=j,y=b(),u=Q(d),n=(m?l.getProject(m):void 0)?.id,i=_||"all",p=k||(y?"hybrid":"lexical"),c={};if((i==="all"||i==="sessions")&&p!=="semantic"){const e=l.searchSessions(u,n);c.session=e.map((t,s)=>({id:t.id,snippet:`${t.date}: ${(t.objectives||t.completed||"No info").slice(0,80)}`,rank:s+1}))}if((i==="all"||i==="quests")&&p!=="semantic"){const e=l.searchQuests(u);c.quest=e.map((t,s)=>({id:t.id,snippet:`#${t.id}: ${t.title}`,rank:s+1}))}if((i==="all"||i==="data")&&p!=="semantic"){const e=l.searchData(u,n);c.note=e.map((t,s)=>({id:t.id,snippet:`${t.category}/${t.key}: ${S(t.value).slice(0,60)}`,rank:s+1}))}if((i==="all"||i==="artifacts")&&p!=="semantic"){const e=l.searchArtifacts(u,n);c.artifact=e.map((t,s)=>({id:t.id,snippet:`#${t.id}: ${(t.problem||"").slice(0,80)}`,rank:s+1}))}let x=[];if(y&&p!=="lexical"&&(i==="all"||i==="data"||i==="sessions"||i==="quests"))try{x=await y.search(d,20,n)}catch{}const T=60,f=new Map,D=(e,t)=>{t.forEach((s,r)=>{const a=`${e}:${s.id}`,h=f.get(a),g=1/(T+(r+1));h?h.score+=g:f.set(a,{score:g,snippet:s.snippet,type:e,id:s.id})})};for(const[e,t]of Object.entries(c))D(e,t);const v={};for(const e of x)v[e.content_type]||(v[e.content_type]=[]),v[e.content_type].push(e);for(const[e,t]of Object.entries(v))t.forEach((s,r)=>{const a=`${e}:${s.content_id}`,h=f.get(a),g=1/(T+(r+1));h?(h.score+=g,h.snippet+=` (semantic: ${(s.similarity*100).toFixed(0)}%)`):f.set(a,{score:g,snippet:`${e} #${s.content_id} (semantic: ${(s.similarity*100).toFixed(0)}%)`,type:e,id:s.content_id})});const R=Array.from(f.values()).sort((e,t)=>t.score-e.score),P=[];if(R.length>0){const e={};for(const t of R)e[t.type]||(e[t.type]=[]),e[t.type].push(t);for(const[t,s]of Object.entries(e)){const r=t==="note"?"Data":t.charAt(0).toUpperCase()+t.slice(1)+"s";P.push({label:r,count:s.length,items:s.slice(0,10).map(a=>a.snippet)})}}let N=[];(i==="all"||i==="skills")&&(N=l.searchSkills(u,10).map(e=>({name:e.name,category:e.category||"uncategorized",active:!!e.is_active,description:e.description})));let F=[];(i==="all"||i==="entities")&&n!==void 0&&(F=$.searchEntities(n,d,10).map(e=>({name:e.name,type:e.type,relationships:$.getRelationships(e.id,"both").length})));const O=I({query:d,mode:p,sections:P,skills:N,entities:F},(e,t)=>{let s=q(t.brand,`**Search Results for "${e.query}"** _(mode: ${e.mode})_`)+`
-
-`;for(const r of e.sections){s+=`## ${r.label} (${r.count})
-`;for(const a of r.items)s+=`${t.bullet} ${a}
-`;s+=`
-`}if(e.skills.length>0){s+=`## Skills (${e.skills.length})
-`;for(const r of e.skills)s+=`${t.bullet} **${r.name}** (${r.category}) [${r.active?"active":"inactive"}]
-`,s+=`  ${r.description}
-`;s+=`
-`}if(e.entities.length>0){s+=`## Entities (${e.entities.length})
-`;for(const r of e.entities)s+=`${t.bullet} **${r.name}** _(${r.type})_ -- ${r.relationships} relationship${r.relationships!==1?"s":""}
-`;s+=`
-`}return e.sections.length===0&&e.skills.length===0&&e.entities.length===0&&(s+=`_No results found._
-`),s},{summary:(e,t)=>{const s=e.sections.reduce((r,a)=>r+a.count,0)+e.skills.length+e.entities.length;return q(t.brand,`**Search Results for "${e.query}"** _(mode: ${e.mode})_ -- ${s} result${s!==1?"s":""}`)}});return w.set(C("wyrm_search",JSON.stringify(j)),O,2e4),O}},{name:"wyrm_constellation",description:"Cross-project memory query \u2014 search every registered Wyrm project at once for matching truths, artifacts, quests, or design references. Returns candidates grouped by project; the calling AI does semantic ranking on its side. Use this when the operator asks 'have I solved this before?', 'where else have I used X?', or any portfolio-spanning question. Honours per-row cross_project_visibility ('within' = invisible across projects, 'org' = shared, 'public' = anywhere).",inputSchema:{type:"object",properties:{query:{type:"string",description:"Search keywords. Sanitised; safe FTS5 subset only."},kinds:{type:"array",items:{type:"string",enum:["truth","artifact","quest","decision","reference"]},description:"Restrict to specific kinds. Default: all kinds."},visibility:{type:"array",items:{type:"string",enum:["within","org","public"]},description:"Which visibility levels to include. Default: ['org','public'] \u2014 strict cross-project gate."},per_project_limit:{type:"number",description:"Max hits per project (default 5, max 50)."},caller_project_path:{type:"string",description:"Caller's current project path \u2014 excluded from results unless 'within' is in visibility."}},required:["query"]},outputSchema:{type:"object",properties:{query:{type:"string"},count:{type:"integer"},projects:{type:"array",items:{type:"object",properties:{project:{type:"string"},hits:{type:"array",items:{type:"object",properties:{kind:{type:"string"},id:{type:"integer"},title:{type:"string"},visibility:{type:"string"},snippet:{type:["string","null"]}},required:["kind","id","title","visibility","snippet"]}}},required:["project","hits"]}}},required:["query","count","projects"]},annotations:A.wyrm_constellation,aliases:[],handler:(j,{store:l,raw:$})=>{const{query:b,kinds:w,visibility:S,per_project_limit:d,caller_project_path:_}=j;let k;_&&(k=l.getProject(_)?.id);const m=E($(),{query:b,kinds:w,visibility:S,perProjectLimit:d,callerProjectId:k}),y=new Map;for(const o of m){const n=y.get(o.projectName)??[];n.push(o),y.set(o.projectName,n)}const u={query:b,count:m.length,projects:Array.from(y.entries()).map(([o,n])=>({project:o,hits:n.map(i=>({kind:i.kind,id:i.id,title:i.title,visibility:i.visibility,snippet:i.snippet??null}))}))};return I(u,(o,n)=>{if(o.count===0)return q(n.brand,`No constellation matches for *${o.query}* across registered projects.`)+"\n\n_Set `cross_project_visibility` to `'org'` or `'public'` on the items you want surfaced here. Default is `'within'` -- within-project only._";const i=[q(n.brand,`**Constellation** -- ${o.count} match(es) for *${o.query}* across ${o.projects.length} project(s)`)];for(const p of o.projects){i.push(""),i.push(`### ${p.project}`);for(const c of p.hits)i.push(`${n.bullet} \`${c.kind}:${c.id}\` **${c.title}** (${c.visibility})`),c.snippet&&i.push(`   ${c.snippet}`)}return i.join(`
-`)})}}];export{B as searchToolSpecs};
+/**
+ * Search domain — ToolSpec contract v2 (v7 F3 T026, hot-path extraction).
+ *
+ * wyrm_search (FTS ⊕ vector RRF fusion across sessions/quests/data + skills
+ * + entities) and wyrm_constellation (the cross-project scoped mode the
+ * search survivor absorbs per the task routing), moved VERBATIM from the
+ * index.ts dispatch switch + buildAllTools(). The constellation alias keeps
+ * routing here through the spine (identity route → this registry entry —
+ * same handler code path, never a reimplementation).
+ *
+ * T019 renderer: the fused section list is the canonical structured body;
+ * the 6.x markdown derives from it (ASCII default — ✅/❌ skill markers
+ * became [active]/[inactive], spec §6 reformat).
+ *
+ * @copyright 2026 Ghost Protocol (Pvt) Ltd.
+ * @license Proprietary — (c) 2026 Ghost Protocol (Pvt) Ltd. All rights reserved. See LICENSE.
+ */
+import { TOOL_ANNOTATIONS } from '../tool-annotations.js';
+import { renderResult, withGlyph, cacheKeyFor } from '../render.js';
+import { sanitizeFtsQuery } from '../security.js';
+import { constellationQuery } from '../constellation.js';
+export const searchToolSpecs = [
+    {
+        name: "wyrm_search",
+        description: "Use to find anything by keyword or meaning across all projects - find past sessions that mention a topic, look up everything we have about a service, or ask have I solved this problem before in any other project (cross-project constellation). FTS/vector/hybrid; where a function or symbol is defined across repos lives here too (wyrm_symbol_search).",
+        inputSchema: {
+            type: "object",
+            properties: {
+                // v7 F3 (T026): property prose compressed to fund the hot-path
+                // outputSchemas under the 8K default-surface pin (T022/T025 trade).
+                query: { type: "string" },
+                type: { type: "string", enum: ["all", "sessions", "quests", "data", "artifacts", "entities"] },
+                mode: { type: "string", enum: ["lexical", "semantic", "hybrid"], description: "Default hybrid, else lexical" },
+                projectPath: { type: "string", description: "Limit to one project" },
+            },
+            required: ["query"],
+        },
+        outputSchema: {
+            type: "object",
+            properties: {
+                query: { type: "string" },
+                mode: { type: "string", enum: ["lexical", "semantic", "hybrid"] },
+                sections: {
+                    type: "array",
+                    items: {
+                        type: "object",
+                        properties: {
+                            label: { type: "string" },
+                            count: { type: "integer" },
+                            items: { type: "array", items: { type: "string" } },
+                        },
+                        required: ["label", "count", "items"],
+                    },
+                },
+                skills: { type: "array" },
+                entities: { type: "array" },
+            },
+            required: ["query", "mode", "sections", "skills", "entities"],
+        },
+        annotations: TOOL_ANNOTATIONS.wyrm_search,
+        aliases: [],
+        handler: async (args, { store, graph, vectors, cache, decrypt }) => {
+            const { query, type, mode, projectPath } = args;
+            const vectorStore = vectors();
+            const sanitizedQuery = sanitizeFtsQuery(query);
+            const project = projectPath ? store.getProject(projectPath) : undefined;
+            const projectId = project?.id;
+            const searchType = type || 'all';
+            const searchMode = mode || (vectorStore ? 'hybrid' : 'lexical');
+            const ftsResultsByType = {};
+            if (searchType === 'all' || searchType === 'sessions') {
+                if (searchMode !== 'semantic') {
+                    const sessions = store.searchSessions(sanitizedQuery, projectId);
+                    ftsResultsByType['session'] = sessions.map((s, i) => ({
+                        id: s.id,
+                        snippet: `${s.date}: ${(s.objectives || s.completed || 'No info').slice(0, 80)}`,
+                        rank: i + 1,
+                    }));
+                }
+            }
+            if (searchType === 'all' || searchType === 'quests') {
+                if (searchMode !== 'semantic') {
+                    const quests = store.searchQuests(sanitizedQuery);
+                    ftsResultsByType['quest'] = quests.map((q, i) => ({
+                        id: q.id,
+                        snippet: `#${q.id}: ${q.title}`,
+                        rank: i + 1,
+                    }));
+                }
+            }
+            if (searchType === 'all' || searchType === 'data') {
+                if (searchMode !== 'semantic') {
+                    const data = store.searchData(sanitizedQuery, projectId);
+                    ftsResultsByType['note'] = data.map((d, i) => ({
+                        id: d.id,
+                        snippet: `${d.category}/${d.key}: ${decrypt(d.value).slice(0, 60)}`,
+                        rank: i + 1,
+                    }));
+                }
+            }
+            // Memory artifacts: lexical peer of the vector 'artifact' channel below,
+            // so keyword search finds patterns/lessons too (RRF-fused by artifact id).
+            if (searchType === 'all' || searchType === 'artifacts') {
+                if (searchMode !== 'semantic') {
+                    const artifacts = store.searchArtifacts(sanitizedQuery, projectId);
+                    ftsResultsByType['artifact'] = artifacts.map((a, i) => ({
+                        id: a.id,
+                        snippet: `#${a.id}: ${(a.problem || '').slice(0, 80)}`,
+                        rank: i + 1,
+                    }));
+                }
+            }
+            let vectorResults = [];
+            if (vectorStore && searchMode !== 'lexical' && (searchType === 'all' || searchType === 'data' || searchType === 'sessions' || searchType === 'quests')) {
+                try {
+                    vectorResults = await vectorStore.search(query, 20, projectId);
+                }
+                catch {
+                    // Vector search failed gracefully — continue with FTS
+                }
+            }
+            // ── Reciprocal Rank Fusion (RRF) ──────────────────────────────────────
+            // Score formula: Σ 1/(k + rank) where k=60
+            const RRF_K = 60;
+            const rrfScores = new Map();
+            const addFtsToRrf = (contentType, results) => {
+                results.forEach((r, idx) => {
+                    const key = `${contentType}:${r.id}`;
+                    const existing = rrfScores.get(key);
+                    const score = 1 / (RRF_K + (idx + 1));
+                    if (existing) {
+                        existing.score += score;
+                    }
+                    else {
+                        rrfScores.set(key, { score, snippet: r.snippet, type: contentType, id: r.id });
+                    }
+                });
+            };
+            for (const [contentType, results] of Object.entries(ftsResultsByType)) {
+                addFtsToRrf(contentType, results);
+            }
+            // Add vector results to RRF (sorted by similarity desc → rank 1=most similar)
+            const vecByType = {};
+            for (const vr of vectorResults) {
+                if (!vecByType[vr.content_type])
+                    vecByType[vr.content_type] = [];
+                vecByType[vr.content_type].push(vr);
+            }
+            for (const [contentType, results] of Object.entries(vecByType)) {
+                results.forEach((vr, idx) => {
+                    const key = `${contentType}:${vr.content_id}`;
+                    const existing = rrfScores.get(key);
+                    const score = 1 / (RRF_K + (idx + 1));
+                    if (existing) {
+                        existing.score += score;
+                        existing.snippet += ` (semantic: ${(vr.similarity * 100).toFixed(0)}%)`;
+                    }
+                    else {
+                        rrfScores.set(key, { score, snippet: `${contentType} #${vr.content_id} (semantic: ${(vr.similarity * 100).toFixed(0)}%)`, type: contentType, id: vr.content_id });
+                    }
+                });
+            }
+            // Sort by RRF score, group fused results by content type
+            const fused = Array.from(rrfScores.values()).sort((a, b) => b.score - a.score);
+            const sections = [];
+            if (fused.length > 0) {
+                const byType = {};
+                for (const r of fused) {
+                    if (!byType[r.type])
+                        byType[r.type] = [];
+                    byType[r.type].push(r);
+                }
+                for (const [contentType, results] of Object.entries(byType)) {
+                    const label = contentType === 'note' ? 'Data' : contentType.charAt(0).toUpperCase() + contentType.slice(1) + 's';
+                    sections.push({ label, count: results.length, items: results.slice(0, 10).map((r) => r.snippet) });
+                }
+            }
+            // Skills search (FTS only — no vector indexing for skills)
+            let skills = [];
+            if (searchType === 'all' || searchType === 'skills') {
+                skills = store.searchSkills(sanitizedQuery, 10).map((s) => ({
+                    name: s.name,
+                    category: s.category || 'uncategorized',
+                    active: !!s.is_active,
+                    description: s.description,
+                }));
+            }
+            // Entity graph search
+            let entities = [];
+            if ((searchType === 'all' || searchType === 'entities') && projectId !== undefined) {
+                entities = graph.searchEntities(projectId, query, 10).map((e) => ({
+                    name: e.name,
+                    type: e.type,
+                    relationships: graph.getRelationships(e.id, 'both').length,
+                }));
+            }
+            const body = { query, mode: searchMode, sections, skills, entities };
+            const response = renderResult(body, (b, g) => {
+                let text = withGlyph(g.brand, `**Search Results for "${b.query}"** _(mode: ${b.mode})_`) + '\n\n';
+                for (const section of b.sections) {
+                    text += `## ${section.label} (${section.count})\n`;
+                    for (const item of section.items)
+                        text += `${g.bullet} ${item}\n`;
+                    text += '\n';
+                }
+                if (b.skills.length > 0) {
+                    text += `## Skills (${b.skills.length})\n`;
+                    for (const s of b.skills) {
+                        text += `${g.bullet} **${s.name}** (${s.category}) [${s.active ? 'active' : 'inactive'}]\n`;
+                        text += `  ${s.description}\n`;
+                    }
+                    text += '\n';
+                }
+                if (b.entities.length > 0) {
+                    text += `## Entities (${b.entities.length})\n`;
+                    for (const e of b.entities) {
+                        text += `${g.bullet} **${e.name}** _(${e.type})_ -- ${e.relationships} relationship${e.relationships !== 1 ? 's' : ''}\n`;
+                    }
+                    text += '\n';
+                }
+                if (b.sections.length === 0 && b.skills.length === 0 && b.entities.length === 0) {
+                    text += '_No results found._\n';
+                }
+                return text;
+            }, {
+                // WYRM_CHANNEL=structured summary seam: grouped section/skill/entity
+                // counts ride structuredContent unchanged; the text channel collapses
+                // to this header + a count roll-up so the model still sees the shape.
+                summary: (b, g) => {
+                    const total = b.sections.reduce((n, s) => n + s.count, 0) + b.skills.length + b.entities.length;
+                    return withGlyph(g.brand, `**Search Results for "${b.query}"** _(mode: ${b.mode})_ -- ${total} result${total !== 1 ? 's' : ''}`);
+                },
+            });
+            // READ_ONLY_TOOLS cache publish — dispatcher key convention.
+            cache.set(cacheKeyFor('wyrm_search', JSON.stringify(args)), response, 20000);
+            return response;
+        },
+    },
+    {
+        name: "wyrm_constellation",
+        description: "Cross-project memory query — search every registered Wyrm project at once for matching truths, artifacts, quests, or design references. Returns candidates grouped by project; the calling AI does semantic ranking on its side. Use this when the operator asks 'have I solved this before?', 'where else have I used X?', or any portfolio-spanning question. Honours per-row cross_project_visibility ('within' = invisible across projects, 'org' = shared, 'public' = anywhere).",
+        inputSchema: {
+            type: "object",
+            properties: {
+                query: { type: "string", description: "Search keywords. Sanitised; safe FTS5 subset only." },
+                kinds: { type: "array", items: { type: "string", enum: ["truth", "artifact", "quest", "decision", "reference"] }, description: "Restrict to specific kinds. Default: all kinds." },
+                visibility: { type: "array", items: { type: "string", enum: ["within", "org", "public"] }, description: "Which visibility levels to include. Default: ['org','public'] — strict cross-project gate." },
+                per_project_limit: { type: "number", description: "Max hits per project (default 5, max 50)." },
+                caller_project_path: { type: "string", description: "Caller's current project path — excluded from results unless 'within' is in visibility." },
+            },
+            required: ["query"],
+        },
+        outputSchema: {
+            type: "object",
+            properties: {
+                query: { type: "string" },
+                count: { type: "integer" },
+                projects: {
+                    type: "array",
+                    items: {
+                        type: "object",
+                        properties: {
+                            project: { type: "string" },
+                            hits: {
+                                type: "array",
+                                items: {
+                                    type: "object",
+                                    properties: {
+                                        kind: { type: "string" },
+                                        id: { type: "integer" },
+                                        title: { type: "string" },
+                                        visibility: { type: "string" },
+                                        snippet: { type: ["string", "null"] },
+                                    },
+                                    required: ["kind", "id", "title", "visibility", "snippet"],
+                                },
+                            },
+                        },
+                        required: ["project", "hits"],
+                    },
+                },
+            },
+            required: ["query", "count", "projects"],
+        },
+        annotations: TOOL_ANNOTATIONS.wyrm_constellation,
+        aliases: [],
+        handler: (args, { store, raw }) => {
+            const { query: cQuery, kinds: cKinds, visibility: cVis, per_project_limit: cPerProjLimit, caller_project_path: cCallerPath, } = args;
+            let callerProjectId;
+            if (cCallerPath) {
+                const proj = store.getProject(cCallerPath);
+                callerProjectId = proj?.id;
+            }
+            const hits = constellationQuery(raw(), {
+                query: cQuery,
+                kinds: cKinds,
+                visibility: cVis,
+                perProjectLimit: cPerProjLimit,
+                callerProjectId,
+            });
+            // Group by project (the 6.x renderConstellation grouping, now body-side).
+            const byProject = new Map();
+            for (const h of hits) {
+                const arr = byProject.get(h.projectName) ?? [];
+                arr.push(h);
+                byProject.set(h.projectName, arr);
+            }
+            const body = {
+                query: cQuery,
+                count: hits.length,
+                projects: Array.from(byProject.entries()).map(([project, items]) => ({
+                    project,
+                    hits: items.map((h) => ({
+                        kind: h.kind, id: h.id, title: h.title, visibility: h.visibility, snippet: h.snippet ?? null,
+                    })),
+                })),
+            };
+            return renderResult(body, (b, g) => {
+                if (b.count === 0) {
+                    return withGlyph(g.brand, `No constellation matches for *${b.query}* across registered projects.`) +
+                        `\n\n_Set \`cross_project_visibility\` to \`'org'\` or \`'public'\` on the items you want surfaced here. Default is \`'within'\` -- within-project only._`;
+                }
+                const lines = [withGlyph(g.brand, `**Constellation** -- ${b.count} match(es) for *${b.query}* across ${b.projects.length} project(s)`)];
+                for (const p of b.projects) {
+                    lines.push('');
+                    lines.push(`### ${p.project}`);
+                    for (const h of p.hits) {
+                        lines.push(`${g.bullet} \`${h.kind}:${h.id}\` **${h.title}** (${h.visibility})`);
+                        if (h.snippet)
+                            lines.push(`   ${h.snippet}`);
+                    }
+                }
+                return lines.join('\n');
+            });
+        },
+    },
+];
+//# sourceMappingURL=search.js.map