plain-design 1.0.0-beta.150 → 1.0.0-beta.151

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "plain-design",
-  "version": "1.0.0-beta.150",
+  "version": "1.0.0-beta.151",
   "description": "",
   "main": "dist/plain-design.min.js",
   "module": "dist/plain-design.commonjs.min.js",

package/src/packages/components/$ai/index.tsx

@@ -13,6 +13,11 @@ import {toArray} from "@peryl/utils/toArray";
 
 export const $ai = (() => {
 
+  /**
+   * 获取模型服务信息
+   * @author  韦胜健
+   * @date    2025/5/20 19:26
+   */
   const getConfig = (): iAiConfiguration => {
     const aiConfig = $configuration.get('aiConfig');
     if (!aiConfig) {
@@ -22,6 +27,11 @@ export const $ai = (() => {
 
   };
 
+  /**
+   * 带对话历史的模型调用
+   * @author  韦胜健
+   * @date    2025/5/20 19:26
+   */
   const chatHistories = async (histories: iAiHistory[], _aiConfig?: iAiConfiguration): Promise<{ message: string, chatData: iAiChatResponse }> => {
     const axios = Axios.create();
     const aiConfig = _aiConfig || getConfig();
@@ -54,6 +64,11 @@ export const $ai = (() => {
     }
   };
 
+  /**
+   * 带提示词的模型调用
+   * @author  韦胜健
+   * @date    2025/5/20 19:26
+   */
   const chatSystem = async (userContent: string, systemContent?: string, _aiConfig?: iAiConfiguration) => {
     const messages = [{ "role": "user", "content": userContent }];
     if (!!systemContent) {
@@ -62,10 +77,121 @@ export const $ai = (() => {
     return chatHistories(messages, _aiConfig);
   };
 
+  /**
+   * 基础的模型调用
+   * @author  韦胜健
+   * @date    2025/5/20 19:27
+   */
   const chat = async (userContent: string, _aiConfig?: iAiConfiguration): Promise<{ message: string, chatData: iAiChatResponse }> => {
     return chatSystem(userContent, '', _aiConfig);
   };
 
+  /**
+   * 流式调用
+   * @author  韦胜健
+   * @date    2025/5/20 19:27
+   */
+  const chatStream = async (
+    {
+      aiConfig: _aiConfig,
+      messages,
+      onReceiving,
+      onFinish,
+      onThinking,
+      onThinkEnd,
+      onThinkStart,
+    }: {
+      aiConfig?: iAiConfiguration,
+      messages: iAiHistory[],
+      onReceiving: (data: { fullText: string, chunkText: string }) => void,
+      onFinish?: (fullText: string) => void,
+      onThinking?: (thinkText: string) => void,
+      onThinkStart?: () => void,
+      onThinkEnd?: () => void,
+    }
+  ) => {
+    const aiConfig = _aiConfig ?? getConfig();
+    // 发送请求
+    const response = await fetch(aiConfig.url, {
+      method: 'POST',
+      headers: {
+        'Authorization': `Bearer ${aiConfig.key}`,
+        "Content-Type": 'application/json'
+      },
+      body: JSON.stringify({
+        "model": aiConfig.model,
+        "messages": messages,
+        // "messages": [{ role: 'user', content: '你好' }],
+        stream: true // 请求流式响应
+      })
+    });
+
+    if (!response.ok || !response.body) {throw response;}
+
+    // 创建一个解码器来处理UTF-8文本
+    const decoder = new TextDecoder("utf-8");
+
+    let isThinking = false;
+    let thinkText = '';
+    let fullText = '';
+
+    // 读取响应流
+    const reader = response.body.getReader();
+    let done = false;
+
+    while (!done) {
+      const { value, done: readerDone } = await reader.read();
+      done = readerDone;
+      if (done) {
+        console.log('Chat stream done...');
+        onFinish?.(fullText);
+      }
+      if (!value?.length) {
+        continue;
+      }
+
+      const receiveText = (() => {
+        let text = '';
+        let chunk = decoder.decode(value, { stream: true });
+        const pattern = /data: (.*)/g;
+        let match = pattern.exec(chunk);
+        while (match) {
+          if (match[1] === '[DONE]') {
+            return text;
+          } else {
+            const obj = JSON.parse(match[1]);
+            text += obj.choices[0].delta.content ?? '';
+          }
+          match = pattern.exec(chunk);
+        }
+        return text;
+      })();
+
+      if (receiveText.slice(0, 8) === '<think>') {
+        isThinking = true;
+        // thinkText += receiveText;
+        onThinkStart?.();
+        onThinking?.(thinkText);
+      } else if (receiveText.slice(-8) === '</think>') {
+        isThinking = false;
+        // thinkText += receiveText;
+        onThinkEnd?.();
+        onThinking?.(thinkText);
+      } else {
+
+        if (isThinking) {
+          thinkText += receiveText;
+          onThinking?.(thinkText);
+        } else {
+          fullText += receiveText;
+          onReceiving({ fullText: fullText, chunkText: receiveText });
+        }
+      }
+    }
+
+    return response;
+  };
+
 
   /**
    * 根据ai对话，将用户描述转化为表单数据对象，
@@ -139,6 +265,7 @@ export const $ai = (() => {
     chatHistories,
     getConfig,
     getFormDataByChat,
+    chatStream,
   };
 })();