kongbrain 0.5.0 → 0.5.1

package/.kongcode-handoff.json

@@ -0,0 +1,8 @@
+{
+  "sessionId": "7a2dcd89-1468-49c3-a1be-24dac5a05662",
+  "timestamp": "2026-05-02T21:39:49.691Z",
+  "userTurnCount": 1,
+  "lastUserText": "Drain the KongCode pending_work queue. Loop: call mcp__plugin_kongcode_kongcode__fetch_pending_work to claim the next item, analyze the data per the work-type instructions, then call mcp__plugin_kongcode_kongcode__commit_work_results with your output. Repeat until fetch_pending_work returns empty. Be efficient: minimize per-item analysis. This is auto-drain, not user-facing — produce no narration, just process items.",
+  "lastAssistantText": "",
+  "unextractedTokens": 0
+}

package/README.github.md

@@ -4,6 +4,8 @@
 
 ![KongBrain](KongClaw.png)
 
+[![VoidOrigin](https://img.shields.io/badge/VOIDORIGIN-voidorigin.com-0a0a0a?style=for-the-badge&logo=data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHZpZXdCb3g9IjAgMCAyNCAyNCIgZmlsbD0ibm9uZSI+PGNpcmNsZSBjeD0iMTIiIGN5PSIxMiIgcj0iMTAiIHN0cm9rZT0iI2ZmNmIzNSIgc3Ryb2tlLXdpZHRoPSIyIi8+PGNpcmNsZSBjeD0iMTIiIGN5PSIxMiIgcj0iNCIgZmlsbD0iI2ZmNmIzNSIvPjwvc3ZnPg==&logoColor=ff6b35&labelColor=0a0a0a)](https://voidorigin.com)
+
 [![npm](https://img.shields.io/npm/v/kongbrain?style=for-the-badge&logo=npm&color=cb3837)](https://www.npmjs.com/package/kongbrain)
 [![ClawHub](https://img.shields.io/badge/ClawHub-kongbrain-ff6b35?style=for-the-badge)](https://clawhub.ai/packages/kongbrain)
 [![GitHub Stars](https://img.shields.io/github/stars/42U/kongbrain?style=for-the-badge&logo=github&color=gold)](https://github.com/42U/kongbrain)
@@ -498,6 +500,6 @@ The lobster doesn't accept contributions. The ape does.
 
 <div align="center">
 
-MIT License | Built by [42U](https://github.com/42U)
+MIT License | Built by [42U](https://github.com/42U) | [VoidOrigin](https://voidorigin.com)
 
 </div>

package/SKILL.md

@@ -1,7 +1,7 @@
 ---
 name: kongbrain
 description: Graph-backed persistent memory engine for OpenClaw. Replaces the default context window with SurrealDB + vector embeddings that learn across sessions.
-version: 0.5.0
+version: 0.5.1
 homepage: https://github.com/42U/kongbrain
 metadata:
   openclaw:

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kongbrain",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "description": "Graph-backed persistent memory engine for OpenClaw. Replaces the default context window with SurrealDB + vector embeddings that learn across sessions.",
   "type": "module",
   "license": "MIT",

package/src/config.ts

@@ -51,6 +51,14 @@ export interface KongBrainConfig {
   thresholds: ThresholdConfig;
 }
 
+const DEFAULT_EMBEDDING_DIMENSIONS = 1024;
+
+function parsePositiveInteger(value: unknown, fallback: number): number {
+  return typeof value === "number" && Number.isInteger(value) && value > 0
+    ? value
+    : fallback;
+}
+
 function parseEmbeddingConfig(raw: Record<string, unknown>): EmbeddingConfig {
   const openaiCompatRaw = (raw.openaiCompat ?? {}) as Record<string, unknown>;
 
@@ -63,7 +71,7 @@ function parseEmbeddingConfig(raw: Record<string, unknown>): EmbeddingConfig {
 
   return {
     provider,
-    dimensions: typeof raw.dimensions === "number" ? raw.dimensions : 1024,
+    dimensions: parsePositiveInteger(raw.dimensions, DEFAULT_EMBEDDING_DIMENSIONS),
     modelPath:
       process.env.EMBED_MODEL_PATH ??
       (typeof raw.modelPath === "string"

package/src/context-engine.ts

@@ -57,7 +57,7 @@ export class KongBrainContextEngine implements ContextEngine {
   readonly info: ContextEngineInfo = {
     id: "kongbrain",
     name: "KongBrain",
-    version: "0.4.2",
+    version: "0.5.1",
     ownsCompaction: true,
   };
 
@@ -76,7 +76,9 @@ export class KongBrainContextEngine implements ContextEngine {
     // Run schema once per process (idempotent but expensive on every bootstrap)
     if (!this.state.schemaApplied) {
       try {
-        const schemaSql = loadSchema();
+        const schemaSql = loadSchema({
+          embeddingDimensions: this.state.config.embedding.dimensions,
+        });
         await store.queryExec(schemaSql);
         this.state.schemaApplied = true;
       } catch (e) {

package/src/index.ts

@@ -12,6 +12,7 @@ import { parsePluginConfig } from "./config.js";
 import { SurrealStore } from "./surreal.js";
 import { createEmbeddingService } from "./embeddings.js";
 import { GlobalPluginState, type CompleteFn } from "./state.js";
+import { resolveModelRef } from "./model-resolution.js";
 import { KongBrainContextEngine } from "./context-engine.js";
 import { createRecallToolDef } from "./tools/recall.js";
 import { createCoreMemoryToolDef } from "./tools/core-memory.js";
@@ -333,6 +334,38 @@ async function checkEmbeddingProviderMismatch(
   }
 }
 
+/**
+ * Detect vectors whose dimensionality doesn't match the configured dimension.
+ * DEFINE INDEX IF NOT EXISTS is non-destructive — changing embedding.dimensions
+ * after initial setup leaves the HNSW index at the old dimension while new
+ * vectors are written at the new size. Warn so the user knows to rebuild.
+ */
+async function checkEmbeddingDimensionMismatch(
+  store: SurrealStore,
+  configuredDimensions: number,
+  logger: { warn: (msg: string) => void },
+): Promise<void> {
+  if (!store.isAvailable()) return;
+  const tables = ["concept", "memory", "turn", "artifact", "identity_chunk", "skill", "reflection", "monologue"];
+  for (const t of tables) {
+    try {
+      const rows = await store.queryFirst<{ len: number }>(
+        `SELECT array::len(embedding) AS len FROM ${t} WHERE embedding != NONE AND array::len(embedding) > 0 LIMIT 1`,
+      );
+      if (rows.length > 0 && rows[0].len !== configuredDimensions) {
+        logger.warn(
+          `Embedding dimension mismatch: existing vectors are ${rows[0].len}-dimensional but embedding.dimensions is configured as ${configuredDimensions}. ` +
+            `HNSW indexes created at the old dimension are not updated by DEFINE INDEX IF NOT EXISTS. ` +
+            `To fix: drop and recreate the vector indexes, then re-embed affected rows.`,
+        );
+        return;
+      }
+    } catch (e) {
+      swallow.warn(`factory:dimensionMismatchCheck:${t}`, e);
+    }
+  }
+}
+
 export default definePluginEntry({
   id: "kongbrain",
   name: "KongBrain",
@@ -349,7 +382,9 @@ export default definePluginEntry({
     // ensure a single instance survives across module reloads.
     let globalState = getGlobalState();
     if (!globalState) {
-      const store = new SurrealStore(config.surreal);
+      const store = new SurrealStore(config.surreal, {
+        embeddingDimensions: config.embedding.dimensions,
+      });
       const embeddings = createEmbeddingService(config.embedding);
       // Tag every embedding write and filter every embedding search by this
       // provider id, so vectors from different models (different vector
@@ -383,27 +418,22 @@ export default definePluginEntry({
           }
           piAi = await import(piAiPath);
         }
-        // Fall back to calling pi-ai directly (runtime.complete not in OpenClaw 2026.3.24)
-        const provider = params.provider ?? apiRef.runtime.agent.defaults.provider;
-        const rawModel = params.model ?? apiRef.runtime.agent.defaults.model;
-        // defaults.model may be an object {primary: '...', fallbacks: []} — unwrap it
-        const modelIdRaw = typeof rawModel === 'object' && rawModel !== null
-          ? (rawModel as any).primary ?? (rawModel as any).id ?? String(rawModel)
-          : rawModel;
-        // modelId may be "provider/model" format — split if provider not set
-        let resolvedProvider = provider;
-        let modelId = modelIdRaw;
-        if (typeof modelId === 'string' && modelId.includes('/') && !resolvedProvider) {
-          const idx = modelId.indexOf('/');
-          resolvedProvider = modelId.slice(0, idx);
-          modelId = modelId.slice(idx + 1);
-        }
-        const model = piAi!.getModel(resolvedProvider, modelId);
+        // Fall back to calling pi-ai directly (runtime.complete not in OpenClaw 2026.3.24).
+        // Fully-qualified OpenClaw refs (provider/model) are authoritative:
+        // stale runtime default providers must not override e.g.
+        // "openrouter/google/gemini-3-flash-preview".
+        const cfg = apiRef.runtime.config.loadConfig();
+        const resolved = resolveModelRef({
+          explicitProvider: params.provider,
+          explicitModel: params.model,
+          config: cfg,
+          runtimeDefaults: apiRef.runtime.agent.defaults,
+        });
+        const model = piAi!.getModel(resolved.provider, resolved.modelId);
         if (!model) {
-          throw new Error(`Model "${modelId}" not found for provider "${provider}"`);
+          throw new Error(`Model "${resolved.modelId}" not found for provider "${resolved.provider}"`);
         }
         // Resolve auth via OpenClaw's runtime (handles profiles, env vars, etc.)
-        const cfg = apiRef.runtime.config.loadConfig();
         const auth = await apiRef.runtime.modelAuth.getApiKeyForModel({ model, cfg });
         // Build context
         const now = Date.now();
@@ -417,7 +447,7 @@ export default definePluginEntry({
         );
         const context = { systemPrompt: params.system, messages };
         // Pass apiKey directly in options so the provider can use it
-        log.info(`complete(): provider=${resolvedProvider} model=${modelId} msgs=${params.messages.length}`);
+        log.info(`complete(): provider=${resolved.provider} model=${resolved.modelId} msgs=${params.messages.length}`);
         // NOTE: outputFormat (structured output) is intentionally NOT passed to pi-ai.
         // pi-ai's SimpleStreamOptions doesn't support it, and injecting it via onPayload
         // causes the Anthropic API to return empty responses. The daemon's JSON parsing
@@ -471,6 +501,8 @@ export default definePluginEntry({
           // re-embedded with the active provider.
           checkEmbeddingProviderMismatch(store, embeddings.providerId, logger)
             .catch(e => swallow.warn("factory:providerMismatchCheck", e));
+          checkEmbeddingDimensionMismatch(store, config.embedding.dimensions, logger)
+            .catch(e => swallow.warn("factory:dimensionMismatchCheck", e));
         }
       } catch (e) {
         logger.warn(`Embeddings init failed — running in degraded mode: ${e}`);

package/src/model-resolution.ts

@@ -0,0 +1,98 @@
+export interface ResolveModelRefInput {
+  explicitProvider?: unknown;
+  explicitModel?: unknown;
+  config?: unknown;
+  runtimeDefaults?: unknown;
+}
+
+export interface ResolvedModelRef {
+  provider: string;
+  modelId: string;
+}
+
+function asRecord(value: unknown): Record<string, unknown> | null {
+  return typeof value === "object" && value !== null
+    ? value as Record<string, unknown>
+    : null;
+}
+
+function nonEmptyString(value: unknown): string | undefined {
+  return typeof value === "string" && value.trim().length > 0
+    ? value.trim()
+    : undefined;
+}
+
+function unwrapModelRef(value: unknown): string | undefined {
+  const direct = nonEmptyString(value);
+  if (direct) return direct;
+
+  const record = asRecord(value);
+  if (!record) return undefined;
+
+  return unwrapModelRef(record.primary) ?? unwrapModelRef(record.id);
+}
+
+function defaultModelFromConfig(config: unknown): string | undefined {
+  const cfg = asRecord(config);
+  const agents = asRecord(cfg?.agents);
+  const defaults = asRecord(agents?.defaults);
+  return unwrapModelRef(defaults?.model);
+}
+
+function defaultModelFromRuntime(runtimeDefaults: unknown): string | undefined {
+  const defaults = asRecord(runtimeDefaults);
+  return unwrapModelRef(defaults?.model);
+}
+
+function providerFromRuntime(runtimeDefaults: unknown): string | undefined {
+  const defaults = asRecord(runtimeDefaults);
+  return nonEmptyString(defaults?.provider);
+}
+
+function splitQualifiedModelRef(modelRef: string): { provider: string; modelId: string } | null {
+  const slash = modelRef.indexOf("/");
+  if (slash <= 0 || slash === modelRef.length - 1) return null;
+  return {
+    provider: modelRef.slice(0, slash),
+    modelId: modelRef.slice(slash + 1),
+  };
+}
+
+function chooseModelRef(input: ResolveModelRefInput): { modelRef: string; explicit: boolean } | null {
+  const explicit = unwrapModelRef(input.explicitModel);
+  if (explicit) return { modelRef: explicit, explicit: true };
+
+  const configDefault = defaultModelFromConfig(input.config);
+  if (configDefault) return { modelRef: configDefault, explicit: false };
+
+  const runtimeDefault = defaultModelFromRuntime(input.runtimeDefaults);
+  if (runtimeDefault) return { modelRef: runtimeDefault, explicit: false };
+
+  return null;
+}
+
+export function resolveModelRef(input: ResolveModelRefInput): ResolvedModelRef {
+  const chosen = chooseModelRef(input);
+
+  if (!chosen) {
+    throw new Error("No LLM model configured for KongBrain internal completion");
+  }
+
+  const explicitProvider = nonEmptyString(input.explicitProvider);
+  if (chosen.explicit && explicitProvider) {
+    return { provider: explicitProvider, modelId: chosen.modelRef };
+  }
+
+  const qualified = splitQualifiedModelRef(chosen.modelRef);
+  if (qualified) return qualified;
+
+  const provider =
+    explicitProvider ??
+    providerFromRuntime(input.runtimeDefaults);
+
+  if (!provider) {
+    throw new Error(`No LLM provider configured for model "${chosen.modelRef}"`);
+  }
+
+  return { provider, modelId: chosen.modelRef };
+}

package/src/schema-loader.ts

@@ -12,12 +12,30 @@ import { fileURLToPath } from "node:url";
 
 const __dirname = dirname(fileURLToPath(import.meta.url));
 
-export function loadSchema(): string {
+const DEFAULT_EMBEDDING_DIMENSIONS = 1024;
+const DIMENSION_PLACEHOLDER = "__KONGBRAIN_EMBEDDING_DIMENSIONS__";
+
+export interface LoadSchemaOptions {
+  embeddingDimensions?: number;
+}
+
+function normalizeDimensions(value: unknown): number {
+  return typeof value === "number" && Number.isInteger(value) && value > 0
+    ? value
+    : DEFAULT_EMBEDDING_DIMENSIONS;
+}
+
+export function loadSchema(options: LoadSchemaOptions = {}): string {
   const primary = join(__dirname, "schema.surql");
+  let schema: string;
   try {
-    return readFileSync(primary, "utf-8");
+    schema = readFileSync(primary, "utf-8");
   } catch {
     // Dev fallback: compiled output lives in dist/, schema source in src/
-    return readFileSync(join(__dirname, "..", "src", "schema.surql"), "utf-8");
+    schema = readFileSync(join(__dirname, "..", "src", "schema.surql"), "utf-8");
   }
+  return schema.replaceAll(
+    DIMENSION_PLACEHOLDER,
+    String(normalizeDimensions(options.embeddingDimensions)),
+  );
 }

package/src/schema.surql

@@ -42,7 +42,7 @@ DEFINE FIELD IF NOT EXISTS embedding ON artifact TYPE option<array<float>>;
 DEFINE FIELD IF NOT EXISTS embedding_provider ON artifact TYPE option<string>;
 DEFINE FIELD IF NOT EXISTS tags ON artifact TYPE option<array>;
 DEFINE FIELD IF NOT EXISTS created_at ON artifact TYPE datetime DEFAULT time::now();
-DEFINE INDEX IF NOT EXISTS artifact_vec_idx ON artifact FIELDS embedding HNSW DIMENSION 1024 DIST COSINE;
+DEFINE INDEX IF NOT EXISTS artifact_vec_idx ON artifact FIELDS embedding HNSW DIMENSION __KONGBRAIN_EMBEDDING_DIMENSIONS__ DIST COSINE;
 DEFINE INDEX IF NOT EXISTS artifact_type_idx ON artifact FIELDS type;
 DEFINE INDEX IF NOT EXISTS artifact_emb_provider_idx ON artifact FIELDS embedding_provider;
 
@@ -62,7 +62,7 @@ DEFINE FIELD IF NOT EXISTS tags ON concept TYPE option<array>;
 DEFINE FIELD IF NOT EXISTS source ON concept TYPE option<string>;
 DEFINE FIELD IF NOT EXISTS created_at ON concept TYPE datetime DEFAULT time::now();
 DEFINE FIELD IF NOT EXISTS last_accessed ON concept TYPE option<datetime>;
-DEFINE INDEX IF NOT EXISTS concept_vec_idx ON concept FIELDS embedding HNSW DIMENSION 1024 DIST COSINE;
+DEFINE INDEX IF NOT EXISTS concept_vec_idx ON concept FIELDS embedding HNSW DIMENSION __KONGBRAIN_EMBEDDING_DIMENSIONS__ DIST COSINE;
 DEFINE INDEX IF NOT EXISTS concept_emb_provider_idx ON concept FIELDS embedding_provider;
 
 -- ============================================================
@@ -82,7 +82,7 @@ DEFINE FIELD IF NOT EXISTS token_count ON turn TYPE option<int>;
 DEFINE FIELD IF NOT EXISTS tool_name ON turn TYPE option<string>;
 DEFINE FIELD IF NOT EXISTS model ON turn TYPE option<string>;
 DEFINE FIELD IF NOT EXISTS usage ON turn TYPE option<object>;
-DEFINE INDEX IF NOT EXISTS turn_vec_idx ON turn FIELDS embedding HNSW DIMENSION 1024 DIST COSINE;
+DEFINE INDEX IF NOT EXISTS turn_vec_idx ON turn FIELDS embedding HNSW DIMENSION __KONGBRAIN_EMBEDDING_DIMENSIONS__ DIST COSINE;
 DEFINE INDEX IF NOT EXISTS turn_session_idx ON turn FIELDS session_id;
 DEFINE INDEX IF NOT EXISTS turn_tool_name_idx ON turn FIELDS tool_name;
 DEFINE INDEX IF NOT EXISTS turn_emb_provider_idx ON turn FIELDS embedding_provider;
@@ -96,7 +96,7 @@ DEFINE FIELD IF NOT EXISTS text ON identity_chunk TYPE string;
 DEFINE FIELD IF NOT EXISTS embedding ON identity_chunk TYPE option<array<float>>;
 DEFINE FIELD IF NOT EXISTS embedding_provider ON identity_chunk TYPE option<string>;
 DEFINE FIELD IF NOT EXISTS importance ON identity_chunk TYPE float DEFAULT 0.5;
-DEFINE INDEX IF NOT EXISTS identity_vec_idx ON identity_chunk FIELDS embedding HNSW DIMENSION 1024 DIST COSINE;
+DEFINE INDEX IF NOT EXISTS identity_vec_idx ON identity_chunk FIELDS embedding HNSW DIMENSION __KONGBRAIN_EMBEDDING_DIMENSIONS__ DIST COSINE;
 DEFINE INDEX IF NOT EXISTS identity_emb_provider_idx ON identity_chunk FIELDS embedding_provider;
 
 -- Sessions (lightweight, links to task for 5-pillar integration)
@@ -125,7 +125,7 @@ DEFINE FIELD IF NOT EXISTS last_accessed ON memory TYPE option<datetime>;
 DEFINE FIELD IF NOT EXISTS status ON memory TYPE option<string> DEFAULT "active";
 DEFINE FIELD IF NOT EXISTS resolved_at ON memory TYPE option<datetime>;
 DEFINE FIELD IF NOT EXISTS resolved_by ON memory TYPE option<string>;
-DEFINE INDEX IF NOT EXISTS memory_vec_idx ON memory FIELDS embedding HNSW DIMENSION 1024 DIST COSINE;
+DEFINE INDEX IF NOT EXISTS memory_vec_idx ON memory FIELDS embedding HNSW DIMENSION __KONGBRAIN_EMBEDDING_DIMENSIONS__ DIST COSINE;
 DEFINE INDEX IF NOT EXISTS memory_category_idx ON memory FIELDS category;
 DEFINE INDEX IF NOT EXISTS memory_emb_provider_idx ON memory FIELDS embedding_provider;
 
@@ -276,7 +276,7 @@ DEFINE FIELD IF NOT EXISTS failure_count ON skill TYPE int DEFAULT 0;
 DEFINE FIELD IF NOT EXISTS avg_duration_ms ON skill TYPE float DEFAULT 0.0;
 DEFINE FIELD IF NOT EXISTS last_used ON skill TYPE option<datetime>;
 DEFINE FIELD IF NOT EXISTS created_at ON skill TYPE datetime DEFAULT time::now();
-DEFINE INDEX IF NOT EXISTS skill_vec_idx ON skill FIELDS embedding HNSW DIMENSION 1024 DIST COSINE;
+DEFINE INDEX IF NOT EXISTS skill_vec_idx ON skill FIELDS embedding HNSW DIMENSION __KONGBRAIN_EMBEDDING_DIMENSIONS__ DIST COSINE;
 DEFINE INDEX IF NOT EXISTS skill_active_idx ON skill FIELDS active;
 DEFINE INDEX IF NOT EXISTS skill_emb_provider_idx ON skill FIELDS embedding_provider;
 
@@ -296,7 +296,7 @@ DEFINE FIELD IF NOT EXISTS severity ON reflection TYPE string DEFAULT "minor"; -
 DEFINE FIELD IF NOT EXISTS importance ON reflection TYPE float DEFAULT 7.0;
 DEFINE FIELD IF NOT EXISTS access_count ON reflection TYPE int DEFAULT 0;
 DEFINE FIELD IF NOT EXISTS created_at ON reflection TYPE datetime DEFAULT time::now();
-DEFINE INDEX IF NOT EXISTS reflection_vec_idx ON reflection FIELDS embedding HNSW DIMENSION 1024 DIST COSINE;
+DEFINE INDEX IF NOT EXISTS reflection_vec_idx ON reflection FIELDS embedding HNSW DIMENSION __KONGBRAIN_EMBEDDING_DIMENSIONS__ DIST COSINE;
 DEFINE INDEX IF NOT EXISTS reflection_emb_provider_idx ON reflection FIELDS embedding_provider;
 
 DEFINE TABLE IF NOT EXISTS reflects_on TYPE RELATION IN reflection OUT session;
@@ -347,7 +347,7 @@ DEFINE FIELD IF NOT EXISTS embedding_provider ON monologue TYPE option<string>;
 DEFINE FIELD IF NOT EXISTS timestamp ON monologue TYPE datetime DEFAULT time::now();
 DEFINE INDEX IF NOT EXISTS monologue_session ON monologue FIELDS session_id;
 DEFINE INDEX IF NOT EXISTS monologue_category ON monologue FIELDS category;
-DEFINE INDEX IF NOT EXISTS monologue_vec_idx ON monologue FIELDS embedding HNSW DIMENSION 1024 DIST COSINE;
+DEFINE INDEX IF NOT EXISTS monologue_vec_idx ON monologue FIELDS embedding HNSW DIMENSION __KONGBRAIN_EMBEDDING_DIMENSIONS__ DIST COSINE;
 DEFINE INDEX IF NOT EXISTS monologue_emb_provider_idx ON monologue FIELDS embedding_provider;
 
 -- Fibonacci resurfacing: proactive memory that fades over time

package/src/surreal.ts

@@ -47,6 +47,10 @@ export interface UtilityCacheEntry {
   retrieval_count: number;
 }
 
+export interface SurrealStoreOptions {
+  embeddingDimensions?: number;
+}
+
 const RECORD_ID_RE = /^[a-zA-Z_][a-zA-Z0-9_]*:[a-zA-Z0-9_]+$/;
 
 function assertRecordId(id: string): void {
@@ -132,9 +136,11 @@ export class SurrealStore {
    * deployments keep working if the wire-up step is ever skipped.
    */
   private activeProvider: string = "local-bge-m3";
+  private schemaOptions: SurrealStoreOptions;
 
-  constructor(config: SurrealConfig) {
+  constructor(config: SurrealConfig, options: SurrealStoreOptions = {}) {
     this.config = config;
+    this.schemaOptions = options;
     this.db = new Surreal();
   }
 
@@ -216,7 +222,9 @@ export class SurrealStore {
   }
 
   private async runSchema(): Promise<void> {
-    const schema = loadSchema();
+    const schema = loadSchema({
+      embeddingDimensions: this.schemaOptions.embeddingDimensions,
+    });
     await this.db.query(schema);
   }