@bodhi-ventures/aiocs 0.6.0 → 0.6.1

package/dist/{chunk-2UWV3O7E.js → chunk-TJPALXTG.js}

@@ -4367,7 +4367,7 @@ async function startDaemon(input) {
 // package.json
 var package_default = {
   name: "@bodhi-ventures/aiocs",
-  version: "0.6.0",
+  version: "0.6.1",
   license: "MIT",
   type: "module",
   description: "Local-only documentation store, fetcher, and search CLI for AI agents.",
@@ -5439,6 +5439,63 @@ function overlapScore(queryTokens, candidate) {
   }
   return matches;
 }
+function uniqueTokens(tokens) {
+  return [...new Set(tokens)];
+}
+function hasTokenCoverage(requiredTokens, candidate) {
+  if (requiredTokens.length === 0) {
+    return false;
+  }
+  const candidateTokens = new Set(tokenize(candidate));
+  return requiredTokens.every((token) => candidateTokens.has(token));
+}
+var NAVIGATIONAL_TERMS = /* @__PURE__ */ new Set([
+  "api",
+  "apis",
+  "auth",
+  "authentication",
+  "docs",
+  "documentation",
+  "endpoint",
+  "endpoints",
+  "overview",
+  "reference",
+  "references",
+  "rest",
+  "sdk",
+  "transport",
+  "websocket",
+  "ws"
+]);
+function classifyRetrievalQuery(query) {
+  const queryTokens = uniqueTokens(tokenize(query));
+  const matchedNavigationalTerms = uniqueTokens(queryTokens.filter((token) => NAVIGATIONAL_TERMS.has(token)));
+  return {
+    isNavigational: matchedNavigationalTerms.length > 0,
+    queryTokens,
+    routingTokens: queryTokens,
+    matchedNavigationalTerms
+  };
+}
+function scorePageCandidate(query, input) {
+  const intent = classifyRetrievalQuery(query);
+  const routingTokens = intent.routingTokens;
+  const titleOverlap = overlapScore(routingTokens, input.pageTitle);
+  const referenceOverlap = overlapScore(routingTokens, input.pageReference);
+  const sectionOverlap = Math.max(0, ...input.sectionTitles.map((sectionTitle) => overlapScore(routingTokens, sectionTitle)));
+  const exactTitleCoverage = hasTokenCoverage(routingTokens, input.pageTitle);
+  const exactReferenceCoverage = hasTokenCoverage(routingTokens, input.pageReference);
+  const lexicalWeight = intent.isNavigational ? 6 : 4;
+  const vectorWeight = intent.isNavigational ? 1 : 3;
+  const titleWeight = intent.isNavigational ? 8 : 4;
+  const referenceWeight = intent.isNavigational ? 6 : 3;
+  const sectionWeight = intent.isNavigational ? 4 : 2;
+  const sourceHintWeight = intent.isNavigational ? 3 : 2;
+  const commonLocationWeight = intent.isNavigational ? 7 : 4;
+  const learningWeight = intent.isNavigational ? 5 : 4;
+  const pureVectorPenalty = intent.isNavigational && input.bestLexicalScore === 0 && titleOverlap === 0 && referenceOverlap === 0 && input.commonLocationScore === 0 ? 8 : 0;
+  return input.bestLexicalScore * lexicalWeight + input.bestVectorScore * vectorWeight + titleOverlap * titleWeight + referenceOverlap * referenceWeight + sectionOverlap * sectionWeight + input.sourceHintScore * sourceHintWeight + input.commonLocationScore * commonLocationWeight + input.learningScore * learningWeight + (exactTitleCoverage ? 12 : 0) + (exactReferenceCoverage ? 10 : 0) - pureVectorPenalty;
+}
 function scoreLearning(query, learning) {
   const queryTokens = tokenize(query);
   const candidates = [learning.intent, ...learning.searchTerms];
@@ -5537,6 +5594,9 @@ function dedupePagesByIdentity(rows) {
   }
   return deduped;
 }
+function pageIdentityKey(input) {
+  return `${input.sourceId}::${input.snapshotId}::${input.filePath ?? input.pageUrl}`;
+}
 function createCatalog() {
   const dataDir = getAiocsDataDir();
   getAiocsConfigDir();
@@ -5796,6 +5856,8 @@ async function retrieveContext(query, options) {
     const avoidedPageKeys = new Set(
       avoidedLearnings.map((learning) => `${learning.sourceId}::${learning.filePath ?? learning.pageUrl ?? ""}`)
     );
+    const sourceHintScores = /* @__PURE__ */ new Map();
+    const commonLocationBoosts = /* @__PURE__ */ new Map();
     const sourceHints = sourceScope.map((sourceId) => {
       const contextRecord = catalog.getSourceContext(sourceId);
       const score = scoreSourceContext(query, contextRecord.context);
@@ -5819,6 +5881,28 @@ async function retrieveContext(query, options) {
         matchedCommonLocations
       };
     }).filter((entry) => Boolean(entry)).sort((left, right) => right.score - left.score || left.sourceId.localeCompare(right.sourceId));
+    for (const sourceHint of sourceHints) {
+      sourceHintScores.set(
+        sourceHint.sourceId,
+        Math.max(sourceHint.score, sourceHintScores.get(sourceHint.sourceId) ?? 0)
+      );
+      const latestSnapshotId = options.snapshot ?? catalog.getSourceById(sourceHint.sourceId)?.lastSuccessfulSnapshotId ?? null;
+      if (!latestSnapshotId) {
+        continue;
+      }
+      for (const location of sourceHint.matchedCommonLocations) {
+        const key = pageIdentityKey({
+          sourceId: sourceHint.sourceId,
+          snapshotId: latestSnapshotId,
+          pageUrl: location.url ?? "",
+          filePath: location.filePath ?? null
+        });
+        commonLocationBoosts.set(
+          key,
+          Math.max(sourceHint.score, commonLocationBoosts.get(key) ?? 0)
+        );
+      }
+    }
     const search = await searchHybridCatalog({
       catalog,
       config: hybridConfig,
@@ -5847,21 +5931,100 @@ async function retrieveContext(query, options) {
       pageUrl: result.pageUrl,
       filePath: result.filePath
     }));
-    const selectedPages = dedupePagesByIdentity([
+    const commonLocationPages = sourceHints.flatMap((sourceHint) => {
+      const snapshotId = options.snapshot ?? catalog.getSourceById(sourceHint.sourceId)?.lastSuccessfulSnapshotId ?? "";
+      if (!snapshotId) {
+        return [];
+      }
+      return sourceHint.matchedCommonLocations.map((location) => ({
+        sourceId: sourceHint.sourceId,
+        snapshotId,
+        pageUrl: location.url ?? "",
+        filePath: location.filePath ?? null
+      }));
+    });
+    const pageCandidates = dedupePagesByIdentity([
+      ...commonLocationPages,
       ...learnedPages,
       ...searchedPages
-    ]).slice(0, Math.max(1, pageLimit));
-    const pages = selectedPages.flatMap((entry) => {
+    ]).filter((entry) => !avoidedPageKeys.has(`${entry.sourceId}::${entry.filePath ?? entry.pageUrl}`));
+    const searchPageSignals = /* @__PURE__ */ new Map();
+    for (const result of search.results) {
+      if (avoidedPageKeys.has(`${result.sourceId}::${result.filePath ?? result.pageUrl}`)) {
+        continue;
+      }
+      const key = pageIdentityKey({
+        sourceId: result.sourceId,
+        snapshotId: result.snapshotId,
+        pageUrl: result.pageUrl,
+        filePath: result.filePath
+      });
+      const record = searchPageSignals.get(key) ?? {
+        bestLexicalScore: 0,
+        bestVectorScore: 0,
+        sectionTitles: /* @__PURE__ */ new Set()
+      };
+      const signalCount = Math.max(1, result.signals.length);
+      const perSignalScore = result.score / signalCount;
+      if (result.signals.includes("lexical")) {
+        record.bestLexicalScore = Math.max(record.bestLexicalScore, perSignalScore);
+      }
+      if (result.signals.includes("vector")) {
+        record.bestVectorScore = Math.max(record.bestVectorScore, perSignalScore);
+      }
+      record.sectionTitles.add(result.sectionTitle);
+      searchPageSignals.set(key, record);
+    }
+    const learningBoosts = /* @__PURE__ */ new Map();
+    for (const learning of matchedLearnings) {
+      if (!learning.pageUrl && !learning.filePath) {
+        continue;
+      }
+      const snapshotId = learning.snapshotId ?? catalog.getSourceById(learning.sourceId)?.lastSuccessfulSnapshotId ?? "";
+      if (!snapshotId) {
+        continue;
+      }
+      const key = pageIdentityKey({
+        sourceId: learning.sourceId,
+        snapshotId,
+        pageUrl: learning.pageUrl ?? "",
+        filePath: learning.filePath ?? null
+      });
+      learningBoosts.set(key, Math.max(learning.score, learningBoosts.get(key) ?? 0));
+    }
+    const rankedPages = pageCandidates.flatMap((entry) => {
       try {
         const page = catalog.getPage({
           sourceId: entry.sourceId,
           snapshotId: entry.snapshotId,
           ...entry.filePath ? { filePath: entry.filePath } : { url: entry.pageUrl }
         });
+        const key = pageIdentityKey({
+          sourceId: page.sourceId,
+          snapshotId: page.snapshotId,
+          pageUrl: page.page.url,
+          filePath: page.page.filePath
+        });
+        const searchSignals = searchPageSignals.get(key) ?? {
+          bestLexicalScore: 0,
+          bestVectorScore: 0,
+          sectionTitles: /* @__PURE__ */ new Set()
+        };
+        const score = scorePageCandidate(query, {
+          pageTitle: page.page.title,
+          pageReference: page.page.filePath ?? page.page.url,
+          sectionTitles: [...searchSignals.sectionTitles],
+          bestLexicalScore: searchSignals.bestLexicalScore,
+          bestVectorScore: searchSignals.bestVectorScore,
+          learningScore: learningBoosts.get(key) ?? 0,
+          sourceHintScore: sourceHintScores.get(page.sourceId) ?? 0,
+          commonLocationScore: commonLocationBoosts.get(key) ?? 0
+        });
         return [{
           sourceId: page.sourceId,
           snapshotId: page.snapshotId,
-          ...page.page
+          score,
+          page: page.page
         }];
       } catch (error) {
         if (error instanceof AiocsError && (error.code === AIOCS_ERROR_CODES.pageNotFound || error.code === AIOCS_ERROR_CODES.snapshotNotFound)) {
@@ -5869,7 +6032,12 @@ async function retrieveContext(query, options) {
         }
         throw error;
       }
-    });
+    }).sort((left, right) => right.score - left.score || left.page.title.localeCompare(right.page.title) || left.page.url.localeCompare(right.page.url)).slice(0, Math.max(1, pageLimit));
+    const pages = rankedPages.map((entry) => ({
+      sourceId: entry.sourceId,
+      snapshotId: entry.snapshotId,
+      ...entry.page
+    }));
     return {
       query,
       modeRequested: search.modeRequested,

package/dist/cli.js

@@ -39,7 +39,7 @@ import {
   upsertSourceContextFromFile,
   upsertSourceFromSpecFile,
   verifyCoverage
-} from "./chunk-2UWV3O7E.js";
+} from "./chunk-TJPALXTG.js";
 
 // src/cli.ts
 import { Command, CommanderError as CommanderError2 } from "commander";

package/dist/mcp-server.js

@@ -34,7 +34,7 @@ import {
   upsertSourceContextFromFile,
   upsertSourceFromSpecFile,
   verifyCoverage
-} from "./chunk-2UWV3O7E.js";
+} from "./chunk-TJPALXTG.js";
 
 // src/mcp-server.ts
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bodhi-ventures/aiocs",
-  "version": "0.6.0",
+  "version": "0.6.1",
   "license": "MIT",
   "type": "module",
   "description": "Local-only documentation store, fetcher, and search CLI for AI agents.",