inspiration-agent 0.0.1

package/dist/tools/search.tools.js

@@ -0,0 +1,208 @@
+const axios = require("axios");
+const logger = require("../utils/logger");
+const cheerio = require("cheerio");
+
+/**
+ * 搜索工具类
+ * 提供联网搜索功能，使用搜狗搜索
+ */
+class SearchTools {
+  /**
+   * 构造函数
+   */
+  constructor() {
+    this.sogouSearchUrl = "https://www.sogou.com/web";
+  }
+
+  /**
+   * 使用搜狗进行网络搜索
+   * @param {Object} args - 参数对象
+   * @param {string} args.query - 搜索查询词
+   * @param {number} args.maxResults - 最大结果数，默认 5
+   * @returns {Promise<Object>} 操作结果
+   */
+  async webSearch(args) {
+    try {
+      const { query, maxResults = 5 } = args;
+
+      if (!query) {
+        return {
+          success: false,
+          error: "缺少 query 参数",
+          message: "搜索失败: 缺少查询词",
+        };
+      }
+
+      logger.info(`\n========== 网络搜索 ==========`);
+      logger.info(`查询: ${query}`);
+      logger.info(`最大结果数: ${maxResults}`);
+      logger.info(`==============================\n`);
+
+      const response = await axios.get(this.sogouSearchUrl, {
+        params: {
+          query: encodeURIComponent(query),
+          ie: "utf8",
+        },
+        timeout: 10000,
+        headers: {
+          "User-Agent":
+            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+        },
+      });
+
+      const $ = cheerio.load(response.data);
+      const results = [];
+
+      $(".results .vr-title, .results h3 a").each((index, element) => {
+        if (results.length >= maxResults) return false;
+
+        const $el = $(element);
+        const title = $el.text().trim();
+        const url = $el.attr("href");
+        const $parent = $el.closest(".vr-title, .results > div");
+        const snippet = $parent.find(".str_info, .str-text, .abstract").text().trim() || "";
+
+        if (title && url) {
+          results.push({
+            title: title,
+            snippet: snippet,
+            url: url,
+            source: "搜狗",
+          });
+        }
+      });
+
+      logger.info(`\n========== 搜索结果 ==========`);
+      logger.info(`找到 ${results.length} 条结果`);
+      logger.info(`============================\n`);
+
+      return {
+        success: true,
+        query: query,
+        count: results.length,
+        results: results,
+        message: `成功搜索 "${query}"，找到 ${results.length} 条结果`,
+      };
+    } catch (error) {
+      logger.info(`\n========== 搜索失败 ==========`);
+      logger.info(`查询: ${args.query}`);
+      logger.info(`错误: ${error.message}`);
+      logger.info(`============================\n`);
+
+      return {
+        success: false,
+        error: error.message,
+        message: `搜索失败: ${error.message}`,
+      };
+    }
+  }
+
+  /**
+   * 获取网页内容
+   * @param {Object} args - 参数对象
+   * @param {string} args.url - 要获取的网页 URL
+   * @returns {Promise<Object>} 操作结果
+   */
+  async fetchWebPage(args) {
+    try {
+      const { url } = args;
+
+      if (!url) {
+        return {
+          success: false,
+          error: "缺少 url 参数",
+          message: "获取网页失败: 缺少 URL",
+        };
+      }
+
+      logger.info(`\n========== 获取网页内容 ==========`);
+      logger.info(`URL: ${url}`);
+      logger.info(`=================================\n`);
+
+      const response = await axios.get(url, {
+        timeout: 15000,
+        headers: {
+          "User-Agent":
+            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+        },
+      });
+
+      const content = response.data;
+      const contentLength = content.length;
+
+      logger.info(`\n========== 获取成功 ==========`);
+      logger.info(`内容长度: ${contentLength} 字符`);
+      logger.info(`=============================\n`);
+
+      return {
+        success: true,
+        url: url,
+        content: content,
+        contentLength: contentLength,
+        message: `成功获取网页: ${url}`,
+      };
+    } catch (error) {
+      logger.info(`\n========== 获取失败 ==========`);
+      logger.info(`URL: ${args.url}`);
+      logger.info(`错误: ${error.message}`);
+      logger.info(`=============================\n`);
+
+      return {
+        success: false,
+        error: error.message,
+        message: `获取网页失败: ${error.message}`,
+      };
+    }
+  }
+
+  /**
+   * 获取函数定义
+   * 返回所有搜索工具的函数定义，用于 OpenAI 函数调用格式
+   * @returns {Array<Object>} 函数定义数组
+   */
+  getFunctionDefinitions() {
+    return [
+      {
+        type: "function",
+        function: {
+          name: "web_search",
+          description: "使用搜狗搜索进行网络搜索，获取相关信息",
+          parameters: {
+            type: "object",
+            properties: {
+              query: {
+                type: "string",
+                description: "搜索查询词，如 'JavaScript 教程', '最新科技新闻' 等",
+              },
+              maxResults: {
+                type: "integer",
+                description: "返回的最大结果数，默认 5",
+                default: 5,
+              },
+            },
+            required: ["query"],
+          },
+        },
+      },
+      {
+        type: "function",
+        function: {
+          name: "fetch_web_page",
+          description: "获取指定 URL 的网页内容",
+          parameters: {
+            type: "object",
+            properties: {
+              url: {
+                type: "string",
+                description: "要获取的网页 URL，必须以 http:// 或 https:// 开头",
+              },
+            },
+            required: ["url"],
+          },
+        },
+      },
+    ];
+  }
+}
+
+module.exports = SearchTools;

package/dist/utils/logger.js

@@ -0,0 +1,52 @@
+const winston = require("winston");
+const path = require("path");
+const fs = require("fs");
+
+const logDir = path.join(__dirname, "../../logs");
+if (!fs.existsSync(logDir)) {
+  fs.mkdirSync(logDir, { recursive: true });
+}
+
+const logger = winston.createLogger({
+  level: process.env.LOG_LEVEL || "info",
+  format: winston.format.combine(
+    winston.format.timestamp({
+      format: "YYYY-MM-DD HH:mm:ss",
+    }),
+    winston.format.errors({ stack: true }),
+    winston.format.splat(),
+    winston.format.json()
+  ),
+  defaultMeta: { service: "ai-agent" },
+  transports: [
+    new winston.transports.File({
+      filename: path.join(logDir, "error.log"),
+      level: "error",
+    }),
+    new winston.transports.File({
+      filename: path.join(logDir, "combined.log"),
+    }),
+  ],
+});
+
+if (process.env.NODE_ENV !== "production") {
+  logger.add(
+    new winston.transports.Console({
+      format: winston.format.combine(
+        winston.format.colorize(),
+        winston.format.timestamp({
+          format: "YYYY-MM-DD HH:mm:ss",
+        }),
+        winston.format.printf(({ level, message, timestamp, ...meta }) => {
+          let msg = `${timestamp} [${level}] ${message}`;
+          if (Object.keys(meta).length > 0) {
+            msg += ` ${JSON.stringify(meta)}`;
+          }
+          return msg;
+        })
+      ),
+    })
+  );
+}
+
+module.exports = logger;

package/dist/utils/markdown-renderer.js

@@ -0,0 +1,207 @@
+/**
+ * 终端 Markdown 渲染器
+ * 使用 ANSI 转义码在终端中渲染 Markdown 格式内容
+ */
+class TerminalMarkdownRenderer {
+  constructor() {
+    this.ansi = {
+      reset: "\x1b[0m",
+      bright: "\x1b[1m",
+      dim: "\x1b[2m",
+      underscore: "\x1b[4m",
+      blink: "\x1b[5m",
+      reverse: "\x1b[7m",
+      hidden: "\x1b[8m",
+
+      fgBlack: "\x1b[30m",
+      fgRed: "\x1b[31m",
+      fgGreen: "\x1b[32m",
+      fgYellow: "\x1b[33m",
+      fgBlue: "\x1b[34m",
+      fgMagenta: "\x1b[35m",
+      fgCyan: "\x1b[36m",
+      fgWhite: "\x1b[37m",
+
+      bgBlack: "\x1b[40m",
+      bgRed: "\x1b[41m",
+      bgGreen: "\x1b[42m",
+      bgYellow: "\x1b[43m",
+      bgBlue: "\x1b[44m",
+      bgMagenta: "\x1b[45m",
+      bgCyan: "\x1b[46m",
+      bgWhite: "\x1b[47m",
+    };
+
+    this.ansiRegex = /\x1b\[[0-9;]*m/g;
+  }
+
+  /**
+   * 计算去除 ANSI 转义码后的实际字符长度
+   * @param {string} text - 包含 ANSI 转义码的文本
+   * @returns {number} 实际字符长度
+   */
+  getVisibleLength(text) {
+    return text.replace(this.ansiRegex, "").length;
+  }
+
+  /**
+   * 根据可见字符长度截取文本
+   * 确保 ANSI 转义码不会被截断
+   * @param {string} text - 包含 ANSI 转义码的文本
+   * @param {number} visibleStart - 可见字符起始位置
+   * @param {number} visibleEnd - 可见字符结束位置
+   * @returns {string} 截取后的文本
+   */
+  sliceByVisibleLength(text, visibleStart, visibleEnd) {
+    let visibleCount = 0;
+    let result = "";
+    let pos = 0;
+
+    while (pos < text.length && visibleCount < visibleEnd) {
+      const remainingText = text.slice(pos);
+      const ansiMatch = remainingText.match(this.ansiRegex);
+
+      if (ansiMatch && ansiMatch.index === 0) {
+        const ansiCode = ansiMatch[0];
+        if (visibleCount >= visibleStart) {
+          result += ansiCode;
+        }
+        pos += ansiCode.length;
+      } else {
+        const char = Array.from(remainingText)[0];
+        if (visibleCount >= visibleStart) {
+          result += char;
+        }
+        visibleCount++;
+        pos += char.length;
+      }
+    }
+
+    return result;
+  }
+
+  /**
+   * 渲染 Markdown 文本
+   * @param {string} markdown - Markdown 格式的文本
+   * @returns {string} 渲染后的文本
+   */
+  render(markdown) {
+    if (!markdown) return "";
+
+    let result = markdown;
+
+    result = this.renderCodeBlocks(result);
+    result = this.renderInlineCode(result);
+    result = this.renderHeaders(result);
+    result = this.renderBold(result);
+    result = this.renderItalic(result);
+    result = this.renderLinks(result);
+    result = this.renderUnorderedLists(result);
+    result = this.renderOrderedLists(result);
+    result = this.renderHorizontalRules(result);
+    result = this.renderBlockquotes(result);
+
+    return result;
+  }
+
+  /**
+   * 渲染代码块
+   */
+  renderCodeBlocks(text) {
+    return text.replace(/```(\w*)\n([\s\S]*?)```/g, (match, lang, code) => {
+      return `\n${this.ansi.bgBlack}${this.ansi.fgGreen}${code.trim()}${this.ansi.reset}\n`;
+    });
+  }
+
+  /**
+   * 渲染行内代码
+   */
+  renderInlineCode(text) {
+    return text.replace(
+      /`([^`]+)`/g,
+      `${this.ansi.bgBlack}${this.ansi.fgCyan}$1${this.ansi.reset}`
+    );
+  }
+
+  /**
+   * 渲染标题
+   */
+  renderHeaders(text) {
+    return text
+      .replace(/^#### (.+)$/gm, `${this.ansi.dim}${this.ansi.fgCyan}    $1${this.ansi.reset}`)
+      .replace(/^### (.+)$/gm, `${this.ansi.dim}${this.ansi.fgCyan}   $1${this.ansi.reset}`)
+      .replace(/^## (.+)$/gm, `${this.ansi.bright}${this.ansi.fgBlue}  $1${this.ansi.reset}`)
+      .replace(/^# (.+)$/gm, `${this.ansi.bright}${this.ansi.fgGreen}$1${this.ansi.reset}`);
+  }
+
+  /**
+   * 渲染粗体
+   */
+  renderBold(text) {
+    return text.replace(/\*\*([^*]+)\*\*/g, `${this.ansi.bright}$1${this.ansi.reset}`);
+  }
+
+  /**
+   * 渲染斜体
+   */
+  renderItalic(text) {
+    return text.replace(/\*([^*]+)\*/g, `${this.ansi.underscore}$1${this.ansi.reset}`);
+  }
+
+  /**
+   * 渲染链接
+   */
+  renderLinks(text) {
+    return text.replace(
+      /\[([^\]]+)\]\(([^)]+)\)/g,
+      `${this.ansi.fgBlue}${this.ansi.underscore}$1${this.ansi.reset} (${this.ansi.dim}$2${this.ansi.reset})`
+    );
+  }
+
+  /**
+   * 渲染无序列表
+   */
+  renderUnorderedLists(text) {
+    return text.replace(/^(\s*)- (.+)$/gm, `${this.ansi.fgYellow}$1• $2${this.ansi.reset}`);
+  }
+
+  /**
+   * 渲染有序列表
+   */
+  renderOrderedLists(text) {
+    return text.replace(/^(\s*)(\d+)\. (.+)$/gm, `${this.ansi.fgYellow}$1$2. $3${this.ansi.reset}`);
+  }
+
+  /**
+   * 渲染分隔线
+   */
+  renderHorizontalRules(text) {
+    return text.replace(/^---$/gm, `${this.ansi.dim}${"─".repeat(50)}${this.ansi.reset}`);
+  }
+
+  /**
+   * 渲染引用块
+   */
+  renderBlockquotes(text) {
+    return text.replace(/^> (.+)$/gm, `${this.ansi.dim}│ $1${this.ansi.reset}`);
+  }
+
+  /**
+   * 流式渲染 Markdown
+   * 用于逐步渲染流式输出的内容
+   * @param {string} chunk - 当前接收的文本块
+   * @param {string} buffer - 之前的缓冲区
+   * @returns {Object} { rendered: 渲染后的文本, buffer: 新的缓冲区 }
+   */
+  renderStream(chunk, buffer = "") {
+    const fullText = buffer + chunk;
+    const rendered = this.render(fullText);
+
+    return {
+      rendered: rendered,
+      buffer: fullText,
+    };
+  }
+}
+
+module.exports = TerminalMarkdownRenderer;

package/package.json

@@ -0,0 +1,51 @@
+{
+  "name": "inspiration-agent",
+  "version": "0.0.1",
+  "description": "Personal AI Assistant with multi-channel support",
+  "main": "dist/index.js",
+  "bin": {
+    "inspiration": "./dist/cli.js"
+  },
+  "files": [
+    "dist/",
+    "README.md"
+  ],
+  "scripts": {
+    "build": "node scripts/build.js",
+    "start": "node src/index.js",
+    "start:dist": "node dist/cli.js",
+    "format": "prettier --write \"src/**/*.js\" \"*.json\" \"*.md\"",
+    "prepublishOnly": "npm run build",
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": [
+    "ai",
+    "agent",
+    "deepseek",
+    "kimi",
+    "zhipu",
+    "minimax",
+    "feishu",
+    "chat"
+  ],
+  "author": "",
+  "license": "MIT",
+  "engines": {
+    "node": ">=18"
+  },
+  "dependencies": {
+    "@larksuiteoapi/node-sdk": "^1.58.0",
+    "axios": "^1.13.4",
+    "cheerio": "^1.2.0",
+    "dotenv": "^17.2.3",
+    "node-cron": "^4.2.1",
+    "puppeteer": "^24.37.2",
+    "sharp": "^0.34.5",
+    "sqlite3": "^5.1.7",
+    "tiktoken": "^1.0.22",
+    "winston": "^3.19.0"
+  },
+  "devDependencies": {
+    "prettier": "^3.8.1"
+  }
+}