@axiom-lattice/gateway 2.1.40 → 2.1.41

package/dist/index.mjs

@@ -4,6 +4,8 @@ import cors from "@fastify/cors";
 import multipart from "@fastify/multipart";
 import sensible from "@fastify/sensible";
 import websocket from "@fastify/websocket";
+import staticPlugin from "@fastify/static";
+import path2 from "path";
 
 // src/services/agent_service.ts
 import {
@@ -1841,6 +1843,215 @@ async function getToolConfigs(request, reply) {
   }
 }
 
+// src/controllers/data-query.ts
+import {
+  getStoreLattice as getStoreLattice4,
+  metricsServerManager
+} from "@axiom-lattice/core";
+function getTenantId5(request) {
+  return request.headers["x-tenant-id"] || "default";
+}
+async function executeDataQuery(request, reply) {
+  const tenantId = getTenantId5(request);
+  const body = request.body;
+  try {
+    if (!body.serverKey && !body.datasourceId) {
+      reply.code(400);
+      return {
+        success: false,
+        message: "Either serverKey or datasourceId must be provided"
+      };
+    }
+    const isSemanticQuery = body.metrics && body.metrics.length > 0;
+    const isSqlQuery = body.customSql && body.customSql.trim().length > 0;
+    if (!isSemanticQuery && !isSqlQuery) {
+      reply.code(400);
+      return {
+        success: false,
+        message: "Either metrics (for semantic query) or customSql (for SQL query) must be provided"
+      };
+    }
+    if (isSemanticQuery && isSqlQuery) {
+      reply.code(400);
+      return {
+        success: false,
+        message: "Cannot provide both metrics and customSql. Use one query type only."
+      };
+    }
+    const storeLattice = getStoreLattice4("default", "metrics");
+    const store = storeLattice.store;
+    if (!body.serverKey) {
+      reply.code(400);
+      return {
+        success: false,
+        message: "serverKey is required"
+      };
+    }
+    const config = await store.getConfigByKey(tenantId, body.serverKey);
+    if (!config) {
+      reply.code(404);
+      return {
+        success: false,
+        message: `Metrics server configuration not found: ${body.serverKey}`
+      };
+    }
+    if (config.config.type !== "semantic") {
+      reply.code(400);
+      return {
+        success: false,
+        message: "This endpoint only supports semantic metrics servers"
+      };
+    }
+    if (!body.datasourceId) {
+      reply.code(400);
+      return {
+        success: false,
+        message: "datasourceId is required"
+      };
+    }
+    if (!metricsServerManager.hasServer(tenantId, body.serverKey)) {
+      reply.code(400);
+      return {
+        success: false,
+        message: `Metrics server not registered: ${body.serverKey}. Please register the server first.`
+      };
+    }
+    const client = metricsServerManager.getClient(tenantId, body.serverKey);
+    if (isSemanticQuery) {
+      return await executeSemanticQuery(client, body, reply);
+    } else {
+      return await executeSqlQuery(client, body, reply);
+    }
+  } catch (error) {
+    console.error("Failed to execute data query:", error);
+    reply.code(500);
+    return {
+      success: false,
+      message: `Failed to execute data query: ${error instanceof Error ? error.message : String(error)}`
+    };
+  }
+}
+async function executeSemanticQuery(client, body, reply) {
+  const semanticFilters = (body.filters || []).map((f) => ({
+    dimension: f.dimension,
+    operator: f.operator,
+    values: f.values
+  }));
+  const result = await client.semanticQuery({
+    datasourceId: body.datasourceId,
+    metrics: body.metrics,
+    groupBy: body.groupBy,
+    filters: semanticFilters,
+    limit: body.limit || 1e3
+  });
+  return {
+    success: true,
+    message: "Semantic query executed successfully",
+    data: {
+      semanticModel: result.semanticModel,
+      columns: result.columns,
+      rows: result.rows,
+      rowsObject: result.rowsObject,
+      metadata: {
+        rowCount: result.rowCount,
+        executionTimeMs: result.executionTimeMs
+      }
+    }
+  };
+}
+async function executeSqlQuery(client, body, reply) {
+  const result = await client.executeSqlQuery({
+    datasourceId: body.datasourceId,
+    customSql: body.customSql,
+    params: body.params,
+    limit: body.limit || 100
+  });
+  return {
+    success: true,
+    message: "SQL query executed successfully",
+    data: {
+      tableName: result.tableName,
+      columns: result.columns.map((name) => ({ name, type: "unknown" })),
+      rows: result.rows,
+      rowsObject: result.rowsObject,
+      executedSql: result.executedSql,
+      metadata: {
+        rowCount: result.rowCount,
+        executionTimeMs: result.executionTimeMs
+      }
+    }
+  };
+}
+
+// src/schemas/data-query.ts
+var dataQuerySchema = {
+  description: "Execute data query (semantic or SQL)",
+  tags: ["Data Query"],
+  summary: "Query Data",
+  body: {
+    type: "object",
+    properties: {
+      serverKey: {
+        type: "string",
+        description: "Target semantic metrics server key (optional if configured in runConfig)"
+      },
+      datasourceId: {
+        type: "string",
+        description: "Data source ID (optional if configured in runConfig)"
+      },
+      // Semantic query parameters
+      metrics: {
+        type: "array",
+        items: { type: "string" },
+        description: "Array of metric names for semantic query"
+      },
+      groupBy: {
+        type: "array",
+        items: { type: "string" },
+        description: "Optional array of dimensions to group by"
+      },
+      filters: {
+        type: "array",
+        items: {
+          type: "object",
+          properties: {
+            dimension: { type: "string" },
+            operator: { type: "string" },
+            values: {
+              type: "array",
+              items: { type: ["string", "number", "boolean"] }
+            }
+          },
+          required: ["dimension", "operator", "values"]
+        },
+        description: "Optional array of filters"
+      },
+      // SQL query parameters
+      customSql: {
+        type: "string",
+        description: "Custom SQL query string with named parameters"
+      },
+      params: {
+        type: "object",
+        additionalProperties: { type: ["string", "number", "boolean"] },
+        description: "Optional parameters for SQL query"
+      },
+      // Common parameters
+      limit: {
+        type: "number",
+        description: "Maximum number of results (default: 1000)"
+      },
+      format: {
+        type: "string",
+        enum: ["echarts", "raw"],
+        description: "Response format (default: echarts)"
+      }
+    }
+  }
+  // Response schema temporarily removed - format not yet finalized
+  // TODO: Add proper response schema once data format is confirmed
+};
+
 // src/schemas/index.ts
 var getAllMemoryItemsSchema = {
   description: "Get all memory items for an assistant thread",
@@ -2302,8 +2513,8 @@ var sandboxService = new SandboxService();
 
 // src/controllers/sandbox.ts
 import { getSandBoxManager as getSandBoxManager2 } from "@axiom-lattice/core";
-function getFilenameFromPath(path2) {
-  const segments = path2.replace(/\/+$/, "").split("/");
+function getFilenameFromPath(path3) {
+  const segments = path3.replace(/\/+$/, "").split("/");
   return segments[segments.length - 1] || "download";
 }
 var EXT_TO_MIME = {
@@ -2355,10 +2566,10 @@ function registerSandboxProxyRoutes(app2) {
         const pathValue = pathEntry && typeof pathEntry === "object" && "value" in pathEntry ? String(pathEntry.value) : typeof pathEntry === "string" ? pathEntry : void 0;
         const formData = new FormData();
         formData.append("file", new Blob([buffer]), data.filename ?? "file");
-        const path2 = `/home/gem/uploads/${pathValue ? pathValue : ""}${data.filename}`;
+        const path3 = `/home/gem/uploads/${pathValue ? pathValue : ""}${data.filename}`;
         const uploadResult = await sandbox.file.uploadFile({
           file: buffer,
-          path: path2
+          path: path3
         });
         if (!uploadResult.ok) {
           return reply.status(502).send({ error: `Upload error: ${uploadResult.error}` });
@@ -2446,14 +2657,14 @@ function registerSandboxProxyRoutes(app2) {
 import * as fs from "fs/promises";
 import * as path from "path";
 import { Readable as Readable2 } from "stream";
-import { getStoreLattice as getStoreLattice4 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice5 } from "@axiom-lattice/core";
 import { SandboxFilesystem, FilesystemBackend } from "@axiom-lattice/core";
 import { getSandBoxManager as getSandBoxManager3 } from "@axiom-lattice/core";
 import { v4 as uuidv4 } from "uuid";
 var WorkspaceController = class {
   constructor() {
-    this.workspaceStore = getStoreLattice4("default", "workspace").store;
-    this.projectStore = getStoreLattice4("default", "project").store;
+    this.workspaceStore = getStoreLattice5("default", "workspace").store;
+    this.projectStore = getStoreLattice5("default", "project").store;
   }
   getTenantId(request) {
     const userTenantId = request.user?.tenantId;
@@ -2731,6 +2942,33 @@ var WorkspaceController = class {
           const webStream = body.stream();
           const nodeStream = Readable2.fromWeb(webStream);
           const contentType2 = body.contentType ?? inferredContentType;
+          console.log(`[viewFile] Sandbox returned stream, contentType: ${contentType2}, filename: ${filename2}`);
+          const isHtml2 = contentType2?.toLowerCase().includes("text/html") || filename2.toLowerCase().endsWith(".html") || filename2.toLowerCase().endsWith(".htm");
+          if (isHtml2) {
+            console.log(`[viewFile] HTML stream detected, collecting for context injection`);
+            const chunks = [];
+            for await (const chunk of nodeStream) {
+              chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
+            }
+            let content2 = Buffer.concat(chunks).toString("utf-8");
+            const contextScript = `<script>window.__AI2APP_CONTEXT__=${JSON.stringify({
+              tenantId,
+              workspaceId,
+              projectId,
+              timestamp: Date.now()
+            })};</script>`;
+            if (content2.toLowerCase().includes("</head>")) {
+              content2 = content2.replace(/<\/head>/i, `${contextScript}</head>`);
+              console.log(`[viewFile] Context script injected before </head> (stream)`);
+            } else if (content2.toLowerCase().includes("<html>")) {
+              content2 = content2.replace(/<html>/i, `<html>${contextScript}`);
+              console.log(`[viewFile] Context script injected after <html> (stream)`);
+            } else {
+              content2 = contextScript + content2;
+              console.log(`[viewFile] Context script prepended to content (stream)`);
+            }
+            return reply.status(200).type(contentType2).header("Content-Disposition", "inline").send(Buffer.from(content2, "utf-8"));
+          }
           return reply.status(200).type(contentType2).header("Content-Disposition", "inline").send(nodeStream);
         }
         const bodyUnknown = downloadResult.body;
@@ -2750,13 +2988,56 @@ var WorkspaceController = class {
         } else {
           return reply.status(502).send({ success: false, error: "Unexpected view response format" });
         }
+        const isHtml = contentType?.toLowerCase().includes("text/html") || filename2.toLowerCase().endsWith(".html") || filename2.toLowerCase().endsWith(".htm");
+        if (isHtml) {
+          console.log(`[viewFile] Injecting AI2APP context for sandbox HTML file: ${filename2}, tenantId: ${tenantId}, contentType: ${contentType}`);
+          let content2 = buf.toString("utf-8");
+          const contextScript = `<script>window.__AI2APP_CONTEXT__=${JSON.stringify({
+            tenantId,
+            workspaceId,
+            projectId,
+            timestamp: Date.now()
+          })};</script>`;
+          if (content2.toLowerCase().includes("</head>")) {
+            content2 = content2.replace(/<\/head>/i, `${contextScript}</head>`);
+            console.log(`[viewFile] Context script injected before </head>`);
+          } else if (content2.toLowerCase().includes("<html>")) {
+            content2 = content2.replace(/<html>/i, `<html>${contextScript}`);
+            console.log(`[viewFile] Context script injected after <html>`);
+          } else {
+            content2 = contextScript + content2;
+            console.log(`[viewFile] Context script prepended to content`);
+          }
+          buf = Buffer.from(content2, "utf-8");
+        }
         return reply.status(200).type(contentType).header("Content-Disposition", "inline").send(buf);
       }
       const { backend } = await this.getBackend(tenantId, workspaceId, projectId);
       const content = await backend.read(resolvedPath, 0, Infinity);
       const filename = this.getFilenameFromPath(resolvedPath);
       const mimeType = this.getMimeType(filename);
-      const buffer = Buffer.from(content, "utf-8");
+      let finalContent = content;
+      const isHtmlFs = mimeType?.toLowerCase().includes("text/html") || filename.toLowerCase().endsWith(".html") || filename.toLowerCase().endsWith(".htm");
+      if (isHtmlFs) {
+        console.log(`[viewFile] Injecting AI2APP context for filesystem HTML file: ${filename}, tenantId: ${tenantId}, mimeType: ${mimeType}`);
+        const contextScript = `<script>window.__AI2APP_CONTEXT__=${JSON.stringify({
+          tenantId,
+          workspaceId,
+          projectId,
+          timestamp: Date.now()
+        })};</script>`;
+        if (content.toLowerCase().includes("</head>")) {
+          finalContent = content.replace(/<\/head>/i, `${contextScript}</head>`);
+          console.log(`[viewFile] Context script injected before </head>`);
+        } else if (content.toLowerCase().includes("</html>")) {
+          finalContent = content.replace(/<\/html>/i, `${contextScript}</html>`);
+          console.log(`[viewFile] Context script injected before </html>`);
+        } else {
+          finalContent = content + contextScript;
+          console.log(`[viewFile] Context script appended to content`);
+        }
+      }
+      const buffer = Buffer.from(finalContent, "utf-8");
       return reply.status(200).type(mimeType).header("Content-Disposition", "inline").send(buffer);
     } catch (error) {
       const message = error instanceof Error ? error.message : String(error);
@@ -2766,17 +3047,17 @@ var WorkspaceController = class {
   async listPath(request) {
     const tenantId = this.getTenantId(request);
     const { workspaceId, projectId } = request.params;
-    const path2 = request.query.path || "/";
+    const path3 = request.query.path || "/";
     const { backend } = await this.getBackend(tenantId, workspaceId, projectId);
-    const files = await backend.lsInfo(path2);
+    const files = await backend.lsInfo(path3);
     return { success: true, data: files };
   }
   async readFile(request) {
     const tenantId = this.getTenantId(request);
     const { workspaceId, projectId } = request.params;
-    const { path: path2, offset = 0, limit = 1e3 } = request.query;
+    const { path: path3, offset = 0, limit = 1e3 } = request.query;
     const { backend } = await this.getBackend(tenantId, workspaceId, projectId);
-    const content = await backend.read(path2, Number(offset), Number(limit));
+    const content = await backend.read(path3, Number(offset), Number(limit));
     return { success: true, data: { content, offset, limit } };
   }
   /**
@@ -2919,11 +3200,11 @@ function registerWorkspaceRoutes(app2) {
 
 // src/controllers/database-configs.ts
 import {
-  getStoreLattice as getStoreLattice5,
+  getStoreLattice as getStoreLattice6,
   sqlDatabaseManager
 } from "@axiom-lattice/core";
 import { randomUUID as randomUUID3 } from "crypto";
-function getTenantId5(request) {
+function getTenantId6(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
     return userTenantId;
@@ -2931,9 +3212,9 @@ function getTenantId5(request) {
   return request.headers["x-tenant-id"] || "default";
 }
 async function getDatabaseConfigList(request, reply) {
-  const tenantId = getTenantId5(request);
+  const tenantId = getTenantId6(request);
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const configs = await store.getAllConfigs(tenantId);
     console.log("Backend: getAllConfigs returned:", configs);
@@ -2961,10 +3242,10 @@ async function getDatabaseConfigList(request, reply) {
   }
 }
 async function getDatabaseConfig(request, reply) {
-  const tenantId = getTenantId5(request);
+  const tenantId = getTenantId6(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -2987,10 +3268,10 @@ async function getDatabaseConfig(request, reply) {
   }
 }
 async function createDatabaseConfig(request, reply) {
-  const tenantId = getTenantId5(request);
+  const tenantId = getTenantId6(request);
   const body = request.body;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, body.key);
     if (existing) {
@@ -3022,11 +3303,11 @@ async function createDatabaseConfig(request, reply) {
   }
 }
 async function updateDatabaseConfig(request, reply) {
-  const tenantId = getTenantId5(request);
+  const tenantId = getTenantId6(request);
   const { key } = request.params;
   const updates = request.body;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, key);
     if (!existing) {
@@ -3064,10 +3345,10 @@ async function updateDatabaseConfig(request, reply) {
   }
 }
 async function deleteDatabaseConfig(request, reply) {
-  const tenantId = getTenantId5(request);
+  const tenantId = getTenantId6(request);
   const { keyOrId } = request.params;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     console.log("Delete request - keyOrId:", keyOrId);
     let config = await store.getConfigByKey(tenantId, keyOrId);
@@ -3113,10 +3394,10 @@ async function deleteDatabaseConfig(request, reply) {
   }
 }
 async function testDatabaseConnection(request, reply) {
-  const tenantId = getTenantId5(request);
+  const tenantId = getTenantId6(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3201,12 +3482,12 @@ function registerDatabaseConfigRoutes(app2) {
 
 // src/controllers/metrics-configs.ts
 import {
-  getStoreLattice as getStoreLattice6,
-  metricsServerManager,
-  SemanticMetricsClient
+  getStoreLattice as getStoreLattice7,
+  metricsServerManager as metricsServerManager2,
+  SemanticMetricsClient as SemanticMetricsClient2
 } from "@axiom-lattice/core";
 import { randomUUID as randomUUID4 } from "crypto";
-function getTenantId6(request) {
+function getTenantId7(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
     return userTenantId;
@@ -3214,9 +3495,9 @@ function getTenantId6(request) {
   return request.headers["x-tenant-id"] || "default";
 }
 async function getMetricsServerConfigList(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const configs = await store.getAllConfigs(tenantId);
     return {
@@ -3240,10 +3521,10 @@ async function getMetricsServerConfigList(request, reply) {
   }
 }
 async function getMetricsServerConfig(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3266,10 +3547,10 @@ async function getMetricsServerConfig(request, reply) {
   }
 }
 async function createMetricsServerConfig(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const body = request.body;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, body.key);
     if (existing) {
@@ -3298,7 +3579,7 @@ async function createMetricsServerConfig(request, reply) {
     };
     const config = await store.createConfig(tenantId, id, configData);
     try {
-      metricsServerManager.registerServer(tenantId, config.key, config.config);
+      metricsServerManager2.registerServer(tenantId, config.key, config.config);
     } catch (error) {
       console.warn("Failed to auto-register metrics server:", error);
     }
@@ -3317,11 +3598,11 @@ async function createMetricsServerConfig(request, reply) {
   }
 }
 async function updateMetricsServerConfig(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   const updates = request.body;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, key);
     if (!existing) {
@@ -3349,7 +3630,7 @@ async function updateMetricsServerConfig(request, reply) {
     }
     if (updates.config) {
       try {
-        metricsServerManager.registerServer(tenantId, updated.key, updated.config);
+        metricsServerManager2.registerServer(tenantId, updated.key, updated.config);
       } catch (error) {
         console.warn("Failed to re-register metrics server:", error);
       }
@@ -3368,10 +3649,10 @@ async function updateMetricsServerConfig(request, reply) {
   }
 }
 async function deleteMetricsServerConfig(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { keyOrId } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     let config = await store.getConfigByKey(tenantId, keyOrId);
     let configKey = keyOrId;
@@ -3396,8 +3677,8 @@ async function deleteMetricsServerConfig(request, reply) {
       };
     }
     try {
-      if (metricsServerManager.hasServer(tenantId, configKey)) {
-        metricsServerManager.removeServer(tenantId, configKey);
+      if (metricsServerManager2.hasServer(tenantId, configKey)) {
+        metricsServerManager2.removeServer(tenantId, configKey);
       }
     } catch (error) {
       console.warn("Failed to remove from MetricsServerManager:", error);
@@ -3415,10 +3696,10 @@ async function deleteMetricsServerConfig(request, reply) {
   }
 }
 async function testMetricsServerConnection(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3429,11 +3710,11 @@ async function testMetricsServerConnection(request, reply) {
       };
     }
     const testKey = `__test_${key}_${Date.now()}`;
-    metricsServerManager.registerServer(tenantId, testKey, config.config);
+    metricsServerManager2.registerServer(tenantId, testKey, config.config);
     try {
-      const client = metricsServerManager.getClient(tenantId, testKey);
+      const client = metricsServerManager2.getClient(tenantId, testKey);
       const result = await client.testConnection();
-      metricsServerManager.removeServer(tenantId, testKey);
+      metricsServerManager2.removeServer(tenantId, testKey);
       return {
         success: true,
         message: result.connected ? "Connection test successful" : "Connection test failed",
@@ -3441,7 +3722,7 @@ async function testMetricsServerConnection(request, reply) {
       };
     } catch (error) {
       try {
-        metricsServerManager.removeServer(tenantId, testKey);
+        metricsServerManager2.removeServer(tenantId, testKey);
       } catch {
       }
       return {
@@ -3466,10 +3747,10 @@ async function testMetricsServerConnection(request, reply) {
   }
 }
 async function listAvailableMetrics(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3479,10 +3760,10 @@ async function listAvailableMetrics(request, reply) {
         message: "Metrics server configuration not found"
       };
     }
-    if (!metricsServerManager.hasServer(tenantId, key)) {
-      metricsServerManager.registerServer(tenantId, key, config.config);
+    if (!metricsServerManager2.hasServer(tenantId, key)) {
+      metricsServerManager2.registerServer(tenantId, key, config.config);
     }
-    const client = metricsServerManager.getClient(tenantId, key);
+    const client = metricsServerManager2.getClient(tenantId, key);
     const metrics = await client.listMetrics();
     return {
       success: true,
@@ -3504,11 +3785,11 @@ async function listAvailableMetrics(request, reply) {
   }
 }
 async function queryMetricsData(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   const { metricName, startTime, endTime, step, labels } = request.body;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3525,10 +3806,10 @@ async function queryMetricsData(request, reply) {
         message: "metricName is required"
       };
     }
-    if (!metricsServerManager.hasServer(tenantId, key)) {
-      metricsServerManager.registerServer(tenantId, key, config.config);
+    if (!metricsServerManager2.hasServer(tenantId, key)) {
+      metricsServerManager2.registerServer(tenantId, key, config.config);
     }
-    const client = metricsServerManager.getClient(tenantId, key);
+    const client = metricsServerManager2.getClient(tenantId, key);
     const result = await client.queryMetricData(metricName, {
       startTime,
       endTime,
@@ -3552,10 +3833,10 @@ async function queryMetricsData(request, reply) {
   }
 }
 async function getDataSources(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3573,7 +3854,7 @@ async function getDataSources(request, reply) {
       };
     }
     const semanticConfig = config.config;
-    const client = new SemanticMetricsClient(semanticConfig);
+    const client = new SemanticMetricsClient2(semanticConfig);
     const allDatasources = await client.getDataSources();
     const selectedIds = semanticConfig.selectedDataSources || [];
     const filteredDatasources = selectedIds.length > 0 ? allDatasources.filter((ds) => selectedIds.includes(String(ds.id))) : allDatasources;
@@ -3593,10 +3874,10 @@ async function getDataSources(request, reply) {
   }
 }
 async function getDatasourceMetrics(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key, datasourceId } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3614,7 +3895,7 @@ async function getDatasourceMetrics(request, reply) {
       };
     }
     const semanticConfig = config.config;
-    const client = new SemanticMetricsClient(semanticConfig);
+    const client = new SemanticMetricsClient2(semanticConfig);
     const metrics = await client.getDatasourceMetrics(datasourceId);
     return {
       success: true,
@@ -3630,11 +3911,11 @@ async function getDatasourceMetrics(request, reply) {
   }
 }
 async function querySemanticMetrics(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   const body = request.body;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3659,33 +3940,37 @@ async function querySemanticMetrics(request, reply) {
       };
     }
     const semanticConfig = config.config;
-    const client = new SemanticMetricsClient(semanticConfig);
+    const client = new SemanticMetricsClient2(semanticConfig);
     const result = await client.semanticQuery(body);
+    const columnNames = result.columns.map((col) => col.name);
     const allDataPoints = [];
-    const metricNames = [];
-    for (const metricResult of result.results) {
-      metricNames.push(metricResult.metricName);
-      for (const row of metricResult.rows) {
-        allDataPoints.push({
-          timestamp: row.timestamp ? new Date(row.timestamp).getTime() : void 0,
-          value: typeof row.value === "number" ? row.value : 0,
-          metricName: metricResult.metricName,
-          labels: Object.fromEntries(
-            Object.entries(row).filter(([k]) => k !== "value" && k !== "timestamp").map(([k, v]) => [k, String(v)])
-          )
-        });
-      }
+    const timestampColIdx = columnNames.findIndex(
+      (col) => col.toLowerCase().includes("date") || col.toLowerCase().includes("time")
+    );
+    const valueColIdx = columnNames.findIndex(
+      (col) => col.toLowerCase().includes("value") || col.toLowerCase().includes("rate") || col.toLowerCase().includes("amt")
+    );
+    for (const row of result.rows) {
+      const timestamp = timestampColIdx >= 0 ? row[timestampColIdx] : void 0;
+      const value = valueColIdx >= 0 ? row[valueColIdx] : row[row.length - 1];
+      allDataPoints.push({
+        timestamp: timestamp ? new Date(String(timestamp)).getTime() : void 0,
+        value: typeof value === "number" ? value : Number(value) || 0,
+        labels: Object.fromEntries(
+          columnNames.map((col, idx) => [col, String(row[idx] ?? "")])
+        )
+      });
     }
     return {
       success: true,
       message: "Semantic query executed successfully",
       data: {
-        datasourceId: result.datasourceId,
-        metrics: metricNames,
+        semanticModel: result.semanticModel,
+        columns: columnNames,
         dataPoints: allDataPoints,
         metadata: {
-          rowCount: allDataPoints.length,
-          queryTimeMs: result.totalExecutionTimeMs
+          rowCount: result.rows.length,
+          columnCount: result.columns.length
         }
       }
     };
@@ -3715,7 +4000,7 @@ async function testSemanticDataSources(request, reply) {
       password: body.password,
       headers: body.headers
     };
-    const client = new SemanticMetricsClient(testConfig);
+    const client = new SemanticMetricsClient2(testConfig);
     const datasources = await client.getDataSources();
     return {
       success: true,
@@ -3751,7 +4036,7 @@ async function testDatasourceMetrics(request, reply) {
       password: body.password,
       headers: body.headers
     };
-    const client = new SemanticMetricsClient(testConfig);
+    const client = new SemanticMetricsClient2(testConfig);
     const metrics = await client.getDatasourceMetrics(datasourceId);
     return {
       success: true,
@@ -3784,12 +4069,12 @@ function registerMetricsServerConfigRoutes(app2) {
 
 // src/controllers/mcp-configs.ts
 import {
-  getStoreLattice as getStoreLattice7,
+  getStoreLattice as getStoreLattice8,
   mcpManager,
   toolLatticeManager as toolLatticeManager2
 } from "@axiom-lattice/core";
 import { randomUUID as randomUUID5 } from "crypto";
-function getTenantId7(request) {
+function getTenantId8(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
     return userTenantId;
@@ -3797,9 +4082,9 @@ function getTenantId7(request) {
   return request.headers["x-tenant-id"] || "default";
 }
 async function getMcpServerConfigList(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const configs = await store.getAllConfigs(tenantId);
     return {
@@ -3823,10 +4108,10 @@ async function getMcpServerConfigList(request, reply) {
   }
 }
 async function getMcpServerConfig(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3849,10 +4134,10 @@ async function getMcpServerConfig(request, reply) {
   }
 }
 async function createMcpServerConfig(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const body = request.body;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, body.key);
     if (existing) {
@@ -3888,11 +4173,11 @@ async function createMcpServerConfig(request, reply) {
   }
 }
 async function updateMcpServerConfig(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   const updates = request.body;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, key);
     if (!existing) {
@@ -3938,10 +4223,10 @@ async function updateMcpServerConfig(request, reply) {
   }
 }
 async function deleteMcpServerConfig(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { keyOrId } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     let config = await store.getConfigByKey(tenantId, keyOrId);
     let configKey = keyOrId;
@@ -3985,10 +4270,10 @@ async function deleteMcpServerConfig(request, reply) {
   }
 }
 async function testMcpServerConnection(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4038,10 +4323,10 @@ async function testMcpServerConnection(request, reply) {
   }
 }
 async function listMcpServerTools(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4071,10 +4356,10 @@ async function listMcpServerTools(request, reply) {
   }
 }
 async function connectMcpServer(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4095,7 +4380,7 @@ async function connectMcpServer(request, reply) {
     };
   } catch (error) {
     console.error("Failed to connect MCP server:", error);
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (config) {
@@ -4108,10 +4393,10 @@ async function connectMcpServer(request, reply) {
   }
 }
 async function disconnectMcpServer(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4216,11 +4501,11 @@ function registerMcpServerConfigRoutes(app2) {
 }
 
 // src/controllers/users.ts
-import { getStoreLattice as getStoreLattice8 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice9 } from "@axiom-lattice/core";
 import { v4 as uuidv42 } from "uuid";
 var UsersController = class {
   constructor() {
-    this.userStore = getStoreLattice8("default", "user").store;
+    this.userStore = getStoreLattice9("default", "user").store;
   }
   async listUsers(request, reply) {
     const { email } = request.query;
@@ -4298,11 +4583,11 @@ function registerUserRoutes(app2) {
 }
 
 // src/controllers/tenants.ts
-import { getStoreLattice as getStoreLattice9 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice10 } from "@axiom-lattice/core";
 import { v4 as uuidv43 } from "uuid";
 var TenantsController = class {
   constructor() {
-    this.tenantStore = getStoreLattice9("default", "tenant").store;
+    this.tenantStore = getStoreLattice10("default", "tenant").store;
   }
   // ==================== Tenant CRUD ====================
   async listTenants(request, reply) {
@@ -4366,7 +4651,7 @@ function registerTenantRoutes(app2) {
 }
 
 // src/controllers/auth.ts
-import { getStoreLattice as getStoreLattice10 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice11 } from "@axiom-lattice/core";
 import { v4 as uuidv44 } from "uuid";
 var defaultAuthConfig = {
   autoApproveUsers: true,
@@ -4375,9 +4660,9 @@ var defaultAuthConfig = {
 };
 var AuthController = class {
   constructor(config = {}) {
-    this.userStore = getStoreLattice10("default", "user").store;
-    this.tenantStore = getStoreLattice10("default", "tenant").store;
-    this.userTenantLinkStore = getStoreLattice10("default", "userTenantLink").store;
+    this.userStore = getStoreLattice11("default", "user").store;
+    this.tenantStore = getStoreLattice11("default", "tenant").store;
+    this.userTenantLinkStore = getStoreLattice11("default", "userTenantLink").store;
     this.config = { ...defaultAuthConfig, ...config };
   }
   async register(request, reply) {
@@ -4814,6 +5099,11 @@ var registerLatticeRoutes = (app2) => {
   registerWorkspaceRoutes(app2);
   registerDatabaseConfigRoutes(app2);
   registerMetricsServerConfigRoutes(app2);
+  app2.post(
+    "/api/data/query",
+    { schema: dataQuerySchema },
+    executeDataQuery
+  );
   registerMcpServerConfigRoutes(app2);
   registerUserRoutes(app2);
   registerTenantRoutes(app2);
@@ -4915,7 +5205,9 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
       }),
       headers: {
         "Content-Type": "application/json",
-        "x-tenant-id": tenant_id
+        "x-tenant-id": tenant_id,
+        "x-workspace-id": runConfig?.workspaceId,
+        "x-project-id": runConfig?.projectId
       }
     }).catch((err) => {
       console.error(`fetch\u8BF7\u6C42\u5931\u8D25: ${err.message || String(err)}`);
@@ -5248,6 +5540,10 @@ app.register(multipart, {
   }
 });
 app.register(websocket);
+app.register(staticPlugin, {
+  root: path2.join(__dirname, "../public"),
+  prefix: "/"
+});
 app.setErrorHandler((error, request, reply) => {
   const getHeaderValue = (header) => {
     if (Array.isArray(header)) {
@@ -5286,6 +5582,7 @@ var start = async (config) => {
       logger = loggerLattice.client;
     }
     app.decorate("loggerLattice", loggerLattice);
+    registerLatticeRoutes(app);
     if (!sandboxLatticeManager2.hasLattice("default")) {
       const sandboxBaseURL = process.env.SANDBOX_BASE_URL || "http://localhost:8080";
       sandboxLatticeManager2.registerLattice("default", {