@yh-ui/hooks 0.1.10 → 0.1.15

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

@@ -0,0 +1,29 @@
+import { ref } from "vue";
+export function useAiRequest(options) {
+  const loading = ref(false);
+  const data = ref("");
+  const error = ref(null);
+  const send = async (query, ...args) => {
+    loading.value = true;
+    error.value = null;
+    try {
+      const result = await options.request(query, ...args);
+      data.value = result;
+      options.onSuccess?.(result);
+      return result;
+    } catch (e) {
+      const err = e instanceof Error ? e : new Error(String(e));
+      error.value = err;
+      options.onError?.(err);
+      throw err;
+    } finally {
+      loading.value = false;
+    }
+  };
+  return {
+    loading,
+    data,
+    error,
+    send
+  };
+}

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

@@ -0,0 +1,223 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.qwenParser = exports.plainTextParser = exports.openaiParser = exports.geminiParser = exports.ernieParser = exports.claudeParser = void 0;
+exports.useAiStream = useAiStream;
+var _vue = require("vue");
+const openaiParser = raw => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    if (!line.startsWith("data: ")) continue;
+    const data = line.slice(6).trim();
+    if (data === "[DONE]") break;
+    try {
+      const json = JSON.parse(data);
+      const delta = json?.choices?.[0]?.delta?.content;
+      if (delta) text += delta;
+    } catch {}
+  }
+  return text || null;
+};
+exports.openaiParser = openaiParser;
+const ernieParser = raw => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    if (!line.startsWith("data: ")) continue;
+    const data = line.slice(6).trim();
+    try {
+      const json = JSON.parse(data);
+      if (json?.result) text += json.result;
+    } catch {}
+  }
+  return text || null;
+};
+exports.ernieParser = ernieParser;
+const qwenParser = raw => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    if (!line.startsWith("data: ")) continue;
+    const data = line.slice(6).trim();
+    try {
+      const json = JSON.parse(data);
+      const t = json?.output?.text;
+      if (t) text += t;
+    } catch {}
+  }
+  return text || null;
+};
+exports.qwenParser = qwenParser;
+const claudeParser = raw => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    if (!line.startsWith("data: ")) continue;
+    const data = line.slice(6).trim();
+    try {
+      const json = JSON.parse(data);
+      if (json?.type === "content_block_delta" && json?.delta?.text) {
+        text += json.delta.text;
+      }
+    } catch {}
+  }
+  return text || null;
+};
+exports.claudeParser = claudeParser;
+const geminiParser = raw => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    const content = line.startsWith("data: ") ? line.slice(6).trim() : line.trim();
+    if (!content) continue;
+    try {
+      const json = JSON.parse(content);
+      const part = json?.candidates?.[0]?.content?.parts?.[0]?.text;
+      if (part) text += part;
+    } catch {}
+  }
+  return text || null;
+};
+exports.geminiParser = geminiParser;
+const plainTextParser = raw => raw || null;
+exports.plainTextParser = plainTextParser;
+class TypewriterThrottle {
+  queue = [];
+  rafId = null;
+  onUpdate;
+  charsPerFrame;
+  constructor(onUpdate, charsPerFrame = 3) {
+    this.onUpdate = onUpdate;
+    this.charsPerFrame = charsPerFrame;
+  }
+  push(text) {
+    this.queue.push(...text.split(""));
+    if (this.rafId === null) {
+      this.schedule();
+    }
+  }
+  schedule() {
+    this.rafId = requestAnimationFrame(() => {
+      this.rafId = null;
+      if (this.queue.length === 0) return;
+      const batch = this.queue.splice(0, this.charsPerFrame).join("");
+      this.onUpdate(batch);
+      if (this.queue.length > 0) {
+        this.schedule();
+      }
+    });
+  }
+  flush() {
+    if (this.rafId !== null) {
+      cancelAnimationFrame(this.rafId);
+      this.rafId = null;
+    }
+    if (this.queue.length > 0) {
+      const remaining = this.queue.splice(0).join("");
+      this.onUpdate(remaining);
+    }
+  }
+  cancel() {
+    if (this.rafId !== null) {
+      cancelAnimationFrame(this.rafId);
+      this.rafId = null;
+    }
+    this.queue = [];
+  }
+}
+function useAiStream(options) {
+  const isStreaming = (0, _vue.ref)(false);
+  const currentContent = (0, _vue.ref)("");
+  let abortController = new AbortController();
+  let typewriter = null;
+  const parser = options.parser ?? plainTextParser;
+  const enableTypewriter = options.typewriter !== false;
+  const charsPerFrame = options.charsPerFrame ?? 3;
+  const stop = () => {
+    if (isStreaming.value) {
+      abortController.abort();
+      isStreaming.value = false;
+      typewriter?.flush();
+    }
+  };
+  const fetchStream = async (query, ...args) => {
+    isStreaming.value = true;
+    currentContent.value = "";
+    abortController = new AbortController();
+    if (enableTypewriter) {
+      typewriter = new TypewriterThrottle(chunk => {
+        currentContent.value += chunk;
+        options.onUpdate?.(chunk, currentContent.value);
+      }, charsPerFrame);
+    }
+    const pushText = text => {
+      if (!text) return;
+      if (enableTypewriter && typewriter) {
+        typewriter.push(text);
+      } else {
+        currentContent.value += text;
+        options.onUpdate?.(text, currentContent.value);
+      }
+    };
+    try {
+      const response = await options.request(query, ...args);
+      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) pushText(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) pushText(parsed);
+        }
+      }
+      if (!abortController.signal.aborted) {
+        if (enableTypewriter && typewriter) {
+          typewriter.flush();
+        }
+        isStreaming.value = false;
+        options.onFinish?.(currentContent.value);
+      }
+    } catch (e) {
+      if (e.name !== "AbortError") {
+        options.onError?.(e);
+      }
+      typewriter?.cancel();
+      isStreaming.value = false;
+    }
+  };
+  return {
+    isStreaming,
+    currentContent,
+    fetchStream,
+    stop,
+    // 暴露解析器供测试/自定义使用
+    parsers: {
+      openaiParser,
+      ernieParser,
+      qwenParser,
+      claudeParser,
+      geminiParser,
+      plainTextParser
+    }
+  };
+}

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

@@ -0,0 +1,76 @@
+export type StreamChunkParser = (raw: string) => string | null;
+/**
+ * OpenAI / DeepSeek 格式解析器
+ * data: {"choices":[{"delta":{"content":"hello"}}]}
+ */
+export declare const openaiParser: StreamChunkParser;
+/**
+ * 文心一言 / ERNIE 格式解析器
+ * data: {"result":"hello","is_end":false}
+ */
+export declare const ernieParser: StreamChunkParser;
+/**
+ * 通义千问 / Qwen 格式解析器
+ * data: {"output":{"text":"hello"},"finish_reason":null}
+ */
+export declare const qwenParser: StreamChunkParser;
+/**
+ * Anthropic / Claude 格式解析器
+ * data: {"type":"content_block_delta","delta":{"type":"text_delta","text":"hello"}}
+ */
+export declare const claudeParser: StreamChunkParser;
+/**
+ * Google / Gemini 格式解析器
+ * data: {"candidates":[{"content":{"parts":[{"text":"hello"}]}}]}
+ */
+export declare const geminiParser: StreamChunkParser;
+/**
+ * 纯文本流解析器（AsyncGenerator 输出的原始字符串）
+ */
+export declare const plainTextParser: StreamChunkParser;
+export interface AiStreamOptions {
+    /**
+     * 请求适配器，返回 AsyncGenerator 或 fetch Response
+     */
+    request: (query: string, ...args: unknown[]) => Promise<Response | AsyncGenerator<string, void, unknown>> | AsyncGenerator<string, void, unknown>;
+    /**
+     * 流式块解析器，用于适配不同厂商的格式
+     * @default plainTextParser（直接输出原始字符串）
+     */
+    parser?: StreamChunkParser;
+    /**
+     * 是否启用打字机平滑节流效果
+     * @default true
+     */
+    typewriter?: boolean;
+    /**
+     * 每帧渲染的字符数（打字机速度控制）
+     * @default 3
+     */
+    charsPerFrame?: number;
+    onUpdate?: (chunk: string, fullContent: string) => void;
+    onFinish?: (content: string) => void;
+    onError?: (err: Error) => void;
+}
+/**
+ * useAiStream - 多厂商兼容流式请求引擎
+ *
+ * 特性：
+ * - 支持 OpenAI / DeepSeek / 文心一言 / 通义千问 等主流格式（Adapter 模式）
+ * - 内置 rAF 打字机节流，保证平滑输出体验
+ * - 完整的 AbortController 取消支持
+ */
+export declare function useAiStream(options: AiStreamOptions): {
+    isStreaming: import("vue").Ref<boolean, boolean>;
+    currentContent: import("vue").Ref<string, string>;
+    fetchStream: (query: string, ...args: unknown[]) => Promise<void>;
+    stop: () => void;
+    parsers: {
+        openaiParser: StreamChunkParser;
+        ernieParser: StreamChunkParser;
+        qwenParser: StreamChunkParser;
+        claudeParser: StreamChunkParser;
+        geminiParser: StreamChunkParser;
+        plainTextParser: StreamChunkParser;
+    };
+};

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

@@ -0,0 +1,210 @@
+import { ref } from "vue";
+export const openaiParser = (raw) => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    if (!line.startsWith("data: ")) continue;
+    const data = line.slice(6).trim();
+    if (data === "[DONE]") break;
+    try {
+      const json = JSON.parse(data);
+      const delta = json?.choices?.[0]?.delta?.content;
+      if (delta) text += delta;
+    } catch {
+    }
+  }
+  return text || null;
+};
+export const ernieParser = (raw) => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    if (!line.startsWith("data: ")) continue;
+    const data = line.slice(6).trim();
+    try {
+      const json = JSON.parse(data);
+      if (json?.result) text += json.result;
+    } catch {
+    }
+  }
+  return text || null;
+};
+export const qwenParser = (raw) => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    if (!line.startsWith("data: ")) continue;
+    const data = line.slice(6).trim();
+    try {
+      const json = JSON.parse(data);
+      const t = json?.output?.text;
+      if (t) text += t;
+    } catch {
+    }
+  }
+  return text || null;
+};
+export const claudeParser = (raw) => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    if (!line.startsWith("data: ")) continue;
+    const data = line.slice(6).trim();
+    try {
+      const json = JSON.parse(data);
+      if (json?.type === "content_block_delta" && json?.delta?.text) {
+        text += json.delta.text;
+      }
+    } catch {
+    }
+  }
+  return text || null;
+};
+export const geminiParser = (raw) => {
+  const lines = raw.split("\n");
+  let text = "";
+  for (const line of lines) {
+    const content = line.startsWith("data: ") ? line.slice(6).trim() : line.trim();
+    if (!content) continue;
+    try {
+      const json = JSON.parse(content);
+      const part = json?.candidates?.[0]?.content?.parts?.[0]?.text;
+      if (part) text += part;
+    } catch {
+    }
+  }
+  return text || null;
+};
+export const plainTextParser = (raw) => raw || null;
+class TypewriterThrottle {
+  queue = [];
+  rafId = null;
+  onUpdate;
+  charsPerFrame;
+  constructor(onUpdate, charsPerFrame = 3) {
+    this.onUpdate = onUpdate;
+    this.charsPerFrame = charsPerFrame;
+  }
+  push(text) {
+    this.queue.push(...text.split(""));
+    if (this.rafId === null) {
+      this.schedule();
+    }
+  }
+  schedule() {
+    this.rafId = requestAnimationFrame(() => {
+      this.rafId = null;
+      if (this.queue.length === 0) return;
+      const batch = this.queue.splice(0, this.charsPerFrame).join("");
+      this.onUpdate(batch);
+      if (this.queue.length > 0) {
+        this.schedule();
+      }
+    });
+  }
+  flush() {
+    if (this.rafId !== null) {
+      cancelAnimationFrame(this.rafId);
+      this.rafId = null;
+    }
+    if (this.queue.length > 0) {
+      const remaining = this.queue.splice(0).join("");
+      this.onUpdate(remaining);
+    }
+  }
+  cancel() {
+    if (this.rafId !== null) {
+      cancelAnimationFrame(this.rafId);
+      this.rafId = null;
+    }
+    this.queue = [];
+  }
+}
+export function useAiStream(options) {
+  const isStreaming = ref(false);
+  const currentContent = ref("");
+  let abortController = new AbortController();
+  let typewriter = null;
+  const parser = options.parser ?? plainTextParser;
+  const enableTypewriter = options.typewriter !== false;
+  const charsPerFrame = options.charsPerFrame ?? 3;
+  const stop = () => {
+    if (isStreaming.value) {
+      abortController.abort();
+      isStreaming.value = false;
+      typewriter?.flush();
+    }
+  };
+  const fetchStream = async (query, ...args) => {
+    isStreaming.value = true;
+    currentContent.value = "";
+    abortController = new AbortController();
+    if (enableTypewriter) {
+      typewriter = new TypewriterThrottle((chunk) => {
+        currentContent.value += chunk;
+        options.onUpdate?.(chunk, currentContent.value);
+      }, charsPerFrame);
+    }
+    const pushText = (text) => {
+      if (!text) return;
+      if (enableTypewriter && typewriter) {
+        typewriter.push(text);
+      } else {
+        currentContent.value += text;
+        options.onUpdate?.(text, currentContent.value);
+      }
+    };
+    try {
+      const response = await options.request(query, ...args);
+      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) pushText(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) pushText(parsed);
+        }
+      }
+      if (!abortController.signal.aborted) {
+        if (enableTypewriter && typewriter) {
+          typewriter.flush();
+        }
+        isStreaming.value = false;
+        options.onFinish?.(currentContent.value);
+      }
+    } catch (e) {
+      if (e.name !== "AbortError") {
+        options.onError?.(e);
+      }
+      typewriter?.cancel();
+      isStreaming.value = false;
+    }
+  };
+  return {
+    isStreaming,
+    currentContent,
+    fetchStream,
+    stop,
+    // 暴露解析器供测试/自定义使用
+    parsers: {
+      openaiParser,
+      ernieParser,
+      qwenParser,
+      claudeParser,
+      geminiParser,
+      plainTextParser
+    }
+  };
+}