@jussmor/commit-memory-mcp 0.3.1 → 0.3.2

package/README.md

@@ -47,3 +47,14 @@ For MCP Registry publication, keep `package.json` `mcpName` and `server.json` `n
 - `OLLAMA_EMBED_MODEL` local embedding model name
 
 If `OLLAMA_EMBED_MODEL` is not set, the package uses deterministic local fallback embeddings.
+
+### Copilot LLM reranking (optional)
+
+Set `COPILOT_TOKEN` to a GitHub token with Copilot access to enable LLM-based reranking.
+After initial vector/keyword retrieval, results are sent to Copilot for semantic scoring and re-sorted.
+
+- `COPILOT_TOKEN` GitHub PAT or token with Copilot access (enables reranking)
+- `COPILOT_MODEL` model slug (default: `gpt-4o-mini`, supports `claude-sonnet-4-5`, `gpt-4o`, etc.)
+- `COPILOT_BASE_URL` API base URL (default: `https://api.githubcopilot.com`)
+
+Reranking works alongside or instead of Ollama — no embedding model required.

package/dist/search/query.js

@@ -1,4 +1,5 @@
 import { embedText } from "./embeddings.js";
+import { copilotRerankEnabled, rerankWithCopilot } from "./rerank.js";
 function scoreWithBoost(base, row, activeFile) {
     let score = base;
     if (activeFile && row.file_path === activeFile) {
@@ -16,6 +17,9 @@ function createPreview(hunkText) {
     return hunkText.split("\n").slice(0, 6).join("\n");
 }
 export async function searchRelatedCommits(db, query, limit, activeFile) {
+    // Fetch extra candidates when Copilot reranking is enabled so the LLM has
+    // more to work with before we trim to the requested limit.
+    const fetchLimit = copilotRerankEnabled() ? limit * 2 : limit;
     const embedding = await embedText(query);
     const embeddingJson = JSON.stringify(embedding);
     try {
@@ -36,8 +40,8 @@ export async function searchRelatedCommits(db, query, limit, activeFile) {
         JOIN commits cm ON cm.sha = c.sha
         WHERE v.embedding MATCH ? AND k = ?
       `)
-            .all(embeddingJson, limit);
-        return rows.map((row) => {
+            .all(embeddingJson, fetchLimit);
+        const candidates = rows.map((row) => {
             const base = 1 / (1 + Math.max(0, row.distance));
             const score = scoreWithBoost(base, row, activeFile);
             return {
@@ -51,6 +55,8 @@ export async function searchRelatedCommits(db, query, limit, activeFile) {
                 preview: createPreview(row.hunk_text),
             };
         });
+        const reranked = await rerankWithCopilot(query, candidates);
+        return reranked.slice(0, limit);
     }
     catch {
         const rows = db
@@ -69,8 +75,8 @@ export async function searchRelatedCommits(db, query, limit, activeFile) {
         ORDER BY cm.date DESC
         LIMIT ?
       `)
-            .all(`%${query}%`, limit);
-        return rows.map((row, idx) => ({
+            .all(`%${query}%`, fetchLimit);
+        const keywordCandidates = rows.map((row, idx) => ({
             chunkId: row.chunk_id,
             sha: row.sha,
             filePath: row.file_path,
@@ -80,6 +86,8 @@ export async function searchRelatedCommits(db, query, limit, activeFile) {
             author: row.author,
             preview: createPreview(row.hunk_text),
         }));
+        const rerankedKeyword = await rerankWithCopilot(query, keywordCandidates);
+        return rerankedKeyword.slice(0, limit);
     }
 }
 export function explainCommitMatch(db, chunkId) {

package/dist/search/rerank.d.ts

@@ -0,0 +1,12 @@
+import type { SearchResult } from "../types.js";
+/**
+ * Rerank search results using GitHub Copilot chat completions (Claude / GPT).
+ *
+ * Activated when COPILOT_TOKEN is set.
+ * Env vars:
+ *   COPILOT_TOKEN    – GitHub PAT or token with Copilot access (required)
+ *   COPILOT_MODEL    – model slug (default: "gpt-4o-mini")
+ *   COPILOT_BASE_URL – API base URL (default: https://api.githubcopilot.com)
+ */
+export declare function rerankWithCopilot(query: string, results: SearchResult[]): Promise<SearchResult[]>;
+export declare function copilotRerankEnabled(): boolean;

package/dist/search/rerank.js

@@ -0,0 +1,72 @@
+/**
+ * Rerank search results using GitHub Copilot chat completions (Claude / GPT).
+ *
+ * Activated when COPILOT_TOKEN is set.
+ * Env vars:
+ *   COPILOT_TOKEN    – GitHub PAT or token with Copilot access (required)
+ *   COPILOT_MODEL    – model slug (default: "gpt-4o-mini")
+ *   COPILOT_BASE_URL – API base URL (default: https://api.githubcopilot.com)
+ */
+export async function rerankWithCopilot(query, results) {
+    const token = process.env.COPILOT_TOKEN;
+    if (!token || results.length === 0) {
+        return results;
+    }
+    const baseUrl = process.env.COPILOT_BASE_URL ?? "https://api.githubcopilot.com";
+    const model = process.env.COPILOT_MODEL ?? "gpt-4o-mini";
+    const commitList = results
+        .map((r, i) => `${i + 1}. [${r.date}] ${r.author} — ${r.subject}\n   File: ${r.filePath}\n   ${r.preview.split("\n").slice(0, 3).join(" | ")}`)
+        .join("\n\n");
+    const prompt = `You are a code search assistant. Rate each commit excerpt's relevance to the query on a scale from 0.0 to 1.0.
+
+Query: "${query}"
+
+Commits:
+${commitList}
+
+Respond with ONLY a JSON array of numbers, one per commit, in the same order. Example: [0.9, 0.2, 0.7]`;
+    const messages = [{ role: "user", content: prompt }];
+    const response = await fetch(`${baseUrl}/chat/completions`, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${token}`,
+            "Copilot-Integration-Id": "commit-memory-mcp",
+        },
+        body: JSON.stringify({
+            model,
+            messages,
+            temperature: 0,
+            max_tokens: 256,
+        }),
+    });
+    if (!response.ok) {
+        // Fall back to original order if the API call fails
+        console.error(`[rerank] Copilot API error ${response.status}, using original order`);
+        return results;
+    }
+    const data = (await response.json());
+    const content = data.choices?.[0]?.message?.content ?? "";
+    // Extract JSON array from the response (tolerates markdown fences)
+    const match = content.match(/\[[\d.,\s]+\]/);
+    if (!match) {
+        console.error("[rerank] Could not parse scores from Copilot response");
+        return results;
+    }
+    let scores;
+    try {
+        scores = JSON.parse(match[0]);
+    }
+    catch {
+        return results;
+    }
+    return results
+        .map((result, i) => ({
+        ...result,
+        score: typeof scores[i] === "number" ? scores[i] : result.score,
+    }))
+        .sort((a, b) => b.score - a.score);
+}
+export function copilotRerankEnabled() {
+    return Boolean(process.env.COPILOT_TOKEN);
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jussmor/commit-memory-mcp",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "mcpName": "io.github.jussmor/commit-memory",
   "description": "Commit-aware RAG with sqlite-vec and MCP tools for local agent workflows",
   "license": "MIT",