@saber2pr/ai-assistant 0.0.4

package/lib/context.d.ts

@@ -0,0 +1,3 @@
+import React from 'react';
+import { AIAssistantConfig } from './types/assistant';
+export declare const AIConfigContext: React.Context<AIAssistantConfig>;

package/lib/context.js

@@ -0,0 +1,8 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIConfigContext = void 0;
+var react_1 = __importDefault(require("react"));
+exports.AIConfigContext = react_1.default.createContext({});

package/lib/hooks/useAsync.d.ts

@@ -0,0 +1,14 @@
+export type UseAsyncOps<T> = {
+    manual: boolean;
+    onSuccess?(result?: T): any;
+    onError?(err?: Error): any;
+};
+export type UseAsyncResult<A extends any[], T> = {
+    data: T;
+    loading: boolean;
+    setData(data: T): void;
+    run: (...args: A) => Promise<T>;
+};
+export declare function useAsync<A extends any[], T = any>(run: (...args: A) => Promise<T>): UseAsyncResult<A, T>;
+export declare function useAsync<A extends any[], T = any>(run: (...args: A) => Promise<T>, deps: any[]): UseAsyncResult<A, T>;
+export declare function useAsync<A extends any[], T = any>(run: (...args: A) => Promise<T>, deps: any[], ops: UseAsyncOps<T>): UseAsyncResult<A, T>;

package/lib/hooks/useAsync.js

@@ -0,0 +1,105 @@
+"use strict";
+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 __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
+    if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
+        if (ar || !(i in from)) {
+            if (!ar) ar = Array.prototype.slice.call(from, 0, i);
+            ar[i] = from[i];
+        }
+    }
+    return to.concat(ar || Array.prototype.slice.call(from));
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useAsync = useAsync;
+var react_1 = require("react");
+function useAsync(run, deps, ops) {
+    var _this = this;
+    if (deps === void 0) { deps = []; }
+    var _a = (0, react_1.useState)(), data = _a[0], setData = _a[1];
+    var _b = (0, react_1.useState)(), loading = _b[0], setLoading = _b[1];
+    var main = function () {
+        var args = [];
+        for (var _i = 0; _i < arguments.length; _i++) {
+            args[_i] = arguments[_i];
+        }
+        return __awaiter(_this, void 0, void 0, function () {
+            var result, error_1;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        _a.trys.push([0, 4, 7, 8]);
+                        setLoading(true);
+                        return [4 /*yield*/, run.apply(void 0, args)];
+                    case 1:
+                        result = _a.sent();
+                        if (!(ops === null || ops === void 0 ? void 0 : ops.onSuccess)) return [3 /*break*/, 3];
+                        return [4 /*yield*/, (ops === null || ops === void 0 ? void 0 : ops.onSuccess(result))];
+                    case 2:
+                        _a.sent();
+                        _a.label = 3;
+                    case 3:
+                        setData(result);
+                        return [2 /*return*/, result];
+                    case 4:
+                        error_1 = _a.sent();
+                        if (!(ops === null || ops === void 0 ? void 0 : ops.onError)) return [3 /*break*/, 6];
+                        return [4 /*yield*/, (ops === null || ops === void 0 ? void 0 : ops.onError(error_1))];
+                    case 5:
+                        _a.sent();
+                        _a.label = 6;
+                    case 6: return [3 /*break*/, 8];
+                    case 7:
+                        setLoading(false);
+                        return [7 /*endfinally*/];
+                    case 8: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    (0, react_1.useEffect)(function () {
+        if (!(ops === null || ops === void 0 ? void 0 : ops.manual)) {
+            main();
+        }
+    }, __spreadArray([ops === null || ops === void 0 ? void 0 : ops.manual], deps, true));
+    return {
+        data: data,
+        loading: loading,
+        run: main,
+        setData: setData,
+    };
+}

package/lib/hooks/useCopy.d.ts

@@ -0,0 +1,5 @@
+import ClipboardJS from 'clipboard';
+export declare const useCopy: <Button extends HTMLElement = any, Target extends HTMLElement = any>(id: string, init?: (cp: ClipboardJS) => void | VoidFunction, deps?: any[]) => {
+    button: import("react").MutableRefObject<Button>;
+    target: import("react").MutableRefObject<Target>;
+};

package/lib/hooks/useCopy.js

@@ -0,0 +1,48 @@
+"use strict";
+var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
+    if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
+        if (ar || !(i in from)) {
+            if (!ar) ar = Array.prototype.slice.call(from, 0, i);
+            ar[i] = from[i];
+        }
+    }
+    return to.concat(ar || Array.prototype.slice.call(from));
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useCopy = void 0;
+var clipboard_1 = __importDefault(require("clipboard"));
+var react_1 = require("react");
+var useCopy = function (id, init, deps) {
+    if (init === void 0) { init = function (cp) {
+        cp.on('success', function () { });
+    }; }
+    if (deps === void 0) { deps = []; }
+    var ref = (0, react_1.useRef)();
+    var targetRef = (0, react_1.useRef)();
+    (0, react_1.useEffect)(function () {
+        var targetId = "Clipboard-Target-".concat(id || Date.now());
+        if (ref.current) {
+            ref.current.setAttribute('data-clipboard-target', "#".concat(targetId));
+        }
+        else {
+            return;
+        }
+        if (targetRef.current) {
+            targetRef.current.id = targetId;
+        }
+        else {
+            ref.current.id = targetId;
+        }
+        var cp = new clipboard_1.default(ref.current);
+        init && init(cp);
+        return function () { return cp.destroy(); };
+    }, __spreadArray(__spreadArray([], deps, true), [id], false));
+    return {
+        button: ref,
+        target: targetRef,
+    };
+};
+exports.useCopy = useCopy;

package/lib/hooks/useI18n.d.ts

@@ -0,0 +1,5 @@
+import { i18nKeys } from '../i18n/locales';
+export declare const useI18n: () => {
+    t: (key: i18nKeys) => string;
+    locale: "zh-CN" | "en-US";
+};

package/lib/hooks/useI18n.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useI18n = void 0;
+var react_1 = require("react");
+var context_1 = require("../context");
+var locales_1 = require("../i18n/locales");
+var useI18n = function () {
+    var config = (0, react_1.useContext)(context_1.AIConfigContext);
+    var locale = config.locale || 'zh-CN';
+    var strings = locales_1.locales[locale] || locales_1.locales['zh-CN'];
+    var t = function (key) {
+        return strings[key] || key;
+    };
+    return { t: t, locale: locale };
+};
+exports.useI18n = useI18n;

package/lib/i18n/i18n.d.ts

@@ -0,0 +1,14 @@
+export interface ITextMap {
+    [key: string]: string;
+}
+export type ILocalesTextMap = {
+    [locale: string]: ITextMap;
+};
+export declare class I18n<T extends ILocalesTextMap, M = T[keyof T]> {
+    private localesTextMap;
+    private currentTextMap;
+    locale: keyof T;
+    constructor(localesTextMap: T);
+    setLocal(locale: keyof T): void;
+    format(contentKey: keyof M, args?: Record<string, string>): string | keyof M;
+}

package/lib/i18n/i18n.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.I18n = void 0;
+var template = function (tpl, values) {
+    return tpl.replace(/{\w+}/g, function (slot) { return values[slot.replace(/{|}/g, '')]; });
+};
+var I18n = /** @class */ (function () {
+    function I18n(localesTextMap) {
+        this.localesTextMap = localesTextMap;
+        this.currentTextMap = {};
+    }
+    I18n.prototype.setLocal = function (locale) {
+        this.locale = locale;
+        this.currentTextMap =
+            this.localesTextMap[locale] ||
+                this.localesTextMap[Object.keys(this.localesTextMap)[0]];
+    };
+    I18n.prototype.format = function (contentKey, args) {
+        var i18nformatString = this.currentTextMap[contentKey];
+        if (!i18nformatString) {
+            var currentTextMap = this.localesTextMap['en'] || {};
+            i18nformatString = currentTextMap[contentKey];
+        }
+        if (!i18nformatString) {
+            return contentKey;
+        }
+        return args ? template(i18nformatString, args) : i18nformatString;
+    };
+    return I18n;
+}());
+exports.I18n = I18n;

package/lib/i18n/index.d.ts

@@ -0,0 +1,6 @@
+import { I18n } from './i18n';
+declare const i18n: I18n<{
+    zh: {};
+    en: {};
+}, {} | {}>;
+export { i18n };

package/lib/i18n/index.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.i18n = void 0;
+var i18n_1 = require("./i18n");
+var locale_1 = require("./locale");
+var i18n = new i18n_1.I18n(locale_1.localeMap);
+exports.i18n = i18n;

package/lib/i18n/locale.d.ts

@@ -0,0 +1,4 @@
+export declare const localeMap: {
+    zh: {};
+    en: {};
+};

package/lib/i18n/locale.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.localeMap = void 0;
+exports.localeMap = {
+    zh: {},
+    en: {}
+};

package/lib/i18n/locales.d.ts

@@ -0,0 +1,36 @@
+export declare const locales: {
+    'zh-CN': {
+        assistantTitle: string;
+        newChat: string;
+        tabChat: string;
+        tabHistory: string;
+        loadingModel: string;
+        loadingTip: string;
+        noHistory: string;
+        defaultWelcome: string;
+        defaultPlaceholder: string;
+        defaultEmpty: string;
+        cancel: string;
+        send: string;
+        thinking: string;
+        newChatFallback: string;
+    };
+    'en-US': {
+        assistantTitle: string;
+        newChat: string;
+        tabChat: string;
+        tabHistory: string;
+        loadingModel: string;
+        loadingTip: string;
+        noHistory: string;
+        defaultWelcome: string;
+        defaultPlaceholder: string;
+        defaultEmpty: string;
+        cancel: string;
+        send: string;
+        thinking: string;
+        newChatFallback: string;
+    };
+};
+export type LocaleType = keyof typeof locales;
+export type i18nKeys = keyof typeof locales['zh-CN'];

package/lib/i18n/locales.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.locales = void 0;
+exports.locales = {
+    'zh-CN': {
+        assistantTitle: 'AI 助手',
+        newChat: '新对话',
+        tabChat: '对话',
+        tabHistory: '历史列表',
+        loadingModel: '正在初始化模型...',
+        loadingTip: '首次加载模型可能需要较长时间，请耐心等待',
+        noHistory: '暂无历史对话',
+        defaultWelcome: '有什么可以帮忙的？',
+        defaultPlaceholder: '给 GPT 发送消息',
+        defaultEmpty: '我是 AI，可以回答你的问题，请在下方输入框输入你的需求～',
+        cancel: '取消',
+        send: '发送',
+        thinking: '思考中',
+        newChatFallback: '新对话',
+    },
+    'en-US': {
+        assistantTitle: 'AI Assistant',
+        newChat: 'New Chat',
+        tabChat: 'Chat',
+        tabHistory: 'History',
+        loadingModel: 'Initializing model...',
+        loadingTip: 'First load may take a while, please wait patiently',
+        noHistory: 'No history',
+        defaultWelcome: 'How can I help you?',
+        defaultPlaceholder: 'Send a message to GPT',
+        defaultEmpty: 'I am an AI, I can answer your questions. Please enter your needs in the input box below~',
+        cancel: 'Cancel',
+        send: 'Send',
+        thinking: 'Thinking...',
+        newChatFallback: 'New Chat',
+    }
+};

package/lib/index.d.ts

@@ -0,0 +1 @@
+export * from './chat';

package/lib/index.js

@@ -0,0 +1,17 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./chat"), exports);

package/lib/llm/context.d.ts

@@ -0,0 +1,3 @@
+import { MLCEngine } from '@mlc-ai/web-llm';
+export declare const LLMContext: import("react").Context<MLCEngine>;
+export declare const useLLm: () => MLCEngine;

package/lib/llm/context.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useLLm = exports.LLMContext = void 0;
+var react_1 = require("react");
+exports.LLMContext = (0, react_1.createContext)(null);
+var useLLm = function () {
+    var context = (0, react_1.useContext)(exports.LLMContext);
+    return context;
+};
+exports.useLLm = useLLm;

package/lib/llm/engine.d.ts

@@ -0,0 +1,6 @@
+import { InitProgressCallback } from "@mlc-ai/web-llm";
+export interface GetLLmengineProps {
+    selectedModel?: string;
+    initProgressCallback?: InitProgressCallback;
+}
+export declare const getLLMengine: ({ selectedModel, initProgressCallback }: GetLLmengineProps) => Promise<import("@mlc-ai/web-llm").MLCEngine>;

package/lib/llm/engine.js

@@ -0,0 +1,53 @@
+"use strict";
+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 __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getLLMengine = void 0;
+var web_llm_1 = require("@mlc-ai/web-llm");
+var getLLMengine = function (_a) { return __awaiter(void 0, [_a], void 0, function (_b) {
+    var engine;
+    var _c = _b.selectedModel, selectedModel = _c === void 0 ? 'Llama-3.1-8B-Instruct-q4f32_1-MLC' : _c, initProgressCallback = _b.initProgressCallback;
+    return __generator(this, function (_d) {
+        switch (_d.label) {
+            case 0: return [4 /*yield*/, (0, web_llm_1.CreateMLCEngine)(selectedModel, { initProgressCallback: initProgressCallback })];
+            case 1:
+                engine = _d.sent();
+                return [2 /*return*/, engine];
+        }
+    });
+}); };
+exports.getLLMengine = getLLMengine;

package/lib/styles.d.ts

@@ -0,0 +1,6 @@
+export declare const Root: import("styled-components").StyledComponent<"div", any, {}, never>;
+export declare const Layout: import("styled-components").StyledComponent<"div", any, {}, never>;
+export declare const Aside: import("styled-components").StyledComponent<"div", any, {}, never>;
+export declare const Content: import("styled-components").StyledComponent<"div", any, {}, never>;
+export declare const MainContent: import("styled-components").StyledComponent<"div", any, {}, never>;
+export declare const FloatButton: import("styled-components").StyledComponent<"div", any, {}, never>;

package/lib/styles.js

@@ -0,0 +1,18 @@
+"use strict";
+var __makeTemplateObject = (this && this.__makeTemplateObject) || function (cooked, raw) {
+    if (Object.defineProperty) { Object.defineProperty(cooked, "raw", { value: raw }); } else { cooked.raw = raw; }
+    return cooked;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FloatButton = exports.MainContent = exports.Content = exports.Aside = exports.Layout = exports.Root = void 0;
+var styled_components_1 = __importDefault(require("styled-components"));
+exports.Root = styled_components_1.default.div(templateObject_1 || (templateObject_1 = __makeTemplateObject(["\n  padding: 48px;\n"], ["\n  padding: 48px;\n"])));
+exports.Layout = styled_components_1.default.div(templateObject_2 || (templateObject_2 = __makeTemplateObject(["\n  margin: 48px;\n  border-radius: 4px;\n  border: 1px solid #e5e5e5;\n  display: flex;\n  height: 600px;\n  background-color: #fff;\n"], ["\n  margin: 48px;\n  border-radius: 4px;\n  border: 1px solid #e5e5e5;\n  display: flex;\n  height: 600px;\n  background-color: #fff;\n"])));
+exports.Aside = styled_components_1.default.div(templateObject_3 || (templateObject_3 = __makeTemplateObject(["\n  display: none;\n"], ["\n  display: none;\n"])));
+exports.Content = styled_components_1.default.div(templateObject_4 || (templateObject_4 = __makeTemplateObject(["\n  padding: 12px 0px;\n  flex-grow: 1;\n"], ["\n  padding: 12px 0px;\n  flex-grow: 1;\n"])));
+exports.MainContent = styled_components_1.default.div(templateObject_5 || (templateObject_5 = __makeTemplateObject(["\n  display: flex;\n  justify-content: center;\n  align-items: center;\n  height: 100%;\n  font-size: 18px;\n  color: #999;\n"], ["\n  display: flex;\n  justify-content: center;\n  align-items: center;\n  height: 100%;\n  font-size: 18px;\n  color: #999;\n"])));
+exports.FloatButton = styled_components_1.default.div(templateObject_6 || (templateObject_6 = __makeTemplateObject(["\n  position: fixed;\n  width: 56px;\n  height: 56px;\n  border-radius: 28px;\n  background: linear-gradient(135deg, #1890ff 0%, #0050b3 100%);\n  box-shadow: 0 4px 16px rgba(24, 144, 255, 0.35);\n  display: flex;\n  justify-content: center;\n  align-items: center;\n  cursor: grab;\n  z-index: 1000;\n  color: #fff;\n  font-size: 24px;\n  transition: all 0.3s cubic-bezier(0.175, 0.885, 0.32, 1.275);\n  border: 1px solid rgba(255, 255, 255, 0.1);\n  backdrop-filter: blur(4px);\n\n  &:active {\n    cursor: grabbing;\n  }\n\n  &:hover {\n    transform: scale(1.1) rotate(5deg);\n    background: linear-gradient(135deg, #40a9ff 0%, #096dd9 100%);\n    box-shadow: 0 8px 24px rgba(24, 144, 255, 0.45);\n  }\n\n  &::after {\n    content: '';\n    position: absolute;\n    top: -2px;\n    left: -2px;\n    right: -2px;\n    bottom: -2px;\n    border-radius: 30px;\n    background: linear-gradient(135deg, rgba(255, 255, 255, 0.4) 0%, rgba(255, 255, 255, 0) 100%);\n    opacity: 0.5;\n    pointer-events: none;\n  }\n\n  .anticon {\n    filter: drop-shadow(0 2px 4px rgba(0, 0, 0, 0.2));\n  }\n"], ["\n  position: fixed;\n  width: 56px;\n  height: 56px;\n  border-radius: 28px;\n  background: linear-gradient(135deg, #1890ff 0%, #0050b3 100%);\n  box-shadow: 0 4px 16px rgba(24, 144, 255, 0.35);\n  display: flex;\n  justify-content: center;\n  align-items: center;\n  cursor: grab;\n  z-index: 1000;\n  color: #fff;\n  font-size: 24px;\n  transition: all 0.3s cubic-bezier(0.175, 0.885, 0.32, 1.275);\n  border: 1px solid rgba(255, 255, 255, 0.1);\n  backdrop-filter: blur(4px);\n\n  &:active {\n    cursor: grabbing;\n  }\n\n  &:hover {\n    transform: scale(1.1) rotate(5deg);\n    background: linear-gradient(135deg, #40a9ff 0%, #096dd9 100%);\n    box-shadow: 0 8px 24px rgba(24, 144, 255, 0.45);\n  }\n\n  &::after {\n    content: '';\n    position: absolute;\n    top: -2px;\n    left: -2px;\n    right: -2px;\n    bottom: -2px;\n    border-radius: 30px;\n    background: linear-gradient(135deg, rgba(255, 255, 255, 0.4) 0%, rgba(255, 255, 255, 0) 100%);\n    opacity: 0.5;\n    pointer-events: none;\n  }\n\n  .anticon {\n    filter: drop-shadow(0 2px 4px rgba(0, 0, 0, 0.2));\n  }\n"])));
+var templateObject_1, templateObject_2, templateObject_3, templateObject_4, templateObject_5, templateObject_6;

package/lib/types/assistant.d.ts

@@ -0,0 +1,53 @@
+import { LocaleType } from '../i18n/locales';
+import { MLCEngine } from '@mlc-ai/web-llm';
+export interface ChatMessage {
+    role: 'system' | 'user' | 'assistant';
+    content: string;
+}
+export interface AIAssistantConfig {
+    /**
+     * The welcome message displayed at the top of the chat.
+     */
+    welcomeMessage?: string;
+    /**
+     * A list of suggested questions to show when the chat is empty.
+     */
+    suggestions?: string[];
+    /**
+     * The placeholder text for the message input box.
+     */
+    placeholder?: string;
+    /**
+     * The message displayed when the thread list is empty.
+     */
+    emptyMessage?: string;
+    /**
+     * The language locale for the UI.
+     * @default 'zh-CN'
+     */
+    locale?: LocaleType;
+    /**
+     * The ID of the container element where the assistant will be mounted.
+     * @default 'ai-assistant-root'
+     */
+    containerId?: string;
+    /**
+     * The initial coordinates of the floating button.
+     * @default { x: window.innerWidth - 100, y: window.innerHeight - 100 }
+     */
+    initialPosition?: {
+        x: number;
+        y: number;
+    };
+    /**
+     * A hook to intercept and modify conversation messages before they are sent to the AI.
+     * Useful for injecting system prompts, background knowledge, or filtering history.
+     *
+     * @example
+     * onBeforeChat: (messages) => {
+     *   const systemPrompt = { role: 'system', content: 'You are a translation assistant' };
+     *   return [systemPrompt, ...messages];
+     * }
+     */
+    onBeforeChat?: (messages: ChatMessage[], llm: MLCEngine) => ChatMessage[] | Promise<ChatMessage[]>;
+}

package/lib/types/assistant.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/utils/event.d.ts

@@ -0,0 +1,8 @@
+export declare class Dispatcher {
+    private constructor();
+    static instance: Dispatcher;
+    private events;
+    addEventListener(type: string, listener: Function): void;
+    removeEventListener(type: string, listener: Function): void;
+    dispatch(type: string, message: any): void;
+}

package/lib/utils/event.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Dispatcher = void 0;
+var Dispatcher = /** @class */ (function () {
+    function Dispatcher() {
+        this.events = {};
+    }
+    Dispatcher.prototype.addEventListener = function (type, listener) {
+        var listeners = this.events[type] || [];
+        listeners.push(listener);
+        this.events[type] = listeners;
+    };
+    Dispatcher.prototype.removeEventListener = function (type, listener) {
+        var listeners = this.events[type] || [];
+        var index = listeners.indexOf(listener);
+        listeners.splice(index, 1);
+    };
+    Dispatcher.prototype.dispatch = function (type, message) {
+        var listeners = this.events[type] || [];
+        for (var _i = 0, listeners_1 = listeners; _i < listeners_1.length; _i++) {
+            var listener = listeners_1[_i];
+            listener(message);
+        }
+    };
+    Dispatcher.instance = new Dispatcher();
+    return Dispatcher;
+}());
+exports.Dispatcher = Dispatcher;

package/lib/utils/parseStreamData.d.ts

@@ -0,0 +1,5 @@
+export declare function parseStreamData(content: string): {
+    event: string;
+    id: string;
+    data: any;
+}[];

package/lib/utils/parseStreamData.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseStreamData = parseStreamData;
+function parseStreamData(content) {
+    // 按行分割数据
+    var chunks = content.split('\n\n').filter(Boolean);
+    var result = [];
+    // 遍历行以解析事件类型、ID和数据
+    for (var _i = 0, chunks_1 = chunks; _i < chunks_1.length; _i++) {
+        var chunk = chunks_1[_i];
+        var event_1 = void 0;
+        var id = void 0;
+        var data = void 0;
+        var lines = chunk.split('\n');
+        for (var _a = 0, lines_1 = lines; _a < lines_1.length; _a++) {
+            var line = lines_1[_a];
+            if (line.startsWith('event:')) {
+                event_1 = line.replace('event:', '').trim();
+            }
+            if (line.startsWith('id:')) {
+                id = line.replace('id:', '').trim();
+            }
+            if (line.startsWith('data:')) {
+                try {
+                    data = JSON.parse(line.replace('data:', '').trim());
+                }
+                catch (error) { }
+            }
+        }
+        result.push({ event: event_1, id: id, data: data });
+    }
+    return result;
+}

package/lib/utils/streamRequest.d.ts

@@ -0,0 +1,4 @@
+import { ChatModelRunResult } from '@assistant-ui/react';
+export declare function streamRequest(url: string, options: RequestInit & {
+    onChange?(type: 'complete' | 'running' | 'incomplete', data: string): void;
+}): AsyncGenerator<ChatModelRunResult>;