filecat 5.32.0 → 5.32.2

package/build/server/main/domain/ai_agent/ai_agent.controller.js

@@ -0,0 +1,136 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Ai_AgentController = void 0;
+const routing_controllers_1 = require("routing-controllers");
+const ai_agent_service_1 = require("./ai_agent.service");
+const user_service_1 = require("../user/user.service");
+const user_req_1 = require("../../../common/req/user.req");
+const router_1 = require("../../../common/frame/router");
+const WsData_1 = require("../../../common/frame/WsData");
+const ws_server_1 = require("../../../common/frame/ws.server");
+const Result_1 = require("../../other/Result");
+const threads_filecat_1 = require("../../threads/filecat/threads.filecat");
+let Ai_AgentController = class Ai_AgentController {
+    async chat(data, res, ctx) {
+        user_service_1.userService.check_user_auth(ctx.headers.authorization, user_req_1.UserAuth.ai_agent_page);
+        res.setHeader("Content-Type", "text/event-stream; charset=utf-8");
+        res.setHeader("Cache-Control", "no-cache");
+        res.setHeader("Connection", "keep-alive");
+        res.setHeader("X-Accel-Buffering", "no");
+        const token = ctx.headers.authorization;
+        let stream;
+        try {
+            stream = await ai_agent_service_1.ai_agentService.chat(data.messages, res, token);
+        }
+        catch (err) {
+            console.log(err);
+            if (!res.writableEnded && !res.destroyed) {
+                const message = typeof err === 'string'
+                    ? err
+                    : err?.message || 'AI service error';
+                ai_agent_service_1.ai_agentService.write_to_res(res, message);
+                res.write(`data: [DONE]\n\n`);
+                res.end();
+            }
+        }
+        return stream;
+    }
+    get_info(data) {
+        const wss = data.wss;
+        user_service_1.userService.check_user_auth(wss.token, user_req_1.UserAuth.ai_agent_setting);
+        ai_agent_service_1.ai_agentService.all_wss_set.add(wss);
+        wss.setClose(() => {
+            ai_agent_service_1.ai_agentService.all_wss_set.delete(wss);
+        });
+        return ai_agent_service_1.ai_agentService.docs_info;
+    }
+    async ai_load_one_file(ctx, data) {
+        user_service_1.userService.check_user_auth(ctx.headers.authorization, user_req_1.UserAuth.ai_agent_setting);
+        await ai_agent_service_1.ai_agentService.load_one_file(ctx.headers.authorization, data.param_path);
+        return (0, Result_1.Sucess)("");
+    }
+    async ai_load_restart(ctx, data) {
+        user_service_1.userService.check_user_auth(ctx.headers.authorization, user_req_1.UserAuth.ai_agent_setting);
+        await ai_agent_service_1.ai_agentService.close_index();
+        await threads_filecat_1.ThreadsFilecat.restart();
+        ws_server_1.Wss.sendToAllClient(WsData_1.CmdType.ai_load_info, ai_agent_service_1.ai_agentService.docs_info, ai_agent_service_1.ai_agentService.all_wss_set);
+        ai_agent_service_1.ai_agentService.init_search_docs().catch(console.error);
+        return (0, Result_1.Sucess)("");
+    }
+    async ai_load_close(ctx, data) {
+        user_service_1.userService.check_user_auth(ctx.headers.authorization, user_req_1.UserAuth.ai_agent_setting);
+        await ai_agent_service_1.ai_agentService.close_index();
+        await threads_filecat_1.ThreadsFilecat.close();
+        ws_server_1.Wss.sendToAllClient(WsData_1.CmdType.ai_load_info, ai_agent_service_1.ai_agentService.docs_info, ai_agent_service_1.ai_agentService.all_wss_set);
+        return (0, Result_1.Sucess)("");
+    }
+    async ai_del(ctx, data) {
+        user_service_1.userService.check_user_auth(ctx.headers.authorization, user_req_1.UserAuth.ai_agent_setting);
+        ai_agent_service_1.ai_agentService.delete_index_with_progress(data.param_path).catch(console.error);
+        return (0, Result_1.Sucess)("");
+    }
+};
+exports.Ai_AgentController = Ai_AgentController;
+__decorate([
+    (0, routing_controllers_1.Post)("/chat"),
+    __param(0, (0, routing_controllers_1.Body)()),
+    __param(1, (0, routing_controllers_1.Res)()),
+    __param(2, (0, routing_controllers_1.Req)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, Object, Object]),
+    __metadata("design:returntype", Promise)
+], Ai_AgentController.prototype, "chat", null);
+__decorate([
+    (0, router_1.msg)(WsData_1.CmdType.ai_load_info),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [WsData_1.WsData]),
+    __metadata("design:returntype", void 0)
+], Ai_AgentController.prototype, "get_info", null);
+__decorate([
+    (0, routing_controllers_1.Post)("/ai_load_one_file"),
+    __param(0, (0, routing_controllers_1.Req)()),
+    __param(1, (0, routing_controllers_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, Object]),
+    __metadata("design:returntype", Promise)
+], Ai_AgentController.prototype, "ai_load_one_file", null);
+__decorate([
+    (0, routing_controllers_1.Post)("/ai_load_restart"),
+    __param(0, (0, routing_controllers_1.Req)()),
+    __param(1, (0, routing_controllers_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, Object]),
+    __metadata("design:returntype", Promise)
+], Ai_AgentController.prototype, "ai_load_restart", null);
+__decorate([
+    (0, routing_controllers_1.Post)("/ai_load_close"),
+    __param(0, (0, routing_controllers_1.Req)()),
+    __param(1, (0, routing_controllers_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, Object]),
+    __metadata("design:returntype", Promise)
+], Ai_AgentController.prototype, "ai_load_close", null);
+__decorate([
+    (0, routing_controllers_1.Post)("/ai_del"),
+    __param(0, (0, routing_controllers_1.Req)()),
+    __param(1, (0, routing_controllers_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, Object]),
+    __metadata("design:returntype", Promise)
+], Ai_AgentController.prototype, "ai_del", null);
+exports.Ai_AgentController = Ai_AgentController = __decorate([
+    (0, routing_controllers_1.JsonController)("/ai_agent")
+], Ai_AgentController);
+//# sourceMappingURL=ai_agent.controller.js.map

package/build/server/main/domain/ai_agent/ai_agent.controller.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai_agent.controller.js","sourceRoot":"","sources":["../../../../../src/main/domain/ai_agent/ai_agent.controller.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,6DAAyE;AAEzE,yDAAmD;AACnD,uDAAiD;AACjD,2DAAsD;AACtD,yDAAiD;AACjD,yDAA6D;AAC7D,+DAAoD;AACpD,+CAA0C;AAC1C,2EAAqE;AAG9D,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;IAGrB,AAAN,KAAK,CAAC,IAAI,CAAS,IAAS,EAAS,GAAa,EAAS,GAAG;QAC1D,0BAAW,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,EAAE,mBAAQ,CAAC,aAAa,CAAC,CAAC;QAG/E,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,kCAAkC,CAAC,CAAC;QAClE,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;QAC3C,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAC1C,GAAG,CAAC,SAAS,CAAC,mBAAmB,EAAE,IAAI,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,aAAa,CAAA;QACvC,IAAI,MAAM,CAAE;QACZ,IAAI,CAAC;YACD,MAAM,GAAI,MAAM,kCAAe,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,KAAK,CAAC,CAAA;QACnE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACjB,IAAI,CAAC,GAAG,CAAC,aAAa,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC;gBACvC,MAAM,OAAO,GACT,OAAO,GAAG,KAAK,QAAQ;oBACnB,CAAC,CAAC,GAAG;oBACL,CAAC,CAAC,GAAG,EAAE,OAAO,IAAI,kBAAkB,CAAC;gBAG7C,kCAAe,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;gBAG3C,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;gBAG9B,GAAG,CAAC,GAAG,EAAE,CAAC;YACd,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAA;IACjB,CAAC;IAGD,QAAQ,CAAC,IAAiB;QACtB,MAAM,GAAG,GAAI,IAAI,CAAC,GAAW,CAAA;QAC7B,0BAAW,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,EAAE,mBAAQ,CAAC,gBAAgB,CAAC,CAAC;QAClE,kCAAe,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QACpC,GAAG,CAAC,QAAQ,CAAC,GAAE,EAAE;YACb,kCAAe,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;QAC3C,CAAC,CAAC,CAAA;QACF,OAAO,kCAAe,CAAC,SAAS,CAAA;IACpC,CAAC;IAGK,AAAN,KAAK,CAAC,gBAAgB,CAAQ,GAAG,EAAU,IAAS;QAChD,0BAAW,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,EAAE,mBAAQ,CAAC,gBAAgB,CAAC,CAAC;QAClF,MAAM,kCAAe,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QAC/E,OAAQ,IAAA,eAAM,EAAC,EAAE,CAAC,CAAA;IACtB,CAAC;IAGK,AAAN,KAAK,CAAC,eAAe,CAAQ,GAAG,EAAU,IAAS;QAC/C,0BAAW,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,EAAE,mBAAQ,CAAC,gBAAgB,CAAC,CAAC;QAClF,MAAM,kCAAe,CAAC,WAAW,EAAE,CAAA;QACnC,MAAM,gCAAc,CAAC,OAAO,EAAE,CAAA;QAC9B,eAAG,CAAC,eAAe,CAAC,gBAAO,CAAC,YAAY,EAAE,kCAAe,CAAC,SAAS,EAAC,kCAAe,CAAC,WAAW,CAAC,CAAA;QAChG,kCAAe,CAAC,gBAAgB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACxD,OAAQ,IAAA,eAAM,EAAC,EAAE,CAAC,CAAA;IACtB,CAAC;IAGK,AAAN,KAAK,CAAC,aAAa,CAAQ,GAAG,EAAU,IAAS;QAC7C,0BAAW,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,EAAE,mBAAQ,CAAC,gBAAgB,CAAC,CAAC;QAClF,MAAM,kCAAe,CAAC,WAAW,EAAE,CAAA;QACnC,MAAM,gCAAc,CAAC,KAAK,EAAE,CAAA;QAC5B,eAAG,CAAC,eAAe,CAAC,gBAAO,CAAC,YAAY,EAAE,kCAAe,CAAC,SAAS,EAAC,kCAAe,CAAC,WAAW,CAAC,CAAA;QAChG,OAAQ,IAAA,eAAM,EAAC,EAAE,CAAC,CAAA;IACtB,CAAC;IAGK,AAAN,KAAK,CAAC,MAAM,CAAQ,GAAG,EAAU,IAAS;QACtC,0BAAW,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,EAAE,mBAAQ,CAAC,gBAAgB,CAAC,CAAC;QAClF,kCAAe,CAAC,0BAA0B,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACjF,OAAQ,IAAA,eAAM,EAAC,EAAE,CAAC,CAAA;IACtB,CAAC;CAGJ,CAAA;AAjFY,gDAAkB;AAGrB;IADL,IAAA,0BAAI,EAAC,OAAO,CAAC;IACF,WAAA,IAAA,0BAAI,GAAE,CAAA;IAAa,WAAA,IAAA,yBAAG,GAAE,CAAA;IAAiB,WAAA,IAAA,yBAAG,GAAE,CAAA;;;;8CA+BzD;AAGD;IADC,IAAA,YAAG,EAAC,gBAAO,CAAC,YAAY,CAAC;;qCACX,eAAM;;kDAQpB;AAGK;IADL,IAAA,0BAAI,EAAC,mBAAmB,CAAC;IACF,WAAA,IAAA,yBAAG,GAAE,CAAA;IAAO,WAAA,IAAA,0BAAI,GAAE,CAAA;;;;0DAIzC;AAGK;IADL,IAAA,0BAAI,EAAC,kBAAkB,CAAC;IACF,WAAA,IAAA,yBAAG,GAAE,CAAA;IAAO,WAAA,IAAA,0BAAI,GAAE,CAAA;;;;yDAOxC;AAGK;IADL,IAAA,0BAAI,EAAC,gBAAgB,CAAC;IACF,WAAA,IAAA,yBAAG,GAAE,CAAA;IAAO,WAAA,IAAA,0BAAI,GAAE,CAAA;;;;uDAMtC;AAGK;IADL,IAAA,0BAAI,EAAC,SAAS,CAAC;IACF,WAAA,IAAA,yBAAG,GAAE,CAAA;IAAO,WAAA,IAAA,0BAAI,GAAE,CAAA;;;;gDAI/B;6BA9EQ,kBAAkB;IAD9B,IAAA,oCAAc,EAAC,WAAW,CAAC;GACf,kBAAkB,CAiF9B"}