@softerist/heuristic-mcp 3.2.2 → 3.2.4

package/lib/workspace-env.js

@@ -5,9 +5,7 @@ import {
   WORKSPACE_ENV_VARS,
 } from './constants.js';
 
-const EXCLUDED_DYNAMIC_WORKSPACE_ENV_KEYS = new Set([
-  'ANTIGRAVITY_EDITOR_APP_ROOT',
-]);
+const EXCLUDED_DYNAMIC_WORKSPACE_ENV_KEYS = new Set(['ANTIGRAVITY_EDITOR_APP_ROOT']);
 
 function isTruthy(value) {
   return /^(1|true|yes|on)$/i.test(String(value || '').trim());
@@ -21,20 +19,20 @@ export function isDynamicWorkspaceEnvEnabled(env = process.env, options = {}) {
 }
 
 export function scoreWorkspaceEnvKey(key) {
-  const upper = String(key || '').toUpperCase();
-  let score = 0;
-  if (upper.includes('WORKSPACE')) score += 8;
-  if (upper.includes('PROJECT')) score += 4;
-  if (upper.includes('ROOT')) score += 3;
-  if (upper.includes('CWD')) score += 2;
-  if (upper.includes('DIR')) score += 1;
-  return score;
-}
-
-function hasDynamicWorkspacePrefix(key) {
-  return DYNAMIC_WORKSPACE_ENV_PREFIXES.some((prefix) => key.startsWith(prefix));
-}
-
+  const upper = String(key || '').toUpperCase();
+  let score = 0;
+  if (upper.includes('WORKSPACE')) score += 8;
+  if (upper.includes('PROJECT')) score += 4;
+  if (upper.includes('ROOT')) score += 3;
+  if (upper.includes('CWD')) score += 2;
+  if (upper.includes('DIR')) score += 1;
+  return score;
+}
+
+function hasDynamicWorkspacePrefix(key) {
+  return DYNAMIC_WORKSPACE_ENV_PREFIXES.some((prefix) => key.startsWith(prefix));
+}
+
 export function getDynamicWorkspaceEnvKeys(env = process.env, options = {}) {
   if (!isDynamicWorkspaceEnvEnabled(env, options)) {
     return [];
@@ -43,27 +41,27 @@ export function getDynamicWorkspaceEnvKeys(env = process.env, options = {}) {
   return Object.keys(env)
     .filter((key) => !EXCLUDED_DYNAMIC_WORKSPACE_ENV_KEYS.has(String(key || '').toUpperCase()))
     .filter((key) => !WORKSPACE_ENV_VARS.includes(key))
-    .filter((key) => {
-      const providerSpecific = hasDynamicWorkspacePrefix(key) && WORKSPACE_ENV_KEY_PATTERN.test(key);
-      const genericWorkspace = WORKSPACE_ENV_GENERIC_DISCOVERY_PATTERN.test(key);
-      return providerSpecific || genericWorkspace;
-    })
-    .sort((a, b) => scoreWorkspaceEnvKey(b) - scoreWorkspaceEnvKey(a));
-}
-
+    .filter((key) => {
+      const providerSpecific =
+        hasDynamicWorkspacePrefix(key) && WORKSPACE_ENV_KEY_PATTERN.test(key);
+      const genericWorkspace = WORKSPACE_ENV_GENERIC_DISCOVERY_PATTERN.test(key);
+      return providerSpecific || genericWorkspace;
+    })
+    .sort((a, b) => scoreWorkspaceEnvKey(b) - scoreWorkspaceEnvKey(a));
+}
+
 export function getWorkspaceEnvKeys(env = process.env, options = {}) {
   return [...WORKSPACE_ENV_VARS, ...getDynamicWorkspaceEnvKeys(env, options)];
 }
 
-
 export function getWorkspaceEnvDiagnosticKeys(env = process.env, options = {}) {
   const prioritized = getWorkspaceEnvKeys(env, options);
   const prioritizedSet = new Set(prioritized);
-
-  const extraKeys = Object.keys(env)
-    .filter((key) => !prioritizedSet.has(key))
-    .filter((key) => WORKSPACE_ENV_KEY_PATTERN.test(key))
-    .sort((a, b) => scoreWorkspaceEnvKey(b) - scoreWorkspaceEnvKey(a));
-
-  return [...prioritized, ...extraKeys];
-}
+
+  const extraKeys = Object.keys(env)
+    .filter((key) => !prioritizedSet.has(key))
+    .filter((key) => WORKSPACE_ENV_KEY_PATTERN.test(key))
+    .sort((a, b) => scoreWorkspaceEnvKey(b) - scoreWorkspaceEnvKey(a));
+
+  return [...prioritized, ...extraKeys];
+}

package/mcp_config.json

@@ -1,9 +1,9 @@
-{
-  "mcpServers": {
-    "heuristic-mcp": {
-      "command": "node",
-      "args": ["scripts\\mcp-launcher.js"],
-      "disabled": false
-    }
-  }
-}
+{
+  "mcpServers": {
+    "heuristic-mcp": {
+      "command": "node",
+      "args": ["scripts\\mcp-launcher.js"],
+      "disabled": false
+    }
+  }
+}

package/package.json

@@ -1,86 +1,86 @@
-{
-  "name": "@softerist/heuristic-mcp",
-  "version": "3.2.2",
-  "description": "An enhanced MCP server providing intelligent semantic code search with find-similar-code, recency ranking, and improved chunking. Fork of smart-coding-mcp.",
-  "type": "module",
-  "main": "index.js",
-  "bin": {
-    "heuristic-mcp": "index.js"
-  },
-  "files": [
-    "index.js",
-    "config.jsonc",
-    "mcp_config.json",
-    "search-configs.js",
-    "lib/",
-    "features/",
-    "scripts/",
-    "README.md",
-    "LICENSE"
-  ],
-  "scripts": {
-    "start": "node --expose-gc index.js",
-    "dev": "node --expose-gc --watch index.js",
-    "test": "vitest run",
-    "test:watch": "vitest",
-    "clean": "node scripts/clear-cache.js",
-    "lint": "eslint .",
-    "format": "prettier --write .",
-    "postinstall": "node scripts/postinstall.js && node scripts/download-model.js"
-  },
-  "keywords": [
-    "mcp",
-    "semantic-search",
-    "code-search",
-    "embeddings",
-    "ai",
-    "model-context-protocol",
-    "hybrid-search",
-    "code-intelligence",
-    "cursor",
-    "vscode",
-    "claude",
-    "codex",
-    "openai",
-    "gemini",
-    "anthropic",
-    "antigravity",
-    "heuristic"
-  ],
-  "author": {
-    "name": "Softerist",
-    "url": "https://github.com/softerist"
-  },
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/softerist/heuristic-mcp.git"
-  },
-  "homepage": "https://github.com/softerist/heuristic-mcp#readme",
-  "license": "MIT",
-  "dependencies": {
-    "@huggingface/transformers": "^3.8.1",
-    "@modelcontextprotocol/sdk": "^1.0.4",
-    "better-sqlite3": "^12.6.2",
-    "chokidar": "^3.5.3",
-    "fdir": "^6.5.0",
-    "ignore": "^7.0.5",
-    "punycode": "^2.3.1"
-  },
-  "optionalDependencies": {
-    "hnswlib-node": "^3.0.0"
-  },
-  "engines": {
-    "node": ">=18.0.0"
-  },
-  "overrides": {
-    "punycode": "^2.3.1"
-  },
-  "devDependencies": {
-    "@eslint/js": "^9.39.2",
-    "@vitest/coverage-v8": "^4.0.18",
-    "eslint": "^9.39.2",
-    "globals": "^17.1.0",
-    "prettier": "^3.8.1",
-    "vitest": "^4.0.16"
-  }
-}
+{
+  "name": "@softerist/heuristic-mcp",
+  "version": "3.2.4",
+  "description": "An enhanced MCP server providing intelligent semantic code search with find-similar-code, recency ranking, and improved chunking. Fork of smart-coding-mcp.",
+  "type": "module",
+  "main": "index.js",
+  "bin": {
+    "heuristic-mcp": "index.js"
+  },
+  "files": [
+    "index.js",
+    "config.jsonc",
+    "mcp_config.json",
+    "search-configs.js",
+    "lib/",
+    "features/",
+    "scripts/",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "start": "node --expose-gc index.js",
+    "dev": "node --expose-gc --watch index.js",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "clean": "node scripts/clear-cache.js",
+    "lint": "eslint .",
+    "format": "prettier --write .",
+    "postinstall": "node scripts/postinstall.js && node scripts/download-model.js"
+  },
+  "keywords": [
+    "mcp",
+    "semantic-search",
+    "code-search",
+    "embeddings",
+    "ai",
+    "model-context-protocol",
+    "hybrid-search",
+    "code-intelligence",
+    "cursor",
+    "vscode",
+    "claude",
+    "codex",
+    "openai",
+    "gemini",
+    "anthropic",
+    "antigravity",
+    "heuristic"
+  ],
+  "author": {
+    "name": "Softerist",
+    "url": "https://github.com/softerist"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/softerist/heuristic-mcp.git"
+  },
+  "homepage": "https://github.com/softerist/heuristic-mcp#readme",
+  "license": "MIT",
+  "dependencies": {
+    "@huggingface/transformers": "^3.8.1",
+    "@modelcontextprotocol/sdk": "^1.0.4",
+    "better-sqlite3": "^12.6.2",
+    "chokidar": "^3.5.3",
+    "fdir": "^6.5.0",
+    "ignore": "^7.0.5",
+    "punycode": "^2.3.1"
+  },
+  "optionalDependencies": {
+    "hnswlib-node": "^3.0.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "overrides": {
+    "punycode": "^2.3.1"
+  },
+  "devDependencies": {
+    "@eslint/js": "^9.39.2",
+    "@vitest/coverage-v8": "^4.0.18",
+    "eslint": "^9.39.2",
+    "globals": "^17.1.0",
+    "prettier": "^3.8.1",
+    "vitest": "^4.0.16"
+  }
+}

package/scripts/clear-cache.js

@@ -1,20 +1,20 @@
-#!/usr/bin/env node
-import fs from 'fs/promises';
-import { loadConfig } from '../lib/config.js';
-
-async function clearCache() {
-  try {
-    const config = await loadConfig(process.cwd());
-    const cacheDir = config.cacheDirectory;
-
-    // Remove cache directory
-    await fs.rm(cacheDir, { recursive: true, force: true });
-    console.info(`Cache cleared successfully: ${cacheDir}`);
-    console.info('Next startup will perform a full reindex.');
-  } catch (error) {
-    console.error(`Error clearing cache: ${error.message}`);
-    process.exit(1);
-  }
-}
-
-clearCache();
+#!/usr/bin/env node
+import fs from 'fs/promises';
+import { loadConfig } from '../lib/config.js';
+
+async function clearCache() {
+  try {
+    const config = await loadConfig(process.cwd());
+    const cacheDir = config.cacheDirectory;
+
+    // Remove cache directory
+    await fs.rm(cacheDir, { recursive: true, force: true });
+    console.info(`Cache cleared successfully: ${cacheDir}`);
+    console.info('Next startup will perform a full reindex.');
+  } catch (error) {
+    console.error(`Error clearing cache: ${error.message}`);
+    process.exit(1);
+  }
+}
+
+clearCache();

package/scripts/download-model.js

@@ -1,43 +1,43 @@
-import path from 'path';
-
-import { getGlobalCacheDir } from '../lib/config.js';
-
-async function downloadModel() {
-  const globalCacheDir = path.join(getGlobalCacheDir(), 'xenova');
-
-  try {
-    const transformers = await import('@huggingface/transformers');
-    const { pipeline, env } = transformers;
-
-    // Force cache directory to global location
-    env.cacheDir = globalCacheDir;
-
-    console.info(`[Model Setup] Pre-caching model to: ${globalCacheDir}`);
-    // Check if network is available by pinging HF (simple check)
-    // Actually, pipeline() will fail fast if network is down
-    console.info(`[Model Setup] Downloading 'jinaai/jina-embeddings-v2-base-code'...`);
-
-    // This will download the model to the cache directory
-    await pipeline('feature-extraction', 'jinaai/jina-embeddings-v2-base-code');
-
-    console.info(`[Model Setup] ✅ Model cached successfully!`);
-  } catch (error) {
-    if (error && error.code === 'ERR_MODULE_NOT_FOUND') {
-      console.warn(
-        '[Model Setup] ⚠️  Transformers not available yet; skipping model pre-download.'
-      );
-      console.warn(
-        '[Model Setup] This is okay! The server will attempt to download it when started.'
-      );
-      return;
-    }
-    console.warn(`[Model Setup] ⚠️  Constructive warning: Failed to pre-download model.`);
-    console.warn(
-      '[Model Setup] This is okay! The server will attempt to download it when started.'
-    );
-    console.warn(`[Model Setup] Error details: ${error.message}`);
-    // Don't fail the install, just warn
-  }
-}
-
-downloadModel();
+import path from 'path';
+
+import { getGlobalCacheDir } from '../lib/config.js';
+
+async function downloadModel() {
+  const globalCacheDir = path.join(getGlobalCacheDir(), 'xenova');
+
+  try {
+    const transformers = await import('@huggingface/transformers');
+    const { pipeline, env } = transformers;
+
+    // Force cache directory to global location
+    env.cacheDir = globalCacheDir;
+
+    console.info(`[Model Setup] Pre-caching model to: ${globalCacheDir}`);
+    // Check if network is available by pinging HF (simple check)
+    // Actually, pipeline() will fail fast if network is down
+    console.info(`[Model Setup] Downloading 'jinaai/jina-embeddings-v2-base-code'...`);
+
+    // This will download the model to the cache directory
+    await pipeline('feature-extraction', 'jinaai/jina-embeddings-v2-base-code');
+
+    console.info(`[Model Setup] ✅ Model cached successfully!`);
+  } catch (error) {
+    if (error && error.code === 'ERR_MODULE_NOT_FOUND') {
+      console.warn(
+        '[Model Setup] ⚠️  Transformers not available yet; skipping model pre-download.'
+      );
+      console.warn(
+        '[Model Setup] This is okay! The server will attempt to download it when started.'
+      );
+      return;
+    }
+    console.warn(`[Model Setup] ⚠️  Constructive warning: Failed to pre-download model.`);
+    console.warn(
+      '[Model Setup] This is okay! The server will attempt to download it when started.'
+    );
+    console.warn(`[Model Setup] Error details: ${error.message}`);
+    // Don't fail the install, just warn
+  }
+}
+
+downloadModel();

package/scripts/mcp-launcher.js

@@ -1,49 +1,49 @@
-import path from 'node:path';
-import { spawn } from 'node:child_process';
-import { fileURLToPath } from 'node:url';
-import { getWorkspaceEnvKeys } from '../lib/workspace-env.js';
-
-const __dirname = path.dirname(fileURLToPath(import.meta.url));
-const repoRoot = path.resolve(__dirname, '..');
-const indexPath = path.join(repoRoot, 'index.js');
-
-let workspace;
-const passthrough = [];
-
-function readWorkspaceFromEnv() {
-  for (const key of getWorkspaceEnvKeys()) {
-    const value = process.env[key];
-    if (!value || value.includes('${')) continue;
-    return value;
-  }
-  return null;
-}
-
-for (let i = 2; i < process.argv.length; i += 1) {
-  const arg = process.argv[i];
-  if (arg === '--workspace' && i + 1 < process.argv.length) {
-    workspace = process.argv[i + 1];
-    i += 1;
-    continue;
-  }
-  passthrough.push(arg);
-}
-
-if (!workspace) {
-  workspace = readWorkspaceFromEnv();
-}
-
-if (!workspace) {
-  workspace = process.cwd();
-}
-
-const args = ['--expose-gc', indexPath, '--workspace', workspace, ...passthrough];
-const child = spawn(process.execPath, args, { stdio: 'inherit' });
-
-child.on('exit', (code, signal) => {
-  if (signal) {
-    process.kill(process.pid, signal);
-    return;
-  }
-  process.exit(code ?? 1);
-});
+import path from 'node:path';
+import { spawn } from 'node:child_process';
+import { fileURLToPath } from 'node:url';
+import { getWorkspaceEnvKeys } from '../lib/workspace-env.js';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const repoRoot = path.resolve(__dirname, '..');
+const indexPath = path.join(repoRoot, 'index.js');
+
+let workspace;
+const passthrough = [];
+
+function readWorkspaceFromEnv() {
+  for (const key of getWorkspaceEnvKeys()) {
+    const value = process.env[key];
+    if (!value || value.includes('${')) continue;
+    return value;
+  }
+  return null;
+}
+
+for (let i = 2; i < process.argv.length; i += 1) {
+  const arg = process.argv[i];
+  if (arg === '--workspace' && i + 1 < process.argv.length) {
+    workspace = process.argv[i + 1];
+    i += 1;
+    continue;
+  }
+  passthrough.push(arg);
+}
+
+if (!workspace) {
+  workspace = readWorkspaceFromEnv();
+}
+
+if (!workspace) {
+  workspace = process.cwd();
+}
+
+const args = ['--expose-gc', indexPath, '--workspace', workspace, ...passthrough];
+const child = spawn(process.execPath, args, { stdio: 'inherit' });
+
+child.on('exit', (code, signal) => {
+  if (signal) {
+    process.kill(process.pid, signal);
+    return;
+  }
+  process.exit(code ?? 1);
+});

package/scripts/postinstall.js

@@ -1,12 +1,12 @@
-import { register } from '../features/register.js';
-
-// Run the registration process - MUST await to ensure file writes complete
-console.info('[PostInstall] Running Heuristic MCP registration...');
-
-try {
-  await register();
-  console.info('[PostInstall] Registration complete.');
-} catch (err) {
-  console.error('[PostInstall] Registration failed:', err.message);
-  // Don't fail the install if registration fails, just warn
-}
+import { register } from '../features/register.js';
+
+// Run the registration process - MUST await to ensure file writes complete
+console.info('[PostInstall] Running Heuristic MCP registration...');
+
+try {
+  await register();
+  console.info('[PostInstall] Registration complete.');
+} catch (err) {
+  console.error('[PostInstall] Registration failed:', err.message);
+  // Don't fail the install if registration fails, just warn
+}

package/search-configs.js

@@ -1,36 +1,36 @@
-import { loadConfig } from './lib/config.js';
-import { EmbeddingsCache } from './lib/cache.js';
-import { HybridSearch } from './features/hybrid-search.js';
-import { pipeline, env } from '@huggingface/transformers';
-
-// Force same thread config as server
-if (env?.backends?.onnx) {
-  env.backends.onnx.numThreads = 2;
-  if (env.backends.onnx.wasm) {
-    env.backends.onnx.wasm.numThreads = 2;
-  }
-}
-
-async function searchConfigs() {
-  const config = await loadConfig(process.cwd());
-  const cache = new EmbeddingsCache(config);
-  await cache.load();
-
-  const embedder = async (text) => {
-    const pipe = await pipeline('feature-extraction', config.embeddingModel, {
-      session_options: { numThreads: 2 },
-      dtype: 'fp32',
-    });
-    return pipe(text, { pooling: 'mean', normalize: true });
-  };
-
-  const searcher = new HybridSearch(embedder, cache, config);
-  const { results } = await searcher.search('configuration files, config, settings');
-
-  console.info(JSON.stringify(results, null, 2));
-}
-
-searchConfigs().catch((err) => {
-  console.error(err);
-  process.exit(1);
-});
+import { loadConfig } from './lib/config.js';
+import { EmbeddingsCache } from './lib/cache.js';
+import { HybridSearch } from './features/hybrid-search.js';
+import { pipeline, env } from '@huggingface/transformers';
+
+// Force same thread config as server
+if (env?.backends?.onnx) {
+  env.backends.onnx.numThreads = 2;
+  if (env.backends.onnx.wasm) {
+    env.backends.onnx.wasm.numThreads = 2;
+  }
+}
+
+async function searchConfigs() {
+  const config = await loadConfig(process.cwd());
+  const cache = new EmbeddingsCache(config);
+  await cache.load();
+
+  const embedder = async (text) => {
+    const pipe = await pipeline('feature-extraction', config.embeddingModel, {
+      session_options: { numThreads: 2 },
+      dtype: 'fp32',
+    });
+    return pipe(text, { pooling: 'mean', normalize: true });
+  };
+
+  const searcher = new HybridSearch(embedder, cache, config);
+  const { results } = await searcher.search('configuration files, config, settings');
+
+  console.info(JSON.stringify(results, null, 2));
+}
+
+searchConfigs().catch((err) => {
+  console.error(err);
+  process.exit(1);
+});