开发日记 | 用cloudflare worker实现telegram bot

1
const TELEGRAM_BOT_TOKEN = "";
2
const CF_API_TOKEN = "";
3

4
async function run(model, input) {
5
  const response = await fetch(
6
    `https://api.cloudflare.com/client/v4/accounts/1f5b3d5417456835c8191a2ec42f449e/ai/run/${model}`,
7
    {
8
      headers: { Authorization: `Bearer ${CF_API_TOKEN}` },
9
      method: "POST",
10
      body: JSON.stringify(input),
11
    }
12
  );
13
  return await response.json();
14
}
15

16
async function sendTelegramMessage(chat_id, text) {
17
  try {
18
    await fetch(`https://api.telegram.org/bot${TELEGRAM_BOT_TOKEN}/sendMessage`, {
19
      method: "POST",
20
      headers: { "Content-Type": "application/json" },
21
      body: JSON.stringify({
22
        chat_id,
23
        text,
24
        parse_mode: "MarkdownV2",
25
      }),
26
    });
27
  } catch (err) {
28
    console.error("发送 Telegram 消息失败:", err);
29
  }
30
}
31

32
export default {
33
  async fetch(request, env, ctx) {
34
    if (request.method !== "POST") return new Response("ok", { status: 200 });
35

36
    try {
37
      const data = await request.json();
38
      console.log("Telegram Update:", JSON.stringify(data));
39

40
      const message = data.message;
41
      if (!message?.text) return new Response("ok", { status: 200 });
42

43
      const text = message.text;
44
      const match = text.match(/^\/chat(?:@\w+)?(?:\s+([\s\S]+))?$/);
45
      const arg = match ? (match[1] || "").trim() : "";
46

47
      console.log("完整消息文本:", text);
48
      console.log("命令参数:", arg);
49

50
      if (!arg) {
51
        await sendTelegramMessage(message.chat.id, "请在 /chat 后输入要生成的内容，例如：/chat 写一个小故事");
52
      } else {
53

54
        let history = await env.bot.get(message.chat.id);
55
        history = history ? JSON.parse(history) : [];
56
        history.push({ role: "user", content: arg });
57
        await env.bot.put(message.chat.id, JSON.stringify(history));
58

59
        const aiResponse = await run("@cf/openai/gpt-oss-120b", {
60
          instructions: '你是liaoke的私人中文AI助手，简洁地回答用户的问题。',
61
          reasoning: { "effort": "medium" },
62
          input: history,
63
        });
64

65
        console.log("AI 原始响应:", JSON.stringify(aiResponse));
66

67

68
        const reasonText = aiResponse?.result?.output
69
          ?.filter(o => o.type === "reasoning")
70
          ?.map(o => o.content?.map(c => c.text).join("\n"))
71
          .filter(Boolean)
72
          .join("\n\n") || "AI 无法生成内容";
73

74
        const outputText = aiResponse?.result?.output
75
          ?.filter(o => o.type === "message")
76
          ?.map(o => o.content?.map(c => c.text).join("\n"))
77
          .filter(Boolean)
78
          .join("\n\n") || "AI 无法生成内容";
79

80

81
        const escapeMD = (text) => {
82
          if (!text) return "";
83
          return text
84
            .replace(/\\/g, "\\\\")
85
            .replace(/`/g, "\\`")
86
            .replace(/_/g, "\\_")
87
            .replace(/\*/g, "\\*")
88
            .replace(/\[/g, "\\[")
89
            .replace(/\]/g, "\\]")
90
            .replace(/~/g, "\\~")
91
            .replace(/>/g, "\\>")
92
            .replace(/#/g, "\\#")
93
            .replace(/\+/g, "\\+")
94
            .replace(/-/g, "\\-")
95
            .replace(/=/g, "\\=")
96
            .replace(/\|/g, "\\|")
97
            .replace(/\{/g, "\\{")
98
            .replace(/\}/g, "\\}")
99
            .replace(/\./g, "\\.")
100
            .replace(/!/g, "\\!");
101
        };
102

103

104
        const messageMD = `
105
🧠 推理过程：
106
\`\`\`
107
${escapeMD(reasonText)}
108
\`\`\`
109

110
💬 最终回答：
111
${escapeMD(outputText)}
112
        `;
113

114

115
        await sendTelegramMessage(message.chat.id, messageMD);
116
        history.push({ role: "assistant", content: outputText });
117
        await env.bot.put(message.chat.id, JSON.stringify(history));
118
        console.log("AI 输出发送给用户:", messageMD);
119
      }
120

121
      return new Response("ok", { status: 200 });
122
    } catch (err) {
123
      console.error("解析 JSON 或处理失败:", err);
124
      return new Response("error", { status: 400 });
125
    }
126
  },
127
};