@axiom-lattice/core 2.1.27 → 2.1.28

package/dist/index.mjs

@@ -1911,6 +1911,20 @@ ${serverKeys.map(
   return tool7(
     async (_input, _exeConfig) => {
       try {
+        const runConfig = _exeConfig?.configurable?.runConfig || {};
+        const metricsDataSource = runConfig.metricsDataSource;
+        if (metricsDataSource) {
+          const { serverKey, datasourceId } = metricsDataSource;
+          return `\u2139\uFE0F **\u5F53\u524D\u5DF2\u9009\u4E2D\u6570\u636E\u6E90**
+
+\u60A8\u5DF2\u901A\u8FC7\u754C\u9762\u9009\u4E2D\u4E86\u7279\u5B9A\u7684\u6570\u636E\u6E90\uFF0C\u6240\u6709 metrics \u67E5\u8BE2\u5C06\u4E13\u6CE8\u4E8E\uFF1A
+- **\u670D\u52A1\u5668**: ${serverKey}
+- **\u6570\u636E\u6E90ID**: ${datasourceId}
+
+\u60A8\u53EF\u4EE5\u76F4\u63A5\u4F7F\u7528\u5176\u4ED6 metrics \u5DE5\u5177\uFF08\u5982 query_metrics_list\u3001query_semantic_metric_data \u7B49\uFF09\uFF0C\u65E0\u9700\u518D\u6307\u5B9A serverKey \u548C datasourceId \u53C2\u6570\u3002
+
+\u5982\u9700\u67E5\u770B\u6240\u6709\u53EF\u7528\u7684\u6570\u636E\u6E90\uFF0C\u8BF7\u6E05\u9664\u5F53\u524D\u9009\u62E9\u6216\u91CD\u65B0\u6253\u5F00\u6570\u636E\u6E90\u9009\u62E9\u5668\u3002`;
+        }
         if (serverKeys.length === 0) {
           return "No metrics servers configured.";
         }
@@ -2006,10 +2020,13 @@ ${serverKeys.map(
   ).join("\n")}` : "";
   return tool8(
     async ({
-      serverKey,
+      serverKey: inputServerKey,
       datasourceIds
     }, _exeConfig) => {
       try {
+        const runConfig = _exeConfig?.configurable?.runConfig || {};
+        const metricsDataSource = runConfig.metricsDataSource;
+        const serverKey = metricsDataSource?.serverKey || inputServerKey;
         if (!serverKey) {
           return "Error: serverKey parameter is required. Available servers: " + serverKeys.join(", ");
         }
@@ -2021,7 +2038,7 @@ ${serverKeys.map(
           return `Error: Server "${serverKey}" is not a semantic metrics server. This tool only works with semantic servers.`;
         }
         const client = metricsServerManager.getClient(serverKey);
-        const targetDatasourceIds = datasourceIds && datasourceIds.length > 0 ? datasourceIds : client.getSelectedDataSources();
+        const targetDatasourceIds = datasourceIds && datasourceIds.length > 0 ? datasourceIds : metricsDataSource?.datasourceId ? [metricsDataSource.datasourceId] : client.getSelectedDataSources();
         if (targetDatasourceIds.length === 0) {
           return `Error: No data sources specified and no default data sources configured for server "${serverKey}".`;
         }
@@ -2069,7 +2086,7 @@ ${serverKeys.map(
         for (const dsId of sortedDatasourceIds) {
           const metrics = metricsByDatasource.get(dsId);
           const sortedMetrics = metrics.sort((a, b) => {
-            if (a.domain !== b.domain) {
+            if (a.domain && a.domain !== b.domain) {
               return a.domain.localeCompare(b.domain);
             }
             return a.metricName.localeCompare(b.metricName);
@@ -2096,8 +2113,8 @@ ${serverKeys.map(
       name: "query_metrics_list",
       description: `${QUERY_METRICS_LIST_DESCRIPTION}${availableServersText}`,
       schema: z9.object({
-        serverKey: z9.string().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}`),
-        datasourceIds: z9.array(z9.string()).optional().describe("Optional array of datasource IDs to query. If not provided, uses all selected datasources.")
+        serverKey: z9.string().optional().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}. Optional if configured in runConfig.metricsDataSource`),
+        datasourceIds: z9.array(z9.string()).optional().describe("Optional array of datasource IDs to query. If not provided, uses all selected datasources or the one configured in runConfig.metricsDataSource.")
       })
     }
   );
@@ -2154,11 +2171,15 @@ ${serverKeys.map(
   ).join("\n")}` : "";
   return tool9(
     async ({
-      serverKey,
+      serverKey: inputServerKey,
       metricName,
-      datasourceId
+      datasourceId: inputDatasourceId
     }, _exeConfig) => {
       try {
+        const runConfig = _exeConfig?.configurable?.runConfig || {};
+        const metricsDataSource = runConfig.metricsDataSource;
+        const serverKey = metricsDataSource?.serverKey || inputServerKey;
+        const datasourceId = inputDatasourceId || metricsDataSource?.datasourceId;
         if (!serverKey) {
           return "Error: serverKey parameter is required. Available servers: " + serverKeys.join(", ");
         }
@@ -2217,12 +2238,20 @@ ${serverKeys.map(
         lines.push(foundDetail.description);
         lines.push("");
         if (foundDetail.defaultTimeContext) {
+          const dtc = foundDetail.defaultTimeContext;
           lines.push(`## \u65F6\u95F4\u4E0A\u4E0B\u6587`);
           lines.push("");
-          lines.push(`- **\u65F6\u95F4\u7EF4\u5EA6**: ${foundDetail.defaultTimeContext.timeDimension} (${foundDetail.defaultTimeContext.label})`);
-          lines.push(`- **\u9ED8\u8BA4\u7C92\u5EA6**: ${foundDetail.defaultTimeContext.granularity}`);
-          lines.push(`- **\u9ED8\u8BA4\u7A97\u53E3**: ${foundDetail.defaultTimeContext.window}`);
-          lines.push(`- **\u652F\u6301\u7684\u7C92\u5EA6**: ${foundDetail.defaultTimeContext.supportedGrains.join("\u3001")}`);
+          if (dtc.timeDimension) {
+            lines.push(`- **\u65F6\u95F4\u7EF4\u5EA6**: ${dtc.timeDimension}${dtc.label ? ` (${dtc.label})` : ""}`);
+          }
+          if (dtc.granularity) {
+            lines.push(`- **\u9ED8\u8BA4\u7C92\u5EA6**: ${dtc.granularity}`);
+          }
+          if (dtc.window) {
+            lines.push(`- **\u9ED8\u8BA4\u7A97\u53E3**: ${dtc.window}`);
+          }
+          const grains = dtc.supportedGrains;
+          lines.push(`- **\u652F\u6301\u7684\u7C92\u5EA6**: ${Array.isArray(grains) && grains.length > 0 ? grains.join("\u3001") : "\u672A\u914D\u7F6E"}`);
           lines.push("");
         }
         if (foundDetail.supportedDimensions && foundDetail.supportedDimensions.length > 0) {
@@ -2286,10 +2315,12 @@ ${serverKeys.map(
           const aiContext = foundDetail.aiAgentContext;
           lines.push(`## AI \u5206\u6790\u4E0A\u4E0B\u6587`);
           lines.push("");
-          lines.push(`### \u6307\u6807\u6781\u6027`);
-          lines.push("");
-          lines.push(aiContext.polarity === "positive" ? "\u6B63\u5411\u6307\u6807\uFF08\u8D8A\u9AD8\u8D8A\u597D\uFF09" : "\u8D1F\u5411\u6307\u6807\uFF08\u8D8A\u4F4E\u8D8A\u597D\uFF09");
-          lines.push("");
+          if (aiContext.polarity) {
+            lines.push(`### \u6307\u6807\u6781\u6027`);
+            lines.push("");
+            lines.push(aiContext.polarity === "positive" ? "\u6B63\u5411\u6307\u6807\uFF08\u8D8A\u9AD8\u8D8A\u597D\uFF09" : "\u8D1F\u5411\u6307\u6807\uFF08\u8D8A\u4F4E\u8D8A\u597D\uFF09");
+            lines.push("");
+          }
           if (aiContext.synonyms && aiContext.synonyms.length > 0) {
             lines.push(`### \u540C\u4E49\u8BCD/\u522B\u540D`);
             lines.push("");
@@ -2312,7 +2343,9 @@ ${serverKeys.map(
             if (aiContext.diagnosticWorkflow.trigger && aiContext.diagnosticWorkflow.trigger.any_of) {
               lines.push(`**\u89E6\u53D1\u6761\u4EF6** (\u6EE1\u8DB3\u4EFB\u4E00):`);
               for (const trigger of aiContext.diagnosticWorkflow.trigger.any_of) {
-                lines.push(`- ${trigger.metric} ${trigger.operator} ${trigger.value}`);
+                if (trigger.metric && trigger.operator) {
+                  lines.push(`- ${trigger.metric} ${trigger.operator} ${trigger.value ?? "N/A"}`);
+                }
               }
               lines.push("");
             }
@@ -2320,14 +2353,17 @@ ${serverKeys.map(
               lines.push(`**\u5206\u6790\u52A8\u4F5C**:`);
               for (let i = 0; i < aiContext.diagnosticWorkflow.actions.length; i++) {
                 const action = aiContext.diagnosticWorkflow.actions[i];
-                lines.push(`${i + 1}. **${action.type}**`);
+                if (!action) continue;
+                lines.push(`${i + 1}. **${action.type ?? "\u672A\u547D\u540D\u52A8\u4F5C"}**`);
                 if (action.metric) {
                   lines.push(`   - \u5BF9\u6BD4\u6307\u6807: ${action.metric}`);
                 }
                 if (action.dimensions && action.dimensions.length > 0) {
                   lines.push(`   - \u4E0B\u94BB\u7EF4\u5EA6: ${action.dimensions.join("\u3001")}`);
                 }
-                lines.push(`   - \u76EE\u7684: ${action.intent}`);
+                if (action.intent) {
+                  lines.push(`   - \u76EE\u7684: ${action.intent}`);
+                }
               }
               lines.push("");
             }
@@ -2361,9 +2397,9 @@ ${serverKeys.map(
       name: "query_metric_definition",
       description: `${QUERY_METRIC_DEFINITION_DESCRIPTION}${availableServersText}`,
       schema: z10.object({
-        serverKey: z10.string().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}`),
+        serverKey: z10.string().optional().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}. Optional if configured in runConfig.metricsDataSource`),
         metricName: z10.string().describe("The name of the metric to get definition for."),
-        datasourceId: z10.string().optional().describe("Optional specific datasource ID to search in. If not provided, searches all selected datasources.")
+        datasourceId: z10.string().optional().describe("Optional specific datasource ID to search in. If not provided, uses the one configured in runConfig.metricsDataSource or searches all selected datasources.")
       })
     }
   );
@@ -2555,14 +2591,18 @@ ${serverKeys.map(
   ).join("\n")}` : "";
   return tool10(
     async ({
-      serverKey,
-      datasourceId,
+      serverKey: inputServerKey,
+      datasourceId: inputDatasourceId,
       metrics,
       groupBy,
       filters,
       limit
     }, _exeConfig) => {
       try {
+        const runConfig = _exeConfig?.configurable?.runConfig || {};
+        const metricsDataSource = runConfig.metricsDataSource;
+        const serverKey = metricsDataSource?.serverKey || inputServerKey;
+        const datasourceId = metricsDataSource?.datasourceId || inputDatasourceId;
         if (!serverKey) {
           return "Error: serverKey parameter is required. Available servers: " + serverKeys.join(", ");
         }
@@ -2585,6 +2625,7 @@ ${serverKeys.map(
           operator: f.operator,
           values: f.values
         }));
+        console.log(`[query_semantic_metric_data] Querying: server=${serverKey}, datasource=${datasourceId}, metrics=[${metrics.join(", ")}], groupBy=[${groupBy?.join(", ") || ""}], filters=${JSON.stringify(filters)}, limit=${limit || 1e3}`);
         const result = await client.semanticQuery({
           datasourceId,
           metrics,
@@ -2605,8 +2646,8 @@ ${serverKeys.map(
       name: "query_semantic_metric_data",
       description: `${QUERY_SEMANTIC_METRIC_DATA_DESCRIPTION}${availableServersText}`,
       schema: z11.object({
-        serverKey: z11.string().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}`),
-        datasourceId: z11.string().describe("The data source ID to query metrics from."),
+        serverKey: z11.string().optional().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}. Optional if configured in runConfig.metricsDataSource`),
+        datasourceId: z11.string().optional().describe("The data source ID to query metrics from. Optional if configured in runConfig.metricsDataSource"),
         metrics: z11.array(z11.string()).describe("Array of metric names to query (e.g., ['net_sales_amt', 'gross_profit'])."),
         groupBy: z11.array(z11.string()).optional().describe("Optional array of dimensions to group by (e.g., ['DocDate__month', 'CustomerCode'])."),
         filters: z11.array(z11.object({
@@ -2633,10 +2674,13 @@ ${serverKeys.map(
   ).join("\n")}` : "";
   return tool11(
     async ({
-      serverKey,
+      serverKey: inputServerKey,
       datasourceIds
     }, _exeConfig) => {
       try {
+        const runConfig = _exeConfig?.configurable?.runConfig || {};
+        const metricsDataSource = runConfig.metricsDataSource;
+        const serverKey = metricsDataSource?.serverKey || inputServerKey;
         if (!serverKey) {
           return "Error: serverKey parameter is required. Available servers: " + serverKeys.join(", ");
         }
@@ -2648,7 +2692,7 @@ ${serverKeys.map(
           return `Error: Server "${serverKey}" is not a semantic metrics server. This tool only works with semantic servers.`;
         }
         const client = metricsServerManager.getClient(serverKey);
-        const targetDatasourceIds = datasourceIds && datasourceIds.length > 0 ? datasourceIds : client.getSelectedDataSources();
+        const targetDatasourceIds = datasourceIds && datasourceIds.length > 0 ? datasourceIds : metricsDataSource?.datasourceId ? [metricsDataSource.datasourceId] : client.getSelectedDataSources();
         if (targetDatasourceIds.length === 0) {
           return `Error: No data sources specified and no default data sources configured for server "${serverKey}".`;
         }
@@ -2708,8 +2752,8 @@ ${serverKeys.map(
       name: "query_tables_list",
       description: `${QUERY_TABLES_LIST_DESCRIPTION}${availableServersText}`,
       schema: z12.object({
-        serverKey: z12.string().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}`),
-        datasourceIds: z12.array(z12.string()).optional().describe("Optional array of datasource IDs to query. If not provided, uses all selected datasources.")
+        serverKey: z12.string().optional().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}. Optional if configured in runConfig.metricsDataSource`),
+        datasourceIds: z12.array(z12.string()).optional().describe("Optional array of datasource IDs to query. If not provided, uses the one configured in runConfig.metricsDataSource or all selected datasources.")
       })
     }
   );
@@ -2728,11 +2772,15 @@ ${serverKeys.map(
   ).join("\n")}` : "";
   return tool12(
     async ({
-      serverKey,
+      serverKey: inputServerKey,
       tableName,
-      datasourceId
+      datasourceId: inputDatasourceId
     }, _exeConfig) => {
       try {
+        const runConfig = _exeConfig?.configurable?.runConfig || {};
+        const metricsDataSource = runConfig.metricsDataSource;
+        const serverKey = metricsDataSource?.serverKey || inputServerKey;
+        const datasourceId = inputDatasourceId || metricsDataSource?.datasourceId;
         if (!serverKey) {
           return "Error: serverKey parameter is required. Available servers: " + serverKeys.join(", ");
         }
@@ -2820,9 +2868,9 @@ ${serverKeys.map(
       name: "query_table_definition",
       description: `${QUERY_TABLE_DEFINITION_DESCRIPTION}${availableServersText}`,
       schema: z13.object({
-        serverKey: z13.string().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}`),
+        serverKey: z13.string().optional().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}. Optional if configured in runConfig.metricsDataSource`),
         tableName: z13.string().describe("The name of the table to get definition for."),
-        datasourceId: z13.string().optional().describe("Optional specific datasource ID to search in. If not provided, searches all selected datasources.")
+        datasourceId: z13.string().optional().describe("Optional specific datasource ID to search in. If not provided, uses the one configured in runConfig.metricsDataSource or searches all selected datasources.")
       })
     }
   );
@@ -2876,13 +2924,17 @@ ${serverKeys.map(
   ).join("\n")}` : "";
   return tool13(
     async ({
-      serverKey,
-      datasourceId,
+      serverKey: inputServerKey,
+      datasourceId: inputDatasourceId,
       customSql,
       params,
       limit
     }, _exeConfig) => {
       try {
+        const runConfig = _exeConfig?.configurable?.runConfig || {};
+        const metricsDataSource = runConfig.metricsDataSource;
+        const serverKey = metricsDataSource?.serverKey || inputServerKey;
+        const datasourceId = metricsDataSource?.datasourceId || inputDatasourceId;
         if (!serverKey) {
           return "Error: serverKey parameter is required. Available servers: " + serverKeys.join(", ");
         }
@@ -2915,8 +2967,8 @@ ${serverKeys.map(
       name: "execute_sql_query",
       description: `${EXECUTE_SQL_QUERY_DESCRIPTION}${availableServersText}`,
       schema: z14.object({
-        serverKey: z14.string().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}`),
-        datasourceId: z14.string().describe("The data source ID to execute SQL against."),
+        serverKey: z14.string().optional().describe(`Target semantic metrics server. Choose from: ${serverKeys.join(", ")}. Optional if configured in runConfig.metricsDataSource`),
+        datasourceId: z14.string().optional().describe("The data source ID to execute SQL against. Optional if configured in runConfig.metricsDataSource"),
         customSql: z14.string().describe("Custom SQL query string with named parameters (e.g., :year, :category). Use double quotes for identifiers."),
         params: z14.record(z14.union([z14.string(), z14.number(), z14.boolean()])).optional().describe("Optional parameters for the SQL query. Keys should match the :paramName placeholders in customSql."),
         limit: z14.number().optional().describe("Maximum number of results to return (default: 1000).")
@@ -7549,6 +7601,11 @@ var AgentWorkerState = Annotation.Root({
     reducer: (x, y) => y ?? x,
     default: () => void 0
   }),
+  // RunConfig for passing configuration through the graph
+  runConfig: Annotation({
+    reducer: (x, y) => y ?? x,
+    default: () => void 0
+  }),
   // Execution result
   result: Annotation({
     reducer: (x, y) => y ?? x,
@@ -7566,12 +7623,13 @@ var AgentWorkerState = Annotation.Root({
   })
 });
 async function executeNode(state) {
-  const { assistant_id, thread_id, input, command2 } = state;
+  const { assistant_id, thread_id, input, command2, runConfig } = state;
   const callParams = {
     assistant_id,
     thread_id,
     input,
-    command: command2
+    command: command2,
+    runConfig
   };
   const result = await AgentManager.getInstance().callAgentInQueue(
     callParams,
@@ -7636,7 +7694,7 @@ var DEFAULT_SUBAGENT_PROMPT = "In order to complete the objective that the user
 var EXCLUDED_STATE_KEYS = ["messages", "todos", "jumpTo"];
 var DEFAULT_GENERAL_PURPOSE_DESCRIPTION = "General-purpose agent for researching complex questions, searching for files and content, and executing multi-step tasks. When you are searching for a keyword or file and are not confident that you will find the right match in the first few tries use this agent to perform the search for you. This agent has access to all tools as the main agent.";
 function getTaskToolDescription(subagentDescriptions) {
-  return `
+  return subagentDescriptions.length > 0 ? `
 Launch an ephemeral subagent to handle complex, multi-step independent tasks with isolated context windows.
 
 Available agent types and the tools they have access to:
@@ -7746,7 +7804,7 @@ Since the user is greeting, use the greeting-responder agent to respond with a f
 </commentary>
 assistant: "I'm going to use the Task tool to launch with the greeting-responder agent"
 </example>
-  `.trim();
+  `.trim() : "No subagents available, you DON'T NEED TO USE THE TASK TOOL, just use the tools directly.";
 }
 var TASK_SYSTEM_PROMPT = `## \`task\` (subagent spawner)
 
@@ -7833,16 +7891,16 @@ function getSubagents(options) {
   }
   for (const agentParams of subagents) {
     subagentDescriptions.push(
-      `- ${agentParams.name}: ${agentParams.description}`
+      `- ${agentParams.key}: ${agentParams.name} - ${agentParams.description}`
     );
     if ("runnable" in agentParams) {
-      agents[agentParams.name] = agentParams.runnable;
+      agents[agentParams.key] = agentParams.runnable;
     } else {
       const middleware = agentParams.middleware ? [...defaultSubagentMiddleware, ...agentParams.middleware] : [...defaultSubagentMiddleware];
       const interruptOn = agentParams.interruptOn || defaultInterruptOn;
       if (interruptOn)
         middleware.push(humanInTheLoopMiddleware({ interruptOn }));
-      agents[agentParams.name] = createAgent2({
+      agents[agentParams.key] = createAgent2({
         model: agentParams.model ?? defaultModel,
         systemPrompt: agentParams.systemPrompt,
         tools: agentParams.tools ?? defaultTools,
@@ -7890,7 +7948,12 @@ function createTaskTool(options) {
         const workerResult = await agentWorkerGraph.invoke({
           assistant_id: subagent_type,
           thread_id: subagent_thread_id,
-          input: { ...subagentState, message: description }
+          input: { ...subagentState, message: description },
+          runConfig: {
+            ...config.configurable?.runConfig,
+            assistant_id: subagent_type,
+            thread_id: subagent_thread_id
+          }
         });
         const result = workerResult.finalState?.values;
         if (!config.toolCall?.id) {
@@ -9509,6 +9572,7 @@ var DeepAgentGraphBuilder = class {
     const subagents = params.subAgents.map((sa) => {
       if (sa.client) {
         return {
+          key: sa.config.key,
           name: sa.config.name,
           description: sa.config.description,
           runnable: sa.client
@@ -9518,6 +9582,7 @@ var DeepAgentGraphBuilder = class {
           config: sa.config
         });
         return {
+          key: sa.config.key,
           name: sa.config.name,
           description: sa.config.description,
           runnable: subagentClient