llm-content-creator 0.1.0

package/dist/domain/workflow/nodes/domain/workflow/nodes/BaseNode.js

@@ -0,0 +1,328 @@
+"use strict";
+/**
+ * BaseNode - 节点基类
+ *
+ * 所有工作流节点的基类，提供通用功能：
+ * - 错误处理
+ * - Token 记录
+ * - 日志记录
+ * - 状态验证
+ * - 重试支持
+ */
+var __assign = (this && this.__assign) || function () {
+    __assign = Object.assign || function(t) {
+        for (var s, i = 1, n = arguments.length; i < n; i++) {
+            s = arguments[i];
+            for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
+                t[p] = s[p];
+        }
+        return t;
+    };
+    return __assign.apply(this, arguments);
+};
+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.NodeContext = exports.BaseNode = void 0;
+var logger_js_1 = require("../../../infrastructure/logging/logger.js");
+var logger = (0, logger_js_1.createLogger)('BaseNode');
+/**
+ * 节点基类
+ */
+var BaseNode = /** @class */ (function () {
+    function BaseNode(config) {
+        var _a, _b;
+        this.name = config.name;
+        this.retryCount = (_a = config.retryCount) !== null && _a !== void 0 ? _a : 0;
+        this.timeout = (_b = config.timeout) !== null && _b !== void 0 ? _b : 60000; // 60 秒默认超时
+        this.logger = (0, logger_js_1.createLogger)("Node:".concat(this.name));
+    }
+    /**
+     * 执行节点（带重试和错误处理）
+     *
+     * @param state - 当前工作流状态
+     * @returns 节点执行结果
+     */
+    BaseNode.prototype.execute = function (state) {
+        return __awaiter(this, void 0, void 0, function () {
+            var startTime, attempt, stateUpdate, duration, error_1, duration, errorMessage;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        startTime = Date.now();
+                        attempt = this.getAttemptCount(state);
+                        this.logger.info('Node execution started', {
+                            nodeId: this.name,
+                            taskId: state.taskId,
+                            attempt: attempt + 1,
+                        });
+                        _a.label = 1;
+                    case 1:
+                        _a.trys.push([1, 3, , 4]);
+                        // 验证状态
+                        this.validateState(state);
+                        return [4 /*yield*/, this.executeWithTimeout(state)];
+                    case 2:
+                        stateUpdate = _a.sent();
+                        duration = Date.now() - startTime;
+                        // 记录成功日志
+                        this.logger.info('Node execution completed', {
+                            nodeId: this.name,
+                            taskId: state.taskId,
+                            duration: duration,
+                        });
+                        // 返回结果
+                        return [2 /*return*/, {
+                                success: true,
+                                stateUpdate: __assign(__assign({}, stateUpdate), { currentStep: this.name }),
+                                duration: duration,
+                            }];
+                    case 3:
+                        error_1 = _a.sent();
+                        duration = Date.now() - startTime;
+                        errorMessage = error_1 instanceof Error ? error_1.message : String(error_1);
+                        // 记录错误日志
+                        this.logger.error('Node execution failed', {
+                            nodeId: this.name,
+                            taskId: state.taskId,
+                            error: errorMessage,
+                            duration: duration,
+                        });
+                        // 返回错误结果
+                        return [2 /*return*/, {
+                                success: false,
+                                stateUpdate: {
+                                    error: errorMessage,
+                                },
+                                error: errorMessage,
+                                duration: duration,
+                            }];
+                    case 4: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+     * 验证状态（可选，由子类重写）
+     *
+     * @param state - 当前工作流状态
+     * @throws Error 如果状态无效
+     */
+    BaseNode.prototype.validateState = function (state) {
+        // 默认验证：检查是否有错误
+        if (state.error) {
+            throw new Error("Previous error: ".concat(state.error));
+        }
+    };
+    /**
+     * 获取当前尝试次数（可选，由子类重写）
+     *
+     * @param state - 当前工作流状态
+     * @returns 尝试次数
+     */
+    BaseNode.prototype.getAttemptCount = function (state) {
+        // 默认：返回 0
+        // 子类可以根据 state 中的重试计数器返回实际值
+        return 0;
+    };
+    /**
+     * 执行节点逻辑（带超时控制）
+     *
+     * @param state - 当前工作流状态
+     * @returns 状态更新
+     */
+    BaseNode.prototype.executeWithTimeout = function (state) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, Promise.race([
+                        this.executeLogic(state),
+                        this.createTimeoutPromise(),
+                    ])];
+            });
+        });
+    };
+    /**
+     * 创建超时 Promise
+     */
+    BaseNode.prototype.createTimeoutPromise = function () {
+        var _this = this;
+        return new Promise(function (_, reject) {
+            setTimeout(function () {
+                reject(new Error("Node timeout: ".concat(_this.name, " exceeded ").concat(_this.timeout, "ms")));
+            }, _this.timeout);
+        });
+    };
+    /**
+     * 记录 LLM Token 使用（可选，由子类调用）
+     *
+     * @param state - 工作流状态
+     * @param tokensIn - 输入 Token 数量
+     * @param tokensOut - 输出 Token 数量
+     */
+    BaseNode.prototype.recordTokenUsage = function (state, tokensIn, tokensOut) {
+        this.logger.debug('Token usage recorded', {
+            nodeId: this.name,
+            taskId: state.taskId,
+            tokensIn: tokensIn,
+            tokensOut: tokensOut,
+            totalTokens: tokensIn + tokensOut,
+        });
+        // TODO: 保存到数据库（后续实现）
+        // await tokenUsageRepo.create({
+        //   taskId: state.taskId,
+        //   traceId: generateTraceId(),
+        //   stepName: this.name,
+        //   apiName: 'deepseek',
+        //   modelName: 'deepseek-chat',
+        //   tokensIn,
+        //   tokensOut,
+        //   totalTokens: tokensIn + tokensOut,
+        //   ...
+        // });
+    };
+    /**
+     * 创建状态更新辅助函数
+     */
+    BaseNode.prototype.updateState = function (field, value) {
+        var _a;
+        return _a = {}, _a[field] = value, _a;
+    };
+    /**
+     * 转换为 LangGraph 节点
+     *
+     * @returns LangGraph 节点函数
+     */
+    BaseNode.prototype.toLangGraphNode = function () {
+        var _this = this;
+        return function (state) { return __awaiter(_this, void 0, void 0, function () {
+            var result;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        logger.debug("Node ".concat(this.name, " received state"), {
+                            taskId: state.taskId,
+                            topic: state.topic,
+                            hasTopic: !!state.topic,
+                            stateKeys: Object.keys(state),
+                        });
+                        return [4 /*yield*/, this.execute(state)];
+                    case 1:
+                        result = _a.sent();
+                        if (!result.success) {
+                            throw new Error(result.error || 'Node execution failed');
+                        }
+                        return [2 /*return*/, result.stateUpdate];
+                }
+            });
+        }); };
+    };
+    return BaseNode;
+}());
+exports.BaseNode = BaseNode;
+/**
+ * 节点执行上下文
+ *
+ * 提供额外的工具和方法供节点使用
+ */
+var NodeContext = /** @class */ (function () {
+    function NodeContext(nodeName) {
+        this.logger = (0, logger_js_1.createLogger)("Node:".concat(nodeName));
+    }
+    /**
+     * 生成 Trace ID（用于链路追踪）
+     */
+    NodeContext.prototype.generateTraceId = function () {
+        return "".concat(Date.now(), "-").concat(Math.random().toString(36).substring(2, 15));
+    };
+    /**
+     * 延迟执行（毫秒）
+     */
+    NodeContext.prototype.delay = function (ms) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, new Promise(function (resolve) { return setTimeout(resolve, ms); })];
+            });
+        });
+    };
+    /**
+     * 计算文本的 Token 数量（估算）
+     */
+    NodeContext.prototype.estimateTokens = function (text) {
+        // 粗略估算：英文约 4 字符/token，中文约 1.5 字符/token
+        var chineseChars = (text.match(/[\u4e00-\u9fa5]/g) || []).length;
+        var englishChars = text.length - chineseChars;
+        return Math.ceil(chineseChars / 1.5 + englishChars / 4);
+    };
+    /**
+     * 截断文本到指定 Token 数量
+     */
+    NodeContext.prototype.truncateToTokens = function (text, maxTokens) {
+        var estimatedTokens = this.estimateTokens(text);
+        if (estimatedTokens <= maxTokens) {
+            return text;
+        }
+        // 粗略截断：按比例缩短
+        var ratio = maxTokens / estimatedTokens;
+        var truncatedLength = Math.floor(text.length * ratio);
+        return text.substring(0, truncatedLength);
+    };
+    /**
+     * 安全地解析 JSON
+     */
+    NodeContext.prototype.safeParseJSON = function (json, fallback) {
+        try {
+            return JSON.parse(json);
+        }
+        catch (error) {
+            this.logger.warn('Failed to parse JSON, using fallback', {
+                error: error instanceof Error ? error.message : String(error),
+            });
+            return fallback;
+        }
+    };
+    /**
+     * 格式化错误消息
+     */
+    NodeContext.prototype.formatError = function (error) {
+        if (error instanceof Error) {
+            return error.message;
+        }
+        return String(error);
+    };
+    return NodeContext;
+}());
+exports.NodeContext = NodeContext;