@mujian/js-sdk 0.0.6-beta.35 → 0.0.6-beta.36

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 import { EVENT } from './events';
+import * as ai from './modules/ai';
 /** @hidden */
 export { EVENT } from './events';
 type Call<T> = {
@@ -58,7 +59,7 @@ declare class MujianSdk {
                     setActive: (personaId: string) => Promise<void>;
                 };
             };
-            complete: (message: string, onData: (data: string) => void, signal?: AbortSignal | undefined) => Promise<string>;
+            complete: (message: string, onData: (data: string | ai.chat.ParsedData) => void, signal?: AbortSignal | undefined, option?: ai.chat.CompleteOption | undefined) => Promise<string>;
             applyRegex: (props: {
                 message: import("./react").Message;
                 index: number;

package/dist/index.js

@@ -18,11 +18,38 @@ var events_EVENT = /*#__PURE__*/ function(EVENT) {
     EVENT["MUJIAN_AI_CHAT_MESSAGE_GET_PROMPT"] = "mujian:ai:chat:message:getPrompt";
     return EVENT;
 }({});
-const chat_complete = async function(message, onData, signal) {
+function wrapOnData(onData) {
+    let buffer = '';
+    return function(data) {
+        buffer += data;
+        const lines = buffer.split('\n');
+        buffer = lines.pop() || '';
+        for (const line of lines)if (line.startsWith('data: ')) try {
+            const parsedData = JSON.parse(line.slice(6));
+            if (parsedData.isFinished) return void onData({
+                isFinished: true,
+                content: ''
+            });
+            const content = parsedData?.choices?.[0]?.delta?.content;
+            if (content?.length > 0) onData({
+                isFinished: false,
+                content
+            });
+        } catch (e) {
+            onData({
+                isFinished: true,
+                error: e,
+                content: ''
+            });
+            return;
+        }
+    };
+}
+const chat_complete = async function(message, onData, signal, option = {}) {
     return await this.call(events_EVENT.MUJIAN_AI_CHAT_COMPLETE, {
         content: message
     }, {
-        onData,
+        onData: option.parseContent ? wrapOnData(onData) : onData,
         signal
     });
 };

package/dist/modules/ai/chat/chat.d.ts

@@ -1,15 +1,46 @@
 import type { MujianSdk } from '../../../index.ts';
 import { Message } from '../../../react';
+/**
+ * LLM生成返回的数据
+ */
+export type ParsedData = {
+    /** 是否已完成生成，最后一块数据时为true */
+    isFinished: boolean;
+    /** LLM生成的delta content */
+    content: string;
+    /** 出错时非空 */
+    error?: unknown;
+};
+/**
+ * 对话补全配置
+ */
+export type CompleteOption = {
+    /** 给onData传递的数据是否是解析后的，推荐true */
+    parseContent?: boolean;
+};
 /**
  * 对话补全
- * @param {string} message 消息
- * @param {Function} onData 数据
- * @param {AbortSignal} signal 信号
  * @returns {Promise<string>} 返回值
  */
 export declare const complete: (
 /** @hidden */
-this: MujianSdk, message: string, onData: (data: string) => void, signal?: AbortSignal) => Promise<string>;
+this: MujianSdk, 
+/**
+ * 消息
+ */
+message: string, 
+/**
+ * 数据生成时的回调
+ */
+onData: (data: string | ParsedData) => void, 
+/**
+ * 停止生成的信号
+ */
+signal?: AbortSignal | undefined, 
+/**
+ * 对话补全配置
+ */
+option?: CompleteOption) => Promise<string>;
 export declare const applyRegex: (this: MujianSdk, props: {
     message: Message;
     index: number;

package/dist/modules/ai/chat/index.d.ts

@@ -1,4 +1,4 @@
-export { applyRegex, complete, continueComplete, regenerate } from './chat';
+export { applyRegex, type CompleteOption, complete, continueComplete, type ParsedData, regenerate, } from './chat';
 export * as message from './message';
 export * as project from './project';
 export * as settings from './settings';

package/dist/react.js

@@ -432,11 +432,38 @@ const MdRendererBase = ({ content })=>{
 const MdRenderer = /*#__PURE__*/ react.memo(MdRendererBase, (prev, next)=>prev.content === next.content);
 MdRendererBase.displayName = "MdRenderer";
 MdRenderer.displayName = "MdRenderer";
-const chat_complete = async function(message, onData, signal) {
+function wrapOnData(onData) {
+    let buffer = '';
+    return function(data) {
+        buffer += data;
+        const lines = buffer.split('\n');
+        buffer = lines.pop() || '';
+        for (const line of lines)if (line.startsWith('data: ')) try {
+            const parsedData = JSON.parse(line.slice(6));
+            if (parsedData.isFinished) return void onData({
+                isFinished: true,
+                content: ''
+            });
+            const content = parsedData?.choices?.[0]?.delta?.content;
+            if (content?.length > 0) onData({
+                isFinished: false,
+                content
+            });
+        } catch (e) {
+            onData({
+                isFinished: true,
+                error: e,
+                content: ''
+            });
+            return;
+        }
+    };
+}
+const chat_complete = async function(message, onData, signal, option = {}) {
     return await this.call("mujian:ai:chat:complete", {
         content: message
     }, {
-        onData,
+        onData: option.parseContent ? wrapOnData(onData) : onData,
         signal
     });
 };
@@ -781,6 +808,7 @@ async function* callSdk(mujian, content, type = 'generate', signal) {
     try {
         const onData = (data)=>{
             if (error) return;
+            if ('string' != typeof data) return;
             queue.push(data);
             if (resolveWait) {
                 resolveWait();

package/dist/umd/index.js

@@ -298,11 +298,38 @@
                 personaId
             });
         };
-        const chat_complete = async function(message, onData, signal) {
+        function wrapOnData(onData) {
+            let buffer = '';
+            return function(data) {
+                buffer += data;
+                const lines = buffer.split('\n');
+                buffer = lines.pop() || '';
+                for (const line of lines)if (line.startsWith('data: ')) try {
+                    const parsedData = JSON.parse(line.slice(6));
+                    if (parsedData.isFinished) return void onData({
+                        isFinished: true,
+                        content: ''
+                    });
+                    const content = parsedData?.choices?.[0]?.delta?.content;
+                    if (content?.length > 0) onData({
+                        isFinished: false,
+                        content
+                    });
+                } catch (e) {
+                    onData({
+                        isFinished: true,
+                        error: e,
+                        content: ''
+                    });
+                    return;
+                }
+            };
+        }
+        const chat_complete = async function(message, onData, signal, option = {}) {
             return await this.call(events_EVENT.MUJIAN_AI_CHAT_COMPLETE, {
                 content: message
             }, {
-                onData,
+                onData: option.parseContent ? wrapOnData(onData) : onData,
                 signal
             });
         };

package/dist/umd/react.js

@@ -6290,11 +6290,38 @@ ${content}
                     personaId
                 });
             };
-            const chat_complete = async function(message, onData, signal) {
+            function wrapOnData(onData) {
+                let buffer = '';
+                return function(data) {
+                    buffer += data;
+                    const lines = buffer.split('\n');
+                    buffer = lines.pop() || '';
+                    for (const line of lines)if (line.startsWith('data: ')) try {
+                        const parsedData = JSON.parse(line.slice(6));
+                        if (parsedData.isFinished) return void onData({
+                            isFinished: true,
+                            content: ''
+                        });
+                        const content = parsedData?.choices?.[0]?.delta?.content;
+                        if (content?.length > 0) onData({
+                            isFinished: false,
+                            content
+                        });
+                    } catch (e) {
+                        onData({
+                            isFinished: true,
+                            error: e,
+                            content: ''
+                        });
+                        return;
+                    }
+                };
+            }
+            const chat_complete = async function(message, onData, signal, option = {}) {
                 return await this.call("mujian:ai:chat:complete", {
                     content: message
                 }, {
-                    onData,
+                    onData: option.parseContent ? wrapOnData(onData) : onData,
                     signal
                 });
             };
@@ -8011,6 +8038,7 @@ ${content}
                 try {
                     const onData = (data)=>{
                         if (error) return;
+                        if ('string' != typeof data) return;
                         queue.push(data);
                         if (resolveWait) {
                             resolveWait();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mujian/js-sdk",
-  "version": "0.0.6-beta.35",
+  "version": "0.0.6-beta.36",
   "type": "module",
   "exports": {
     ".": {