feat: 支持使用json的意图识别 & 支持reportAgent

2025-01-27 14:26:11 +00:00 · 2025-01-27 14:26:11 +00:00 · fe3783bdea
commit fe3783bdea
parent 10d37e5f73
16 changed files with 555 additions and 105 deletions
--- a/bun.lockb
+++ b/bun.lockb
--- a/constant/function.ts
+++ b/constant/function.ts
@ -1,9 +1,14 @@
 const functionMap = {
  egg: {
    xName: "小煎蛋",
-    xAuthor: "zhaoyingbo",
+    xAuthor: "Yingbo",
    xIcon: "🍳",
  },
+  webAgent: {
+    xName: "小煎蛋 简报助手",
+    xAuthor: "Yingbo",
+    xIcon: "📎",
+  },
  groupAgent: {
    xName: "小煎蛋 Group Agent",
    xAuthor: "YIBinary ❤️ Yingbo",
@ -11,12 +16,12 @@ const functionMap = {
  },
  sheetDB: {
    xName: "小煎蛋 Sheet DB",
-    xAuthor: "zhaoyingbo",
+    xAuthor: "Yingbo",
    xIcon: "🍪",
  },
  gitlabAgent: {
    xName: "小煎蛋 Gitlab Agent",
-    xAuthor: "zhaoyingbo",
+    xAuthor: "Yingbo",
    xIcon: "🐙",
  },
  soupAgent: {
--- a/controller/intentAgent/index.ts
+++ b/controller/intentAgent/index.ts
@ -0,0 +1,122 @@
+import { parseJsonString } from "@egg/hooks"
+import { z } from "zod"
+import { zodToJsonSchema } from "zod-to-json-schema"
+
+import { Context } from "../../types"
+import llm from "../../utils/llm"
+import { cleanLLMRes } from "../../utils/llm/base"
+
+/**
+ * 基础意图模式
+ */
+const BaseIntentSchema = z.object({
+  intent: z.number().min(1).max(13),
+})
+
+/**
+ * 简报模式
+ */
+const BriefingSchema = BaseIntentSchema.extend({
+  intent: z.literal(3),
+  link: z.string().url(),
+  userDescription: z.string().min(1),
+})
+
+/**
+ * 通用响应模式
+ */
+const CommonResponseSchema = BaseIntentSchema.extend({
+  intent: z.literal(12),
+  message: z.string().min(1),
+})
+
+/**
+ * 意图模式
+ */
+const IntentSchema = z.union([
+  BriefingSchema,
+  CommonResponseSchema,
+  BaseIntentSchema,
+])
+
+type BaseIntent = z.infer<typeof BaseIntentSchema>
+type Briefing = z.infer<typeof BriefingSchema>
+type CommonResponse = z.infer<typeof CommonResponseSchema>
+export type Intent = z.infer<typeof IntentSchema>
+
+/**
+ * JSON模式
+ */
+const jsonSchema = zodToJsonSchema(IntentSchema)
+
+/**
+ * 代理函数
+ * @param {Context} ctx - 上下文对象
+ * @returns {Promise<Intent>} - 返回意图对象
+ */
+const agent = async (ctx: Context): Promise<Intent> => {
+  const {
+    larkBody: { msgText },
+    logger,
+  } = ctx
+
+  let attempts = 0
+  while (attempts < 3) {
+    const res = await llm.invoke(
+      "intentRecognitionNext",
+      {
+        userInput: msgText,
+        time: new Date().toLocaleString("zh-CN", { timeZone: "Asia/Shanghai" }),
+        jsonSchema,
+      },
+      "test",
+      0,
+      true
+    )
+
+    const rawIntent = cleanLLMRes(res as string)
+    const parsedIntent = parseJsonString(rawIntent, null)
+
+    if (parsedIntent) {
+      try {
+        IntentSchema.parse(parsedIntent)
+        logger.debug("Intent is valid: " + JSON.stringify(parsedIntent))
+        return parsedIntent
+      } catch (e: any) {
+        logger.error("Invalid intent: " + String(e.errors))
+      }
+    } else {
+      logger.error("Parsed intent is null")
+    }
+
+    attempts++
+  }
+
+  return {
+    intent: 13,
+  }
+}
+
+const isBaseIntent = (intent: Intent): intent is BaseIntent => {
+  return intent.intent !== 3 && intent.intent !== 12
+}
+
+const isBriefing = (intent: Intent): intent is Briefing => {
+  return intent.intent === 3
+}
+
+const isCommonResponse = (intent: Intent): intent is CommonResponse => {
+  return intent.intent === 12
+}
+
+/**
+ * 意图代理对象
+ */
+const intentAgent = {
+  agent,
+  isBaseIntent,
+  isBriefing,
+  isCommonResponse,
+}
+
+export default intentAgent
--- a/controller/reportAgent/index.ts
+++ b/controller/reportAgent/index.ts
@ -0,0 +1,198 @@
+import db from "../../db"
+import { Context } from "../../types"
+import llm from "../../utils/llm"
+import { extractSheetIds, validateLink } from "../../utils/string"
+
+/**
+ * 爬取网页内容
+ * @param {Context} ctx - 上下文对象
+ * @param {string} link - 网页链接
+ * @returns {Promise<any>} - 返回爬取结果
+ * @throws {Error} - 当爬取失败时抛出错误
+ */
+const crawlWebPage = async (ctx: Context, link: string): Promise<any> => {
+  const { attachService } = ctx
+  const crawRes = await attachService.crawlWeb(link)
+  if (!crawRes || crawRes?.code) throw new Error("网页抓取失败")
+  return crawRes
+}
+
+/**
+ * 生成网页简报
+ * @param {Context} ctx - 上下文对象
+ * @param {string} userDescription - 用户描述
+ * @param {any} content - 网页内容
+ * @returns {Promise<string>} - 返回简报内容
+ * @throws {Error} - 当生成简报失败时抛出错误
+ */
+const generateSummary = async (
+  ctx: Context,
+  userDescription: string,
+  content: any
+): Promise<string> => {
+  const { requestId } = ctx
+  const llmRes = (await llm.invoke(
+    "summaryWeb",
+    {
+      description: userDescription,
+      content: content,
+    },
+    requestId,
+    1
+  )) as string
+  if (!llmRes) throw new Error("模型总结失败")
+  return llmRes
+}
+
+/**
+ * 插入到表格
+ * @param {Context} ctx - 上下文对象
+ * @param {string} link - 网页链接
+ * @param {string} userDescription - 用户描述
+ * @param {string} llmRes - 简报内容
+ * @returns {Promise<string>} - 返回表格链接
+ */
+const insert2Sheet = async (
+  ctx: Context,
+  link: string,
+  userDescription: string,
+  llmRes: string
+) => {
+  const {
+    larkBody: { userId },
+    logger,
+    larkService,
+  } = ctx
+  try {
+    const chat = await db.chat.getAndCreate(ctx)
+    if (!chat?.webSummarySheetLink) {
+      logger.info("No webSummarySheetLink found, skip insert2Sheet")
+      return ""
+    }
+    let sheetToken = ""
+    let range = ""
+    if (chat.webSummarySheetLink.includes("wiki")) {
+      const extractRes = extractSheetIds(chat.webSummarySheetLink)
+      if (!extractRes) {
+        logger.error("Failed to extract sheetToken and range")
+        return ""
+      }
+      const wikiData = await larkService.wiki.getNodeInfo(extractRes.sheetToken)
+      if (!wikiData || wikiData.code) {
+        logger.error("Failed to get wiki data")
+        return ""
+      }
+      sheetToken = wikiData.data.obj_token
+      range = extractRes.range
+    } else {
+      const extractRes = extractSheetIds(chat.webSummarySheetLink)
+      if (!extractRes) {
+        logger.error("Failed to extract sheetToken and range")
+        return ""
+      }
+      sheetToken = extractRes.sheetToken
+      range = extractRes.range
+    }
+
+    await larkService.sheet.insertRows(sheetToken, range, [
+      [
+        userId || "",
+        link,
+        userDescription,
+        llmRes,
+        new Date().toLocaleString("zh-CN", { timeZone: "Asia/Shanghai" }),
+      ],
+    ])
+    return chat.webSummarySheetLink
+  } catch (error: any) {
+    logger.error(`Failed to insert2Sheet: ${error}`)
+    return ""
+  }
+}
+
+/**
+ * 生成简报卡片
+ * @param {Context} ctx - 上下文对象
+ * @param {string} link - 网页链接
+ * @param {string} userDescription - 用户描述
+ * @param {string} llmRes - 简报内容
+ * @param {string} sheetLink - 表格链接
+ * @returns {any} - 返回卡片对象
+ */
+const genReportCard = (
+  ctx: Context,
+  link: string,
+  userDescription: string,
+  llmRes: string,
+  sheetLink: string
+) => {
+  const { larkCard } = ctx
+  const cardGender = larkCard.child("webAgent")
+  const description = userDescription
+    ? `**用户描述📝**\n${userDescription}\n`
+    : ""
+  const sheetLinkMd = sheetLink
+    ? `**小提示🔍**
+您可以在[这里](${sheetLink})查看往期简报哦！`
+    : ""
+  const content = `
+感谢您使用小煎蛋简报🍳！以下是为您总结的简报内容：
+
+**网站地址🌟**
+[${link}](${link})
+
+${description}
+
+**AI简报🌈**
+${llmRes}
+
+${sheetLinkMd}
+`
+  return cardGender.genCard("markdownSuccessCard", {
+    content,
+  })
+}
+
+/**
+ * 生成简报
+ * @param {Context} ctx - 上下文对象
+ * @param {string} link - 网页链接
+ * @param {string} userDescription - 用户描述
+ */
+const agent = async (ctx: Context, link: string, userDescription: string) => {
+  const {
+    larkService: { message },
+    larkCard,
+    logger,
+  } = ctx
+  const cardGender = larkCard.child("webAgent")
+  try {
+    // 校验链接是否合法
+    validateLink(link)
+    // 发送一个loading卡片
+    await message.updateOrReply(
+      cardGender.genSuccessCard("正在为您收集简报，请稍等片刻~")
+    )
+    // 抓取网页
+    const crawRes = await crawlWebPage(ctx, link)
+    // 调用模型生成简报
+    const llmRes = await generateSummary(ctx, userDescription, crawRes)
+    // 插入到表格
+    const sheetLink = await insert2Sheet(ctx, link, userDescription, llmRes)
+    // 发送简报卡片
+    await message.updateOrReply(
+      genReportCard(ctx, link, userDescription, llmRes, sheetLink)
+    )
+  } catch (error: any) {
+    logger.error(`Failed gen report: ${error}`)
+    await message.updateOrReply(
+      cardGender.genErrorCard(`简报生成失败: ${error.message}`)
+    )
+  }
+}
+
+const reportAgent = {
+  agent,
+}
+
+export default reportAgent
--- a/controller/soupAgent/index.ts
+++ b/controller/soupAgent/index.ts
@ -131,10 +131,8 @@ const chat2Soup = async (ctx: Context) => {
    larkBody: { msgText, chatId, messageId },
    logger,
    attachService,
-    larkCard,
    larkService: { message },
  } = ctx
-  const cardGender = larkCard.child("soupAgent")
  message.setReplyMessage(messageId, "text")
  const activeGame = await db.soupGame.getActiveOneByChatId(chatId)
  if (!activeGame) {
@ -150,9 +148,7 @@ const chat2Soup = async (ctx: Context) => {
  })
  if (!res) {
    logger.error(`chatId: ${chatId} failed to get soup result`)
-    await message.updateOrReply(
-      cardGender.genErrorCard(SoupGameMessage.chatFailed)
-    )
+    await message.updateOrReply(SoupGameMessage.chatFailed)
    return
  }
  // 用户答对了
--- a/db/chat/index.ts
+++ b/db/chat/index.ts
@ -13,6 +13,7 @@ export interface Chat {
  mode: "group" | "p2p" | "topic"
  weeklySummary: boolean
  dailySummary: boolean
+  webSummarySheetLink: string
 }

 export type ChatModel = Chat & RecordModel
@ -60,6 +61,7 @@ const getAndCreate = async ({ larkService, logger, larkBody }: Context) => {
    mode: chat_mode,
    weeklySummary: false,
    dailySummary: false,
+    webSummarySheetLink: "",
  }
  return create(newChat)
 }
--- a/index.ts
+++ b/index.ts
@ -12,8 +12,10 @@ initSchedule()

 await initAppConfig()

-const server = Bun.serve({
-  async fetch(req) {
+const bunServer = Bun.serve({
+  fetch: async (req, server) => {
+    // 设置超时时间
+    server.timeout(req, 30)
    // 生成上下文
    const ctx = await genContext(req)
    const { path, genResp, logger } = ctx
@ -43,6 +45,7 @@ const server = Bun.serve({
      return genResp.healthCheck("hello, there is egg, glade to serve you!")
    } catch (error: any) {
      // 错误处理
+      logger.error(error.message)
      return genResp.serverError(error.message || "server error")
    }
  },
@ -54,4 +57,4 @@ const server = Bun.serve({
  port: 3000,
 })

-logger.info(`Listening on ${server.hostname}:${server.port}`)
+logger.info(`Listening on ${bunServer.hostname}:${bunServer.port}`)
--- a/package.json
+++ b/package.json
@ -27,7 +27,7 @@
    "eslint-plugin-simple-import-sort": "^12.1.1",
    "eslint-plugin-unused-imports": "^4.1.4",
    "husky": "^9.1.7",
-    "lint-staged": "^15.4.2",
+    "lint-staged": "^15.4.3",
    "oxlint": "^0.13.2",
    "prettier": "^3.4.2",
    "typescript-eslint": "^8.21.0"
@ -39,7 +39,7 @@
    "@egg/hooks": "^1.2.0",
    "@egg/lark-msg-tool": "^1.21.0",
    "@egg/logger": "^1.6.0",
-    "@egg/net-tool": "^1.28.2",
+    "@egg/net-tool": "^1.30.2",
    "@egg/path-tool": "^1.4.1",
    "@langchain/core": "^0.3.36",
    "@langchain/langgraph": "^0.2.41",
--- a/routes/bot/eventMsg.ts
+++ b/routes/bot/eventMsg.ts
@ -1,8 +1,8 @@
 import groupAgent from "../../controller/groupAgent"
+import intentAgent from "../../controller/intentAgent"
+import reportAgent from "../../controller/reportAgent"
 import soupAgent from "../../controller/soupAgent"
 import { Context } from "../../types"
-import llm from "../../utils/llm"
-import { cleanLLMRes } from "../../utils/llm/base"
 import { isNotP2POrAtBot } from "../../utils/message"

 /**
@ -69,91 +69,66 @@ const manageIdMsg = ({
 */
 const manageIntent = async (ctx: Context) => {
  const {
-    body,
    logger,
    larkService: { message },
    attachService,
    larkBody: { msgText, chatId },
    larkCard,
-    requestId,
  } = ctx
  logger.info(`bot req text: ${msgText}`)
  await message.updateOrReply(
    larkCard.genPendingCard("正在理解您的意图，请稍等...")
  )
  try {
-    const llmRes = (await llm.invoke(
-      "intentRecognition",
-      {
-        userInput: msgText,
-        time: new Date().toLocaleString("zh-CN", { timeZone: "Asia/Shanghai" }),
-      },
-      requestId
-    )) as string
-    const cleanedLlmRes = cleanLLMRes(llmRes)
-    logger.info(`intentRecognition llm res: ${cleanedLlmRes}`)
-    // 返回值不是数字，说明是通识回答
-    const intent = Number(cleanedLlmRes)
-    if (isNaN(intent)) {
-      await message.updateOrReply(larkCard.genSuccessCard(cleanedLlmRes))
+    const intentRes = await intentAgent.agent(ctx)
+    logger.info(`intentRes: ${JSON.stringify(intentRes)}`)
+    if (intentAgent.isBriefing(intentRes)) {
+      reportAgent.agent(ctx, intentRes.link, intentRes.userDescription)
      return
    }
-
-    switch (intent) {
-      // 获取聊天ID
-      case 1:
-        await manageIdMsg(ctx)
-        break
-      // CI监控
-      case 2:
-        await attachService.ciMonitor(chatId)
-        break
-      // 生成简报
-      case 3:
-        await message.updateOrReply(
-          larkCard.genSuccessCard("正在为您收集简报，请稍等片刻~")
-        )
-        await attachService.reportCollector(body)
-        break
-      // 获取帮助
-      case 4:
-        await message.updateOrReply(
-          larkCard.genTempCard("eggGuide", { chat_id: chatId }) as string
-        )
-        break
-      // 开启日报订阅
-      case 5:
-        groupAgent.report.setSubscription(ctx, "daily", true)
-        break
-      // 开启周报订阅
-      case 6:
-        groupAgent.report.setSubscription(ctx, "weekly", true)
-        break
-      // 关闭日报订阅
-      case 7:
-        groupAgent.report.setSubscription(ctx, "daily", false)
-        break
-      // 关闭周报订阅
-      case 8:
-        groupAgent.report.setSubscription(ctx, "weekly", false)
-        break
-      // 立即发送日报
-      case 9:
-        groupAgent.report.gen4Test(ctx, "daily")
-        break
-      // 立即发送周报
-      case 10:
-        groupAgent.report.gen4Test(ctx, "weekly")
-        break
-      // 开始海龟汤游戏
-      case 11:
-        soupAgent.startOrStopGame(ctx, true, "manual")
-        break
-      // 通识回答
-      case 12:
-      default:
-        groupAgent.agent(ctx)
-        break
+    if (intentAgent.isCommonResponse(intentRes)) {
+      await message.updateOrReply(larkCard.genSuccessCard(intentRes.message))
+      return
+    }
+    if (intentAgent.isBaseIntent(intentRes)) {
+      switch (intentRes.intent) {
+        case 1:
+          await manageIdMsg(ctx)
+          break
+        case 2:
+          await attachService.ciMonitor(chatId)
+          break
+        case 4:
+          await message.updateOrReply(
+            larkCard.genTempCard("eggGuide", { chat_id: chatId }) as string
+          )
+          break
+        case 5:
+          groupAgent.report.setSubscription(ctx, "daily", true)
+          break
+        case 6:
+          groupAgent.report.setSubscription(ctx, "weekly", true)
+          break
+        case 7:
+          groupAgent.report.setSubscription(ctx, "daily", false)
+          break
+        case 8:
+          groupAgent.report.setSubscription(ctx, "weekly", false)
+          break
+        case 9:
+          groupAgent.report.gen4Test(ctx, "daily")
+          break
+        case 10:
+          groupAgent.report.gen4Test(ctx, "weekly")
+          break
+        case 11:
+          soupAgent.startOrStopGame(ctx, true, "manual")
+          break
+        case 13:
+        default:
+          groupAgent.agent(ctx)
+          break
+      }
    }
  } catch (error) {
    logger.error(`manageIntent error: ${error}`)
--- a/services/attach/index.ts
+++ b/services/attach/index.ts
@ -1,6 +1,8 @@
 import type { LarkEvent } from "@egg/lark-msg-tool"
 import { NetToolBase } from "@egg/net-tool"

+import { LarkServer } from "../../types"
+
 interface Chat2SoupParams {
  user_query: string
  soup_id: string
@ -84,6 +86,16 @@ class AttachService extends NetToolBase {
    const URL = "https://lark-egg.ai.xiaomi.com/soup/chat"
    return this.post<Chat2SoupResp>(URL, body).catch(() => null)
  }
+
+  /**
+   * 抓取网页内容
+   * @param {string} url - 网页URL。
+   * @returns {Promise<string>} 返回抓取的网页内容。
+   */
+  async crawlWeb(url: string) {
+    const URL = "https://lark-egg.ai.xiaomi.com/tools/web/crawl"
+    return this.get<LarkServer.BaseRes<string>>(URL, { url }).catch(() => null)
+  }
 }

 export default AttachService
--- a/test/archive/wiki.ts
+++ b/test/archive/wiki.ts
@ -0,0 +1,8 @@
+import initAppConfig from "../../constant/config"
+import genLarkService from "../../utils/genLarkService"
+
+await initAppConfig()
+
+const service = genLarkService("egg", "test")
+
+service.wiki.getNodeInfo("V4ZkwhDR8iRCqIk7X81k1rBj4Sc").then(console.log)
--- a/test/llm/intentRecognition.ts
+++ b/test/llm/intentRecognition.ts
@ -1,15 +1,59 @@
+import { parseJsonString } from "@egg/hooks"
+import { z } from "zod"
+import { zodToJsonSchema } from "zod-to-json-schema"
+
 import initAppConfig from "../../constant/config"
 import llm from "../../utils/llm"
+import { cleanLLMRes } from "../../utils/llm/base"

 await initAppConfig()

+const BaseIntentSchema = z.object({
+  intent: z.number().min(1).max(13),
+})
+
+const BriefingSchema = BaseIntentSchema.extend({
+  intent: z.literal(3),
+  link: z.string().url(),
+  userDescription: z.string().min(1),
+})
+
+const CommonResponseSchema = BaseIntentSchema.extend({
+  intent: z.literal(12),
+  message: z.string().min(1),
+})
+
+const IntentSchema = z.union([
+  BriefingSchema,
+  CommonResponseSchema,
+  BaseIntentSchema,
+])
+
+const jsonSchema = zodToJsonSchema(IntentSchema)
+
 const res = await llm.invoke(
-  "intentRecognition",
+  "intentRecognitionNext",
  {
-    userInput: "你是干嘛的",
+    userInput:
+      "https://mp.weixin.qq.com/s/-0J8XbXJU6Bu-UihRtgGAQ Airbnb死磕React Native惨败，微软却玩出花！Office、Outlook全线接入，Copilot成最大赢家 推荐大家看一下，rn助力微软copilot 跨平台实现",
    time: new Date().toLocaleString("zh-CN", { timeZone: "Asia/Shanghai" }),
+    jsonSchema,
  },
-  "test"
+  "test",
+  0,
+  true
 )

-console.log(res)
+const rawIntent = cleanLLMRes(res as string)
+
+const parsedIntent = parseJsonString(rawIntent, null)
+console.log("🚀 ~ parsedIntent:", parsedIntent)
+
+try {
+  IntentSchema.parse(parsedIntent)
+  console.log("Intent is valid:", parsedIntent)
+} catch (e: any) {
+  console.error("Invalid intent:", e.errors)
+}
+
+console.log(cleanLLMRes(res as string))
--- a/test/llm/summaryWeb.ts
+++ b/test/llm/summaryWeb.ts
--- a/test/reportAgent/agent.http
+++ b/test/reportAgent/agent.http
@ -0,0 +1,4 @@
+POST http://localhost:3000/bot?app=egg HTTP/1.1
+content-type: application/json
+
+{"schema":"2.0","header":{"event_id":"c94518fbcb9d66cc93b4144cc69e4f0c","token":"tV9djUKSjzVnekV7xTg2Od06NFTcsBnj","create_time":"1732612280153","event_type":"im.message.receive_v1","tenant_key":"2ee61fe50f4f1657","app_id":"cli_a1eff35b43b89063"},"event":{"message":{"chat_id":"oc_8c789ce8f4ecc6695bb63ca6ec4c61ea","chat_type":"group","content":"{\"text\":\"@_user_1  https://juejin.cn/post/7463301526800826404 openai agent实现，大概是云端起个chrome用playwright搞得，感觉vm技术是ai重要基建之一\"}","create_time":"1732612279943","mentions":[{"id":{"open_id":"ou_032f507d08f9a7f28b042fcd086daef5","union_id":"on_7111660fddd8302ce47bf1999147c011","user_id":""},"key":"@_user_1","name":"小煎蛋","tenant_key":"2ee61fe50f4f1657"}],"message_id":"om_4ab57cb6e889eeb81ca061b137238189","message_type":"text","update_time":"1732612279943"},"sender":{"sender_id":{"open_id":"ou_470ac13b8b50fc472d9d8ee71e03de26","union_id":"on_9dacc59a539023df8b168492f5e5433c","user_id":"zhaoyingbo"},"sender_type":"user","tenant_key":"2ee61fe50f4f1657"}}}
--- a/utils/llm/index.ts
+++ b/utils/llm/index.ts
@ -1,3 +1,5 @@
+import loggerIns from "@egg/logger"
+import { JsonOutputParser } from "@langchain/core/output_parsers"
 import { PromptTemplate } from "@langchain/core/prompts"

 import { adjustTimeRange, getSpecificTime, getTimeRange } from "../time"
@ -15,24 +17,51 @@ const invoke = async (
  promptName: string,
  variables: Record<string, any>,
  requestId: string,
-  temperature = 0
+  temperature = 0,
+  jsonMode = false
 ) => {
-  const { langfuse, langfuseHandler } = await getLangfuse("invoke", requestId)
-  const prompt = await langfuse.getPrompt(promptName)
+  const logger = loggerIns.child({ requestId })
+  const attemptInvoke = async () => {
+    const { langfuse, langfuseHandler } = await getLangfuse("invoke", requestId)
+    const prompt = await langfuse.getPrompt(promptName)

-  const langchainTextPrompt = PromptTemplate.fromTemplate(
-    prompt.getLangchainPrompt()
-  ).withConfig({
-    metadata: { langfusePrompt: prompt },
-  })
+    const langchainTextPrompt = PromptTemplate.fromTemplate(
+      prompt.getLangchainPrompt()
+    ).withConfig({
+      metadata: { langfusePrompt: prompt },
+    })

-  const chain = langchainTextPrompt.pipe(await getModel(temperature))
+    const chain = langchainTextPrompt.pipe(await getModel(temperature))

-  const { content } = await chain.invoke(variables, {
-    callbacks: [langfuseHandler],
-  })
+    if (jsonMode) {
+      chain.pipe(new JsonOutputParser())
+    }

-  return content
+    const { content } = await chain.invoke(variables, {
+      callbacks: [langfuseHandler],
+    })
+
+    return content
+  }
+
+  let result
+  let attempts = 0
+  do {
+    try {
+      result = await attemptInvoke()
+      break
+    } catch (e) {
+      logger.error(`🚀 ~ invoke ~ attemptInvoke ~ e: ${e}`)
+      attempts++
+    }
+  } while (attempts < 3)
+
+  if (!result) {
+    logger.error("Failed to invoke after 3 attempts")
+    return ""
+  }
+
+  return result
 }

 /**
@ -42,6 +71,7 @@ const invoke = async (
 * @returns
 */
 const timeParser = async (userInput: string, requestId: string) => {
+  const logger = loggerIns.child({ requestId })
  const time = new Date().toLocaleString("zh-CN", { timeZone: "Asia/Shanghai" })
  const weekDay = `星期${"日一二三四五六"[new Date().getDay()]}`
  const invokeParser = async () => {
@ -63,7 +93,7 @@ const timeParser = async (userInput: string, requestId: string) => {
      )) as string
      return JSON.parse(res.replaceAll("`", ""))
    } catch (e) {
-      console.error("🚀 ~ timeParser ~ invokeParser ~ e", e)
+      logger.error(`🚀 ~ timeParser ~ invokeParser ~ e: ${e}`)
      // 如果解析失败，则返回空字符串
      return { s: "", e: "" }
    }
--- a/utils/string.ts
+++ b/utils/string.ts
@ -0,0 +1,33 @@
+/**
+ * 从小米链接中提取 sheetToken 和 range。
+ *
+ * @param {string} url - 要提取数据的URL。
+ * @returns {{sheetToken: string, range: string} | null} - 包含 sheetToken 和 range 的对象，如果没有匹配则返回 null。
+ */
+export const extractSheetIds = (url: string) => {
+  // 定义匹配 wiki 和 sheets 两种URL格式的正则表达式
+  const pattern =
+    /wiki\/([\w\d]+)\?sheet=([\w\d]+)|sheets\/([\w\d]+)\?sheet=([\w\d]+)/
+  const match = url.match(pattern)
+  if (match) {
+    return {
+      sheetToken: match[1] || match[3], // 对于第一种URL格式，sheetToken 在组1；对于第二种格式，在组3
+      range: match[2] || match[4], // range 在第一种URL格式中是组2，在第二种格式是组4
+    }
+  }
+  return null // 如果没有匹配，则返回 null
+}
+
+/**
+ * 校验链接是否合法
+ * @param {string} link - 网页链接
+ * @throws {Error} - 当链接为空或格式不正确时抛出错误
+ */
+export const validateLink = (link: string): void => {
+  if (!link) throw new Error("链接不能为空")
+  try {
+    new URL(link)
+  } catch {
+    throw new Error("链接格式不正确")
+  }
+}