code-graph-context 2.12.4 → 2.12.6

package/README.md

@@ -190,6 +190,7 @@ If you prefer to edit the config files directly:
 | `NEO4J_USER` | No | `neo4j` | Neo4j username |
 | `NEO4J_PASSWORD` | No | `PASSWORD` | Neo4j password |
 | `EMBEDDING_MODEL` | No | `codesage/codesage-base-v2` | Local embedding model (see [Embedding Configuration](#embedding-configuration)) |
+| `EMBEDDING_BATCH_SIZE` | No | `16` | Texts per embedding batch (lower = less memory, higher = faster) |
 | `EMBEDDING_SIDECAR_PORT` | No | `8787` | Port for local embedding server |
 | `EMBEDDING_DEVICE` | No | auto (`mps`/`cpu`) | Device for embeddings. Auto-detects MPS on Apple Silicon |
 | `EMBEDDING_HALF_PRECISION` | No | `false` | Set `true` for float16 (uses ~0.5x memory) |

package/dist/core/config/schema.js

@@ -432,7 +432,6 @@ export const CORE_TYPESCRIPT_SCHEMA = {
                 labels: ['Enum', 'TypeScript'],
                 primaryLabel: 'Enum',
                 indexed: ['name', 'isExported'],
-                skipEmbedding: true,
             },
         },
         [CoreNodeType.FUNCTION_DECLARATION]: {
@@ -493,7 +492,6 @@ export const CORE_TYPESCRIPT_SCHEMA = {
                 labels: ['Variable', 'TypeScript'],
                 primaryLabel: 'Variable',
                 indexed: ['name'],
-                skipEmbedding: true,
             },
         },
         [CoreNodeType.TYPE_ALIAS]: {
@@ -514,7 +512,6 @@ export const CORE_TYPESCRIPT_SCHEMA = {
                 labels: ['TypeAlias', 'TypeScript'],
                 primaryLabel: 'TypeAlias',
                 indexed: ['name'],
-                skipEmbedding: true,
             },
         },
         [CoreNodeType.CONSTRUCTOR_DECLARATION]: {

package/dist/core/embeddings/local-embeddings.service.js

@@ -1,12 +1,12 @@
 /**
  * Local Embeddings Service
- * Uses a Python sidecar running Qodo-Embed-1-1.5B (or configurable model).
+ * Uses a Python sidecar running CodeSage-Base-v2 (or configurable model).
  * Default provider — no API key required.
  */
 import { debugLog } from '../../mcp/utils.js';
 import { getEmbeddingSidecar } from './embedding-sidecar.js';
 const BATCH_CONFIG = {
-    maxBatchSize: 8, // Small batches — 1.5B model on MPS OOMs at higher values on 16GB machines
+    maxBatchSize: parseInt(process.env.EMBEDDING_BATCH_SIZE ?? '', 10) || 16,
 };
 export class LocalEmbeddingsService {
     async embedText(text) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-graph-context",
-  "version": "2.12.4",
+  "version": "2.12.6",
   "description": "MCP server that builds code graphs to provide rich context to LLMs",
   "type": "module",
   "homepage": "https://github.com/drewdrewH/code-graph-context#readme",