@mujian/js-sdk 0.0.6-beta.1

package/README.md

@@ -0,0 +1,23 @@
+# Rslib project
+
+## Setup
+
+Install the dependencies:
+
+```bash
+pnpm install
+```
+
+## Get started
+
+Build the library:
+
+```bash
+pnpm build
+```
+
+Build the library in watch mode:
+
+```bash
+pnpm dev
+```

package/dist/events/index.d.ts

@@ -0,0 +1,71 @@
+/**
+ * @hidden
+ * 事件
+ * @enum {string}
+ */
+export declare enum EVENT {
+    /**
+     * Doc comments may be included here.
+     */
+    MUJIAN_INIT = "mujian:init",
+    /**
+     * 调用中断信号
+     */
+    MUJIAN_AI_CHAT_STOP = "mujian:ai:chat:stop",
+    /**
+     * 发送消息
+     */
+    MUJIAN_AI_CHAT_COMPLETE = "mujian:ai:chat:complete",
+    /**
+     * 对消息应用正则表达式
+     */
+    MUJIAN_AI_CHAT_APPLY_REGEX = "mujian:ai:chat:applyRegex",
+    /**
+     * 生成文本
+     */
+    MUJIAN_AI_TEXT_GENERATE = "mujian:ai:text:generate",
+    /**
+     * 获取对话
+     */
+    MUJIAN_AI_CHAT_MESSAGE_GET_ALL = "mujian:ai:chat:message:getAll",
+    /**
+     * 获取项目信息
+     */
+    MUJIAN_AI_CHAT_PROJECT_GET_INFO = "mujian:ai:chat:project:getInfo",
+    /**
+     * 获取个人设置
+     */
+    MUJIAN_AI_SETTINGS_PERSONA_GET_ACTIVE = "mujian:ai:settings:persona:getActive",
+    /**
+     * 设置个人设置
+     */
+    MUJIAN_AI_SETTINGS_PERSONA_SET_ACTIVE = "mujian:ai:settings:persona:setActive",
+    /**
+     * 获取模型列表
+     */
+    MUJIAN_AI_SETTINGS_MODEL_GET_ALL = "mujian:ai:settings:model:getAll",
+    /**
+     * 设置模型设置
+     */
+    MUJIAN_AI_SETTINGS_MODEL_SET_ACTIVE = "mujian:ai:settings:model:setActive",
+    /**
+     * 获取模型设置
+     */
+    MUJIAN_AI_SETTINGS_MODEL_GET_ACTIVE = "mujian:ai:settings:model:getActive",
+    /**
+     * 删除消息
+     */
+    MUJIAN_AI_CHAT_MESSAGE_DELETE_ONE = "mujian:ai:chat:message:deleteOne",
+    /**
+     * 编辑消息
+     */
+    MUJIAN_AI_CHAT_MESSAGE_EDIT_ONE = "mujian:ai:chat:message:editOne",
+    /**
+     * 滑动消息
+     */
+    MUJIAN_AI_CHAT_MESSAGE_SWIPE = "mujian:ai:chat:message:swipe",
+    /**
+     * 获取消息提示词
+     */
+    MUJIAN_AI_CHAT_MESSAGE_GET_PROMPT = "mujian:ai:chat:message:getPrompt"
+}

package/dist/index.d.ts

@@ -0,0 +1,96 @@
+import { EVENT } from './events';
+/** @hidden */
+export { EVENT } from './events';
+type Call<T> = {
+    resolve: (value: T) => void;
+    reject: (reason?: unknown) => void;
+    onData?: (data: T) => void;
+    onComplete?: () => void;
+    signal?: AbortSignal;
+};
+declare global {
+    interface Window {
+        $mujian: MujianSdk;
+    }
+}
+declare class MujianSdk {
+    private constructor();
+    /** @hidden */
+    static getInstance(): MujianSdk;
+    /** @hidden */
+    private parent;
+    /** @hidden */
+    private ready;
+    /**
+     * 是否初始化
+     */
+    get isReady(): boolean;
+    /** @hidden */
+    private pendingRequests;
+    /**
+     * 初始化
+     */
+    init(): Promise<void>;
+    /** @hidden */
+    emit<T>(event: EVENT, data?: T): void;
+    /** @hidden */
+    private nextCallId;
+    /** @hidden */
+    private getCallId;
+    /** @hidden */
+    call<Req, Resp>(event: EVENT, data?: Req, controller?: Pick<Call<Resp>, 'onData' | 'onComplete' | 'signal'>): Promise<Resp>;
+    /**
+     * AI
+     */
+    ai: {
+        chat: {
+            project: {
+                getInfo: () => Promise<import("./types").ProjectInfo>;
+            };
+            settings: {
+                model: {
+                    getAll: () => Promise<import("./types").ModelInfo[]>;
+                    getActive: () => Promise<string>;
+                    setActive: (modelId: string) => Promise<string>;
+                };
+                persona: {
+                    getActive: () => Promise<import("./types").PersonaInfo>;
+                    setActive: (personaId: string) => Promise<void>;
+                };
+            };
+            complete: (message: string, onData: (data: string) => void, signal?: AbortSignal | undefined) => Promise<string>;
+            applyRegex: (props: {
+                message: import("./react").Message;
+                index: number;
+                depth: number;
+            }) => Promise<import("./react").Message>;
+            continue: (onData: (data: string) => void, signal?: AbortSignal | undefined) => Promise<string>;
+            regenerate: (onData: (data: string) => void, signal?: AbortSignal | undefined) => Promise<string>;
+            message: {
+                getAll: () => Promise<import("./react").Message[]>;
+                deleteOne: (messageId: string) => Promise<unknown>;
+                editOne: (messageId: string, content: string) => Promise<unknown>;
+                swipe: (messageId: string, swipeId: number) => Promise<unknown>;
+                getPrompt: (messageId: string) => Promise<unknown>;
+            };
+        };
+        text: {
+            complete: () => Promise<void>;
+        };
+    };
+    ui: {};
+    util: {
+        cn: () => void;
+    };
+    hybrid: {};
+    player: {};
+    game: {
+        assets: () => void;
+        saves: {
+            saveGame: () => Promise<void>;
+            loadGame: () => Promise<void>;
+        };
+        ranking: () => void;
+    };
+}
+export { MujianSdk };

package/dist/index.js

@@ -0,0 +1,220 @@
+import postmate from "postmate";
+var events_EVENT = /*#__PURE__*/ function(EVENT) {
+    EVENT["MUJIAN_INIT"] = "mujian:init";
+    EVENT["MUJIAN_AI_CHAT_STOP"] = "mujian:ai:chat:stop";
+    EVENT["MUJIAN_AI_CHAT_COMPLETE"] = "mujian:ai:chat:complete";
+    EVENT["MUJIAN_AI_CHAT_APPLY_REGEX"] = "mujian:ai:chat:applyRegex";
+    EVENT["MUJIAN_AI_TEXT_GENERATE"] = "mujian:ai:text:generate";
+    EVENT["MUJIAN_AI_CHAT_MESSAGE_GET_ALL"] = "mujian:ai:chat:message:getAll";
+    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";
+    EVENT["MUJIAN_AI_SETTINGS_MODEL_GET_ALL"] = "mujian:ai:settings:model:getAll";
+    EVENT["MUJIAN_AI_SETTINGS_MODEL_SET_ACTIVE"] = "mujian:ai:settings:model:setActive";
+    EVENT["MUJIAN_AI_SETTINGS_MODEL_GET_ACTIVE"] = "mujian:ai:settings:model:getActive";
+    EVENT["MUJIAN_AI_CHAT_MESSAGE_DELETE_ONE"] = "mujian:ai:chat:message:deleteOne";
+    EVENT["MUJIAN_AI_CHAT_MESSAGE_EDIT_ONE"] = "mujian:ai:chat:message:editOne";
+    EVENT["MUJIAN_AI_CHAT_MESSAGE_SWIPE"] = "mujian:ai:chat:message:swipe";
+    EVENT["MUJIAN_AI_CHAT_MESSAGE_GET_PROMPT"] = "mujian:ai:chat:message:getPrompt";
+    return EVENT;
+}({});
+const chat_complete = async function(message, onData, signal) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_COMPLETE, {
+        content: message
+    }, {
+        onData,
+        signal
+    });
+};
+const applyRegex = async function(props) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_APPLY_REGEX, props);
+};
+const continueComplete = async function(onData, signal) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_COMPLETE, {
+        isContinue: true
+    }, {
+        onData,
+        signal
+    });
+};
+const regenerate = async function(onData, signal) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_COMPLETE, {
+        isRegenerate: true
+    }, {
+        onData,
+        signal
+    });
+};
+const getAll = async function() {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_MESSAGE_GET_ALL);
+};
+const messageDeleteOne = async function(messageId) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_MESSAGE_DELETE_ONE, {
+        messageId
+    });
+};
+const messageEditOne = async function(messageId, content) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_MESSAGE_EDIT_ONE, {
+        messageId,
+        content
+    });
+};
+const messageSwipe = async function(messageId, swipeId) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_MESSAGE_SWIPE, {
+        messageId,
+        swipeId
+    });
+};
+const getPrompt = async function(messageId) {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_MESSAGE_GET_PROMPT, {
+        messageId
+    });
+};
+const getInfo = async function() {
+    return await this.call(events_EVENT.MUJIAN_AI_CHAT_PROJECT_GET_INFO);
+};
+const getActive = async function() {
+    return await this.call(events_EVENT.MUJIAN_AI_SETTINGS_PERSONA_GET_ACTIVE);
+};
+const setActive = async function(personaId) {
+    return await this.call(events_EVENT.MUJIAN_AI_SETTINGS_PERSONA_SET_ACTIVE, {
+        personaId
+    });
+};
+const model_getActive = async function() {
+    return await this.call(events_EVENT.MUJIAN_AI_SETTINGS_MODEL_GET_ACTIVE);
+};
+const model_setActive = async function(modelId) {
+    return await this.call(events_EVENT.MUJIAN_AI_SETTINGS_MODEL_SET_ACTIVE, {
+        modelId
+    });
+};
+const model_getAll = async function() {
+    return await this.call(events_EVENT.MUJIAN_AI_SETTINGS_MODEL_GET_ALL);
+};
+const generate = async function() {
+    return await this.call(events_EVENT.MUJIAN_AI_TEXT_GENERATE, {
+        content: 'q'
+    });
+};
+const saveGame = async function() {};
+const loadGame = async function() {};
+class MujianSdk {
+    constructor(){}
+    static getInstance() {
+        if (!window.$mujian) window.$mujian = new MujianSdk();
+        return window.$mujian;
+    }
+    parent = null;
+    ready = false;
+    get isReady() {
+        return this.ready;
+    }
+    pendingRequests = new Map();
+    async init() {
+        const handshake = new postmate.Model({
+            reply: ({ id, complete, data })=>{
+                const call = this.pendingRequests.get(id);
+                if (!call) return;
+                call.onData?.(data);
+                if (complete) {
+                    call.onComplete?.();
+                    call.resolve(data);
+                    this.pendingRequests.delete(id);
+                }
+            }
+        });
+        try {
+            const parent = await handshake;
+            this.ready = true;
+            this.parent = parent;
+            console.log('mujian sdk client init');
+            await this.call(events_EVENT.MUJIAN_INIT);
+        } catch (error) {
+            console.log('init error', error);
+        }
+    }
+    emit(event, data) {
+        if (!this.ready) throw new Error('Mujian is not initialized');
+        this.parent?.emit(event, data);
+    }
+    nextCallId = 0;
+    getCallId() {
+        const id = this.nextCallId;
+        this.nextCallId += 1;
+        return id;
+    }
+    async call(event, data, controller) {
+        if (!this.ready) throw new Error('Mujian is not initialized');
+        const callId = this.getCallId();
+        return new Promise((resolve, reject)=>{
+            this.pendingRequests.set(callId, {
+                resolve,
+                reject,
+                ...controller
+            });
+            this.emit(event, {
+                id: callId,
+                data
+            });
+            controller?.signal?.addEventListener('abort', ()=>{
+                this.emit(events_EVENT.MUJIAN_AI_CHAT_STOP, {
+                    id: callId
+                });
+            });
+        });
+    }
+    ai = {
+        chat: {
+            project: {
+                getInfo: getInfo.bind(this)
+            },
+            settings: {
+                model: {
+                    getAll: model_getAll.bind(this),
+                    getActive: model_getActive.bind(this),
+                    setActive: model_setActive.bind(this)
+                },
+                persona: {
+                    getActive: getActive.bind(this),
+                    setActive: setActive.bind(this)
+                }
+            },
+            complete: chat_complete.bind(this),
+            applyRegex: applyRegex.bind(this),
+            continue: continueComplete.bind(this),
+            regenerate: regenerate.bind(this),
+            message: {
+                getAll: getAll.bind(this),
+                deleteOne: messageDeleteOne.bind(this),
+                editOne: messageEditOne.bind(this),
+                swipe: messageSwipe.bind(this),
+                getPrompt: getPrompt.bind(this)
+            }
+        },
+        text: {
+            complete: generate.bind(this)
+        }
+    };
+    ui = {};
+    util = {
+        cn: ()=>{
+            console.log('cn');
+        }
+    };
+    hybrid = {};
+    player = {};
+    game = {
+        assets: ()=>{
+            console.log('assets');
+        },
+        saves: {
+            saveGame: saveGame.bind(this),
+            loadGame: loadGame.bind(this)
+        },
+        ranking: ()=>{
+            console.log('ranking');
+        }
+    };
+}
+export { events_EVENT as EVENT, MujianSdk };

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

@@ -0,0 +1,35 @@
+import type { MujianSdk } from '../../../index.ts';
+import { Message } from '../../../react';
+/**
+ * 对话补全
+ * @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>;
+export declare const applyRegex: (this: MujianSdk, props: {
+    message: Message;
+    index: number;
+    depth: number;
+}) => Promise<Message>;
+/**
+ * 继续
+ * @param onData 数据
+ * @param signal 信号
+ * @returns
+ */
+export declare const continueComplete: (
+/** @hidden */
+this: MujianSdk, onData: (data: string) => void, signal?: AbortSignal) => Promise<string>;
+/**
+ * 重新生成
+ * @param onData 数据
+ * @param signal 信号
+ * @returns
+ */
+export declare const regenerate: (
+/** @hidden */
+this: MujianSdk, onData: (data: string) => void, signal?: AbortSignal) => Promise<string>;

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

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

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

@@ -0,0 +1,43 @@
+import type { MujianSdk } from '../../../../index.ts';
+import type { Message } from '../../../../react/chat/useChat/message';
+/**
+ * 获取消息列表
+ * @returns 消息当前列表
+ */
+export declare const getAll: (
+/** @hidden */
+this: MujianSdk) => Promise<Message[]>;
+/**
+ * 删除消息
+ * @param messageId 消息ID
+ * @returns
+ */
+export declare const messageDeleteOne: (
+/** @hidden */
+this: MujianSdk, messageId: string) => Promise<unknown>;
+/**
+ * 编辑消息
+ * @param messageId 消息ID
+ * @param content 内容
+ * @returns
+ */
+export declare const messageEditOne: (
+/** @hidden */
+this: MujianSdk, messageId: string, content: string) => Promise<unknown>;
+/**
+ * 滑动消息
+ * @param messageId 消息ID
+ * @param swipeId 滑动ID
+ * @returns
+ */
+export declare const messageSwipe: (
+/** @hidden */
+this: MujianSdk, messageId: string, swipeId: number) => Promise<unknown>;
+/**
+ * 获取消息提示词
+ * @param messageId 消息ID
+ * @returns
+ */
+export declare const getPrompt: (
+/** @hidden */
+this: MujianSdk, messageId: string) => Promise<unknown>;

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

@@ -0,0 +1,3 @@
+import type { MujianSdk } from '../../../../index.ts';
+import type { ProjectInfo } from '../../../../types';
+export declare const getInfo: (this: MujianSdk) => Promise<ProjectInfo>;

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

@@ -0,0 +1,3 @@
+import * as persona from './persona';
+import * as model from './model';
+export { persona, model };

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

@@ -0,0 +1,24 @@
+import type { MujianSdk } from '../../../../index.ts';
+import type { ModelInfo } from '../../../../types';
+/**
+ * 获取当前激活的模型
+ * @returns 模型ID
+ */
+export declare const getActive: (
+/** @hidden */
+this: MujianSdk) => Promise<string>;
+/**
+ * 设置当前激活的模型
+ * @param modelId
+ * @returns 模型ID
+ */
+export declare const setActive: (
+/** @hidden */
+this: MujianSdk, modelId: string) => Promise<string>;
+/**
+ * 获取模型列表
+ * @returns 模型列表
+ */
+export declare const getAll: (
+/** @hidden */
+this: MujianSdk) => Promise<ModelInfo[]>;

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

@@ -0,0 +1,17 @@
+import type { MujianSdk } from '../../../../index.ts';
+import type { PersonaInfo } from '../../../../types';
+/**
+ * 获取当前激活的个人设置
+ * @returns 个人设置
+ */
+export declare const getActive: (
+/** @hidden */
+this: MujianSdk) => Promise<PersonaInfo>;
+/**
+ * 设置当前激活的个人设置
+ * @param personaId 个人设置ID
+ * @returns
+ */
+export declare const setActive: (
+/** @hidden */
+this: MujianSdk, personaId: string) => Promise<void>;

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

@@ -0,0 +1,5 @@
+/**
+ * @module ai
+ */
+export * as chat from './chat';
+export * as text from './text';

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

@@ -0,0 +1,2 @@
+import type { MujianSdk } from '../../../index';
+export declare const generate: (this: MujianSdk) => Promise<void>;

package/dist/modules/game/index.d.ts

@@ -0,0 +1,4 @@
+/**
+ * @module game
+ */
+export * as saves from './saves';

package/dist/modules/game/saves.d.ts

@@ -0,0 +1,3 @@
+import type { MujianSdk } from '../../index.ts';
+export declare const saveGame: (this: MujianSdk) => Promise<void>;
+export declare const loadGame: (this: MujianSdk) => Promise<void>;

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

@@ -0,0 +1 @@
+export { type Message, useChat } from './useChat';

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

@@ -0,0 +1,15 @@
+export declare class CreateConversationError extends Error {
+    constructor(message: string, cause?: unknown);
+}
+export declare class SendMessageError extends Error {
+    constructor(message: string, cause?: unknown);
+}
+export declare class InvalidDeltaContentError extends Error {
+    constructor(message: string, cause?: unknown);
+}
+export declare class UnexpectedSseEndingError extends Error {
+    constructor(message: string, cause?: unknown);
+}
+export declare class InsufficientBalanceError extends Error {
+    constructor(message?: string, cause?: unknown);
+}

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

@@ -0,0 +1,55 @@
+import { type Dispatch, type SetStateAction } from 'react';
+import type { Message } from './message';
+export type { Message };
+/**
+ * Use chat
+ * @param {object} body - The body of the chat.
+ * @param {function} onError - The function to call when an error occurs.
+ * @param {function} onFinish - The function to call when the chat is finished.
+ * @returns {UseChatReturn} The chat.
+ */
+export type UseChatProps = {
+    body?: object;
+    onError?: (e: Error) => void;
+    onFinish?: (message: Message) => void;
+};
+/**
+ * Use chat return
+ * @param {Message[]} messages - The messages of the chat.
+ * @param {string} status - The status of the chat.
+ * @param {Error} error - The error of the chat.
+ * @param {function} append - The function to call when the chat is finished.
+ * @param {function} regenerate - The function to call when the chat is finished.
+ */
+export type UseChatReturn = {
+    messages: Message[];
+    status: 'uninitialized' | 'ready' | 'streaming' | 'error';
+    error?: Error;
+    /** 提交用户输入，让AI回答 */
+    append: (query: string) => Promise<void>;
+    /** 重新生成最后一条回答 */
+    regenerate: () => Promise<void>;
+    /** 让AI继续说 */
+    continueGenerate: () => Promise<void>;
+    stop: () => void;
+    setMessages: Dispatch<SetStateAction<Message[]>>;
+    input: string;
+    setInput: Dispatch<SetStateAction<string>>;
+    handleSubmit: () => void;
+    setSwipe: (messageId: string, swipeId: number) => Promise<void>;
+    editMessage: (messageId: string, content: string) => Promise<void>;
+    deleteMessage: (messageId: string) => Promise<void>;
+};
+/**
+ * Use chat common
+ * @param {object} body - The body of the chat.
+ * @returns {UseChatCommon} The chat.
+ */
+export type UseChatCommon = {
+    body?: object;
+};
+/**
+ * Use chat
+ * @returns {UseChatReturn} The chat.
+ */
+export declare const useChat: ({ body, onError, onFinish, }: UseChatProps) => UseChatReturn;

package/dist/react/chat/useChat/inner/chatStreaming.d.ts

@@ -0,0 +1,23 @@
+import { type Dispatch, type SetStateAction } from 'react';
+import type { MujianSdk } from '../../../../index.ts';
+import type { UseChatCommon } from '..';
+import type { Message } from '../message';
+interface ChatStreamingProps {
+    common: UseChatCommon;
+    setError: (error: Error | undefined) => void;
+    setMessages: Dispatch<SetStateAction<Message[]>>;
+    mujian: MujianSdk;
+}
+export declare const useChatStreaming: ({ common, setError, setMessages, mujian, }: ChatStreamingProps) => {
+    chatStreaming: ({ query, regenerate, is_continue, signal, }: {
+        /** 用户输入 */
+        query?: string;
+        /** 是否重说 */
+        regenerate?: boolean;
+        /** 是否继续 */
+        is_continue?: boolean;
+        signal?: AbortSignal;
+    }) => Promise<void>;
+    isStreaming: boolean;
+};
+export {};

package/dist/react/chat/useChat/inner/stream.d.ts

@@ -0,0 +1,3 @@
+import type { MujianSdk } from '../../../../index.ts';
+export declare function callSdk(mujian: MujianSdk, content: string, type?: 'generate' | 'regenerate' | 'continue', signal?: AbortSignal): AsyncGenerator<string | undefined, void, unknown>;
+export declare function callSdk2(mujian: MujianSdk, content: string, type?: 'generate' | 'regenerate' | 'continue', signal?: AbortSignal): ReadableStream;