@axiom-lattice/core 2.1.36 → 2.1.37

package/dist/index.d.mts

@@ -400,6 +400,7 @@ declare class SqlDatabaseManager {
     private static instance;
     private databases;
     private defaultDatabaseKeys;
+    private configStore;
     private constructor();
     /**
      * Get the singleton instance
@@ -422,14 +423,20 @@ declare class SqlDatabaseManager {
      * @param key - Database key to set as default
      */
     setDefaultDatabase(tenantId: string, key: string): void;
+    /**
+     * Set the configuration store for on-demand database loading
+     * @param store - The database configuration store
+     */
+    setConfigStore(store: _axiom_lattice_protocols.DatabaseConfigStore): void;
     /**
      * Get a database by key for a specific tenant
+     * If database is not registered and configStore is set, will try to load from store
      * @param tenantId - Tenant identifier (required)
      * @param key - Database key (optional, uses default if not provided)
      * @returns ISqlDatabase instance
      * @throws Error if tenant or database not found
      */
-    getDatabase(tenantId: string, key?: string): ISqlDatabase;
+    getDatabase(tenantId: string, key?: string): Promise<ISqlDatabase>;
     /**
      * Check if a database is registered for a tenant
      * @param tenantId - Tenant identifier

package/dist/index.d.ts

@@ -400,6 +400,7 @@ declare class SqlDatabaseManager {
     private static instance;
     private databases;
     private defaultDatabaseKeys;
+    private configStore;
     private constructor();
     /**
      * Get the singleton instance
@@ -422,14 +423,20 @@ declare class SqlDatabaseManager {
      * @param key - Database key to set as default
      */
     setDefaultDatabase(tenantId: string, key: string): void;
+    /**
+     * Set the configuration store for on-demand database loading
+     * @param store - The database configuration store
+     */
+    setConfigStore(store: _axiom_lattice_protocols.DatabaseConfigStore): void;
     /**
      * Get a database by key for a specific tenant
+     * If database is not registered and configStore is set, will try to load from store
      * @param tenantId - Tenant identifier (required)
      * @param key - Database key (optional, uses default if not provided)
      * @returns ISqlDatabase instance
      * @throws Error if tenant or database not found
      */
-    getDatabase(tenantId: string, key?: string): ISqlDatabase;
+    getDatabase(tenantId: string, key?: string): Promise<ISqlDatabase>;
     /**
      * Check if a database is registered for a tenant
      * @param tenantId - Tenant identifier

package/dist/index.js

@@ -1081,6 +1081,7 @@ var SqlDatabaseManager = class _SqlDatabaseManager {
   constructor() {
     this.databases = /* @__PURE__ */ new Map();
     this.defaultDatabaseKeys = /* @__PURE__ */ new Map();
+    this.configStore = null;
   }
   /**
    * Get the singleton instance
@@ -1139,27 +1140,53 @@ var SqlDatabaseManager = class _SqlDatabaseManager {
     }
     this.defaultDatabaseKeys.set(tenantId, key);
   }
+  /**
+   * Set the configuration store for on-demand database loading
+   * @param store - The database configuration store
+   */
+  setConfigStore(store) {
+    this.configStore = store;
+  }
   /**
    * Get a database by key for a specific tenant
+   * If database is not registered and configStore is set, will try to load from store
    * @param tenantId - Tenant identifier (required)
    * @param key - Database key (optional, uses default if not provided)
    * @returns ISqlDatabase instance
    * @throws Error if tenant or database not found
    */
-  getDatabase(tenantId, key) {
+  async getDatabase(tenantId, key) {
     const tenantDbs = this.databases.get(tenantId);
-    if (!tenantDbs) {
-      throw new Error(`No databases registered for tenant '${tenantId}'`);
-    }
     const dbKey = key || this.defaultDatabaseKeys.get(tenantId);
     if (!dbKey) {
       throw new Error(`No default database set for tenant '${tenantId}'`);
     }
-    const database = tenantDbs.get(dbKey);
-    if (!database) {
-      throw new Error(`Database '${dbKey}' not found for tenant '${tenantId}'`);
+    if (tenantDbs) {
+      const database = tenantDbs.get(dbKey);
+      if (database) {
+        return database;
+      }
+    }
+    if (this.configStore) {
+      const configEntry = await this.configStore.getConfigByKey(tenantId, dbKey);
+      if (configEntry) {
+        this.registerDatabase(tenantId, dbKey, configEntry.config);
+        if (!this.defaultDatabaseKeys.has(tenantId)) {
+          this.defaultDatabaseKeys.set(tenantId, dbKey);
+        }
+        const newTenantDbs = this.databases.get(tenantId);
+        if (newTenantDbs) {
+          const database = newTenantDbs.get(dbKey);
+          if (database) {
+            return database;
+          }
+        }
+      }
+    }
+    if (!tenantDbs) {
+      throw new Error(`No databases registered for tenant '${tenantId}'`);
     }
-    return database;
+    throw new Error(`Database '${dbKey}' not found for tenant '${tenantId}'`);
   }
   /**
    * Check if a database is registered for a tenant
@@ -1298,7 +1325,7 @@ ${databaseKeys.map(
         if (!databaseKeys.includes(databaseKey)) {
           return `Error: databaseKey "${databaseKey}" is not in the allowed list: [${databaseKeys.join(", ")}]`;
         }
-        const database = sqlDatabaseManager.getDatabase(databaseKey, tenantId);
+        const database = await sqlDatabaseManager.getDatabase(tenantId, databaseKey);
         const tables = await database.listTables();
         if (tables.length === 0) {
           return "No tables found in the database.";
@@ -1345,7 +1372,7 @@ ${databaseKeys.map(
         if (!databaseKeys.includes(databaseKey)) {
           return `Error: databaseKey "${databaseKey}" is not in the allowed list: [${databaseKeys.join(", ")}]`;
         }
-        const database = sqlDatabaseManager.getDatabase(databaseKey, tenantId);
+        const database = await sqlDatabaseManager.getDatabase(tenantId, databaseKey);
         const tableNames = tables.split(",").map((t) => t.trim()).filter((t) => t.length > 0);
         if (tableNames.length === 0) {
           return "Error: No table names provided. Please provide a comma-separated list of table names.";
@@ -1502,7 +1529,7 @@ ${trimmedQuery}
         }
         if (databaseKey && databaseKeys.includes(databaseKey)) {
           try {
-            const database = sqlDatabaseManager.getDatabase(databaseKey, tenantId);
+            const database = await sqlDatabaseManager.getDatabase(tenantId, databaseKey);
             const dbType = database.getDatabaseType();
             if (dbType === "postgres") {
               try {
@@ -1603,7 +1630,7 @@ ${databaseKeys.map(
         if (!trimmedQuery) {
           return "Error: Empty query provided. Please provide a valid SQL query.";
         }
-        const database = sqlDatabaseManager.getDatabase(databaseKey, tenantId);
+        const database = await sqlDatabaseManager.getDatabase(tenantId, databaseKey);
         const result = await database.executeQuery(trimmedQuery);
         return formatQueryResult(result.rows, result.fields);
       } catch (error) {
@@ -5064,11 +5091,20 @@ var SandboxFilesystem = class {
 
 // src/middlewares/codeEvalMiddleware.ts
 var import_langchain37 = require("langchain");
+
+// src/middlewares/contextSchema.ts
+var import_zod41 = __toESM(require("zod"));
+var contextSchema = import_zod41.default.object({
+  runConfig: import_zod41.default.any()
+});
+
+// src/middlewares/codeEvalMiddleware.ts
 function createCodeEvalMiddleware(params = { isolatedLevel: "global" }) {
   const codeEvalTool = createCodeEvalTool({ isolatedLevel: params.isolatedLevel });
   const codeExecuteFileTool = createCodeExecuteFileTool({ isolatedLevel: params.isolatedLevel });
   return (0, import_langchain37.createMiddleware)({
     name: "codeEvalMiddleware",
+    contextSchema,
     tools: [codeEvalTool, codeExecuteFileTool, getToolClient("convert_to_markdown")]
   });
 }
@@ -5103,6 +5139,7 @@ function createBrowserMiddleware(params = { isolatedLevel: "global" }) {
   ];
   return (0, import_langchain38.createMiddleware)({
     name: "browserMiddleware",
+    contextSchema,
     tools
   });
 }
@@ -5123,6 +5160,7 @@ function createSqlMiddleware(params) {
   };
   return (0, import_langchain39.createMiddleware)({
     name: "sqlMiddleware",
+    contextSchema,
     tools: [
       createListTablesSqlTool(toolParams),
       createInfoSqlTool(toolParams),
@@ -7142,11 +7180,11 @@ ${body}` : `${frontmatter}
 };
 
 // src/tool_lattice/skill/load_skills.ts
-var import_zod41 = __toESM(require("zod"));
+var import_zod42 = __toESM(require("zod"));
 var import_langchain40 = require("langchain");
 
 // src/tool_lattice/skill/load_skill_content.ts
-var import_zod42 = __toESM(require("zod"));
+var import_zod43 = __toESM(require("zod"));
 var import_langchain41 = require("langchain");
 var LOAD_SKILL_CONTENT_DESCRIPTION = `
 Execute a skill within the main conversation
@@ -7233,15 +7271,15 @@ ${content}`;
     {
       name: "skill",
       description: LOAD_SKILL_CONTENT_DESCRIPTION,
-      schema: import_zod42.default.object({
-        skill_name: import_zod42.default.string().describe("The name of the skill to load")
+      schema: import_zod43.default.object({
+        skill_name: import_zod43.default.string().describe("The name of the skill to load")
       })
     }
   );
 };
 
 // src/tool_lattice/skill/load_skill_resource.ts
-var import_zod43 = __toESM(require("zod"));
+var import_zod44 = __toESM(require("zod"));
 var import_langchain42 = require("langchain");
 var LOAD_SKILL_RESOURCE_DESCRIPTION = `Load a specific resource file from a skill's resources directory. Use this tool when you need to access template files, example data, or other resources bundled with a skill. The resource paths are listed in the skill content when using the load_skill_content tool.`;
 var createLoadSkillResourceTool = () => {
@@ -7267,9 +7305,9 @@ var createLoadSkillResourceTool = () => {
     {
       name: "load_skill_resource",
       description: LOAD_SKILL_RESOURCE_DESCRIPTION,
-      schema: import_zod43.default.object({
-        skill_name: import_zod43.default.string().describe("The name of the skill containing the resource"),
-        resource_path: import_zod43.default.string().describe("The path to the resource relative to the skill's resources/ directory")
+      schema: import_zod44.default.object({
+        skill_name: import_zod44.default.string().describe("The name of the skill containing the resource"),
+        resource_path: import_zod44.default.string().describe("The path to the resource relative to the skill's resources/ directory")
       })
     }
   );
@@ -7284,6 +7322,7 @@ function createSkillMiddleware(params = {}) {
   let latestSkills = [];
   return (0, import_langchain43.createMiddleware)({
     name: "skillMiddleware",
+    contextSchema,
     tools: [
       // createLoadSkillsTool({ skills: readAll ? undefined : skills }),
       createLoadSkillContentTool(),
@@ -7291,25 +7330,28 @@ function createSkillMiddleware(params = {}) {
     ],
     beforeAgent: async (state, runtime) => {
       try {
+        console.log("latestSkills_before", latestSkills);
         const storeLattice = getStoreLattice("default", "skill");
         const skillStore = storeLattice?.store;
-        const tenantId = runtime?.context?.tenantId || state?.tenantId || "default";
+        const tenantId = runtime?.context?.runConfig?.tenantId;
+        if (!tenantId) {
+          throw new Error("tenantId is missing");
+        }
         if (readAll) {
-          latestSkills = await skillStore.getAllSkills(tenantId);
+          const allSkills = await skillStore.getAllSkills(tenantId);
+          latestSkills = allSkills.filter((skill) => skill !== void 0 && skill !== null);
         } else if (skills && skills.length > 0) {
-          const skillLatticePromises = skills.map(
-            (skillId) => skillStore.getSkillById(tenantId, skillId)
-          );
-          const skillLattices = await Promise.all(skillLatticePromises);
-          latestSkills = skillLattices.filter((skill) => skill !== void 0);
+          const allSkills = await skillStore.getAllSkills(tenantId);
+          latestSkills = allSkills.filter((skill) => skills.includes(skill.id));
         }
       } catch (error) {
         console.error("Error fetching skills:", error);
       }
     },
     wrapModelCall: (request, handler) => {
-      const skillsPrompt = latestSkills.map((skill) => `## ${skill.name}
-${skill.description}`).join("\n");
+      console.log("latestSkills_after", latestSkills);
+      const skillsPrompt = latestSkills.filter((skill) => skill !== void 0 && skill !== null && !!skill.name).map((skill) => `## ${skill.name}
+${skill.description || ""}`).join("\n");
       const skillsAddendum = `
 
 <available_skills>
@@ -7331,7 +7373,7 @@ ${skillsPrompt}
 var import_langchain44 = require("langchain");
 var import_langgraph4 = require("@langchain/langgraph");
 var import_v3 = require("zod/v3");
-var import_zod44 = require("@langchain/langgraph/zod");
+var import_zod45 = require("@langchain/langgraph/zod");
 
 // src/deep_agent_new/backends/utils.ts
 var import_micromatch = __toESM(require("micromatch"));
@@ -7795,7 +7837,7 @@ function fileDataReducer(left, right) {
   return result;
 }
 var FilesystemStateSchema = import_v3.z.object({
-  files: (0, import_zod44.withLangGraph)(
+  files: (0, import_zod45.withLangGraph)(
     import_v3.z.record(import_v3.z.string(), FileDataSchema).default({}),
     {
       reducer: {
@@ -8071,6 +8113,7 @@ function createFilesystemMiddleware(options = {}) {
   ];
   return (0, import_langchain44.createMiddleware)({
     name: "FilesystemMiddleware",
+    contextSchema,
     stateSchema: FilesystemStateSchema,
     tools,
     wrapModelCall: systemPrompt ? async (request, handler) => {
@@ -8201,6 +8244,7 @@ function createMetricsMiddleware(params) {
   };
   return (0, import_langchain45.createMiddleware)({
     name: "metricsMiddleware",
+    contextSchema,
     tools: [
       createListMetricsDataSourcesTool(toolParams),
       createQueryMetricsListTool(toolParams),
@@ -8219,16 +8263,16 @@ var import_langgraph5 = require("@langchain/langgraph");
 
 // src/tool_lattice/ask_user_to_clarify/index.ts
 var import_langchain46 = require("langchain");
-var import_zod45 = __toESM(require("zod"));
-var questionSchema = import_zod45.default.object({
-  question: import_zod45.default.string().describe("The question text to ask the user"),
-  options: import_zod45.default.array(import_zod45.default.string()).describe("List of EXACT, selectable values. Maximum 3 options allowed. DO NOT include any 'placeholder' options that require the user to type (e.g., do NOT add 'Enter manual value'). If manual input is needed, set allowOther to true instead."),
-  type: import_zod45.default.enum(["single", "multiple"]).describe("Whether the question allows single or multiple selections"),
-  required: import_zod45.default.boolean().optional().default(false).describe("Whether this question must be answered"),
-  allowOther: import_zod45.default.boolean().optional().default(true).describe("Set to true to append an 'Other' option that opens a free-text input field. Use this for open-ended answers or when the 3 options cannot cover all possibilities.")
+var import_zod46 = __toESM(require("zod"));
+var questionSchema = import_zod46.default.object({
+  question: import_zod46.default.string().describe("The question text to ask the user"),
+  options: import_zod46.default.array(import_zod46.default.string()).describe("List of EXACT, selectable values. Maximum 3 options allowed. DO NOT include any 'placeholder' options that require the user to type (e.g., do NOT add 'Enter manual value'). If manual input is needed, set allowOther to true instead."),
+  type: import_zod46.default.enum(["single", "multiple"]).describe("Whether the question allows single or multiple selections"),
+  required: import_zod46.default.boolean().optional().default(false).describe("Whether this question must be answered"),
+  allowOther: import_zod46.default.boolean().optional().default(true).describe("Set to true to append an 'Other' option that opens a free-text input field. Use this for open-ended answers or when the 3 options cannot cover all possibilities.")
 });
-var inputSchema = import_zod45.default.object({
-  questions: import_zod45.default.array(questionSchema).min(1, "At least one question is required").describe("A structured sequence of clarification questions. Use these to gather missing parameters or disambiguate user intent before proceeding.")
+var inputSchema = import_zod46.default.object({
+  questions: import_zod46.default.array(questionSchema).min(1, "At least one question is required").describe("A structured sequence of clarification questions. Use these to gather missing parameters or disambiguate user intent before proceeding.")
 });
 function createAskUserToClarifyTool() {
   return (0, import_langchain46.tool)(
@@ -10355,7 +10399,7 @@ var MemoryBackend = class {
 
 // src/deep_agent_new/middleware/todos.ts
 var import_langgraph9 = require("@langchain/langgraph");
-var import_zod46 = require("zod");
+var import_zod47 = require("zod");
 var import_langchain51 = require("langchain");
 var WRITE_TODOS_DESCRIPTION = `Use this tool to create and manage a structured task list for your current work session. This helps you track progress, organize complex tasks, and demonstrate thoroughness to the user.
 It also helps the user understand the progress of the task and overall progress of their requests.
@@ -10583,12 +10627,12 @@ Writing todos takes time and tokens, use it when it is helpful for managing comp
 ## Important To-Do List Usage Notes to Remember
 - The \`write_todos\` tool should never be called multiple times in parallel.
 - Don't be afraid to revise the To-Do list as you go. New information may reveal new tasks that need to be done, or old tasks that are irrelevant.`;
-var TodoStatus = import_zod46.z.enum(["pending", "in_progress", "completed"]).describe("Status of the todo");
-var TodoSchema = import_zod46.z.object({
-  content: import_zod46.z.string().describe("Content of the todo item"),
+var TodoStatus = import_zod47.z.enum(["pending", "in_progress", "completed"]).describe("Status of the todo");
+var TodoSchema = import_zod47.z.object({
+  content: import_zod47.z.string().describe("Content of the todo item"),
   status: TodoStatus
 });
-var stateSchema = import_zod46.z.object({ todos: import_zod46.z.array(TodoSchema).default([]) });
+var stateSchema = import_zod47.z.object({ todos: import_zod47.z.array(TodoSchema).default([]) });
 function todoListMiddleware(options) {
   const writeTodos = (0, import_langchain51.tool)(
     ({ todos }, config) => {
@@ -10607,8 +10651,8 @@ function todoListMiddleware(options) {
     {
       name: "write_todos",
       description: options?.toolDescription ?? WRITE_TODOS_DESCRIPTION,
-      schema: import_zod46.z.object({
-        todos: import_zod46.z.array(TodoSchema).describe("List of todo items to update")
+      schema: import_zod47.z.object({
+        todos: import_zod47.z.array(TodoSchema).describe("List of todo items to update")
       })
     }
   );
@@ -10635,7 +10679,7 @@ function createDeepAgent(params = {}) {
     middleware: customMiddleware = [],
     subagents = [],
     responseFormat,
-    contextSchema,
+    contextSchema: contextSchema2,
     checkpointer,
     store,
     backend,
@@ -10704,7 +10748,7 @@ ${BASE_PROMPT}` : BASE_PROMPT;
     tools,
     middleware,
     responseFormat,
-    contextSchema,
+    contextSchema: contextSchema2,
     checkpointer,
     store,
     name