@saber2pr/ai-assistant 0.0.4

package/lib/utils/streamRequest.js

@@ -0,0 +1,150 @@
+"use strict";
+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 __await = (this && this.__await) || function (v) { return this instanceof __await ? (this.v = v, this) : new __await(v); }
+var __asyncGenerator = (this && this.__asyncGenerator) || function (thisArg, _arguments, generator) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var g = generator.apply(thisArg, _arguments || []), i, q = [];
+    return i = Object.create((typeof AsyncIterator === "function" ? AsyncIterator : Object).prototype), verb("next"), verb("throw"), verb("return", awaitReturn), i[Symbol.asyncIterator] = function () { return this; }, i;
+    function awaitReturn(f) { return function (v) { return Promise.resolve(v).then(f, reject); }; }
+    function verb(n, f) { if (g[n]) { i[n] = function (v) { return new Promise(function (a, b) { q.push([n, v, a, b]) > 1 || resume(n, v); }); }; if (f) i[n] = f(i[n]); } }
+    function resume(n, v) { try { step(g[n](v)); } catch (e) { settle(q[0][3], e); } }
+    function step(r) { r.value instanceof __await ? Promise.resolve(r.value.v).then(fulfill, reject) : settle(q[0][2], r); }
+    function fulfill(value) { resume("next", value); }
+    function reject(value) { resume("throw", value); }
+    function settle(f, v) { if (f(v), q.shift(), q.length) resume(q[0][0], q[0][1]); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.streamRequest = streamRequest;
+var parseStreamData_1 = require("./parseStreamData");
+function streamRequest(url, options) {
+    return __asyncGenerator(this, arguments, function streamRequest_1() {
+        var result, reader, currentContent, reasonContent, chunks, _a, done, value, text, result_2, _i, result_1, parserRes, _b, content, reasoning_content, error_1;
+        return __generator(this, function (_c) {
+            switch (_c.label) {
+                case 0: return [4 /*yield*/, __await(fetch(url, options))];
+                case 1:
+                    result = _c.sent();
+                    reader = result.body.getReader();
+                    currentContent = '';
+                    reasonContent = '';
+                    chunks = '';
+                    _c.label = 2;
+                case 2:
+                    if (!true) return [3 /*break*/, 18];
+                    _c.label = 3;
+                case 3:
+                    _c.trys.push([3, 13, , 17]);
+                    return [4 /*yield*/, __await(reader.read()
+                        // 流读取完成
+                        )];
+                case 4:
+                    _a = _c.sent(), done = _a.done, value = _a.value;
+                    if (!done) return [3 /*break*/, 7];
+                    options.onChange && options.onChange('complete', currentContent);
+                    return [4 /*yield*/, __await({
+                            status: {
+                                type: 'complete',
+                                reason: 'stop',
+                            },
+                            content: [
+                                {
+                                    text: currentContent,
+                                    type: 'text',
+                                },
+                            ],
+                        })];
+                case 5: return [4 /*yield*/, _c.sent()];
+                case 6:
+                    _c.sent();
+                    return [3 /*break*/, 18];
+                case 7:
+                    text = new TextDecoder().decode(value, { stream: true });
+                    chunks += text;
+                    result_2 = (0, parseStreamData_1.parseStreamData)(chunks);
+                    _i = 0, result_1 = result_2;
+                    _c.label = 8;
+                case 8:
+                    if (!(_i < result_1.length)) return [3 /*break*/, 12];
+                    parserRes = result_1[_i];
+                    if (!(parserRes.event === 'answer')) return [3 /*break*/, 11];
+                    _b = parserRes.data.delta, content = _b.content, reasoning_content = _b.reasoning_content;
+                    currentContent += content || '';
+                    reasonContent += reasoning_content || '';
+                    options.onChange && options.onChange('running', currentContent);
+                    return [4 /*yield*/, __await({
+                            status: {
+                                type: 'running',
+                            },
+                            content: [
+                                {
+                                    text: currentContent,
+                                    type: 'text',
+                                },
+                            ],
+                        })];
+                case 9: return [4 /*yield*/, _c.sent()];
+                case 10:
+                    _c.sent();
+                    _c.label = 11;
+                case 11:
+                    _i++;
+                    return [3 /*break*/, 8];
+                case 12:
+                    chunks = '';
+                    return [3 /*break*/, 17];
+                case 13:
+                    error_1 = _c.sent();
+                    // 点击了取消
+                    console.error('Stream reading error:', error_1);
+                    if (!((error_1 === null || error_1 === void 0 ? void 0 : error_1.name) === 'AbortError')) return [3 /*break*/, 16];
+                    options.signal && options.signal.throwIfAborted();
+                    options.onChange && options.onChange('incomplete', reasonContent);
+                    return [4 /*yield*/, __await({
+                            status: {
+                                type: 'incomplete',
+                                reason: 'error',
+                            },
+                            content: [
+                                {
+                                    text: reasonContent,
+                                    type: 'reasoning',
+                                },
+                            ],
+                        })];
+                case 14: return [4 /*yield*/, _c.sent()];
+                case 15:
+                    _c.sent();
+                    return [3 /*break*/, 18];
+                case 16: return [3 /*break*/, 17];
+                case 17: return [3 /*break*/, 2];
+                case 18: return [2 /*return*/];
+            }
+        });
+    });
+}

package/package.json

@@ -0,0 +1,75 @@
+{
+  "name": "@saber2pr/ai-assistant",
+  "version": "0.0.4",
+  "description": "AI Assistant",
+  "files": [
+    "lib"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "start": "cross-env NODE_ENV=development webpack -w",
+    "dev": "cross-env NODE_ENV=development webpack-dev-server",
+    "lint": "prettier --write ./src",
+    "build": "rimraf ./build && cross-env NODE_ENV=production webpack",
+    "prepublishOnly": "yarn tsc",
+    "prepare": "husky install"
+  },
+  "license": "MIT",
+  "keywords": [
+    "AI Assistant"
+  ],
+  "dependencies": {
+    "@assistant-ui/react": "^0.8.6",
+    "@assistant-ui/react-ai-sdk": "^0.8.0",
+    "@assistant-ui/react-markdown": "^0.8.0",
+    "@mlc-ai/web-llm": "^0.2.80",
+    "antd": "~4",
+    "clipboard": "^2.0.11",
+    "normalize.css": "^8.0.1",
+    "react": "^18.1.0",
+    "react-dom": "^18.1.0",
+    "react-hook-speech-to-text": "^0.8.0",
+    "react-syntax-highlighter": "^15.6.1",
+    "remark-gfm": "^4.0.1",
+    "styled-components": "^5.3.5"
+  },
+  "devDependencies": {
+    "@babel/core": "^7.17.5",
+    "dotenv": "^16.4.7",
+    "@babel/preset-env": "^7.18.2",
+    "@babel/preset-react": "^7.17.12",
+    "@babel/preset-typescript": "^7.17.12",
+    "@types/react": "^18.0.9",
+    "@types/react-dom": "^18.0.5",
+    "@types/styled-components": "^5.1.25",
+    "@types/webpack": "^5.28.0",
+    "@types/webpack-dev-server": "^4.7.2",
+    "babel-loader": "^8.2.3",
+    "babel-plugin-import": "^1.13.3",
+    "babel-plugin-module-resolver": "^4.1.0",
+    "cross-env": "^6.0.0",
+    "css-loader": "^2.1.1",
+    "html-webpack-plugin": "^5.5.0",
+    "husky": "^7.0.4",
+    "less": "^3.9.0",
+    "less-loader": "^5.0.0",
+    "lint-staged": "^12.3.4",
+    "mini-css-extract-plugin": "^1.6.1",
+    "prettier": "^2.5.1",
+    "rimraf": "^3.0.2",
+    "style-loader": "^0.23.1",
+    "typescript": "^5.9.3",
+    "url-loader": "^2.0.0",
+    "webpack": "^5.69.1",
+    "webpack-cli": "^4.9.2",
+    "webpack-dev-server": "^4.7.4"
+  },
+  "lint-staged": {
+    "*.{js,jsx,ts,tsx}": [
+      "yarn lint",
+      "git add ."
+    ]
+  }
+}