@axiom-lattice/examples-deep_research 1.0.27 → 1.0.28

package/src/index-with-auth.ts

@@ -0,0 +1,122 @@
+import dotenv from "dotenv";
+import path from "path";
+import { LatticeGateway } from "@axiom-lattice/gateway";
+import {
+  storeLatticeManager,
+  registerStoreLattice,
+} from "@axiom-lattice/core";
+import { LoggerConfig, LoggerType } from "@axiom-lattice/protocols";
+import {
+  PostgreSQLThreadStore,
+  PostgreSQLDatabaseConfigStore,
+  PostgreSQLWorkspaceStore,
+  PostgreSQLProjectStore,
+  PostgreSQLUserStore,
+  PostgreSQLTenantStore,
+} from "@axiom-lattice/pg-stores";
+import "./agents";
+
+// Load environment variables
+dotenv.config({ path: path.resolve(__dirname, "../.env") });
+
+console.log(`
+╔════════════════════════════════════════════════╗
+║  Lattice Deep Research Server                  ║
+║  With Multi-Tenant Auth Support                ║
+╚════════════════════════════════════════════════╝
+`);
+
+// Auth Configuration
+const AUTH_CONFIG = {
+  // Auto approve new user registrations (true = no admin approval needed)
+  autoApproveUsers: process.env.AUTO_APPROVE_USERS !== "false",
+  
+  // Allow users to create new tenants during registration
+  allowTenantRegistration: process.env.ALLOW_TENANT_REGISTRATION !== "false",
+  
+  // JWT secret for token signing (change in production!)
+  jwtSecret: process.env.JWT_SECRET || "your-secret-key-change-in-production",
+  
+  // Token expiration in seconds (default: 24 hours)
+  tokenExpiration: parseInt(process.env.TOKEN_EXPIRATION || "86400", 10),
+};
+
+console.log("Auth Configuration:");
+console.log(`  - Auto Approve Users: ${AUTH_CONFIG.autoApproveUsers}`);
+console.log(`  - Allow Tenant Registration: ${AUTH_CONFIG.allowTenantRegistration}`);
+console.log(`  - Token Expiration: ${AUTH_CONFIG.tokenExpiration}s`);
+
+// Initialize PostgreSQL Stores with multi-tenant support
+const dbConfig = process.env.DATABASE_URL || "";
+
+// 1. User Store (NEW - for authentication)
+const userStore = new PostgreSQLUserStore({
+  poolConfig: dbConfig,
+  autoMigrate: true,
+});
+storeLatticeManager.removeLattice("default", "user");
+registerStoreLattice("default", "user", userStore);
+console.log("✓ PostgreSQL UserStore initialized");
+
+// 2. Tenant Store (NEW - for multi-tenancy)
+const tenantStore = new PostgreSQLTenantStore({
+  poolConfig: dbConfig,
+  autoMigrate: true,
+});
+storeLatticeManager.removeLattice("default", "tenant");
+registerStoreLattice("default", "tenant", tenantStore);
+console.log("✓ PostgreSQL TenantStore initialized");
+
+// 3. Existing stores...
+const threadStore = new PostgreSQLThreadStore({ poolConfig: dbConfig });
+storeLatticeManager.removeLattice("default", "thread");
+registerStoreLattice("default", "thread", threadStore);
+
+const databaseConfigStore = new PostgreSQLDatabaseConfigStore({
+  poolConfig: dbConfig,
+  autoMigrate: true,
+});
+storeLatticeManager.removeLattice("default", "database");
+registerStoreLattice("default", "database", databaseConfigStore);
+
+const workspaceStore = new PostgreSQLWorkspaceStore({
+  poolConfig: dbConfig,
+  autoMigrate: true,
+});
+storeLatticeManager.removeLattice("default", "workspace");
+registerStoreLattice("default", "workspace", workspaceStore);
+
+const projectStore = new PostgreSQLProjectStore({
+  poolConfig: dbConfig,
+  autoMigrate: true,
+});
+storeLatticeManager.removeLattice("default", "project");
+registerStoreLattice("default", "project", projectStore);
+
+// Parse port
+const port = parseInt(process.env.PORT || "4001", 10);
+
+// Logger config
+const loggerConfig: LoggerConfig = {
+  name: "default",
+  description: "Default logger for lattice-gateway",
+  type: LoggerType.PINO,
+  serviceName: "lattice/deep_research",
+  loggerName: "lattice/deep_research",
+};
+
+// Start server with auth routes enabled
+LatticeGateway.startAsHttpEndpoint({
+  port,
+  queueServiceConfig: { type: "memory", defaultStartPollingQueue: true },
+  loggerConfig,
+});
+
+console.log(`\n🚀 Server running at http://localhost:${port}`);
+console.log("\nAuth Endpoints:");
+console.log(`  POST http://localhost:${port}/api/auth/register`);
+console.log(`  POST http://localhost:${port}/api/auth/login`);
+console.log(`  GET  http://localhost:${port}/api/auth/pending`);
+console.log(`  POST http://localhost:${port}/api/auth/approve`);
+console.log(`  GET  http://localhost:${port}/api/tenants`);
+console.log(`  GET  http://localhost:${port}/api/users`);

package/src/index.ts

@@ -15,6 +15,7 @@ import {
   registerStoreLattice,
   sandboxLatticeManager,
   sqlDatabaseManager,
+  metricsServerManager,
 } from "@axiom-lattice/core";
 import { McpServerConfig } from "@axiom-lattice/protocols";
 import "./agents";
@@ -25,7 +26,7 @@ import {
   LoggerType,
   ScheduleType,
 } from "@axiom-lattice/protocols";
-import { PostgreSQLAssistantStore, PostgreSQLScheduleStorage, PostgreSQLThreadStore, PostgreSQLDatabaseConfigStore, PostgreSQLWorkspaceStore, PostgreSQLProjectStore } from "@axiom-lattice/pg-stores";
+import { PostgreSQLAssistantStore, PostgreSQLScheduleStorage, PostgreSQLThreadStore, PostgreSQLDatabaseConfigStore, PostgreSQLMetricsServerConfigStore, PostgreSQLWorkspaceStore, PostgreSQLProjectStore, PostgreSQLUserStore, PostgreSQLTenantStore, PostgreSQLUserTenantLinkStore } from "@axiom-lattice/pg-stores";
 
 // 在文件开头添加
 const fs = require("fs");
@@ -69,6 +70,19 @@ registerStoreLattice("default", "database", databaseConfigStore);
 sqlDatabaseManager.loadAllConfigsFromStore(databaseConfigStore)
 console.log("PostgreSQL DatabaseConfigStore initialized with auto-migration");
 
+// Initialize and register PostgreSQL MetricsServerConfigStore
+// This stores metrics server configurations with apiKey and password encryption
+const metricsConfigStore = new PostgreSQLMetricsServerConfigStore({
+  poolConfig: process.env.DATABASE_URL || "",
+  autoMigrate: true,
+});
+
+// Register metricsConfigStore to replace the default in-memory store
+storeLatticeManager.removeLattice("default", "metrics");
+registerStoreLattice("default", "metrics", metricsConfigStore);
+metricsServerManager.loadAllConfigsFromStore(metricsConfigStore)
+console.log("PostgreSQL MetricsServerConfigStore initialized with auto-migration");
+
 // Initialize and register PostgreSQL WorkspaceStore
 const workspaceStore = new PostgreSQLWorkspaceStore({
   poolConfig: process.env.DATABASE_URL || "",
@@ -91,6 +105,46 @@ storeLatticeManager.removeLattice("default", "project");
 registerStoreLattice("default", "project", projectStore);
 console.log("PostgreSQL ProjectStore initialized with auto-migration");
 
+// Initialize and register PostgreSQL UserStore (for authentication)
+const userStore = new PostgreSQLUserStore({
+  poolConfig: process.env.DATABASE_URL || "",
+  autoMigrate: true,
+});
+storeLatticeManager.removeLattice("default", "user");
+registerStoreLattice("default", "user", userStore);
+console.log("PostgreSQL UserStore initialized with auto-migration");
+
+// Initialize and register PostgreSQL TenantStore (for multi-tenancy)
+const tenantStore = new PostgreSQLTenantStore({
+  poolConfig: process.env.DATABASE_URL || "",
+  autoMigrate: true,
+});
+storeLatticeManager.removeLattice("default", "tenant");
+registerStoreLattice("default", "tenant", tenantStore);
+console.log("PostgreSQL TenantStore initialized with auto-migration");
+
+// Initialize and register PostgreSQL UserTenantLinkStore (for user-tenant relationships)
+const userTenantLinkStore = new PostgreSQLUserTenantLinkStore({
+  poolConfig: process.env.DATABASE_URL || "",
+  autoMigrate: true,
+});
+storeLatticeManager.removeLattice("default", "userTenantLink");
+registerStoreLattice("default", "userTenantLink", userTenantLinkStore);
+console.log("PostgreSQL UserTenantLinkStore initialized with auto-migration");
+
+// Auth configuration
+const AUTH_CONFIG = {
+  autoApproveUsers: process.env.AUTO_APPROVE_USERS !== "false",
+  allowTenantRegistration: process.env.ALLOW_TENANT_REGISTRATION !== "false",
+  jwtSecret: process.env.JWT_SECRET || "your-secret-key-change-in-production",
+  tokenExpiration: parseInt(process.env.TOKEN_EXPIRATION || "86400", 10),
+};
+
+console.log("Auth Configuration:");
+console.log(`  - Auto Approve Users: ${AUTH_CONFIG.autoApproveUsers}`);
+console.log(`  - Allow Tenant Registration: ${AUTH_CONFIG.allowTenantRegistration}`);
+console.log(`  - Token Expiration: ${AUTH_CONFIG.tokenExpiration}s`);
+
 // Parse command line arguments for port configuration
 function parsePort(): number {
   const args = process.argv.slice(2);

package/src/agents/data_agent/skills/generate-chart-md/SKILL.md

@@ -1,176 +0,0 @@
----
-name: generate-chart-md
-description: Generate chart markdown content fragments (with ECharts config) for analysis results. Use when output should include bar, line, pie, scatter, or heatmap charts as embeddable markdown snippets.
-metadata:
-  category: generate-report
----
-
-## 图表类型选择指南
-
-根据数据特征和业务问题选择最合适的图表类型：
-
-- **柱状图** (bar): 比较类别或时间周期
-  - 使用 category xAxis，value yAxis
-  - 多系列用于分组/堆叠柱状图
-- **折线图** (line): 展示时间趋势
-  - 使用 category/time xAxis，value yAxis
-  - 多系列展示多个指标
-- **饼图** (pie): 展示构成/百分比
-  - 无需 xAxis/yAxis
-  - 数据格式: [{value: number, name: string}, ...]
-  - 使用 radius: ["40%", "70%"] 创建环形图
-- **散点图** (scatter): 相关性分析
-  - 使用 value xAxis 和 value yAxis
-  - 数据格式: [[x, y], [x, y], ...]
-- **热力图** (heatmap): 多维数据
-  - 需要 category xAxis 和 yAxis
-  - 数据格式: [[xIndex, yIndex, value], ...]
-
-## ECharts 配置要求
-
-生成完整的 ECharts 配置，必须包含：
-
-\`\`\`chart
-{
-"table": [...], // 原始数据表格
-"echarts": {
-"title": {"text": "清晰的图表标题"},
-"tooltip": {
-"trigger": "axis", // bar/line 用 "axis", pie/scatter 用 "item"
-"formatter": "..." // 可选：自定义格式化
-},
-"legend": {...}, // 多系列时必需
-"xAxis": {
-"type": "category", // 或 "time", "value"
-"name": "X 轴名称",
-"data": [...] // category 类型时必需
-},
-"yAxis": {
-"type": "value",
-"name": "Y 轴名称"
-},
-"series": [{
-"type": "bar|line|pie|scatter|heatmap",
-"name": "系列名称",
-"data": [...],
-"label": {...} // 可选：显示数值
-}],
-"grid": {...} // 可选：控制边距
-}
-}
-\`\`\`
-
-## 最佳实践
-
-- 图表标题清晰描述业务问题
-- 轴标签使用业务术语，而非技术字段名
-- 数值格式化：百分比、货币、千分位
-- 时间序列使用 "xAxis.type: 'time'" 并正确格式化日期
-- 多系列时使用 legend 区分
-- 重要数值在图表上直接标注（series.label）
-
-## 使用场景
-
-### 场景1：销售数据可视化
-用户需要展示各区域销售额对比，或月度销售趋势。
-
-### 场景2：财务指标展示
-展示收入构成、成本结构或利润率变化趋势。
-
-### 场景3：运营数据分析
-展示用户增长、留存率、转化率等运营指标。
-
-## 完整示例
-
-### 示例1：柱状图 - 销售额对比
-
-```chart
-{
-  "table": [
-    {"region": "华北", "sales": 120},
-    {"region": "华东", "sales": 200},
-    {"region": "华南", "sales": 150}
-  ],
-  "echarts": {
-    "title": {"text": "各区域销售额对比"},
-    "tooltip": {"trigger": "axis"},
-    "xAxis": {
-      "type": "category",
-      "data": ["华北", "华东", "华南"],
-      "name": "区域"
-    },
-    "yAxis": {
-      "type": "value",
-      "name": "销售额(万元)"
-    },
-    "series": [{
-      "type": "bar",
-      "data": [120, 200, 150],
-      "label": {"show": true, "position": "top"}
-    }]
-  }
-}
-```
-
-### 示例2：折线图 - 月度趋势
-
-```chart
-{
-  "table": [
-    {"month": "1月", "revenue": 100},
-    {"month": "2月", "revenue": 120},
-    {"month": "3月", "revenue": 150}
-  ],
-  "echarts": {
-    "title": {"text": "月度收入趋势"},
-    "tooltip": {"trigger": "axis"},
-    "xAxis": {
-      "type": "category",
-      "data": ["1月", "2月", "3月"],
-      "name": "月份"
-    },
-    "yAxis": {
-      "type": "value",
-      "name": "收入(万元)"
-    },
-    "series": [{
-      "type": "line",
-      "data": [100, 120, 150],
-      "smooth": true,
-      "label": {"show": true}
-    }]
-  }
-}
-```
-
-### 示例3：饼图 - 收入构成
-
-```chart
-{
-  "table": [
-    {"category": "产品A", "value": 40},
-    {"category": "产品B", "value": 30},
-    {"category": "产品C", "value": 20},
-    {"category": "其他", "value": 10}
-  ],
-  "echarts": {
-    "title": {"text": "收入构成分析"},
-    "tooltip": {"trigger": "item", "formatter": "{b}: {c} ({d}%)"},
-    "series": [{
-      "type": "pie",
-      "radius": ["40%", "70%"],
-      "data": [
-        {"value": 40, "name": "产品A"},
-        {"value": 30, "name": "产品B"},
-        {"value": 20, "name": "产品C"},
-        {"value": 10, "name": "其他"}
-      ],
-      "label": {"show": true, "formatter": "{b}: {d}%"}
-    }]
-  }
-}
-```
-
-## 输出格式
-
-提供完整的 chart JSON 配置，可直接用于渲染。代码块语言标识必须为 `chart`。