@mujian/js-sdk 0.0.6-beta.26 → 0.0.6-beta.28

package/dist/react/chat/useChat/index.d.ts

@@ -10,7 +10,7 @@ export type { Message };
  */
 export type UseChatProps = {
     body?: object;
-    onError?: (e: Error) => void;
+    onError?: (e: unknown) => void;
     onFinish?: (message: Message) => void;
 };
 /**
@@ -24,7 +24,7 @@ export type UseChatProps = {
 export type UseChatReturn = {
     messages: Message[];
     status: 'uninitialized' | 'ready' | 'streaming' | 'error';
-    error?: Error;
+    error?: unknown;
     /** 提交用户输入，让AI回答 */
     append: (query: string) => Promise<void>;
     /** 重新生成最后一条回答 */

package/dist/react.js

@@ -96,7 +96,7 @@ function addDOMPurifyHooks() {
  * @returns {string} Encoded message text
  * @copyright https://github.com/kwaroran/risuAI
  */ function encodeStyleTags(text) {
-    const styleRegex = /<style>(.+?)<\/style>/gi;
+    const styleRegex = /<style>([\s\S]+?)<\/style>/gi;
     return text.replaceAll(styleRegex, (_, match)=>`<custom-style>${encodeURIComponent(match)}</custom-style>`);
 }
 /**
@@ -947,39 +947,67 @@ const useChat = ({ body, onError, onFinish })=>{
     }, [
         isStreaming
     ]);
-    const append = async (query)=>{
+    const append = useCallback(async (query)=>{
         if (isStreaming) throw new Error('useChat is streaming already.');
         const controller = new AbortController();
         setAbortController(controller);
-        await chatStreaming({
-            query,
-            signal: controller.signal
-        });
-        setAbortController(void 0);
-    };
-    const regenerate = async ()=>{
+        try {
+            await chatStreaming({
+                query,
+                signal: controller.signal
+            });
+        } catch (e) {
+            setError(e);
+            throw e;
+        } finally{
+            setAbortController(void 0);
+        }
+    }, [
+        isStreaming,
+        chatStreaming
+    ]);
+    const regenerate = useCallback(async ()=>{
         if (isStreaming) throw new Error('useChat is streaming already.');
         const controller = new AbortController();
         setAbortController(controller);
         const lastMessage = messages.findLast((message)=>'user' === message.role);
-        await chatStreaming({
-            query: lastMessage?.content || '',
-            regenerate: true,
-            signal: controller.signal
-        });
-        setAbortController(void 0);
-    };
-    const continueGenerate = async ()=>{
+        try {
+            await chatStreaming({
+                query: lastMessage?.content || '',
+                regenerate: true,
+                signal: controller.signal
+            });
+        } catch (e) {
+            setError(e);
+            throw e;
+        } finally{
+            setAbortController(void 0);
+        }
+    }, [
+        isStreaming,
+        messages,
+        chatStreaming
+    ]);
+    const continueGenerate = useCallback(async ()=>{
         if (isStreaming) throw new Error('useChat is streaming already.');
         const controller = new AbortController();
         setAbortController(controller);
-        await chatStreaming({
-            query: '',
-            is_continue: true,
-            signal: controller.signal
-        });
-        setAbortController(void 0);
-    };
+        try {
+            await chatStreaming({
+                query: '',
+                is_continue: true,
+                signal: controller.signal
+            });
+        } catch (e) {
+            setError(e);
+            throw e;
+        } finally{
+            setAbortController(void 0);
+        }
+    }, [
+        isStreaming,
+        chatStreaming
+    ]);
     const stop = ()=>{
         abortController?.abort();
     };

package/package.json

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