@axlsdk/axl 0.6.0 → 0.7.0

package/dist/index.cjs

@@ -2128,10 +2128,13 @@ function resolveConfig(config) {
     resolved.defaultProvider = process.env.AXL_DEFAULT_PROVIDER;
   }
   if (process.env.AXL_STATE_STORE) {
-    resolved.state = {
-      ...resolved.state,
-      store: process.env.AXL_STATE_STORE
-    };
+    const envStore = process.env.AXL_STATE_STORE;
+    if (envStore === "memory" || envStore === "sqlite") {
+      resolved.state = {
+        ...resolved.state,
+        store: envStore
+      };
+    }
   }
   if (process.env.AXL_TRACE_ENABLED !== void 0) {
     resolved.trace = { ...resolved.trace, enabled: process.env.AXL_TRACE_ENABLED === "true" };
@@ -4258,122 +4261,6 @@ var SQLiteStore = class {
   }
 };
 
-// src/state/redis.ts
-var RedisStore = class {
-  client;
-  constructor(url) {
-    let Redis;
-    try {
-      const mod = require("ioredis");
-      Redis = mod.default ?? mod;
-    } catch {
-      throw new Error("ioredis is required for RedisStore. Install it with: npm install ioredis");
-    }
-    this.client = url ? new Redis(url) : new Redis();
-  }
-  // ── Key helpers ──────────────────────────────────────────────────────
-  checkpointKey(executionId) {
-    return `axl:checkpoint:${executionId}`;
-  }
-  sessionKey(sessionId) {
-    return `axl:session:${sessionId}`;
-  }
-  sessionMetaKey(sessionId) {
-    return `axl:session-meta:${sessionId}`;
-  }
-  decisionsKey() {
-    return "axl:decisions";
-  }
-  executionStateKey(executionId) {
-    return `axl:exec-state:${executionId}`;
-  }
-  pendingExecSetKey() {
-    return "axl:pending-executions";
-  }
-  // ── Checkpoints ──────────────────────────────────────────────────────
-  async saveCheckpoint(executionId, step, data) {
-    await this.client.hset(this.checkpointKey(executionId), String(step), JSON.stringify(data));
-  }
-  async getCheckpoint(executionId, step) {
-    const raw = await this.client.hget(this.checkpointKey(executionId), String(step));
-    return raw !== null ? JSON.parse(raw) : null;
-  }
-  async getLatestCheckpoint(executionId) {
-    const all = await this.client.hgetall(this.checkpointKey(executionId));
-    if (!all || Object.keys(all).length === 0) return null;
-    let maxStep = -1;
-    let maxData = null;
-    for (const [stepStr, raw] of Object.entries(all)) {
-      const step = Number(stepStr);
-      if (step > maxStep) {
-        maxStep = step;
-        maxData = JSON.parse(raw);
-      }
-    }
-    return { step: maxStep, data: maxData };
-  }
-  // ── Sessions ─────────────────────────────────────────────────────────
-  async saveSession(sessionId, history) {
-    await this.client.set(this.sessionKey(sessionId), JSON.stringify(history));
-    await this.client.sadd("axl:session-ids", sessionId);
-  }
-  async getSession(sessionId) {
-    const raw = await this.client.get(this.sessionKey(sessionId));
-    return raw ? JSON.parse(raw) : [];
-  }
-  async deleteSession(sessionId) {
-    await this.client.del(this.sessionKey(sessionId));
-    await this.client.del(this.sessionMetaKey(sessionId));
-    await this.client.srem("axl:session-ids", sessionId);
-  }
-  async saveSessionMeta(sessionId, key, value) {
-    await this.client.hset(this.sessionMetaKey(sessionId), key, JSON.stringify(value));
-  }
-  async getSessionMeta(sessionId, key) {
-    const raw = await this.client.hget(this.sessionMetaKey(sessionId), key);
-    return raw !== null ? JSON.parse(raw) : null;
-  }
-  // ── Pending Decisions ────────────────────────────────────────────────
-  async savePendingDecision(executionId, decision) {
-    await this.client.hset(this.decisionsKey(), executionId, JSON.stringify(decision));
-  }
-  async getPendingDecisions() {
-    const all = await this.client.hgetall(this.decisionsKey());
-    if (!all) return [];
-    return Object.values(all).map((raw) => JSON.parse(raw));
-  }
-  async resolveDecision(executionId, _result) {
-    await this.client.hdel(this.decisionsKey(), executionId);
-  }
-  // ── Execution State ──────────────────────────────────────────────────
-  async saveExecutionState(executionId, state) {
-    await this.client.set(this.executionStateKey(executionId), JSON.stringify(state));
-    if (state.status === "waiting") {
-      await this.client.sadd(this.pendingExecSetKey(), executionId);
-    } else {
-      await this.client.srem(this.pendingExecSetKey(), executionId);
-    }
-  }
-  async getExecutionState(executionId) {
-    const raw = await this.client.get(this.executionStateKey(executionId));
-    return raw ? JSON.parse(raw) : null;
-  }
-  async listPendingExecutions() {
-    return this.client.smembers(this.pendingExecSetKey());
-  }
-  // ── Sessions (Studio introspection) ────────────────────────────────────
-  async listSessions() {
-    return this.client.smembers("axl:session-ids");
-  }
-  /** Close the Redis connection. */
-  async close() {
-    await this.client.quit();
-  }
-  async deleteCheckpoints(executionId) {
-    await this.client.del(this.checkpointKey(executionId));
-  }
-};
-
 // src/session.ts
 var Session = class _Session {
   constructor(sessionId, runtime, store, options) {
@@ -5260,12 +5147,11 @@ var AxlRuntime = class extends import_node_events2.EventEmitter {
     return this.mcpManager;
   }
   createStateStore() {
-    const storeType = this.config.state?.store ?? "memory";
-    switch (storeType) {
+    const storeOption = this.config.state?.store ?? "memory";
+    if (typeof storeOption !== "string") return storeOption;
+    switch (storeOption) {
       case "sqlite":
         return new SQLiteStore(this.config.state?.sqlite?.path ?? "./data/axl.db");
-      case "redis":
-        return new RedisStore(this.config.state?.redis?.url);
       case "memory":
       default:
         return new MemoryStore();
@@ -5896,6 +5782,137 @@ var AxlRuntime = class extends import_node_events2.EventEmitter {
   }
 };
 
+// src/state/redis.ts
+var RedisStore = class _RedisStore {
+  constructor(client) {
+    this.client = client;
+  }
+  /**
+   * Create a connected RedisStore instance.
+   *
+   * @param url - Redis connection URL (e.g. `redis://localhost:6379`). Defaults to `redis://localhost:6379`.
+   */
+  static async create(url) {
+    let createClient;
+    try {
+      const mod = require("redis");
+      createClient = mod.createClient ?? mod.default?.createClient;
+      if (typeof createClient !== "function") {
+        throw new Error(
+          "redis package does not export createClient. Ensure you have redis ^5.0.0 installed: npm install redis"
+        );
+      }
+    } catch (err) {
+      if (err instanceof Error && err.message.includes("createClient")) throw err;
+      throw new Error("redis is required for RedisStore. Install it with: npm install redis");
+    }
+    const client = url ? createClient({ url }) : createClient();
+    await client.connect();
+    return new _RedisStore(client);
+  }
+  // ── Key helpers ──────────────────────────────────────────────────────
+  checkpointKey(executionId) {
+    return `axl:checkpoint:${executionId}`;
+  }
+  sessionKey(sessionId) {
+    return `axl:session:${sessionId}`;
+  }
+  sessionMetaKey(sessionId) {
+    return `axl:session-meta:${sessionId}`;
+  }
+  decisionsKey() {
+    return "axl:decisions";
+  }
+  executionStateKey(executionId) {
+    return `axl:exec-state:${executionId}`;
+  }
+  pendingExecSetKey() {
+    return "axl:pending-executions";
+  }
+  // ── Checkpoints ──────────────────────────────────────────────────────
+  async saveCheckpoint(executionId, step, data) {
+    await this.client.hSet(this.checkpointKey(executionId), String(step), JSON.stringify(data));
+  }
+  async getCheckpoint(executionId, step) {
+    const raw = await this.client.hGet(this.checkpointKey(executionId), String(step));
+    return raw != null ? JSON.parse(raw) : null;
+  }
+  async getLatestCheckpoint(executionId) {
+    const all = await this.client.hGetAll(this.checkpointKey(executionId));
+    if (!all || Object.keys(all).length === 0) return null;
+    let maxStep = -1;
+    let maxData = null;
+    for (const [stepStr, raw] of Object.entries(all)) {
+      const step = Number(stepStr);
+      if (step > maxStep) {
+        maxStep = step;
+        maxData = JSON.parse(raw);
+      }
+    }
+    return { step: maxStep, data: maxData };
+  }
+  // ── Sessions ─────────────────────────────────────────────────────────
+  async saveSession(sessionId, history) {
+    await this.client.set(this.sessionKey(sessionId), JSON.stringify(history));
+    await this.client.sAdd("axl:session-ids", sessionId);
+  }
+  async getSession(sessionId) {
+    const raw = await this.client.get(this.sessionKey(sessionId));
+    return raw ? JSON.parse(raw) : [];
+  }
+  async deleteSession(sessionId) {
+    await this.client.del(this.sessionKey(sessionId));
+    await this.client.del(this.sessionMetaKey(sessionId));
+    await this.client.sRem("axl:session-ids", sessionId);
+  }
+  async saveSessionMeta(sessionId, key, value) {
+    await this.client.hSet(this.sessionMetaKey(sessionId), key, JSON.stringify(value));
+  }
+  async getSessionMeta(sessionId, key) {
+    const raw = await this.client.hGet(this.sessionMetaKey(sessionId), key);
+    return raw != null ? JSON.parse(raw) : null;
+  }
+  // ── Pending Decisions ────────────────────────────────────────────────
+  async savePendingDecision(executionId, decision) {
+    await this.client.hSet(this.decisionsKey(), executionId, JSON.stringify(decision));
+  }
+  async getPendingDecisions() {
+    const all = await this.client.hGetAll(this.decisionsKey());
+    if (!all) return [];
+    return Object.values(all).map((raw) => JSON.parse(raw));
+  }
+  async resolveDecision(executionId, _result) {
+    await this.client.hDel(this.decisionsKey(), executionId);
+  }
+  // ── Execution State ──────────────────────────────────────────────────
+  async saveExecutionState(executionId, state) {
+    await this.client.set(this.executionStateKey(executionId), JSON.stringify(state));
+    if (state.status === "waiting") {
+      await this.client.sAdd(this.pendingExecSetKey(), executionId);
+    } else {
+      await this.client.sRem(this.pendingExecSetKey(), executionId);
+    }
+  }
+  async getExecutionState(executionId) {
+    const raw = await this.client.get(this.executionStateKey(executionId));
+    return raw ? JSON.parse(raw) : null;
+  }
+  async listPendingExecutions() {
+    return this.client.sMembers(this.pendingExecSetKey());
+  }
+  // ── Sessions (Studio introspection) ────────────────────────────────────
+  async listSessions() {
+    return this.client.sMembers("axl:session-ids");
+  }
+  /** Close the Redis connection. */
+  async close() {
+    await this.client.quit();
+  }
+  async deleteCheckpoints(executionId) {
+    await this.client.del(this.checkpointKey(executionId));
+  }
+};
+
 // src/memory/embedder-openai.ts
 var OpenAIEmbedder = class {
   apiKey;