chatccc 0.2.92 → 0.2.94

package/src/builtin/index.ts

@@ -1,167 +1,167 @@
-/**
- * builtin/index.ts — ChatCCC 内置 Agent 核心 API
- *
- * ChatSession 是程序化入口，既可以被 CLI 调用，也可以被其他模块（如 ToolAdapter）调用。
- */
-
-import { streamText } from "ai";
-
-// ---------------------------------------------------------------------------
-// 系统提示词 — 编译期冻结常量
-// ---------------------------------------------------------------------------
-
-const SYSTEM_PROMPT = [
-  "你是 ChatCCC 内置 AI 编程助手，运行在终端环境中。",
-  "",
-  "## 基本规则",
-  "- 用中文回复，但代码、命令、文件名保持原文",
-  "- 优先给出直接可用的方案，而非长篇解释",
-  "- 如果用户的问题涉及代码，直接给出代码并说明用法",
-  "- 保持简洁，一次聚焦一个问题",
-].join("\n");
-
-// ---------------------------------------------------------------------------
-// 类型定义
-// ---------------------------------------------------------------------------
-
-export interface ChatSessionConfig {
-  /** DeepSeek API 兼容的服务地址，默认 https://api.deepseek.com/v1 */
-  baseURL?: string;
-  /** API Key（优先用 DEEPSEEK_API_KEY 环境变量） */
-  apiKey?: string;
-  /** 模型名称，默认 deepseek-chat */
-  model?: string;
-}
-
-export interface ChatSessionOptions {
-  /** 会话工作目录 */
-  cwd?: string;
-  /** 自定义系统提示词（会拼接到默认提示词之后） */
-  systemPrompt?: string;
-}
-
-/**
- * 流式响应事件
- */
-export type ChatEvent =
-  | { type: "text"; text: string; accumulated: string }
-  | { type: "done"; text: string }
-  | { type: "error"; message: string };
-
-// ---------------------------------------------------------------------------
-// ChatSession
-// ---------------------------------------------------------------------------
-
-/** 消息角色 */
-type MessageRole = "system" | "user" | "assistant" | "tool";
-
-/** 内部消息类型 */
-interface ChatMessage {
-  role: MessageRole;
-  content: string;
-}
-
-export class ChatSession {
-  private model: any;
-  private messages: ChatMessage[];
-
-  constructor(
-    config: ChatSessionConfig = {},
-    options: ChatSessionOptions = {},
-  ) {
-    const apiKey = config.apiKey ?? process.env.DEEPSEEK_API_KEY;
-    if (!apiKey) {
-      throw new Error(
-        "DEEPSEEK_API_KEY 未设置。请设置环境变量 DEEPSEEK_API_KEY 或传入 apiKey",
-      );
-    }
-
-    const baseURL = config.baseURL ?? "https://api.deepseek.com/v1";
-    const modelId = config.model ?? "deepseek-chat";
-
-    // 动态 import 以支持 ESM 包在 CJS 环境下的加载（tsx 处理）
-    const { createOpenAICompatible } = require("@ai-sdk/openai-compatible");
-    const provider = createOpenAICompatible({
-      name: "deepseek",
-      baseURL,
-      apiKey,
-    });
-    this.model = provider(modelId);
-
-    // 构建系统提示词
-    const systemContent = [SYSTEM_PROMPT];
-    if (options?.systemPrompt) {
-      systemContent.push("", options.systemPrompt);
-    }
-    if (options?.cwd) {
-      systemContent.push("", `当前工作目录: ${options.cwd}`);
-    }
-
-    this.messages = [{ role: "system", content: systemContent.join("\n") }];
-  }
-
-  /**
-   * 发送用户消息，返回异步可迭代的文本流。
-   *
-   * 使用方式：
-   * ```typescript
-   * const session = new ChatSession();
-   * for await (const event of session.chat("帮我看看 package.json")) {
-   *   if (event.type === "text") process.stdout.write(event.text);
-   * }
-   * console.log("完成");
-   * ```
-   */
-  async *chat(
-    userMessage: string,
-    signal?: AbortSignal,
-  ): AsyncIterable<ChatEvent> {
-    this.messages.push({ role: "user", content: userMessage });
-
-    let fullText = "";
-
-    try {
-      const result = streamText({
-        model: this.model,
-        messages: this.messages as any,
-        abortSignal: signal,
-      });
-
-      for await (const chunk of result.textStream) {
-        fullText += chunk;
-        yield { type: "text", text: chunk, accumulated: fullText };
-      }
-
-      this.messages.push({ role: "assistant", content: fullText });
-      yield { type: "done", text: fullText };
-    } catch (err) {
-      const message = err instanceof Error ? err.message : String(err);
-      if ((err as Error).name === "AbortError" || signal?.aborted) {
-        // 被中断时，不保存不完整的助手消息
-        if (fullText) {
-          this.messages.push({ role: "assistant", content: fullText + "\n[已中断]" });
-        }
-        yield { type: "done", text: fullText };
-        return;
-      }
-      yield { type: "error", message };
-      throw err;
-    }
-  }
-
-  /** 返回当前的会话历史（只读） */
-  get history(): ReadonlyArray<ChatMessage> {
-    return this.messages;
-  }
-
-  /** 返回当前轮数（不含 system 消息） */
-  get turnCount(): number {
-    return this.messages.length - 1;
-  }
-
-  /** 清空会话历史，保留 system 消息 */
-  reset(): void {
-    const system = this.messages[0];
-    this.messages = [system];
-  }
+/**
+ * builtin/index.ts — ChatCCC 内置 Agent 核心 API
+ *
+ * ChatSession 是程序化入口，既可以被 CLI 调用，也可以被其他模块（如 ToolAdapter）调用。
+ */
+
+import { streamText } from "ai";
+
+// ---------------------------------------------------------------------------
+// 系统提示词 — 编译期冻结常量
+// ---------------------------------------------------------------------------
+
+const SYSTEM_PROMPT = [
+  "你是 ChatCCC 内置 AI 编程助手，运行在终端环境中。",
+  "",
+  "## 基本规则",
+  "- 用中文回复，但代码、命令、文件名保持原文",
+  "- 优先给出直接可用的方案，而非长篇解释",
+  "- 如果用户的问题涉及代码，直接给出代码并说明用法",
+  "- 保持简洁，一次聚焦一个问题",
+].join("\n");
+
+// ---------------------------------------------------------------------------
+// 类型定义
+// ---------------------------------------------------------------------------
+
+export interface ChatSessionConfig {
+  /** DeepSeek API 兼容的服务地址，默认 https://api.deepseek.com/v1 */
+  baseURL?: string;
+  /** API Key（优先用 DEEPSEEK_API_KEY 环境变量） */
+  apiKey?: string;
+  /** 模型名称，默认 deepseek-chat */
+  model?: string;
+}
+
+export interface ChatSessionOptions {
+  /** 会话工作目录 */
+  cwd?: string;
+  /** 自定义系统提示词（会拼接到默认提示词之后） */
+  systemPrompt?: string;
+}
+
+/**
+ * 流式响应事件
+ */
+export type ChatEvent =
+  | { type: "text"; text: string; accumulated: string }
+  | { type: "done"; text: string }
+  | { type: "error"; message: string };
+
+// ---------------------------------------------------------------------------
+// ChatSession
+// ---------------------------------------------------------------------------
+
+/** 消息角色 */
+type MessageRole = "system" | "user" | "assistant" | "tool";
+
+/** 内部消息类型 */
+interface ChatMessage {
+  role: MessageRole;
+  content: string;
+}
+
+export class ChatSession {
+  private model: any;
+  private messages: ChatMessage[];
+
+  constructor(
+    config: ChatSessionConfig = {},
+    options: ChatSessionOptions = {},
+  ) {
+    const apiKey = config.apiKey ?? process.env.DEEPSEEK_API_KEY;
+    if (!apiKey) {
+      throw new Error(
+        "DEEPSEEK_API_KEY 未设置。请设置环境变量 DEEPSEEK_API_KEY 或传入 apiKey",
+      );
+    }
+
+    const baseURL = config.baseURL ?? "https://api.deepseek.com/v1";
+    const modelId = config.model ?? "deepseek-chat";
+
+    // 动态 import 以支持 ESM 包在 CJS 环境下的加载（tsx 处理）
+    const { createOpenAICompatible } = require("@ai-sdk/openai-compatible");
+    const provider = createOpenAICompatible({
+      name: "deepseek",
+      baseURL,
+      apiKey,
+    });
+    this.model = provider(modelId);
+
+    // 构建系统提示词
+    const systemContent = [SYSTEM_PROMPT];
+    if (options?.systemPrompt) {
+      systemContent.push("", options.systemPrompt);
+    }
+    if (options?.cwd) {
+      systemContent.push("", `当前工作目录: ${options.cwd}`);
+    }
+
+    this.messages = [{ role: "system", content: systemContent.join("\n") }];
+  }
+
+  /**
+   * 发送用户消息，返回异步可迭代的文本流。
+   *
+   * 使用方式：
+   * ```typescript
+   * const session = new ChatSession();
+   * for await (const event of session.chat("帮我看看 package.json")) {
+   *   if (event.type === "text") process.stdout.write(event.text);
+   * }
+   * console.log("完成");
+   * ```
+   */
+  async *chat(
+    userMessage: string,
+    signal?: AbortSignal,
+  ): AsyncIterable<ChatEvent> {
+    this.messages.push({ role: "user", content: userMessage });
+
+    let fullText = "";
+
+    try {
+      const result = streamText({
+        model: this.model,
+        messages: this.messages as any,
+        abortSignal: signal,
+      });
+
+      for await (const chunk of result.textStream) {
+        fullText += chunk;
+        yield { type: "text", text: chunk, accumulated: fullText };
+      }
+
+      this.messages.push({ role: "assistant", content: fullText });
+      yield { type: "done", text: fullText };
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      if ((err as Error).name === "AbortError" || signal?.aborted) {
+        // 被中断时，不保存不完整的助手消息
+        if (fullText) {
+          this.messages.push({ role: "assistant", content: fullText + "\n[已中断]" });
+        }
+        yield { type: "done", text: fullText };
+        return;
+      }
+      yield { type: "error", message };
+      throw err;
+    }
+  }
+
+  /** 返回当前的会话历史（只读） */
+  get history(): ReadonlyArray<ChatMessage> {
+    return this.messages;
+  }
+
+  /** 返回当前轮数（不含 system 消息） */
+  get turnCount(): number {
+    return this.messages.length - 1;
+  }
+
+  /** 清空会话历史，保留 system 消息 */
+  reset(): void {
+    const system = this.messages[0];
+    this.messages = [system];
+  }
 }

package/src/index.ts

@@ -92,6 +92,7 @@ import {
   rebuildBindingsFromRegistry,
   resetState,
   setSessionPlatform,
+  startUnifiedDisplayLoop,
 } from "./session.ts";
 import {
   rebuildSessionChatsFromRegistry,
@@ -595,13 +596,14 @@ async function startBotServiceCore(): Promise<void> {
     // 进程首次启动:此时所有 Map 都是空的,resetState 主要是打个 LOG 标识"开始
     // 干净状态"。修正残留的 running stream-state 并重建 session→chat 映射。
     resetState();
+    startUnifiedDisplayLoop();
     fixStaleStreamStates().then(async () => {
       const registry = await loadSessionRegistryForBinding();
       rebuildSessionChatsFromRegistry(registry);
     }).catch((err) => console.error(`[${ts()}] Init bindings failed: ${(err as Error).message}`));
 
     // ⚠️ 关键设计:onReady / onReconnected 只重建只读映射,**绝不**清空运行时
-    // 状态(activePrompts、displayLoops、sessionInfoMap、processedMessages)。
+    // 状态(activePrompts、sessionInfoMap、processedMessages)。
     // SDK 重连只是底层 WebSocket 抖动,业务层不受影响:
     //   - 后台 generator 仍在跑、stream-state 仍在被写
     //   - display loop 仍在向群推送

package/src/orchestrator.ts

@@ -740,11 +740,6 @@ export async function handleCommand(
         .setChatAvatar(chatId, descriptionTool, "new")
         .catch(() => {});
 
-      if (isSessionRunning(newSessionId)) {
-        const { ensureDisplayLoop } = await import("./session.ts");
-        ensureDisplayLoop(newSessionId);
-      }
-
       await platform.sendCard(
         chatId,
         `${toolLabel} Session Reset`,
@@ -901,11 +896,6 @@ export async function handleCommand(
 
       platform.setChatAvatar(chatId, target.tool, "new").catch(() => {});
 
-      if (isSessionRunning(target.sessionId)) {
-        const { ensureDisplayLoop } = await import("./session.ts");
-        ensureDisplayLoop(target.sessionId);
-      }
-
       const targetToolLabel = toolDisplayName(target.tool);
       const busyNote = isSessionRunning(target.sessionId)
         ? "\n\n⚠️ 该会话当前正在生成中，请等待完成后再发送消息。"

package/src/session-chat-binding.ts

@@ -130,6 +130,8 @@ export interface DisplayCardState {
   cardCreatedAt: number;
   lastSentContent: string;
   streamErrorNotified: boolean;
+  /** 所属 session */
+  sessionId: string;
   /** 所属 turn */
   turnCount: number;
   /** 卡片轮转时记录的基线，轮转后只展示增量 */
@@ -139,12 +141,18 @@ export interface DisplayCardState {
   lastSentAccLen?: number;
   /** WeChat delta: 上次发送时的 finalReply */
   lastSentFinalReply?: string;
+  /** 点点点动画计数器（统一 display loop 每个卡片独立计数） */
+  dotCount: number;
 }
 
 export const displayCards = new Map<string, DisplayCardState>();
 
-/** displayLoops: sessionId → 展示循环的 stop 函数 */
-export const displayLoops = new Map<string, () => void>();
+/** 统一 display loop 的 interval handle，由 session.ts 管理 */
+export let unifiedDisplayLoopHandle: ReturnType<typeof setInterval> | null = null;
+
+export function setUnifiedDisplayLoopHandle(h: ReturnType<typeof setInterval> | null): void {
+  unifiedDisplayLoopHandle = h;
+}
 
 // ---------------------------------------------------------------------------
 // queuedMessages: sessionId → 缓存消息（生成中排队，队列最大长度 1）
@@ -201,6 +209,8 @@ export function resetBindingState(): void {
   activePrompts.clear();
   queuedMessages.clear();
   displayCards.clear();
-  for (const stop of displayLoops.values()) stop();
-  displayLoops.clear();
+  if (unifiedDisplayLoopHandle !== null) {
+    clearInterval(unifiedDisplayLoopHandle);
+    unifiedDisplayLoopHandle = null;
+  }
 }