@mujian/js-sdk 0.0.6-beta.50 → 0.0.6-beta.52

package/dist/events/index.d.ts

@@ -40,6 +40,10 @@ export declare enum EVENT {
      * 获取对话
      */
     MUJIAN_AI_CHAT_MESSAGE_GET_ALL = "mujian:ai:chat:message:getAll",
+    /**
+     * 获取对话分页
+     */
+    MUJIAN_AI_CHAT_MESSAGE_GET_PAGE = "mujian:ai:chat:message:getPage",
     /**
      * 获取项目信息
      */

package/dist/index.js

@@ -9,6 +9,7 @@ var events_EVENT = /*#__PURE__*/ function(EVENT) {
     EVENT["MUJIAN_AI_OPENAI_CHAT_COMPLETIONS_CREATE"] = "mujian:ai:openai:chat:completions:create";
     EVENT["MUJIAN_AI_OPENAI_RESPONSES_CREATE"] = "mujian:ai:openai:responses:create";
     EVENT["MUJIAN_AI_CHAT_MESSAGE_GET_ALL"] = "mujian:ai:chat:message:getAll";
+    EVENT["MUJIAN_AI_CHAT_MESSAGE_GET_PAGE"] = "mujian:ai:chat:message:getPage";
     EVENT["MUJIAN_AI_CHAT_PROJECT_GET_INFO"] = "mujian:ai:chat:project:getInfo";
     EVENT["MUJIAN_AI_SETTINGS_PERSONA_GET_ACTIVE"] = "mujian:ai:settings:persona:getActive";
     EVENT["MUJIAN_AI_SETTINGS_PERSONA_SET_ACTIVE"] = "mujian:ai:settings:persona:setActive";
@@ -118,6 +119,12 @@ const getPrompt = async function(messageId) {
         messageId
     });
 };
+async function getPage(fromCursor, pageSize) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_MESSAGE_GET_PAGE, {
+        fromCursor,
+        pageSize
+    });
+}
 const getInfo = async function() {
     return await this.call(events_EVENT.MUJIAN_AI_CHAT_PROJECT_GET_INFO);
 };
@@ -254,7 +261,8 @@ class MujianSdk {
                 deleteOne: messageDeleteOne.bind(this),
                 editOne: messageEditOne.bind(this),
                 swipe: messageSwipe.bind(this),
-                getPrompt: getPrompt.bind(this)
+                getPrompt: getPrompt.bind(this),
+                getPage: getPage.bind(this)
             }
         },
         text: {
@@ -453,7 +461,8 @@ class src_MujianSdk {
                 deleteOne: messageDeleteOne.bind(this),
                 editOne: messageEditOne.bind(this),
                 swipe: messageSwipe.bind(this),
-                getPrompt: getPrompt.bind(this)
+                getPrompt: getPrompt.bind(this),
+                getPage: getPage.bind(this)
             }
         },
         text: {

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

@@ -1,5 +1,6 @@
 import type { MujianSdk } from '../../../../index.ts';
 import type { Message } from '../../../../react/chat/useChat/message';
+import { Cursor } from '../../../../types';
 /**
  * 获取消息列表
  * @returns 消息当前列表
@@ -41,3 +42,6 @@ this: MujianSdk, messageId: string, swipeId: number) => Promise<unknown>;
 export declare const getPrompt: (
 /** @hidden */
 this: MujianSdk, messageId: string) => Promise<unknown>;
+export declare function getPage(
+/** @hidden */
+this: MujianSdk, fromCursor: Cursor, pageSize: number): Promise<unknown>;

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

@@ -12,6 +12,8 @@ export type UseChatProps = {
     body?: object;
     onError?: (e: unknown) => void;
     onFinish?: (message: Message) => void;
+    /** 消息分页大小，不传时不分页 */
+    pageSize?: number;
 };
 /**
  * Use chat return
@@ -39,6 +41,8 @@ export type UseChatReturn = {
     setSwipe: (messageId: string, swipeId: number) => Promise<void>;
     editMessage: (messageId: string, content: string) => Promise<void>;
     deleteMessage: (messageId: string) => Promise<void>;
+    loadMoreMessage: () => Promise<void>;
+    messagesStatus: 'all-loaded' | 'loading' | 'has-more' | 'error';
 };
 /**
  * Use chat common
@@ -52,4 +56,4 @@ export type UseChatCommon = {
  * Use chat
  * @returns {UseChatReturn} The chat.
  */
-export declare const useChat: ({ body, onError, onFinish, }: UseChatProps) => UseChatReturn;
+export declare const useChat: ({ body, onError, onFinish, pageSize, }: UseChatProps) => UseChatReturn;

package/dist/react.js

@@ -1,4 +1,4 @@
-import { useMount, useRequest, useUpdateEffect } from "ahooks";
+import { useInfiniteScroll, useRequest, useUpdateEffect } from "ahooks";
 import react, { createContext, forwardRef, useCallback, useContext, useEffect, useRef, useState } from "react";
 import { Fragment, jsx, jsxs } from "react/jsx-runtime";
 import css_tools from "@adobe/css-tools";
@@ -507,6 +507,7 @@ var events_EVENT = /*#__PURE__*/ function(EVENT) {
     EVENT["MUJIAN_AI_OPENAI_CHAT_COMPLETIONS_CREATE"] = "mujian:ai:openai:chat:completions:create";
     EVENT["MUJIAN_AI_OPENAI_RESPONSES_CREATE"] = "mujian:ai:openai:responses:create";
     EVENT["MUJIAN_AI_CHAT_MESSAGE_GET_ALL"] = "mujian:ai:chat:message:getAll";
+    EVENT["MUJIAN_AI_CHAT_MESSAGE_GET_PAGE"] = "mujian:ai:chat:message:getPage";
     EVENT["MUJIAN_AI_CHAT_PROJECT_GET_INFO"] = "mujian:ai:chat:project:getInfo";
     EVENT["MUJIAN_AI_SETTINGS_PERSONA_GET_ACTIVE"] = "mujian:ai:settings:persona:getActive";
     EVENT["MUJIAN_AI_SETTINGS_PERSONA_SET_ACTIVE"] = "mujian:ai:settings:persona:setActive";
@@ -616,6 +617,12 @@ const getPrompt = async function(messageId) {
         messageId
     });
 };
+async function getPage(fromCursor, pageSize) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_MESSAGE_GET_PAGE, {
+        fromCursor,
+        pageSize
+    });
+}
 const getInfo = async function() {
     return await this.call(events_EVENT.MUJIAN_AI_CHAT_PROJECT_GET_INFO);
 };
@@ -795,7 +802,8 @@ class MujianSdk {
                 deleteOne: messageDeleteOne.bind(this),
                 editOne: messageEditOne.bind(this),
                 swipe: messageSwipe.bind(this),
-                getPrompt: getPrompt.bind(this)
+                getPrompt: getPrompt.bind(this),
+                getPage: getPage.bind(this)
             }
         },
         text: {
@@ -1170,9 +1178,8 @@ const useChatStreaming = ({ common, setError, setMessages, mujian })=>{
         isStreaming
     };
 };
-const useChat = ({ body, onError, onFinish })=>{
+const useChat = ({ body, onError, onFinish, pageSize })=>{
     const [error, _setError] = useState();
-    const [messages, setMessages] = useState([]);
     const [input, setInput] = useState('');
     const [abortController, setAbortController] = useState();
     const mujian = useMujian();
@@ -1180,6 +1187,45 @@ const useChat = ({ body, onError, onFinish })=>{
         _setError(error);
         if (error) onError?.(error);
     };
+    const isLoadingMore = useRef(false);
+    const { data, loadMoreAsync, loadingMore, loading, mutate, error: messageLoadError } = useInfiniteScroll(async (current)=>{
+        isLoadingMore.current = true;
+        if (current?.noMore) return {
+            noMore: true,
+            list: []
+        };
+        try {
+            const resp = await mujian.ai.chat.message.getPage(current?.cursor, pageSize);
+            return {
+                cursor: resp.nextCursor,
+                list: resp.messages,
+                noMore: !resp.nextCursor
+            };
+        } finally{
+            isLoadingMore.current = false;
+        }
+    }, {
+        direction: 'top'
+    });
+    const loadMoreMessage = async ()=>{
+        if (isLoadingMore.current) return;
+        try {
+            isLoadingMore.current = true;
+            await loadMoreAsync();
+        } finally{
+            isLoadingMore.current = false;
+        }
+    };
+    const messages = data?.list ?? [];
+    const setMessages = (messages)=>{
+        'function' == typeof messages ? mutate({
+            cursor: data?.cursor,
+            list: messages(data?.list ?? [])
+        }) : mutate({
+            cursor: data?.cursor,
+            list: messages
+        });
+    };
     const { chatStreaming, isStreaming } = useChatStreaming({
         common: {
             body
@@ -1188,10 +1234,6 @@ const useChat = ({ body, onError, onFinish })=>{
         setMessages,
         mujian
     });
-    useMount(async ()=>{
-        const resp = await mujian.ai.chat.message.getAll();
-        setMessages(resp);
-    });
     useUpdateEffect(()=>{
         if (!isStreaming) {
             const lastMessage = messages[messages.length - 1];
@@ -1316,7 +1358,9 @@ const useChat = ({ body, onError, onFinish })=>{
         },
         setSwipe,
         editMessage,
-        deleteMessage
+        deleteMessage,
+        loadMoreMessage,
+        messagesStatus: messageLoadError ? 'error' : loadingMore || loading ? 'loading' : data?.noMore ? 'all-loaded' : 'has-more'
     };
 };
 export { MdRenderer, MujianProvider, MujianSpinner, Thread, useChat, useMujian };

package/dist/types/index.d.ts

@@ -58,3 +58,7 @@ export type QuickReplyConfig = {
     qrList: Array<QuickReply>;
     idIndex: number;
 };
+export type Cursor = {
+    index: number;
+    messageId: string;
+};

package/dist/umd/index.js

@@ -267,6 +267,7 @@
             EVENT["MUJIAN_AI_OPENAI_CHAT_COMPLETIONS_CREATE"] = "mujian:ai:openai:chat:completions:create";
             EVENT["MUJIAN_AI_OPENAI_RESPONSES_CREATE"] = "mujian:ai:openai:responses:create";
             EVENT["MUJIAN_AI_CHAT_MESSAGE_GET_ALL"] = "mujian:ai:chat:message:getAll";
+            EVENT["MUJIAN_AI_CHAT_MESSAGE_GET_PAGE"] = "mujian:ai:chat:message:getPage";
             EVENT["MUJIAN_AI_CHAT_PROJECT_GET_INFO"] = "mujian:ai:chat:project:getInfo";
             EVENT["MUJIAN_AI_SETTINGS_PERSONA_GET_ACTIVE"] = "mujian:ai:settings:persona:getActive";
             EVENT["MUJIAN_AI_SETTINGS_PERSONA_SET_ACTIVE"] = "mujian:ai:settings:persona:setActive";
@@ -376,6 +377,12 @@
                 messageId
             });
         };
+        async function getPage(fromCursor, pageSize) {
+            return await this.call(events_EVENT.MUJIAN_AI_CHAT_MESSAGE_GET_PAGE, {
+                fromCursor,
+                pageSize
+            });
+        }
         const getInfo = async function() {
             return await this.call(events_EVENT.MUJIAN_AI_CHAT_PROJECT_GET_INFO);
         };
@@ -512,7 +519,8 @@
                         deleteOne: messageDeleteOne.bind(this),
                         editOne: messageEditOne.bind(this),
                         swipe: messageSwipe.bind(this),
-                        getPrompt: getPrompt.bind(this)
+                        getPrompt: getPrompt.bind(this),
+                        getPage: getPage.bind(this)
                     }
                 },
                 text: {
@@ -711,7 +719,8 @@
                         deleteOne: messageDeleteOne.bind(this),
                         editOne: messageEditOne.bind(this),
                         swipe: messageSwipe.bind(this),
-                        getPrompt: getPrompt.bind(this)
+                        getPrompt: getPrompt.bind(this),
+                        getPage: getPage.bind(this)
                     }
                 },
                 text: {