@blocklet/pages-kit 0.2.291 → 0.2.293

package/lib/esm/builtin/async/ai-runtime/components/runtime/state/session.js

@@ -0,0 +1,209 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
+import { create } from 'zustand';
+import { immer } from 'zustand/middleware/immer';
+import { getMessages } from '../../../api/message';
+import { createSession, getSessions, runAssistant } from '../../../api/session';
+import { parseIdentity } from '../../../utils';
+const STATES = {};
+const GET_MESSAGES_LIMIT = 100;
+const GET_MESSAGES_ORDER_DIRECTION = 'desc';
+export const createSessionState = ({ aid }) => {
+    var _a;
+    (_a = STATES[aid]) !== null && _a !== void 0 ? _a : (STATES[aid] = create()(immer((set, get) => ({
+        reset: () => {
+            set((state) => {
+                state.loading = undefined;
+                state.loaded = undefined;
+                state.running = undefined;
+                state.session = undefined;
+                state.messages = undefined;
+                state.noMoreMessage = undefined;
+                state.messageLoading = undefined;
+                state.error = undefined;
+            });
+        },
+        load: () => __awaiter(void 0, void 0, void 0, function* () {
+            var _b;
+            set((state) => {
+                state.loading = true;
+            });
+            try {
+                const { sessions } = yield getSessions(parseIdentity(aid, { rejectWhenError: true }));
+                const session = (_b = sessions[0]) !== null && _b !== void 0 ? _b : (yield createSession({ assistantId: aid })).created;
+                const { messages } = yield getMessages({
+                    sessionId: session.id,
+                    orderDirection: GET_MESSAGES_ORDER_DIRECTION,
+                    limit: GET_MESSAGES_LIMIT,
+                });
+                set((state) => {
+                    state.session = session;
+                    state.messages = messages.toReversed();
+                    state.noMoreMessage = messages.length < GET_MESSAGES_LIMIT;
+                });
+            }
+            catch (error) {
+                set((state) => {
+                    state.error = error;
+                });
+                throw error;
+            }
+            finally {
+                set((state) => {
+                    state.loaded = true;
+                    state.loading = false;
+                });
+            }
+        }),
+        loadMoreMessages: () => __awaiter(void 0, void 0, void 0, function* () {
+            var _c;
+            set((state) => {
+                state.messageLoading = true;
+            });
+            const { session, messages } = get();
+            if (!session)
+                return;
+            try {
+                const result = yield getMessages({
+                    sessionId: session.id,
+                    before: (_c = messages === null || messages === void 0 ? void 0 : messages.slice(-1)[0]) === null || _c === void 0 ? void 0 : _c.id,
+                    orderDirection: GET_MESSAGES_ORDER_DIRECTION,
+                    limit: GET_MESSAGES_LIMIT,
+                });
+                set((state) => {
+                    var _a;
+                    state.messages = ((_a = state.messages) !== null && _a !== void 0 ? _a : []).concat(result.messages.toReversed());
+                    state.noMoreMessage = result.messages.length < GET_MESSAGES_LIMIT;
+                });
+            }
+            finally {
+                set((state) => {
+                    state.messageLoading = false;
+                });
+            }
+        }),
+        execute: (_d) => __awaiter(void 0, [_d], void 0, function* ({ aid, working, parameters, onResponseStart, }) {
+            var _e, e_1, _f, _g;
+            let { session } = get();
+            if (!session) {
+                yield get().load();
+                session = get().session;
+            }
+            if (!session)
+                throw new Error('This session is not initialized');
+            const { id: sessionId } = session;
+            const identity = parseIdentity(aid, { rejectWhenError: true });
+            let message;
+            set((state) => {
+                state.running = true;
+                state.error = undefined;
+            });
+            try {
+                const stream = runAssistant({
+                    sessionId,
+                    identity,
+                    working,
+                    parameters: Object.assign(Object.assign({}, parameters), { $clientTime: new Date().toISOString() }),
+                });
+                let responseStarted = false;
+                try {
+                    for (var _h = true, stream_1 = __asyncValues(stream), stream_1_1; stream_1_1 = yield stream_1.next(), _e = stream_1_1.done, !_e; _h = true) {
+                        _g = stream_1_1.value;
+                        _h = false;
+                        const value = _g;
+                        if (!responseStarted) {
+                            responseStarted = true;
+                            onResponseStart === null || onResponseStart === void 0 ? void 0 : onResponseStart();
+                        }
+                        if ((value === null || value === void 0 ? void 0 : value.type) === 'CHUNK') {
+                            if (!message) {
+                                message = {
+                                    taskId: value.taskId,
+                                    assistantId: identity.assistantId,
+                                    sessionId,
+                                    parameters,
+                                    createdAt: new Date().toISOString(),
+                                    updatedAt: new Date().toISOString(),
+                                    loading: true,
+                                };
+                                set((state) => {
+                                    var _a;
+                                    (_a = state.messages) !== null && _a !== void 0 ? _a : (state.messages = []);
+                                    state.messages.push(message);
+                                });
+                            }
+                            if (message.taskId === value.taskId) {
+                                requestAnimationFrame(() => {
+                                    set((state) => {
+                                        var _a, _b, _c, _d, _e, _f;
+                                        var _g;
+                                        const msg = (_a = state.messages) === null || _a === void 0 ? void 0 : _a.findLast((i) => i.taskId === message.taskId);
+                                        if (!msg)
+                                            return;
+                                        (_b = msg.result) !== null && _b !== void 0 ? _b : (msg.result = {});
+                                        msg.result.content = (msg.result.content || '') + (value.delta.content || '');
+                                        if ((_d = (_c = value.delta) === null || _c === void 0 ? void 0 : _c.images) === null || _d === void 0 ? void 0 : _d.length) {
+                                            msg.result.images = ((_e = msg.result.images) !== null && _e !== void 0 ? _e : []).concat(value.delta.images);
+                                        }
+                                        if (value.delta.object) {
+                                            (_f = (_g = msg.result).objects) !== null && _f !== void 0 ? _f : (_g.objects = []);
+                                            msg.result.objects.push({ taskId: value.taskId, data: value.delta.object });
+                                        }
+                                    });
+                                });
+                            }
+                        }
+                        if ((value === null || value === void 0 ? void 0 : value.type) === 'ERROR') {
+                            set((state) => {
+                                var _a;
+                                const msg = (_a = state.messages) === null || _a === void 0 ? void 0 : _a.findLast((i) => i.taskId === message.taskId);
+                                if (msg)
+                                    msg.error = value.error;
+                                else
+                                    throw new Error(value.error.message);
+                            });
+                        }
+                    }
+                }
+                catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                finally {
+                    try {
+                        if (!_h && !_e && (_f = stream_1.return)) yield _f.call(stream_1);
+                    }
+                    finally { if (e_1) throw e_1.error; }
+                }
+            }
+            catch (error) {
+                set((state) => {
+                    state.error = error;
+                });
+            }
+            finally {
+                set((state) => {
+                    var _a;
+                    state.running = false;
+                    if (message) {
+                        const msg = (_a = state.messages) === null || _a === void 0 ? void 0 : _a.findLast((i) => i.taskId === message.taskId);
+                        if (msg)
+                            msg.loading = false;
+                    }
+                });
+            }
+        }),
+    }))));
+    return STATES[aid];
+};

package/lib/esm/builtin/async/ai-runtime/components/runtime/utils/map-right.js

@@ -0,0 +1,7 @@
+export default function mapRight(array, fn) {
+    const result = new Array(array.length);
+    for (let i = array.length - 1; i >= 0; i--) {
+        result.push(fn(array[i], i, array));
+    }
+    return result;
+}

package/lib/esm/builtin/async/ai-runtime/components/runtime-chat/InitialForm.js

@@ -19,9 +19,8 @@ var __rest = (this && this.__rest) || function (s, e) {
     return t;
 };
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
-import { Alert, Box, Stack, alpha } from '@mui/material';
-import omit from 'lodash/omit';
-import { useMemo, useState } from 'react';
+import { Box, Stack, alpha } from '@mui/material';
+import { useEffect, useMemo, useState } from 'react';
 import { useSessionContext } from '../../../../session';
 import { createSession } from '../../api/session';
 import { useAssistantState } from '../../state/assistant';
@@ -29,16 +28,22 @@ import { globalLoadingState } from '../../state/loading';
 import { createSessionState } from '../../state/session';
 import { MessageSuggestedQuestion } from '../common/MessageSuggestedQuestions';
 import ParameterForm from '../form/ParameterForm';
-import MessageItemView from '../message/MessageItemView';
+import MessageErrorView from '../message/MessageErrorView';
+import MessageItemView, { MessageItemWrapper } from '../message/MessageItemView';
 import PoweredBy from '../PoweredBy';
 export default function InitialForm({ page }) {
-    var _a;
+    var _a, _b;
     const { assistantId } = page;
     const { session: authSession } = useSessionContext();
     const [assistant] = useAssistantState({ assistantId: page.assistantId });
-    const { parameters } = assistant;
     const [error, setError] = useState();
-    const isChat = parameters === null || parameters === void 0 ? void 0 : parameters.some((i) => i.key === 'question');
+    useEffect(() => {
+        var _a;
+        if (error && authSession.user && ((_a = error === null || error === void 0 ? void 0 : error.error) === null || _a === void 0 ? void 0 : _a.status) === 401) {
+            onSubmit(error.parameters);
+            setError(undefined);
+        }
+    }, [authSession.user]);
     const opener = useMemo(() => {
         var _a;
         const { openerMessage } = (_a = assistant.release) !== null && _a !== void 0 ? _a : {};
@@ -46,6 +51,7 @@ export default function InitialForm({ page }) {
             return {
                 taskId: '',
                 sessionId: '',
+                assistantId: '',
                 createdAt: new Date().toISOString(),
                 updatedAt: new Date().toISOString(),
                 result: { content: openerMessage },
@@ -54,14 +60,7 @@ export default function InitialForm({ page }) {
         return undefined;
     }, [assistant.release]);
     const [creating, setCreating] = useState(false);
-    const onSubmit = (...args_1) => __awaiter(this, [...args_1], void 0, function* (parameters = {}, entry) {
-        if (!authSession.user) {
-            yield new Promise((resolve) => {
-                authSession.login(() => {
-                    resolve();
-                });
-            });
-        }
+    const onSubmit = (parameters) => __awaiter(this, void 0, void 0, function* () {
         if (creating)
             return;
         setCreating(true);
@@ -69,20 +68,20 @@ export default function InitialForm({ page }) {
             yield globalLoadingState.getState().run(() => __awaiter(this, void 0, void 0, function* () {
                 const { created: session } = yield createSession({
                     assistantId,
-                    name: parameters.question,
-                    parameters: omit(parameters, isChat ? ['question'] : []),
-                    entry,
+                    name: typeof (parameters === null || parameters === void 0 ? void 0 : parameters.question) === 'string' ? parameters.question : undefined,
+                    parameters,
                 });
                 createSessionState({ sessionId: session.id, session }).getState().execute({
                     assistantId,
                     parameters,
                 });
-                setError(undefined);
                 page.navigate({ page: 'session', assistantId, sessionId: session.id });
             }));
+            setError(undefined);
         }
         catch (error) {
-            setError(error);
+            setError({ error, parameters });
+            throw error;
         }
         finally {
             setCreating(false);
@@ -90,11 +89,11 @@ export default function InitialForm({ page }) {
     });
     if (authSession.loading)
         return null;
-    return (_jsxs(_Fragment, { children: [opener && (_jsx(Box, { my: 2, mx: 2, children: _jsx(MessageItemView, { assistantId: assistantId, hideAvatar: true, assistant: assistant, message: opener }) })), ((_a = assistant.entries) === null || _a === void 0 ? void 0 : _a.length) && (_jsx(EntryListView, { assistant: assistant, onEntryClick: (entry) => onSubmit(entry.parameters, { id: entry.id, title: entry.title }), sx: { px: 2 } })), _jsx(Box, { flex: 1 }), _jsxs(Box, { className: "ai-runtime-form-container", sx: {
+    return (_jsxs(_Fragment, { children: [_jsxs(Box, { sx: { mt: 5, mb: 5, px: 2 }, children: [opener && (_jsx(Box, { mb: 2, children: _jsx(MessageItemView, { page: page, message: opener }) })), ((_a = assistant.entries) === null || _a === void 0 ? void 0 : _a.length) && (_jsx(EntryListView, { assistant: assistant, onEntryClick: (entry) => onSubmit(entry.parameters), sx: { pl: 6.5, pr: 2.5 } })), error && (_jsx(MessageItemWrapper, { assistant: assistant, agentMessage: _jsx(MessageErrorView, { error: error.error, page: page }), sx: { mt: 2 } }))] }), _jsx(Box, { flex: 1 }), _jsxs(Box, { className: "ai-runtime-form-container", sx: {
                     position: 'sticky',
                     bottom: 0,
                     background: (theme) => `linear-gradient(180deg, ${alpha(theme.palette.background.default, 0.1)}, ${alpha(theme.palette.background.default, 1)} 16px)`,
-                }, children: [_jsx(ParameterForm, { assistant: assistant, onSubmit: onSubmit }), error && _jsx(Alert, { severity: "error", children: error.message }), _jsx(PoweredBy, { assistantId: page.assistantId, sx: { my: 1, px: 2 } })] })] }));
+                }, children: [_jsx(ParameterForm, { disabled: ((_b = error === null || error === void 0 ? void 0 : error.error) === null || _b === void 0 ? void 0 : _b.status) === 401, page: page, assistant: assistant, onSubmit: onSubmit }), _jsx(PoweredBy, { assistantId: page.assistantId, sx: { my: 1, px: 2 } })] })] }));
 }
 export function EntryListView(_a) {
     var _b;

package/lib/esm/builtin/async/ai-runtime/components/runtime-chat/RuntimeChatSession.js

@@ -1,11 +1,13 @@
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
 import { RuntimeOutputVariable } from '@blocklet/ai-runtime/types';
-import { Alert, Box, alpha } from '@mui/material';
+import { Box, alpha } from '@mui/material';
 import { useState } from 'react';
 import { useAssistantState } from '../../state/assistant';
 import { useSessionState } from '../../state/session';
 import { MessageSuggestedQuestions } from '../common';
 import ParameterForm from '../form/ParameterForm';
+import MessageErrorView from '../message/MessageErrorView';
+import { MessageItemWrapper } from '../message/MessageItemView';
 import MessageListView from '../message/MessageListView';
 import PoweredBy from '../PoweredBy';
 import { useScrollToBottomModule } from '../ScrollView';
@@ -31,16 +33,21 @@ export default function RuntimeChatSession({ page }) {
             setError(undefined);
         })
             .catch((error) => {
-            setError(error);
+            setError({ error, parameters });
         });
     };
-    return (_jsxs(_Fragment, { children: [_jsx(MessageListView, { className: "ai-chat-message-list", my: 5, px: 2, assistantId: page.assistantId, sessionId: page.sessionId }), _jsx(Box, { ml: 6.5, mr: 2.5, mt: -4, px: 2, children: (suggestedQuestions === null || suggestedQuestions === void 0 ? void 0 : suggestedQuestions.length) ? (_jsx(MessageSuggestedQuestions, { dataSource: suggestedQuestions, onClick: (item) => {
-                        onSubmit(Object.assign(Object.assign({}, lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.parameters), { question: item.question }));
-                    } })) : !((_e = state.messages) === null || _e === void 0 ? void 0 : _e.length) && ((_f = assistant.entries) === null || _f === void 0 ? void 0 : _f.length) ? (_jsx(EntryListView, { assistant: assistant, onEntryClick: (entry) => onSubmit(Object.assign(Object.assign({}, lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.parameters), entry.parameters)) })) : null }), _jsx(Box, { flex: 1 }), error && (_jsx(Alert, { severity: "error", sx: { mb: 1, mx: 2 }, children: error.message })), _jsxs(Box, { className: "ai-runtime-form-container", sx: {
+    return (_jsxs(_Fragment, { children: [_jsxs(Box, { sx: {
+                    mb: {
+                        xs: 1,
+                        md: 2,
+                    },
+                }, children: [_jsx(MessageListView, { className: "ai-chat-message-list", my: 5, px: 2, page: page }), _jsx(Box, { ml: 6.5, mr: 2.5, mt: -4, px: 2, children: (suggestedQuestions === null || suggestedQuestions === void 0 ? void 0 : suggestedQuestions.length) ? (_jsx(MessageSuggestedQuestions, { dataSource: suggestedQuestions, onClick: (item) => {
+                                onSubmit(Object.assign(Object.assign({}, lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.parameters), { question: item.question }));
+                            } })) : !((_e = state.messages) === null || _e === void 0 ? void 0 : _e.length) && ((_f = assistant.entries) === null || _f === void 0 ? void 0 : _f.length) ? (_jsx(EntryListView, { assistant: assistant, onEntryClick: (entry) => onSubmit(Object.assign(Object.assign({}, lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.parameters), entry.parameters)) })) : null }), error && (_jsx(MessageItemWrapper, { assistant: assistant, agentMessage: _jsx(MessageErrorView, { error: error.error, page: page }), sx: { mt: 2, px: 2 } }))] }), _jsx(Box, { flex: 1 }), _jsxs(Box, { className: "ai-runtime-form-container", sx: {
                     pt: 2,
                     position: 'sticky',
                     bottom: 0,
                     // 从上透明到白色 0.72 的径向渐变
                     background: (theme) => `linear-gradient(180deg, ${alpha(theme.palette.background.default, 0.1)}, ${alpha(theme.palette.background.default, 1)} 16px)`,
-                }, children: [_jsx(ParameterForm, { sessionId: page.sessionId, defaultValues: lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.parameters, assistant: assistant, onSubmit: onSubmit, isSubmitting: state.running }), _jsx(PoweredBy, { assistantId: page.assistantId, sx: { my: 0.5, px: 2 } })] })] }));
+                }, children: [_jsx(ParameterForm, { page: page, defaultValues: lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.parameters, assistant: assistant, onSubmit: onSubmit, isSubmitting: state.running }), _jsx(PoweredBy, { assistantId: page.assistantId, sx: { my: 0.5, px: 2 } })] })] }));
 }

package/lib/esm/builtin/async/ai-runtime/components/runtime-form/RuntimeFormForms.js

@@ -62,12 +62,16 @@ export default function RuntimeFormForms(_a) {
     const form = useForm({ defaultValues: defaultForm });
     const [error, setError] = useState();
     const [expand, setExpand] = useState(false);
-    const { session } = useSessionContext();
+    const { session: authSession } = useSessionContext();
     const [submitting, setSubmitting] = useState(false);
     useEffect(() => {
         form.reset(defaultForm);
     }, [defaultForm]);
     const onSubmit = (parameters) => __awaiter(this, void 0, void 0, function* () {
+        if (!authSession.user) {
+            // TODO:
+            return;
+        }
         try {
             setSubmitting(true);
             let { sessionId } = page;
@@ -99,11 +103,6 @@ export default function RuntimeFormForms(_a) {
     const handleSubmit = (e) => __awaiter(this, void 0, void 0, function* () {
         var _e;
         e.preventDefault();
-        if (!session.user) {
-            yield new Promise((resolve) => {
-                session.login(() => resolve());
-            });
-        }
         (_e = agent.parameters) === null || _e === void 0 ? void 0 : _e.some((parameter) => {
             if (parameter.required && isEmpty(form.getValues(parameter.key))) {
                 setExpand(true);

package/lib/esm/builtin/async/ai-runtime/components/runtime-form/RuntimeFormResult.js

@@ -12,17 +12,14 @@ var __rest = (this && this.__rest) || function (s, e) {
 import { jsx as _jsx } from "react/jsx-runtime";
 import { Box, Stack } from '@mui/material';
 import { Suspense } from 'react';
-import { useAssistantState } from '../../state/assistant';
 import { useSessionState } from '../../state/session';
 import MessageItemView from '../message/MessageItemView';
 export default function RuntimeFormResult(_a) {
     var { page } = _a, props = __rest(_a, ["page"]);
-    return (_jsx(Stack, Object.assign({}, props, { children: _jsx(Box, { children: _jsx(Suspense, { children: page.sessionId ? _jsx(ResultView, { page: page }) : null }) }) })));
+    return (_jsx(Stack, Object.assign({}, props, { children: _jsx(Box, { children: _jsx(Suspense, { children: page.page === 'session' && _jsx(ResultView, { page: page }) }) }) })));
 }
 function ResultView({ page }) {
-    const { sessionId, assistantId } = page;
-    const [assistant] = useAssistantState({ assistantId });
-    const [, { messages = [] }] = useSessionState({ sessionId });
+    const [, { messages = [] }] = useSessionState(page);
     const message = messages[messages.length - 1];
-    return (_jsx(Stack, { gap: 2, children: message && (_jsx(MessageItemView, { hideAvatar: true, assistantId: assistantId, assistant: assistant, message: message }, message.taskId)) }));
+    return (_jsx(Stack, { gap: 2, children: message && _jsx(MessageItemView, { hideAvatar: true, page: page, message: message }, message.taskId) }));
 }

package/lib/esm/builtin/async/ai-runtime/components/runtime-form/index.js

@@ -30,6 +30,10 @@ export default function RuntimeForm({ page }) {
                         display: 'flex',
                         flexDirection: 'column',
                         justifyContent: 'space-between',
+                        mb: {
+                            xs: 1,
+                            md: 0,
+                        },
                     }, children: [_jsxs(Stack, { gap: isMobile ? spacing : 1, spacing: spacing, children: [_jsx(RuntimeFormHeader, { agent: agent }), _jsx(RuntimeFormResult, { page: page })] }), !isMobile && _jsx(PoweredBy, { assistantId: page.assistantId, sx: { mt: spacing, mb: 0.5, px: 2, pt: 0 } })] }), _jsxs(Box, { sx: {
                         flex: {
                             md: 1,

package/lib/esm/builtin/async/ai-runtime/components/session/CheckSession.js

@@ -8,6 +8,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 import { useEffect } from 'react';
+import { useSessionContext } from '../../../../session';
 import { createSession, getSessions } from '../../api/session';
 import useAsync from '../../hooks/useAsync';
 import { useAssistantState } from '../../state/assistant';
@@ -15,6 +16,7 @@ import { globalLoadingState } from '../../state/loading';
 import { createSessionState } from '../../state/session';
 import { parseIdentity } from '../../utils';
 export default function CheckSession({ cacheId, page, autoCreateSession, children, }) {
+    var _a;
     const { assistantId, sessionId } = page;
     useAssistantState({ assistantId });
     const result = useAsync(() => {
@@ -53,6 +55,13 @@ export default function CheckSession({ cacheId, page, autoCreateSession, childre
         if (result)
             page.navigate(...result);
     }, [result, page.navigate]);
+    const { session: authSession } = useSessionContext();
+    useEffect(() => {
+        var _a;
+        if (!((_a = authSession.user) === null || _a === void 0 ? void 0 : _a.did)) {
+            page.navigate({ page: 'assistant', assistantId: page.assistantId, sessionId: undefined }, { replace: true });
+        }
+    }, [(_a = authSession.user) === null || _a === void 0 ? void 0 : _a.did]);
     if (result)
         return null;
     return children;

package/lib/esm/builtin/async/ai-runtime/hooks/auth.js

@@ -0,0 +1,27 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { useCallback } from 'react';
+import { useSessionContext } from '../../../session';
+import { createAssistantState } from '../state/assistant';
+import { createSessionState } from '../state/session';
+import { createSubscriptionState } from '../state/subscription';
+export function useLogin({ page }) {
+    const { session: authSession } = useSessionContext();
+    return useCallback(() => __awaiter(this, void 0, void 0, function* () {
+        yield new Promise((resolve) => {
+            authSession.login(() => resolve());
+        });
+        yield Promise.all([
+            createAssistantState({ assistantId: page.assistantId }).getState().load(),
+            createSubscriptionState({ aid: page.assistantId }).getState().load(),
+            page.sessionId && createSessionState({ sessionId: page.sessionId }).getState().load(),
+        ]);
+    }), [authSession, page]);
+}

package/lib/esm/builtin/async/ai-runtime/hooks/header.js

@@ -1,17 +1,22 @@
 import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import DID from '@arcblock/ux/lib/DID';
-import { Box, Button } from '@mui/material';
+import { Box, Button, useMediaQuery, useTheme } from '@mui/material';
 import { useEffect } from 'react';
 import { Avatar } from '../../../arcblock/ux';
 import { useLocaleContext } from '../../../locale';
 import { useHeaderState } from '../../../page/header';
+import { useSessionContext } from '../../../session';
 import SocialShare from '../components/SocialShare';
 import ThemeProvider from '../components/ThemeProvider';
 import { useSubscribeLink } from './subscribe';
 const logoSize = 44;
 export function useSetHeader({ assistant, assistantId }) {
-    const { subscribeLink, requireSubscribe } = useSubscribeLink({ assistantId });
+    var _a;
+    const { subscribeLink, isPaymentProject } = useSubscribeLink({ assistantId });
+    const theme = useTheme();
+    const isMobile = useMediaQuery(theme.breakpoints.down('sm'));
     const { t, locale } = useLocaleContext();
+    const { session: authSession } = useSessionContext();
     // set logo
     useEffect(() => {
         const { release } = assistant;
@@ -20,27 +25,32 @@ export function useSetHeader({ assistant, assistantId }) {
                     width: logoSize,
                     height: logoSize,
                 } }));
-            state.brand = (release === null || release === void 0 ? void 0 : release.title) || assistant.name;
-            state.description = assistant.createdBy && (_jsxs(Box, { sx: {
-                    display: 'inline-flex',
-                    alignItems: 'center',
-                    gap: 0.5,
-                    mt: 0.5,
-                }, children: [t('by'), " ", _jsx(DID, { did: assistant.createdBy, copyable: false })] }));
+            if (!isMobile) {
+                state.brand = (release === null || release === void 0 ? void 0 : release.title) || assistant.name;
+                state.description = assistant.createdBy && (_jsxs(Box, { sx: {
+                        display: 'inline-flex',
+                        alignItems: 'center',
+                        gap: 0.5,
+                        mt: 0.5,
+                    }, children: [t('by'), " ", _jsx(DID, { did: assistant.createdBy, copyable: false })] }));
+            }
             const shareContent = `${(release === null || release === void 0 ? void 0 : release.title) || assistant.name}
 
 ${release === null || release === void 0 ? void 0 : release.description}
 `;
-            state.addons = (...exists) => [
-                _jsx(ThemeProvider, { children: _jsx(Button, { disabled: !requireSubscribe, variant: "outlined", href: subscribeLink, 
-                        // @ts-ignore
-                        color: "black", sx: {
-                            // 字母不大写
-                            textTransform: 'none',
-                        }, children: subscribeLink ? t('subscribe') : t('subscribed') }) }),
-                _jsx(SocialShare, { content: shareContent }),
-                ...exists,
-            ].filter(Boolean);
+            state.addons = (...exists) => {
+                var _a;
+                return [
+                    ((_a = authSession === null || authSession === void 0 ? void 0 : authSession.user) === null || _a === void 0 ? void 0 : _a.did) && isPaymentProject && (_jsx(ThemeProvider, { children: _jsx(Button, { disabled: !subscribeLink, variant: "outlined", href: subscribeLink, 
+                            // @ts-ignore
+                            color: "black", sx: {
+                                // 字母不大写
+                                textTransform: 'none',
+                            }, children: subscribeLink ? t('subscribe') : t('subscribed') }) })),
+                    _jsx(SocialShare, { content: shareContent }),
+                    ...exists,
+                ].filter(Boolean);
+            };
         });
         return () => useHeaderState.setState((state) => {
             state.logo = undefined;
@@ -48,5 +58,5 @@ ${release === null || release === void 0 ? void 0 : release.description}
             state.description = undefined;
             state.addons = (...exists) => [...exists];
         });
-    }, [locale]);
+    }, [locale, isMobile, (_a = authSession.user) === null || _a === void 0 ? void 0 : _a.did]);
 }

package/lib/esm/builtin/async/ai-runtime/hooks/subscribe.js

@@ -5,21 +5,23 @@ import { PAYMENT_KIT_DID } from '../constants';
 import { useAssistantState } from '../state/assistant';
 import { useSubscriptionState } from '../state/subscription';
 export function useSubscribeLink({ assistantId }) {
-    var _a, _b, _c;
+    var _a, _b, _c, _d, _e;
     const [, state] = useSubscriptionState({ aid: assistantId });
     const [assistant] = useAssistantState({ assistantId });
-    const { session } = useSessionContext();
+    const { session: authSession } = useSessionContext();
     const requireSubscribe = ((_b = (_a = assistant.release) === null || _a === void 0 ? void 0 : _a.payment) === null || _b === void 0 ? void 0 : _b.enable) && !state.subscription;
-    const subscribeLink = requireSubscribe && session.user && ((_c = state.release) === null || _c === void 0 ? void 0 : _c.paymentLinkId)
+    const isPaymentProject = (_d = (_c = assistant.release) === null || _c === void 0 ? void 0 : _c.payment) === null || _d === void 0 ? void 0 : _d.enable;
+    const subscribeLink = requireSubscribe && authSession.user && ((_e = state.release) === null || _e === void 0 ? void 0 : _e.paymentLinkId)
         ? getSubscribeLink({
             aid: assistantId,
-            userId: session.user.did,
+            userId: authSession.user.did,
             paymentLinkId: state.release.paymentLinkId,
         })
         : undefined;
     return {
         requireSubscribe,
         subscribeLink,
+        isPaymentProject,
     };
 }
 function getSubscribeLink({ aid, userId, paymentLinkId }) {