@amaster.ai/components-templates 1.3.0

package/components/ai-assistant/template/hooks/useConversationProcessor.ts

@@ -0,0 +1,649 @@
+import { useState, useCallback, useMemo, useRef, useEffect } from "react";
+import { produce } from "immer";
+import type {
+  MessagesItem,
+  Conversation,
+  Role,
+  TextMessage,
+  ThoughtMessage,
+  ToolMessage,
+  ErrorMessage,
+  UIRenderMessage,
+} from "../types";
+import type {
+  SendStreamingMessageResponse,
+  SendStreamingMessageSuccessResponse,
+} from "@a2a-js/sdk";
+import { client } from "@/lib/client";
+import { getText } from "../i18n";
+
+export interface UpdateMessageInput {
+  taskId: string;
+  messageId: string;
+  content?: string;
+  partial?: boolean;
+  status?: string;
+  response?: any;
+  metadata?: Record<string, any>;
+}
+
+// ================= 消息处理器 =================
+
+type MessageHandler = (
+  conv: Conversation,
+  part: any,
+  id: string,
+  role: Role,
+  status?: any,
+) => Conversation;
+
+const handlers: Record<string, MessageHandler> = {
+  // 文本内容（逐步追加 content）
+  // {"jsonrpc":"2.0","id":"6d90729b-dd32-46c7-8f48-502850dd33a5","result":{"kind":"status-update","taskId":"b7f71d08-3d35-4cca-88f8-75864632b787","contextId":"9436a03e-ae32-43b7-97cf-309667580efc:anonymous-ef04dcc5-c314-4494-8fd6-3d342bc07dfc","status":{"state":"working","message":{"kind":"message","role":"agent","parts":[{"kind":"text","text":"自动回复"}],"messageId":"e7f45bfa-8273-4c3d-b635-125b10a32a63","taskId":"b7f71d08-3d35-4cca-88f8-75864632b787","contextId":"9436a03e-ae32-43b7-97cf-309667580efc:anonymous-ef04dcc5-c314-4494-8fd6-3d342bc07dfc"},"timestamp":"2026-03-11T11:29:50.614Z"},"final":false,"metadata":{"coderAgent":{"kind":"text-content"},"model":"MiniMax-M2.5","traceId":"msg_36e36c20-f7c7-4b05-b884-c9b21337d2ed"}}}
+  "text-content": (conv, part, id, role, status) => {
+    if (!id) return conv;
+    const existing = conv.messages.find((m) => m.messageId === id);
+    const newText = part.text || "";
+    if (existing && "content" in existing) {
+      return {
+        ...conv,
+        messages: conv.messages.map((m) =>
+          m.messageId === id
+            ? { ...m, content: (m as TextMessage).content + newText }
+            : m,
+        ),
+        lastUpdated: new Date().toISOString(),
+      };
+    }
+
+    return {
+      ...conv,
+      messages: [
+        ...conv.messages,
+        {
+          messageId: id,
+          role,
+          kind: "text-content",
+          content: newText,
+          timestamp: status.timestamp || new Date().toISOString(),
+        } as TextMessage,
+      ],
+      lastUpdated: new Date().toISOString(),
+    };
+  },
+
+  // thought 类型（逐步追加 description）
+  // {"jsonrpc":"2.0","id":"6d90729b-dd32-46c7-8f48-502850dd33a5","result":{"kind":"status-update","taskId":"b7f71d08-3d35-4cca-88f8-75864632b787","contextId":"9436a03e-ae32-43b7-97cf-309667580efc:anonymous-ef04dcc5-c314-4494-8fd6-3d342bc07dfc","status":{"state":"working","message":{"kind":"message","role":"agent","parts":[{"kind":"data","data":{"subject":"","description":"技能"}}],"messageId":"2b5db798-052c-4616-8299-722af26d13ed","taskId":"b7f71d08-3d35-4cca-88f8-75864632b787","contextId":"9436a03e-ae32-43b7-97cf-309667580efc:anonymous-ef04dcc5-c314-4494-8fd6-3d342bc07dfc"},"timestamp":"2026-03-11T11:29:44.295Z"},"final":false,"metadata":{"coderAgent":{"kind":"thought"},"model":"MiniMax-M2.5","traceId":"msg_36e36c20-f7c7-4b05-b884-c9b21337d2ed"}}}
+  thought: (conv, part, id, role, status) => {
+    if (!id) return conv;
+    const newChunk = part?.data?.description || "";
+    const existing = conv.messages.find((m) => m.messageId === id);
+    if (existing && "thought" in existing) {
+      return {
+        ...conv,
+        messages: conv.messages.map((m) =>
+          m.messageId === id
+            ? { ...m, thought: (m as ThoughtMessage).thought + newChunk }
+            : m,
+        ),
+        lastUpdated: new Date().toISOString(),
+      };
+    }
+
+    return {
+      ...conv,
+      messages: [
+        ...conv.messages,
+        {
+          messageId: id,
+          role,
+          kind: "thought",
+          thought: newChunk,
+          timestamp: status?.timestamp || new Date().toISOString(),
+        } as ThoughtMessage,
+      ],
+      lastUpdated: new Date().toISOString(),
+    };
+  },
+
+  // 工具调用（状态变更 + 最终结果）
+  // {"jsonrpc":"2.0","id":"e76855a5-db73-4fa9-ac09-a795512d3579","result":{"kind":"status-update","taskId":"0d9abf02-b15d-4ef3-92e6-b8e8e68eda34","contextId":"9436a03e-ae32-43b7-97cf-309667580efc:anonymous-ef04dcc5-c314-4494-8fd6-3d342bc07dfc","status":{"state":"working","message":{"kind":"message","role":"agent","parts":[{"kind":"data","data":{"request":{"callId":"toolu_0048e8fad3514a358b031792","name":"WebSearch","args":{"query":"北京天气 今天 2026年3月11日"},"isClientInitiated":false,"prompt_id":"9436a03e-ae32-43b7-97cf-309667580efc:anonymous-ef04dcc5-c314-4494-8fd6-3d342bc07dfc########0","traceId":"msg_1183218d-0e0f-48fe-98c9-817f2992c742"},"status":"error","response":{"callId":"toolu_0048e8fad3514a358b031792","error":{},"responseParts":[{"functionResponse":{"id":"toolu_0048e8fad3514a358b031792","name":"WebSearch","response":{"error":"Tool \"WebSearch\" not found in registry. Tools must use the exact names that are registered. Did you mean \"activate_skill\"?"}}}],"resultDisplay":"Tool \"WebSearch\" not found in registry. Tools must use the exact names that are registered. Did you mean \"activate_skill\"?","errorType":"tool_not_registered","contentLength":122}}}],"messageId":"89431f8f-277b-4e78-87f8-4fb6cc29ffd5","taskId":"0d9abf02-b15d-4ef3-92e6-b8e8e68eda34","contextId":"9436a03e-ae32-43b7-97cf-309667580efc:anonymous-ef04dcc5-c314-4494-8fd6-3d342bc07dfc"},"timestamp":"2026-03-11T11:31:56.090Z"},"final":false,"metadata":{"coderAgent":{"kind":"tool-call-update"},"model":"MiniMax-M2.5"}}}
+  tool: (conv, part, _id, role, status) => {
+    const partData = part?.data || {};
+    const tool = partData?.tool;
+    const name = tool?.displayName || tool?.name || "";
+    const request = partData?.request;
+    const id = request?.callId;
+    if (!id || !name) return conv; // 必须有 callId 来识别消息
+
+    const toolStatus = partData?.status;
+    const existing = conv.messages.find((m) => m.messageId === id);
+
+    if (existing && "toolStatus" in existing) {
+      return {
+        ...conv,
+        messages: conv.messages.map((m) =>
+          m.messageId === id
+            ? {
+                ...m,
+                toolStatus: toolStatus || (m as ToolMessage).toolStatus,
+              }
+            : m,
+        ),
+        lastUpdated: new Date().toISOString(),
+      };
+    }
+
+    return {
+      ...conv,
+      messages: [
+        ...conv.messages,
+        {
+          messageId: id,
+          role,
+          kind: "tool",
+          toolName: name,
+          toolDescription: request?.args?.query || "",
+          toolStatus: toolStatus || "pending",
+          timestamp: status?.timestamp || new Date().toISOString(),
+        } as ToolMessage,
+      ],
+      lastUpdated: new Date().toISOString(),
+    };
+  },
+
+  error: (conv, part, id, role, status) => {
+    const errorMsg = part?.text || part?.data?.error || "发生错误";
+    return {
+      ...conv,
+      messages: [
+        ...conv.messages,
+        {
+          messageId: id || `error-${Date.now()}`,
+          role,
+          kind: "error",
+          content: errorMsg,
+          timestamp: new Date().toISOString(),
+        } as ErrorMessage,
+      ],
+      status: "error",
+      lastUpdated: new Date().toISOString(),
+    };
+  },
+
+  "ui-render": (conv, part, id, role, status) => {
+    if (!id) return conv;
+    const existing = conv.messages.find((m) => m.messageId === id);
+    const partData = part?.data || {};
+    const spec = partData.spec || { root: "", elements: {} };
+
+    if (existing && existing.kind === "ui-render") {
+      return {
+        ...conv,
+        messages: conv.messages.map((m) =>
+          m.messageId === id ? { ...(m as UIRenderMessage), spec } : m,
+        ),
+        lastUpdated: new Date().toISOString(),
+      };
+    }
+
+    return {
+      ...conv,
+      messages: [
+        ...conv.messages,
+        {
+          messageId: id,
+          role,
+          kind: "ui-render",
+          spec,
+          timestamp: status?.timestamp || new Date().toISOString(),
+        } as UIRenderMessage,
+      ],
+      lastUpdated: new Date().toISOString(),
+    };
+  },
+};
+
+// 用于生成唯一 assistant message id
+const generateId = () =>
+  `msg-${Date.now()}-${Math.random().toString(36).slice(2, 10)}`;
+
+// ================= 主 Hook =================
+
+type ConversationsState = {
+  byId: Record<string, Conversation>;
+  order: string[]; // taskId 数组，按所需顺序（最近活跃的在前面）
+};
+
+export function useConversationProcessor() {
+  const [state, setState] = useState<ConversationsState>({
+    byId: {},
+    order: [],
+  });
+  const [isLoading, setIsLoading] = useState(false);
+  const loadingRef = useRef(false);
+  const abortControllerRef = useRef<AbortController | null>(null);
+  const forceStopRef = useRef(false);
+  const lastMessageTypeRef = useRef<string>("unknown");
+  const lastMessageIdRef = useRef<string>("");
+
+  useEffect(() => {
+    loadingRef.current = isLoading;
+  }, [isLoading]);
+
+  const handleEnd = useCallback(() => {
+    abort();
+    lastMessageTypeRef.current = "";
+    lastMessageIdRef.current = "";
+  }, []);
+
+  const processDataLine = useCallback((data: SendStreamingMessageResponse) => {
+    if (!(data as any)?.result) return;
+    const result = (data as SendStreamingMessageSuccessResponse).result;
+
+    const { taskId, status, metadata, final } = result as any;
+
+    if (final) {
+      handleEnd();
+      return;
+    }
+
+    if (!taskId) return;
+
+    const parts = status?.message?.parts || [];
+    const firstPart = parts?.[0];
+
+    let kind = "unknown";
+    if (firstPart?.data?.tool) {
+      kind = "tool";
+    } else if (firstPart?.data?.type === "ui") {
+      kind = "ui-render";
+    } else if (firstPart?.kind === "text") {
+      kind = "text-content";
+    } else if (metadata?.coderAgent?.kind === "thought") {
+      kind = "thought";
+    } else if (metadata?.error) {
+      kind = "error";
+    }
+
+    const handler = handlers[kind];
+    if (!handler) {
+      return;
+    }
+    removeLoadingPlaceholder();
+
+    setState(
+      produce((draft) => {
+        let id = lastMessageIdRef.current;
+        if (
+          kind === "ui-render" ||
+          (kind !== "unknown" && kind !== lastMessageTypeRef.current)
+        ) {
+          lastMessageTypeRef.current = kind;
+          id = generateId();
+          lastMessageIdRef.current = id;
+        }
+
+        const prevConv = draft.byId[taskId] ?? {
+          taskId,
+          status: "submitted" as const,
+          messages: [],
+          lastUpdated: new Date().toISOString(),
+        };
+
+        const nextConv = handler(
+          prevConv,
+          firstPart,
+          id,
+          (status?.message?.role === "agent" ? "assistant" : "") || "assistant",
+          status,
+        );
+
+        draft.byId[taskId] = {
+          ...nextConv,
+          status: status?.state || prevConv.status,
+          lastUpdated: new Date().toISOString(),
+        };
+
+        // 调整 order：将更新的 taskId 移到最前面（最近活跃置顶）
+        const index = draft.order.indexOf(taskId);
+        if (index !== -1) {
+          draft.order.splice(index, 1);
+        } else {
+          // 如果是新会话，添加到最前面
+        }
+        draft.order.push(taskId);
+      }),
+    );
+  }, []);
+
+  const conversations = useMemo(() => {
+    return state.order
+      .map((id) => state.byId[id])
+      .filter(Boolean) as Conversation[];
+  }, [state]);
+
+  const getConversation = useCallback(
+    (taskId: string) => {
+      return state.byId[taskId];
+    },
+    [state],
+  );
+
+  const clearConversation = useCallback((taskId: string) => {
+    setState(
+      produce((draft) => {
+        if (!draft.byId[taskId]) return;
+        delete draft.byId[taskId];
+        draft.order = draft.order.filter((id) => id !== taskId);
+      }),
+    );
+  }, []);
+
+  const removeMessage = useCallback((taskId: string, messageId: string) => {
+    setState(
+      produce((draft) => {
+        const conv = draft.byId[taskId];
+        if (!conv) return;
+
+        const nextMessages = conv.messages.filter(
+          (m) => m.messageId !== messageId,
+        );
+
+        draft.byId[taskId] = {
+          ...conv,
+          messages: nextMessages,
+          lastUpdated: new Date().toISOString(),
+        };
+
+        // 调整 order：更新后置顶
+        const index = draft.order.indexOf(taskId);
+        if (index !== -1) {
+          draft.order.splice(index, 1);
+        }
+        draft.order.push(taskId);
+      }),
+    );
+  }, []);
+
+  // 新增：更新已有消息
+  const updateMessage = useCallback((input: UpdateMessageInput) => {
+    const {
+      taskId,
+      messageId,
+      content,
+      partial = false,
+      status,
+      response,
+      metadata = {},
+    } = input;
+
+    setState(
+      produce((draft) => {
+        const conv = draft.byId[taskId];
+        if (!conv) return;
+
+        const msgIndex = conv.messages.findIndex(
+          (m) => m.messageId === messageId,
+        );
+        if (msgIndex === -1) {
+          return;
+        }
+
+        const oldMsg = conv.messages[msgIndex];
+        let updatedMsg: MessagesItem = { ...oldMsg, ...metadata };
+
+        // 处理文本 / thought 内容更新
+        if (content !== undefined) {
+          if ("content" in updatedMsg) {
+            updatedMsg.content = partial
+              ? (updatedMsg as TextMessage).content + content
+              : content;
+          } else if ("thought" in updatedMsg) {
+            updatedMsg.thought = partial
+              ? (updatedMsg as ThoughtMessage).thought + content
+              : content;
+          } else if (oldMsg.kind === "text-content") {
+            // 兼容旧消息类型
+            (updatedMsg as any).content = partial
+              ? ((oldMsg as any).content || "") + content
+              : content;
+          }
+        }
+
+        // 处理 tool 消息的特殊字段
+        if ("toolStatus" in updatedMsg || status) {
+          (updatedMsg as any).toolStatus =
+            status || (updatedMsg as any).toolStatus;
+        }
+        if (response !== undefined) {
+          (updatedMsg as any).response = response;
+        }
+
+        const nextMessages = [...conv.messages];
+        nextMessages[msgIndex] = updatedMsg;
+
+        draft.byId[taskId] = {
+          ...conv,
+          messages: nextMessages,
+          lastUpdated: new Date().toISOString(),
+        };
+
+        // 调整 order：更新后置顶
+        const index = draft.order.indexOf(taskId);
+        if (index !== -1) {
+          draft.order.splice(index, 1);
+        }
+        draft.order.push(taskId);
+      }),
+    );
+  }, []);
+
+  const addConversation = useCallback((conversation: Conversation) => {
+    setState(
+      produce((draft) => {
+        const id = conversation.taskId;
+        draft.byId[id] = conversation;
+        // 新会话添加到最前面
+        const index = draft.order.indexOf(id);
+        if (index !== -1) {
+          draft.order.splice(index, 1);
+        }
+        draft.order.push(id);
+      }),
+    );
+  }, []);
+
+  const addLoadingPlaceholder = useCallback(() => {
+    addConversation({
+      taskId: "loading-placeholder",
+      status: "working",
+      messages: [
+        {
+          messageId: "loading-placeholder",
+          role: "assistant",
+          kind: "text-content",
+          content: getText().thinking,
+          timestamp: new Date().toISOString(),
+        },
+      ],
+      lastUpdated: new Date().toISOString(),
+    });
+  }, []);
+
+  useEffect(() => {
+    if (!isLoading) {
+      removeLoadingPlaceholder();
+    }
+  }, [isLoading]);
+
+  const removeLoadingPlaceholder = useCallback(() => {
+    clearConversation("loading-placeholder");
+  }, []);
+
+  const onSendError = useCallback(() => {
+    addConversation({
+      taskId: generateId(),
+      status: "error",
+      messages: [
+        {
+          messageId: `error-${Date.now()}`,
+          role: "assistant",
+          kind: "error",
+          content: getText().errorMessage,
+          timestamp: new Date().toISOString(),
+        } as ErrorMessage,
+      ],
+      lastUpdated: new Date().toISOString(),
+    });
+    setIsLoading(false);
+  }, []);
+
+  // 发送用户消息，启动流式对话
+  const sendMessage = useCallback(
+    async (userContent: string, enableMock = false) => {
+      if (loadingRef.current) {
+        return;
+      }
+      if (!userContent.trim()) return;
+
+      const taskId = `conv-${Date.now()}`;
+
+      // 添加用户消息
+      const userMsgId = generateId();
+
+      addConversation({
+        taskId,
+        status: "submitted",
+        messages: [
+          {
+            messageId: userMsgId,
+            role: "user",
+            kind: "text-content",
+            content: userContent,
+            timestamp: new Date().toISOString(),
+          },
+        ],
+        lastUpdated: new Date().toISOString(),
+      });
+
+      addLoadingPlaceholder();
+
+      setIsLoading(true);
+      let hasResponse = false;
+
+      if (abortControllerRef.current) {
+        abortControllerRef.current.abort();
+      }
+      if (typeof AbortController !== "undefined") {
+        abortControllerRef.current = new AbortController();
+      }
+      const controller = abortControllerRef.current;
+
+      forceStopRef.current = false;
+
+      if (enableMock || userContent === "MAGA") {
+        const { generateMockUIStream } = await import("../mock/mock-data");
+        const cancelMock = generateMockUIStream(taskId, (chunk) => {
+          if (forceStopRef.current) return;
+          hasResponse = true;
+          processDataLine(chunk as any);
+          if (chunk.result.final) {
+            setIsLoading(false);
+          }
+        });
+        abortControllerRef.current?.signal.addEventListener(
+          "abort",
+          cancelMock,
+        );
+        return;
+      }
+
+      try {
+        const stream = client.copilot.chat([
+          { role: "user", content: userContent },
+        ]);
+
+        for await (const chunk of stream) {
+          if (controller?.signal.aborted || forceStopRef.current) break;
+          hasResponse = true;
+
+          // 根据你的实际 stream 返回格式调整
+          processDataLine(chunk);
+
+          // 检查是否结束
+          if ((chunk as any)?.isFinal || (chunk as any)?.final) {
+            break;
+          }
+        }
+      } catch (err: any) {
+        if (err.name === "AbortError" || forceStopRef.current) {
+          console.error("[Chat] Request aborted", err);
+        } else {
+          console.error("[Chat] Stream error:", err);
+          onSendError();
+        }
+      } finally {
+        abort();
+        if (!hasResponse) {
+          // 如果没有任何响应，移除加载占位
+          onSendError();
+        }
+      }
+    },
+    [processDataLine],
+  );
+
+  // 中止当前请求
+  const abort = useCallback(() => {
+    if (abortControllerRef.current) {
+      abortControllerRef.current.abort();
+      abortControllerRef.current = null;
+    }
+    forceStopRef.current = true;
+    setIsLoading(false);
+    console.debug("[Chat] Conversation aborted");
+  }, []);
+
+  // 重置会话
+  const resetConversation = useCallback((greeting?: string) => {
+    abort();
+    setState({ byId: {}, order: [] });
+
+    if (greeting) {
+      const taskId = `conv-${Date.now()}`;
+      setState({
+        byId: {
+          [taskId]: {
+            taskId,
+            status: "submitted",
+            messages: [
+              {
+                messageId: `greeting-${Date.now()}`,
+                role: "assistant",
+                kind: "text-content",
+                content: greeting,
+                timestamp: new Date().toISOString(),
+              } as TextMessage,
+            ],
+            lastUpdated: new Date().toISOString(),
+          },
+        },
+        order: [taskId],
+      });
+    }
+  }, []);
+
+  return {
+    isLoading,
+    conversations,
+    sendMessage,
+    abort,
+    resetConversation,
+    processDataLine,
+    getConversation,
+    clearConversation,
+    removeMessage,
+    updateMessage,
+  };
+}