@yh-ui/hooks 0.1.10 → 0.1.12

package/dist/use-ai/use-ai-chat.cjs

@@ -0,0 +1,193 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.useAiChat = useAiChat;
+var _vue = require("vue");
+var _useAiStream = require("./use-ai-stream.cjs");
+function createTypewriter(onChar, charsPerFrame) {
+  const queue = [];
+  let rafId = null;
+  const schedule = () => {
+    rafId = requestAnimationFrame(() => {
+      rafId = null;
+      if (queue.length === 0) return;
+      const batch = queue.splice(0, charsPerFrame).join("");
+      onChar(batch);
+      if (queue.length > 0) schedule();
+    });
+  };
+  return {
+    push(text) {
+      queue.push(...text.split(""));
+      if (rafId === null) schedule();
+    },
+    flush() {
+      if (rafId !== null) {
+        cancelAnimationFrame(rafId);
+        rafId = null;
+      }
+      if (queue.length > 0) {
+        onChar(queue.splice(0).join(""));
+      }
+    },
+    cancel() {
+      if (rafId !== null) {
+        cancelAnimationFrame(rafId);
+        rafId = null;
+      }
+      queue.length = 0;
+    }
+  };
+}
+function useAiChat(options = {}) {
+  const {
+    idGenerator = () => Math.random().toString(36).substring(2, 9),
+    parser = _useAiStream.plainTextParser,
+    typewriter: enableTypewriter = true,
+    charsPerFrame = 3,
+    systemPrompt
+  } = options;
+  const messages = (0, _vue.ref)(options.initialMessages ?? []);
+  const isGenerating = (0, _vue.ref)(false);
+  const isSending = (0, _vue.computed)(() => isGenerating.value);
+  let abortController = null;
+  const stop = () => {
+    if (abortController && isGenerating.value) {
+      abortController.abort();
+      isGenerating.value = false;
+      const lastMsg = messages.value[messages.value.length - 1];
+      if (lastMsg?.role === "assistant" && lastMsg.status === "generating") {
+        lastMsg.status = "stopped";
+      }
+    }
+  };
+  const clear = () => {
+    stop();
+    messages.value = [];
+  };
+  const removeMessage = id => {
+    const idx = messages.value.findIndex(m => m.id === id);
+    if (idx !== -1) messages.value.splice(idx, 1);
+  };
+  const updateMessage = (id, patch) => {
+    const msg = messages.value.find(m => m.id === id);
+    if (msg) Object.assign(msg, patch);
+  };
+  const sendMessage = async content => {
+    if (!content.trim() || isGenerating.value) return;
+    messages.value.push({
+      id: idGenerator(),
+      role: "user",
+      content,
+      createAt: Date.now(),
+      status: "success"
+    });
+    if (!options.request) return;
+    const assId = idGenerator();
+    const assistantMsg = {
+      id: assId,
+      role: "assistant",
+      content: "",
+      createAt: Date.now(),
+      status: "loading"
+    };
+    messages.value.push(assistantMsg);
+    isGenerating.value = true;
+    abortController = new AbortController();
+    const history = [];
+    if (systemPrompt) {
+      history.push({
+        id: "system",
+        role: "system",
+        content: systemPrompt,
+        createAt: 0,
+        status: "success"
+      });
+    }
+    history.push(...messages.value.slice(0, -2));
+    try {
+      const response = await options.request(content, history, abortController.signal);
+      const targetMsg = messages.value.find(m => m.id === assId);
+      targetMsg.status = "generating";
+      let typewriterInstance = null;
+      if (enableTypewriter && typeof requestAnimationFrame !== "undefined") {
+        typewriterInstance = createTypewriter(chars => {
+          targetMsg.content += chars;
+        }, charsPerFrame);
+      }
+      const pushChunk = text => {
+        if (!text) return;
+        if (typewriterInstance) {
+          typewriterInstance.push(text);
+        } else {
+          targetMsg.content += text;
+        }
+      };
+      if (typeof response === "object" && response !== null && Symbol.asyncIterator in response) {
+        for await (const chunk of response) {
+          if (abortController.signal.aborted) break;
+          const parsed = parser(chunk);
+          if (parsed) pushChunk(parsed);
+        }
+      } else if (response instanceof Response && response.body) {
+        const reader = response.body.getReader();
+        const decoder = new TextDecoder("utf-8");
+        while (true) {
+          if (abortController.signal.aborted) {
+            reader.cancel();
+            break;
+          }
+          const {
+            done,
+            value
+          } = await reader.read();
+          if (done) break;
+          const chunkStr = decoder.decode(value, {
+            stream: true
+          });
+          const parsed = parser(chunkStr);
+          if (parsed) pushChunk(parsed);
+        }
+      } else if (typeof response === "string") {
+        pushChunk(response);
+      }
+      if (typewriterInstance) {
+        typewriterInstance.flush();
+      }
+      if (!abortController.signal.aborted) {
+        targetMsg.status = "success";
+        options.onFinish?.(targetMsg);
+      }
+    } catch (e) {
+      if (e.name !== "AbortError") {
+        const targetMsg = messages.value.find(m => m.id === assId);
+        if (targetMsg) targetMsg.status = "error";
+        options.onError?.(e);
+      }
+    } finally {
+      if (isGenerating.value) {
+        isGenerating.value = false;
+      }
+    }
+  };
+  return {
+    /** 会话历史 */
+    messages,
+    /** 是否正在生成（等同 isSending，别名友好） */
+    isGenerating,
+    /** 同 isGenerating，语义别名 */
+    isSending,
+    /** 触发发送（自动处理流、打字机） */
+    sendMessage,
+    /** 停止/中断当前生成 */
+    stop,
+    /** 移除单条消息 */
+    removeMessage,
+    /** 修改单条消息内容 */
+    updateMessage,
+    /** 重置清空所有会话 */
+    clear
+  };
+}

package/dist/use-ai/use-ai-chat.d.ts

@@ -0,0 +1,115 @@
+import { type StreamChunkParser } from './use-ai-stream';
+export interface AiChatMessage {
+    /** 唯一标识，避免使用 index 做 key */
+    id: string;
+    /** 消息发送方 */
+    role: 'user' | 'assistant' | 'system';
+    /** 消息内容 */
+    content: string;
+    /**
+     * 消息状态
+     * - loading: 初始加载中（占位）
+     * - generating: 流式生成中
+     * - success: 已成功完成
+     * - error: 发生错误
+     * - stopped: 被用户中途打断
+     */
+    status?: 'loading' | 'generating' | 'success' | 'error' | 'stopped';
+    /** 消息时间戳（ms） */
+    createAt: number;
+    /** 用于展示的时间字符串（可选，不传则自动格式化） */
+    time?: string;
+    /** 扩展字段 */
+    [key: string]: unknown;
+}
+export interface UseAiChatOptions {
+    /** 初始化的消息列表 */
+    initialMessages?: AiChatMessage[];
+    /** 自定义生成 ID 的函数 */
+    idGenerator?: () => string;
+    /**
+     * 请求适配器
+     * 支持：
+     * - AsyncGenerator<string>：最原始的字符流，每次 yield 一段字符
+     * - Promise<string>：直接返回完整回复
+     * - Promise<Response>：SSE 流式响应
+     */
+    request?: (message: string, history: AiChatMessage[], abortSignal: AbortSignal) => AsyncGenerator<string, void, unknown> | Promise<string | Response>;
+    /**
+     * SSE / 流式块的解析器（适配不同厂商格式）
+     * 传入各厂商对应的 parser（来自 useAiStream 模块）
+     * @default plainTextParser
+     */
+    parser?: StreamChunkParser;
+    /**
+     * 是否启用打字机平滑输出效果
+     * @default true
+     */
+    typewriter?: boolean;
+    /**
+     * 打字机每帧输出字符数（越大越快）
+     * @default 3
+     */
+    charsPerFrame?: number;
+    /**
+     * 系统 Prompt（会自动插入到历史的首位）
+     */
+    systemPrompt?: string;
+    /** 出现错误时的回调 */
+    onError?: (err: Error) => void;
+    /** 消息发送完成回调 */
+    onFinish?: (message: AiChatMessage) => void;
+}
+/**
+ * useAiChat - 核心 AI 会话管理 Hook
+ *
+ * 特性：
+ * - 消息列表 CRUD + 状态机管理
+ * - 支持流式 / 非流式响应
+ * - 内置多厂商适配器接口（通过 parser 选项传入）
+ * - rAF 打字机平滑效果（可关闭）
+ * - 完整的 AbortController 取消支持
+ * - 系统 Prompt 自动注入
+ */
+export declare function useAiChat(options?: UseAiChatOptions): {
+    /** 会话历史 */
+    messages: import("vue").Ref<{
+        [x: string]: unknown;
+        id: string;
+        role: "user" | "assistant" | "system";
+        content: string;
+        status?: "loading" | "generating" | "success" | "error" | "stopped"
+        /** 消息时间戳（ms） */
+         | undefined;
+        createAt: number;
+        time?: string
+        /** 扩展字段 */
+         | undefined;
+    }[], AiChatMessage[] | {
+        [x: string]: unknown;
+        id: string;
+        role: "user" | "assistant" | "system";
+        content: string;
+        status?: "loading" | "generating" | "success" | "error" | "stopped"
+        /** 消息时间戳（ms） */
+         | undefined;
+        createAt: number;
+        time?: string
+        /** 扩展字段 */
+         | undefined;
+    }[]>;
+    /** 是否正在生成（等同 isSending，别名友好） */
+    isGenerating: import("vue").Ref<boolean, boolean>;
+    /** 同 isGenerating，语义别名 */
+    isSending: import("vue").ComputedRef<boolean>;
+    /** 触发发送（自动处理流、打字机） */
+    sendMessage: (content: string) => Promise<void>;
+    /** 停止/中断当前生成 */
+    stop: () => void;
+    /** 移除单条消息 */
+    removeMessage: (id: string) => void;
+    /** 修改单条消息内容 */
+    updateMessage: (id: string, patch: Partial<AiChatMessage>) => void;
+    /** 重置清空所有会话 */
+    clear: () => void;
+};

package/dist/use-ai/use-ai-chat.mjs

@@ -0,0 +1,182 @@
+import { ref, computed } from "vue";
+import { plainTextParser } from "./use-ai-stream.mjs";
+function createTypewriter(onChar, charsPerFrame) {
+  const queue = [];
+  let rafId = null;
+  const schedule = () => {
+    rafId = requestAnimationFrame(() => {
+      rafId = null;
+      if (queue.length === 0) return;
+      const batch = queue.splice(0, charsPerFrame).join("");
+      onChar(batch);
+      if (queue.length > 0) schedule();
+    });
+  };
+  return {
+    push(text) {
+      queue.push(...text.split(""));
+      if (rafId === null) schedule();
+    },
+    flush() {
+      if (rafId !== null) {
+        cancelAnimationFrame(rafId);
+        rafId = null;
+      }
+      if (queue.length > 0) {
+        onChar(queue.splice(0).join(""));
+      }
+    },
+    cancel() {
+      if (rafId !== null) {
+        cancelAnimationFrame(rafId);
+        rafId = null;
+      }
+      queue.length = 0;
+    }
+  };
+}
+export function useAiChat(options = {}) {
+  const {
+    idGenerator = () => Math.random().toString(36).substring(2, 9),
+    parser = plainTextParser,
+    typewriter: enableTypewriter = true,
+    charsPerFrame = 3,
+    systemPrompt
+  } = options;
+  const messages = ref(options.initialMessages ?? []);
+  const isGenerating = ref(false);
+  const isSending = computed(() => isGenerating.value);
+  let abortController = null;
+  const stop = () => {
+    if (abortController && isGenerating.value) {
+      abortController.abort();
+      isGenerating.value = false;
+      const lastMsg = messages.value[messages.value.length - 1];
+      if (lastMsg?.role === "assistant" && lastMsg.status === "generating") {
+        lastMsg.status = "stopped";
+      }
+    }
+  };
+  const clear = () => {
+    stop();
+    messages.value = [];
+  };
+  const removeMessage = (id) => {
+    const idx = messages.value.findIndex((m) => m.id === id);
+    if (idx !== -1) messages.value.splice(idx, 1);
+  };
+  const updateMessage = (id, patch) => {
+    const msg = messages.value.find((m) => m.id === id);
+    if (msg) Object.assign(msg, patch);
+  };
+  const sendMessage = async (content) => {
+    if (!content.trim() || isGenerating.value) return;
+    messages.value.push({
+      id: idGenerator(),
+      role: "user",
+      content,
+      createAt: Date.now(),
+      status: "success"
+    });
+    if (!options.request) return;
+    const assId = idGenerator();
+    const assistantMsg = {
+      id: assId,
+      role: "assistant",
+      content: "",
+      createAt: Date.now(),
+      status: "loading"
+    };
+    messages.value.push(assistantMsg);
+    isGenerating.value = true;
+    abortController = new AbortController();
+    const history = [];
+    if (systemPrompt) {
+      history.push({
+        id: "system",
+        role: "system",
+        content: systemPrompt,
+        createAt: 0,
+        status: "success"
+      });
+    }
+    history.push(...messages.value.slice(0, -2));
+    try {
+      const response = await options.request(content, history, abortController.signal);
+      const targetMsg = messages.value.find((m) => m.id === assId);
+      targetMsg.status = "generating";
+      let typewriterInstance = null;
+      if (enableTypewriter && typeof requestAnimationFrame !== "undefined") {
+        typewriterInstance = createTypewriter((chars) => {
+          targetMsg.content += chars;
+        }, charsPerFrame);
+      }
+      const pushChunk = (text) => {
+        if (!text) return;
+        if (typewriterInstance) {
+          typewriterInstance.push(text);
+        } else {
+          targetMsg.content += text;
+        }
+      };
+      if (typeof response === "object" && response !== null && Symbol.asyncIterator in response) {
+        for await (const chunk of response) {
+          if (abortController.signal.aborted) break;
+          const parsed = parser(chunk);
+          if (parsed) pushChunk(parsed);
+        }
+      } else if (response instanceof Response && response.body) {
+        const reader = response.body.getReader();
+        const decoder = new TextDecoder("utf-8");
+        while (true) {
+          if (abortController.signal.aborted) {
+            reader.cancel();
+            break;
+          }
+          const { done, value } = await reader.read();
+          if (done) break;
+          const chunkStr = decoder.decode(value, { stream: true });
+          const parsed = parser(chunkStr);
+          if (parsed) pushChunk(parsed);
+        }
+      } else if (typeof response === "string") {
+        pushChunk(response);
+      }
+      if (typewriterInstance) {
+        typewriterInstance.flush();
+      }
+      if (!abortController.signal.aborted) {
+        targetMsg.status = "success";
+        options.onFinish?.(targetMsg);
+      }
+    } catch (e) {
+      if (e.name !== "AbortError") {
+        const targetMsg = messages.value.find((m) => m.id === assId);
+        if (targetMsg) targetMsg.status = "error";
+        options.onError?.(e);
+      }
+    } finally {
+      if (isGenerating.value) {
+        isGenerating.value = false;
+      }
+    }
+  };
+  return {
+    /** 会话历史 */
+    messages,
+    /** 是否正在生成（等同 isSending，别名友好） */
+    isGenerating,
+    /** 同 isGenerating，语义别名 */
+    isSending,
+    /** 触发发送（自动处理流、打字机） */
+    sendMessage,
+    /** 停止/中断当前生成 */
+    stop,
+    /** 移除单条消息 */
+    removeMessage,
+    /** 修改单条消息内容 */
+    updateMessage,
+    /** 重置清空所有会话 */
+    clear
+  };
+}