@blocklet/pages-kit 0.2.312 → 0.2.313

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

@@ -15,8 +15,19 @@ var __asyncValues = (this && this.__asyncValues) || function (o) {
     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); }
 };
+var __rest = (this && this.__rest) || function (s, e) {
+    var t = {};
+    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
+        t[p] = s[p];
+    if (s != null && typeof Object.getOwnPropertySymbols === "function")
+        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
+            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
+                t[p[i]] = s[p[i]];
+        }
+    return t;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.useSessionState = exports.createSessionState = void 0;
+exports.useCurrentSessionState = exports.useSessionState = exports.useAutoLoadSessionState = exports.createSessionState = void 0;
 const aid_1 = require("@blocklet/ai-runtime/common/aid");
 const react_1 = require("react");
 const zustand_1 = require("zustand");
@@ -33,88 +44,190 @@ const createSessionState = ({ aid }) => {
             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;
+                state.sessions = undefined;
+                state.sessionMap = undefined;
+                state.currentSessionId = undefined;
             });
         },
         load: () => __awaiter(void 0, void 0, void 0, function* () {
-            var _a;
-            set((state) => {
-                state.loading = true;
-            });
-            try {
-                const { sessions } = yield (0, session_1.getSessions)({ aid });
-                const session = (_a = sessions[0]) !== null && _a !== void 0 ? _a : (yield (0, session_1.createSession)({ aid })).created;
-                const { messages } = yield (0, message_1.getMessages)({
-                    sessionId: session.id,
-                    orderDirection: GET_MESSAGES_ORDER_DIRECTION,
-                    limit: GET_MESSAGES_LIMIT,
-                });
+            const { loading, loaded } = get();
+            if (!loading && !loaded) {
                 set((state) => {
-                    state.session = session;
-                    state.messages = messages.toReversed();
-                    state.noMoreMessage = messages.length < GET_MESSAGES_LIMIT;
+                    state.loading = true;
                 });
+                try {
+                    const { sessions } = yield (0, session_1.getSessions)({ aid });
+                    set((state) => {
+                        var _a;
+                        state.sessions = sessions;
+                        state.currentSessionId = (_a = sessions[0]) === null || _a === void 0 ? void 0 : _a.id;
+                        state.sessionMap = Object.fromEntries(sessions.map((i) => [i.id, { session: i }]));
+                    });
+                }
+                catch (error) {
+                    set((state) => {
+                        state.error = error;
+                    });
+                    throw error;
+                }
+                finally {
+                    set((state) => {
+                        state.loading = false;
+                        state.loaded = true;
+                    });
+                }
             }
-            catch (error) {
-                set((state) => {
-                    state.error = error;
-                });
-                throw error;
-            }
-            finally {
-                set((state) => {
-                    state.loaded = true;
-                    state.loading = false;
-                });
+            const { sessionMap, currentSessionId: sessionId } = get();
+            const s = sessionMap === null || sessionMap === void 0 ? void 0 : sessionMap[sessionId];
+            if (!s || !sessionId)
+                return;
+            if (!s.loading && !s.loaded) {
+                try {
+                    set((state) => {
+                        var _a;
+                        const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[state.currentSessionId];
+                        if (!s)
+                            return;
+                        s.loading = true;
+                    });
+                    const { messages } = yield (0, message_1.getMessages)({
+                        sessionId,
+                        orderDirection: GET_MESSAGES_ORDER_DIRECTION,
+                        limit: GET_MESSAGES_LIMIT,
+                    });
+                    set((state) => {
+                        var _a;
+                        const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[state.currentSessionId];
+                        if (!s)
+                            return;
+                        s.messages = messages.toReversed();
+                        s.noMoreMessage = messages.length < GET_MESSAGES_LIMIT;
+                    });
+                }
+                catch (error) {
+                    set((state) => {
+                        var _a;
+                        const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[state.currentSessionId];
+                        if (!s)
+                            return;
+                        s.error = error;
+                    });
+                    throw error;
+                }
+                finally {
+                    set((state) => {
+                        var _a;
+                        const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[state.currentSessionId];
+                        if (!s)
+                            return;
+                        s.loaded = true;
+                        s.loading = false;
+                    });
+                }
             }
         }),
-        loadMoreMessages: () => __awaiter(void 0, void 0, void 0, function* () {
-            var _b;
+        setCurrentSessionId: (sessionId) => {
             set((state) => {
-                state.messageLoading = true;
+                state.currentSessionId = sessionId;
             });
-            const { session, messages } = get();
-            if (!session)
+        },
+        createSession: () => __awaiter(void 0, void 0, void 0, function* () {
+            const { created, sessions } = yield (0, session_1.createSession)({ aid });
+            set((state) => {
+                state.sessions = sessions;
+                state.currentSessionId = created.id;
+                state.sessionMap = Object.fromEntries(sessions.map((i) => {
+                    var _a;
+                    return [
+                        i.id,
+                        Object.assign(Object.assign(Object.assign({}, (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[i.id]), { session: i }), (i.id === created.id ? { messages: [], loaded: true } : {})),
+                    ];
+                }));
+            });
+            return created;
+        }),
+        deleteSession: (...args_1) => __awaiter(void 0, [...args_1], void 0, function* (args = {}) {
+            const sessionId = args.sessionId || get().currentSessionId;
+            if (!sessionId)
+                return;
+            const { sessions } = yield (0, session_1.deleteSession)({ sessionId });
+            set((state) => {
+                var _a;
+                state.sessions = sessions;
+                if (!sessions.some((i) => i.id === state.currentSessionId))
+                    state.currentSessionId = (_a = sessions[0]) === null || _a === void 0 ? void 0 : _a.id;
+            });
+        }),
+        clearSession: (...args_2) => __awaiter(void 0, [...args_2], void 0, function* (args = {}) {
+            const sessionId = args.sessionId || get().currentSessionId;
+            if (!sessionId)
+                return;
+            yield (0, session_1.clearSession)({ sessionId });
+            set((state) => {
+                var _a;
+                const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                if (!s)
+                    return;
+                s.messages = undefined;
+                s.noMoreMessage = undefined;
+                s.messageLoading = undefined;
+            });
+        }),
+        loadMoreMessages: (...args_3) => __awaiter(void 0, [...args_3], void 0, function* (args = {}) {
+            var _a, _b;
+            const sessionId = args.sessionId || get().currentSessionId;
+            if (!sessionId)
+                return;
+            set((state) => {
+                var _a;
+                const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                if (!s)
+                    return;
+                s.messageLoading = true;
+            });
+            const { sessionMap } = get();
+            const s = sessionMap === null || sessionMap === void 0 ? void 0 : sessionMap[sessionId];
+            if (!s)
                 return;
             try {
                 const result = yield (0, message_1.getMessages)({
-                    sessionId: session.id,
-                    before: (_b = messages === null || messages === void 0 ? void 0 : messages.slice(-1)[0]) === null || _b === void 0 ? void 0 : _b.id,
+                    sessionId: s.session.id,
+                    before: (_b = (_a = s.messages) === null || _a === void 0 ? void 0 : _a.slice(-1)[0]) === null || _b === void 0 ? void 0 : _b.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;
+                    var _a, _b;
+                    const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                    if (!s)
+                        return;
+                    s.messages = ((_b = s.messages) !== null && _b !== void 0 ? _b : []).concat(result.messages.toReversed());
+                    s.noMoreMessage = result.messages.length < GET_MESSAGES_LIMIT;
                 });
             }
             finally {
                 set((state) => {
-                    state.messageLoading = false;
+                    var _a;
+                    const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                    if (!s)
+                        return;
+                    s.messageLoading = false;
                 });
             }
         }),
-        execute: (_c) => __awaiter(void 0, [_c], void 0, function* ({ aid, working, parameters, onResponseStart, }) {
+        execute: (_c) => __awaiter(void 0, void 0, void 0, function* () {
             var _d, e_1, _e, _f;
-            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;
+            var { aid, working, parameters, onResponseStart } = _c, args = __rest(_c, ["aid", "working", "parameters", "onResponseStart"]);
+            const sessionId = args.sessionId || get().currentSessionId || (yield get().createSession()).id;
             const identity = (0, aid_1.parseIdentity)(aid, { rejectWhenError: true });
             let message;
             set((state) => {
-                state.running = true;
-                state.error = undefined;
+                var _a;
+                const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                if (!s)
+                    return;
+                s.running = true;
+                s.error = undefined;
             });
             try {
                 const stream = (0, session_1.runAgent)({
@@ -145,26 +258,32 @@ const createSessionState = ({ aid }) => {
                                     loading: true,
                                 };
                                 set((state) => {
-                                    var _a;
-                                    (_a = state.messages) !== null && _a !== void 0 ? _a : (state.messages = []);
-                                    state.messages.push(message);
+                                    var _a, _b;
+                                    const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                                    if (!s)
+                                        return;
+                                    (_b = s.messages) !== null && _b !== void 0 ? _b : (s.messages = []);
+                                    s.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);
+                                        var _a, _b, _c, _d, _e, _f, _g;
+                                        var _h;
+                                        const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                                        if (!s)
+                                            return;
+                                        const msg = (_b = s.messages) === null || _b === void 0 ? void 0 : _b.findLast((i) => i.taskId === message.taskId);
                                         if (!msg)
                                             return;
-                                        (_b = msg.result) !== null && _b !== void 0 ? _b : (msg.result = {});
+                                        (_c = msg.result) !== null && _c !== void 0 ? _c : (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 ((_e = (_d = value.delta) === null || _d === void 0 ? void 0 : _d.images) === null || _e === void 0 ? void 0 : _e.length) {
+                                            msg.result.images = ((_f = msg.result.images) !== null && _f !== void 0 ? _f : []).concat(value.delta.images);
                                         }
                                         if (value.delta.object) {
-                                            (_f = (_g = msg.result).objects) !== null && _f !== void 0 ? _f : (_g.objects = []);
+                                            (_g = (_h = msg.result).objects) !== null && _g !== void 0 ? _g : (_h.objects = []);
                                             msg.result.objects.push({ taskId: value.taskId, data: value.delta.object });
                                         }
                                     });
@@ -173,8 +292,11 @@ const createSessionState = ({ aid }) => {
                         }
                         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);
+                                var _a, _b;
+                                const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                                if (!s)
+                                    return;
+                                const msg = (_b = s.messages) === null || _b === void 0 ? void 0 : _b.findLast((i) => i.taskId === message.taskId);
                                 if (msg)
                                     msg.error = value.error;
                                 else
@@ -193,15 +315,22 @@ const createSessionState = ({ aid }) => {
             }
             catch (error) {
                 set((state) => {
-                    state.error = error;
+                    var _a;
+                    const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                    if (!s)
+                        return;
+                    s.error = error;
                 });
             }
             finally {
                 set((state) => {
-                    var _a;
-                    state.running = false;
+                    var _a, _b;
+                    const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[sessionId];
+                    if (!s)
+                        return;
+                    s.running = false;
                     if (message) {
-                        const msg = (_a = state.messages) === null || _a === void 0 ? void 0 : _a.findLast((i) => i.taskId === message.taskId);
+                        const msg = (_b = s.messages) === null || _b === void 0 ? void 0 : _b.findLast((i) => i.taskId === message.taskId);
                         if (msg)
                             msg.loading = false;
                     }
@@ -211,16 +340,25 @@ const createSessionState = ({ aid }) => {
     }))));
 };
 exports.createSessionState = createSessionState;
-function useSessionState({ autoLoad } = {}) {
+function useAutoLoadSessionState() {
     const { aid } = (0, Runtime_1.useRuntimeContext)();
     const state = (0, react_1.useMemo)(() => (0, exports.createSessionState)({ aid }), [aid]);
+    const currentSessionId = state((s) => s.currentSessionId);
     (0, react_1.useEffect)(() => {
-        if (!autoLoad)
-            return;
-        const { loading, loaded, load } = state.getState();
-        if (!loading && !loaded)
-            load();
-    }, [state, autoLoad]);
-    return state;
+        state.getState().load();
+    }, [state, currentSessionId]);
+}
+exports.useAutoLoadSessionState = useAutoLoadSessionState;
+function useSessionState(selector) {
+    const { aid } = (0, Runtime_1.useRuntimeContext)();
+    return (0, react_1.useMemo)(() => (0, exports.createSessionState)({ aid }), [aid])(selector);
 }
 exports.useSessionState = useSessionState;
+function useCurrentSessionState(selector) {
+    return useSessionState((state) => {
+        var _a;
+        const s = (_a = state.sessionMap) === null || _a === void 0 ? void 0 : _a[state.currentSessionId];
+        return selector(s);
+    });
+}
+exports.useCurrentSessionState = useCurrentSessionState;

package/lib/cjs/builtin/page/header.js

@@ -1,6 +1,48 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.useHeaderState = void 0;
+exports.useHeader = exports.useHeaderState = void 0;
+const immer_1 = require("immer");
+const react_1 = require("react");
 const zustand_1 = require("zustand");
-const immer_1 = require("zustand/middleware/immer");
-exports.useHeaderState = (0, zustand_1.create)()((0, immer_1.immer)(() => ({})));
+const immer_2 = require("zustand/middleware/immer");
+(0, immer_1.enableMapSet)();
+exports.useHeaderState = (0, zustand_1.create)()((0, immer_2.immer)((set, get) => ({
+    stack: new Set(),
+    add: (creator) => {
+        set((state) => {
+            state.stack.add(creator);
+        });
+        get().refresh();
+    },
+    delete: (creator) => {
+        set((state) => {
+            state.stack.delete(creator);
+        });
+        get().refresh();
+    },
+    refresh: () => {
+        const widgets = [];
+        for (const i of get().stack) {
+            widgets.push(i());
+        }
+        const r = widgets.reduce((res, i) => (Object.assign(Object.assign(Object.assign({}, res), i), { addons: (...exists) => {
+                var _a, _b, _c, _d;
+                const a = (_b = (_a = res.addons) === null || _a === void 0 ? void 0 : _a.call(res, ...exists)) !== null && _b !== void 0 ? _b : exists;
+                const b = (_d = (_c = i.addons) === null || _c === void 0 ? void 0 : _c.call(i, ...a)) !== null && _d !== void 0 ? _d : a;
+                return b;
+            } })), {});
+        set((state) => {
+            state.logo = r.logo;
+            state.description = r.description;
+            state.brand = r.brand;
+            state.addons = r.addons;
+        });
+    },
+})));
+function useHeader(creator, deps) {
+    (0, react_1.useEffect)(() => {
+        exports.useHeaderState.getState().add(creator);
+        return () => exports.useHeaderState.getState().delete(creator);
+    }, deps);
+}
+exports.useHeader = useHeader;