ework-qq-bridge 0.1.4 → 0.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ework-qq-bridge",
-  "version": "0.1.4",
+  "version": "0.2.0",
   "description": "QQ group <-> ework issue bridge (OneBot 11 reverse WebSocket)",
   "type": "module",
   "main": "src/index.ts",

package/src/chat.ts

@@ -0,0 +1,72 @@
+// Chat mode: instant Q&A directly against the OpenAI-compatible endpoint.
+// Issue mode (任务/#N) stays the path for long agent work — chat is for the
+// conversational 90% of group traffic that never needs a session.
+
+export interface ChatTurn {
+  role: "user" | "assistant";
+  name: string;
+  content: string;
+}
+
+export const CHAT_SYSTEM_PROMPT = [
+  "你是 QQ 群里的即时问答助手，背后是 ework 开发平台。",
+  "风格：简短直接，能用一两句话说清的就别铺开；技术问题给结论和关键理由，需要展开再展开。",
+  "群里成员通过 @你 提问。你看到的多轮对话里每条 user 消息前缀了提问者的昵称，注意区分不同人。",
+  "如果请求明显是需要长时间执行的开发任务（改代码、查仓库、提交 PR），不要假装去做——建议对方发「任务 <标题>」创建 issue，AI agent 会接单处理。",
+  "不知道就直说，不要编造。",
+].join("\n");
+
+export function buildChatMessages(
+  history: ChatTurn[],
+  question: ChatTurn,
+  maxHistory: number,
+): { role: string; name?: string; content: string }[] {
+  const kept = history.slice(-maxHistory);
+  return [
+    { role: "system", content: CHAT_SYSTEM_PROMPT },
+    ...kept.map((t) => ({ role: t.role, name: t.name, content: t.content })),
+    { role: question.role, name: question.name, content: question.content },
+  ];
+}
+
+export async function chatComplete(
+  apiBase: string,
+  apiKey: string,
+  model: string,
+  messages: { role: string; name?: string; content: string }[],
+  timeoutMs: number,
+): Promise<string> {
+  const ctrl = new AbortController();
+  const timer = setTimeout(() => ctrl.abort(), timeoutMs);
+  try {
+    const res = await fetch(`${apiBase.replace(/\/+$/, "")}/chat/completions`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json", Authorization: `Bearer ${apiKey}` },
+      body: JSON.stringify({ model, messages, max_tokens: 1024, temperature: 0.4 }),
+      signal: ctrl.signal,
+    });
+    if (!res.ok) {
+      throw new Error(`LLM ${res.status}: ${(await res.text()).slice(0, 120)}`);
+    }
+    const data = (await res.json()) as { choices?: { message?: { content?: string } }[] };
+    const text = data.choices?.[0]?.message?.content?.trim();
+    if (!text) throw new Error("LLM returned empty content");
+    return text;
+  } finally {
+    clearTimeout(timer);
+  }
+}
+
+export function splitForQQ(text: string, maxLen = 1500): string[] {
+  if (text.length <= maxLen) return [text];
+  const parts: string[] = [];
+  let rest = text;
+  while (rest.length > maxLen) {
+    let cut = rest.lastIndexOf("\n", maxLen);
+    if (cut < maxLen * 0.5) cut = maxLen;
+    parts.push(rest.slice(0, cut));
+    rest = rest.slice(cut).replace(/^\n+/, "");
+  }
+  if (rest) parts.push(rest);
+  return parts;
+}

package/src/config.ts

@@ -36,6 +36,15 @@ const Schema = z.object({
 
   DB_PATH: z.string().default(""),
 
+  // Chat mode (instant Q&A). When WORK_CHAT_API is set, @bot messages without
+  // a recognized command are answered directly by the LLM instead of usage
+  // help. Empty = chat mode disabled.
+  WORK_CHAT_API: z.string().default(""),
+  WORK_CHAT_API_KEY: z.string().default("sk-vllm"),
+  WORK_CHAT_MODEL: z.string().default("qwen3.8-27b"),
+  WORK_CHAT_TIMEOUT_MS: z.coerce.number().int().positive().default(90_000),
+  WORK_CHAT_MAX_HISTORY: z.coerce.number().int().positive().default(20),
+
   VERBOSE: z.coerce.boolean().default(false),
 
   // Deployment-specific hostnames scrubbed from outbound QQ messages.

package/src/ingest.ts

@@ -19,6 +19,23 @@ interface IngestDeps {
   send(groupId: number, text: string): Promise<void>;
 }
 
+// Looping agent sessions can emit a reply every few seconds; cap forwards
+// per issue so a runaway session cannot flood the group.
+const FORWARD_WINDOW_MS = 60_000;
+const FORWARD_MAX_PER_WINDOW = 5;
+const forwardTimestamps = new Map<string, number[]>();
+
+function forwardAllowed(key: string, now = Date.now()): boolean {
+  const list = (forwardTimestamps.get(key) ?? []).filter((t) => now - t < FORWARD_WINDOW_MS);
+  if (list.length >= FORWARD_MAX_PER_WINDOW) {
+    forwardTimestamps.set(key, list);
+    return false;
+  }
+  list.push(now);
+  forwardTimestamps.set(key, list);
+  return true;
+}
+
 export function verifySignature(secret: string, body: string, header: string | null): boolean {
   if (!secret) return true;
   if (!header) return false;
@@ -74,6 +91,10 @@ export function createIngest(deps: IngestDeps) {
       return new Response("skipped:dup", { status: 200 });
     }
 
+    if (!forwardAllowed(`${owner}/${String(repo.name)}#${number}`)) {
+      console.warn(`[qq-bridge] forward rate-limited for ${owner}/${String(repo.name)}#${number}`);
+      return new Response("rate-limited", { status: 200 });
+    }
     const text = deps.scrub(`[#${number}] ${body}`);
     try {
       await deps.send(groupId, text);

package/src/router.ts

@@ -2,12 +2,14 @@ import type { GroupBinding, Config } from "./config";
 import type { EworkClient } from "./ework";
 import type { GroupMessageEvent } from "./onebot";
 import type { BridgeStore } from "./db";
+import { buildChatMessages, chatComplete, splitForQQ, type ChatTurn } from "./chat";
 
 const HELP_TEXT = [
   "用法：",
   "  任务 <标题> —— 新建 issue，AI 自动接单",
   "  #<编号> <内容> —— 给指定 issue 追加内容",
   "  查询 —— 列出最近 issue",
+  "  @我 + 任意问题 —— 即时问答（不建 issue）",
 ].join("\n");
 
 export interface RouterDeps {
@@ -39,6 +41,23 @@ export function parseCommand(raw: string): ParsedCommand | null {
 
 export function createRouter(deps: RouterDeps) {
   const { cfg, bindings, wakeList, ework, store } = deps;
+  const chatHistory = new Map<number, ChatTurn[]>();
+
+  async function answerChat(ev: GroupMessageEvent, question: string): Promise<void> {
+    const turn: ChatTurn = { role: "user", name: ev.nickname, content: question };
+    const history = chatHistory.get(ev.groupId) ?? [];
+    const messages = buildChatMessages(history, turn, cfg.WORK_CHAT_MAX_HISTORY);
+    const answer = await chatComplete(cfg.WORK_CHAT_API, cfg.WORK_CHAT_API_KEY, cfg.WORK_CHAT_MODEL, messages, cfg.WORK_CHAT_TIMEOUT_MS);
+    history.push(turn, { role: "assistant", name: "bot", content: answer });
+    if (history.length > cfg.WORK_CHAT_MAX_HISTORY * 2) {
+      chatHistory.set(ev.groupId, history.slice(-cfg.WORK_CHAT_MAX_HISTORY * 2));
+    } else {
+      chatHistory.set(ev.groupId, history);
+    }
+    for (const part of splitForQQ(answer)) {
+      await deps.reply(ev.groupId, part);
+    }
+  }
 
   async function handleGroupMessage(ev: GroupMessageEvent): Promise<void> {
     if (store.seenPost(ev.postId)) return;
@@ -57,7 +76,16 @@ export function createRouter(deps: RouterDeps) {
       return;
     }
     if (!cmd) {
-      // Wake-word without a verb: silent-ignore hides the syntax from users.
+      const question = ev.rawMessage.replace(/\[CQ:[^\]]*\]/g, "").trim();
+      if (cfg.WORK_CHAT_API && question) {
+        try {
+          await answerChat(ev, question);
+        } catch (err) {
+          console.error(`[qq-bridge] chat failed: ${err instanceof Error ? err.message : err}`);
+          await deps.reply(ev.groupId, "🤖 回答失败了，稍后再试一次。");
+        }
+        return;
+      }
       await deps.reply(ev.groupId, "没看懂指令。\n" + HELP_TEXT);
       return;
     }

package/tests/bridge.test.ts

@@ -75,3 +75,72 @@ describe("verifySignature", () => {
 test("helps when @bot without verb", () => {
   expect(parseCommand("[CQ:at,qq=2661222094] 测试2")).toBeNull();
 });
+
+describe("chat mode", () => {
+  const { buildChatMessages, splitForQQ, CHAT_SYSTEM_PROMPT } = require("../src/chat");
+
+  test("buildChatMessages: system + capped history + new turn", () => {
+    const hist = Array.from({ length: 30 }, (_, i) => ({ role: "user" as const, name: `u${i}`, content: `m${i}` }));
+    const msgs = buildChatMessages(hist, { role: "user", name: "dog", content: "q" }, 20);
+    expect(msgs[0].role).toBe("system");
+    expect(msgs[0].content).toBe(CHAT_SYSTEM_PROMPT);
+    expect(msgs).toHaveLength(22);
+    expect(msgs[1].name).toBe("u10");
+    expect(msgs[msgs.length - 1]).toEqual({ role: "user", name: "dog", content: "q" });
+  });
+
+  test("splitForQQ keeps <=1500 chunks and prefers newline cuts", () => {
+    const short = splitForQQ("hello");
+    expect(short).toEqual(["hello"]);
+    const long = "line\n".repeat(600);
+    const parts = splitForQQ(long);
+    expect(parts.length).toBeGreaterThan(1);
+    for (const p of parts) expect(p.length).toBeLessThanOrEqual(1500);
+    expect(parts.join("\n")).toBe(long.replace(/\n$/, ""));
+  });
+
+  test("router: @bot + question hits chat when configured, help when not", async () => {
+    const { createRouter } = require("../src/router");
+    const replies: string[] = [];
+    const chatCalls: string[] = [];
+    const baseDeps = (chatApi: string) => ({
+      cfg: {
+        VERBOSE: false,
+        WORK_CHAT_API: chatApi,
+        WORK_CHAT_API_KEY: "k",
+        WORK_CHAT_MODEL: "m",
+        WORK_CHAT_TIMEOUT_MS: 1000,
+        WORK_CHAT_MAX_HISTORY: 20,
+      },
+      bindings: [{ groupId: 1, owner: "o", repo: "r" }],
+      wakeList: new Set(["1403558951"]),
+      ework: { createIssue: async () => 1, addComment: async () => {} },
+      store: { seenPost: () => false },
+      reply: async (_g: number, text: string) => { replies.push(text); },
+    });
+    // chat disabled → usage help
+    let router = createRouter(baseDeps(""));
+    await router.handleGroupMessage({ groupId: 1, userId: 1403558951, nickname: "dog", postId: "p1", rawMessage: "[CQ:at,qq=2661222094] 你好呀" });
+    expect(replies[0]).toContain("没看懂指令");
+    // chat enabled → mocked LLM answer (patch chatComplete via env endpoint failure is overkill; test buildChatMessages path instead)
+    replies.length = 0;
+    router = createRouter(baseDeps("http://invalid.test/v1"));
+    await router.handleGroupMessage({ groupId: 1, userId: 1403558951, nickname: "dog", postId: "p2", rawMessage: "[CQ:at,qq=2661222094] 你好呀" });
+    expect(replies[0]).toContain("回答失败");
+  });
+
+  test("router: @bot with CQ-only payload (no text) still gets help", async () => {
+    const { createRouter } = require("../src/router");
+    const replies: string[] = [];
+    const router = createRouter({
+      cfg: { VERBOSE: false, WORK_CHAT_API: "http://x/v1", WORK_CHAT_API_KEY: "k", WORK_CHAT_MODEL: "m", WORK_CHAT_TIMEOUT_MS: 1000, WORK_CHAT_MAX_HISTORY: 20 },
+      bindings: [{ groupId: 1, owner: "o", repo: "r" }],
+      wakeList: new Set(["1"]),
+      ework: { createIssue: async () => 1, addComment: async () => {} },
+      store: { seenPost: () => false },
+      reply: async (_g: number, text: string) => { replies.push(text); },
+    });
+    await router.handleGroupMessage({ groupId: 1, userId: 1, nickname: "u", postId: "p3", rawMessage: "[CQ:at,qq=2661222094]" });
+    expect(replies[0]).toContain("没看懂指令");
+  });
+});