@axiom-lattice/examples-deep_research 1.0.13 → 1.0.16

package/src/agents/data_agent/skills/sql-query.ts

@@ -0,0 +1,58 @@
+export const sqlQuery = {
+  name: "sql-query",
+  description:
+    "生成和执行 SQL 查询以检索业务数据。适用于需要从数据库获取数据、探索表结构、验证查询正确性的场景。委托给 sql-builder-agent 执行。",
+  prompt: `## 委托给 sql-builder-agent
+
+所有 SQL 相关操作都委托给 sql-builder-agent 子代理执行。
+
+## 数据库模式探索
+
+**请求模式信息**：
+- "请列出数据库中所有可用的表"
+- "请显示表 [X] 的模式，包括列、数据类型和关系"
+
+**检查现有文档**：
+- 先读取 \`/db_schema.md\`（如存在）
+- 仅在需要时请求新的模式探索
+
+## 查询生成与执行
+
+**提供清晰的业务需求**：
+1. **业务问题**：明确要回答的问题
+2. **指标**：需要计算的业务指标（收入、订单数、转化率等）
+3. **维度**：分组维度（地区、渠道、产品类别等）
+4. **筛选条件**：时间范围、状态、类别等
+5. **比较需求**：同比、环比、目标对比等
+
+**请求格式示例**：
+"我需要按地区比较 2024 年第三季度与 2023 年第三季度的收入。请生成并执行 SQL 查询。"
+
+"请查询过去 6 个月每个月的订单量和平均订单金额，按渠道分组。"
+
+## 接收与验证结果
+
+sql-builder-agent 会返回：
+- **SQL 查询**：格式清晰的完整查询
+- **查询结果**：返回的数据
+- **模式信息**：使用的表结构信息
+
+**验证要点**：
+- 查询是否正确回答了业务问题
+- 数据质量（NULL 值、异常值）
+- 结果完整性（行数、时间范围）
+- 列名是否业务友好
+
+## 错误处理
+
+如遇到查询错误：
+- 分析错误信息
+- 检查表名、列名是否正确
+- 验证 JOIN 条件和数据类型
+- 请求 sql-builder-agent 修正并重新执行
+
+## 文档化
+
+将使用的 SQL 查询和结果保存到分析文档中，便于后续参考和复现。
+`,
+};

package/src/agents/data_agent/tools/load_skills.ts

@@ -0,0 +1,88 @@
+/**
+ * Load Skills Tools
+ * Tools for loading skill metadata and content
+ */
+
+import z from "zod";
+import { registerToolLattice } from "@axiom-lattice/core";
+import * as analystSkill from "../skills/analyst";
+import * as dataVisualizationSkill from "../skills/data-visualization";
+import * as sqlQuerySkill from "../skills/sql-query";
+import * as analysisMethodologySkill from "../skills/analysis-methodology";
+import * as notebookReportSkill from "../skills/notebook-report";
+import * as infographicCreatorSkill from "../skills/infographic-creator";
+
+// Type definition for skill structure
+interface Skill {
+  name: string;
+  description: string;
+  prompt: string;
+}
+
+// Registry of all available skills
+const skillsRegistry: Record<string, Skill> = {
+  analyst: analystSkill.analyst,
+  "data-visualization": dataVisualizationSkill.dataVisualization,
+  "sql-query": sqlQuerySkill.sqlQuery,
+  "analysis-methodology": analysisMethodologySkill.analysisMethodology,
+  "notebook-report": notebookReportSkill.notebookReport,
+  "infographic-creator": infographicCreatorSkill.infographicCreator,
+};
+
+/**
+ * Load all skills and return their metadata (name and description, without prompt)
+ */
+registerToolLattice(
+  "load_skills",
+  {
+    name: "load_skills",
+    description:
+      "Load all available skills and return their metadata (name and description). This tool returns skill information without the prompt content. Use this to discover what skills are available.",
+    needUserApprove: false,
+    schema: z.object({}),
+  },
+  async (_input: Record<string, never>, _config: any) => {
+    try {
+      const skillsMeta = Object.values(skillsRegistry).map((skill) => ({
+        name: skill.name,
+        description: skill.description,
+      }));
+
+      return JSON.stringify(skillsMeta, null, 2);
+    } catch (error) {
+      return `Error loading skills: ${error instanceof Error ? error.message : String(error)
+        }`;
+    }
+  }
+);
+
+/**
+ * Load a specific skill's content and return its prompt
+ */
+registerToolLattice(
+  "load_skill_content",
+  {
+    name: "load_skill_content",
+    description:
+      "Load a specific skill's content by name and return its prompt. Use this tool to get the full prompt content for a skill that you want to use.",
+    needUserApprove: false,
+    schema: z.object({
+      skill_name: z.string().describe("The name of the skill to load"),
+    }),
+  },
+  async (input: { skill_name: string }, _config: any) => {
+    try {
+      const skill = skillsRegistry[input.skill_name];
+
+      if (!skill) {
+        const availableSkills = Object.keys(skillsRegistry).join(", ");
+        return `Skill "${input.skill_name}" not found. Available skills: ${availableSkills}`;
+      }
+
+      return skill.prompt;
+    } catch (error) {
+      return `Error loading skill content: ${error instanceof Error ? error.message : String(error)
+        }`;
+    }
+  }
+);

package/src/agents/index.ts

@@ -1,4 +1,3 @@
 //import "./todo";
 import "./research";
-
-// import "./property_viewing";
+import "./data_agent";

package/src/index.ts

@@ -2,11 +2,20 @@ import dotenv from "dotenv";
 
 import { LatticeGateway } from "@axiom-lattice/gateway";
 import {
+  getScheduleLattice,
   registerCheckpointSaver,
   registerModelLattice,
+  registerScheduleLattice,
 } from "@axiom-lattice/core";
 import "./agents";
+
 import path from "path";
+import {
+  LoggerConfig,
+  LoggerType,
+  ScheduleType,
+} from "@axiom-lattice/protocols";
+import { PostgreSQLScheduleStorage } from "@axiom-lattice/pg-stores";
 // 在文件开头添加
 const PACKAGE_VERSION = require("../package.json").version;
 const BUILD_TIME = new Date().toISOString();
@@ -96,10 +105,51 @@ registerModelLattice(
 );
 LatticeGateway.registerLatticeRoutes(LatticeGateway.app);
 
+// // 注册 Schedule Lattice（使用 PostgreSQL 持久化）
+// registerScheduleLattice("default", {
+//   name: "Default Scheduler",
+//   description: "Production scheduler with PostgreSQL",
+//   type: ScheduleType.POSTGRES,
+//   storage: new PostgreSQLScheduleStorage({
+//     poolConfig: process.env.DATABASE_URL || "",
+//   }),
+// });
+
+// const scheduler = getScheduleLattice("default");
+// // 注册 Handler（每次服务启动时都需要注册）
+// scheduler.registerHandler("send-email", async (payload, taskInfo) => {
+//   console.log("send-email", payload, taskInfo);
+// });
+// scheduler.restore();
+
+// // 一次性延迟任务
+// scheduler.scheduleOnce(
+//   "email-123",
+//   "send-email",
+//   { to: "user@example.com", subject: "Welcome!" },
+//   { delayMs: 60000 } // 1分钟后执行
+// );
+
 const port = parsePort();
 console.log(`Starting server on port ${port}`);
 
+// Default logger configuration
+const DEFAULT_LOGGER_CONFIG: LoggerConfig = {
+  name: "default",
+  description: "Default logger for lattice-gateway service",
+  type: LoggerType.PINO,
+  serviceName: "lattice/deep_research",
+  loggerName: "lattice/deep_research",
+  // file: {
+  //   file: "./logs/lattice" + "/deep_research",
+  //   frequency: "daily",
+  //   mkdir: true,
+  //   maxFiles: 30,
+  // },
+};
+
 LatticeGateway.startAsHttpEndpoint({
   port,
   queueServiceConfig: { type: "memory", defaultStartPollingQueue: true },
+  loggerConfig: DEFAULT_LOGGER_CONFIG,
 });