llm-content-creator 0.1.0

package/src/domain/entities/Task.ts

@@ -0,0 +1,178 @@
+/**
+ * Task 领域模型
+ *
+ * 表示一个内容创作任务
+ */
+
+/**
+ * 任务状态
+ */
+export enum TaskStatus {
+  PENDING = 'pending',           // 待处理
+  RUNNING = 'running',           // 运行中
+  WAITING = 'waiting',           // 等待中（等待人工审核）
+  COMPLETED = 'completed',       // 已完成
+  FAILED = 'failed',             // 失败
+  CANCELLED = 'cancelled',       // 已取消
+}
+
+/**
+ * 任务类型
+ */
+export enum TaskType {
+  ARTICLE = 'article',           // 文章
+  SOCIAL_MEDIA = 'social_media', // 社交媒体内容
+  MARKETING = 'marketing',       // 营销文案
+  OTHER = 'other',               // 其他
+}
+
+/**
+ * 任务优先级
+ */
+export enum TaskPriority {
+  LOW = 1,
+  NORMAL = 2,
+  HIGH = 3,
+  URGENT = 4,
+}
+
+/**
+ * 执行模式
+ */
+export enum ExecutionMode {
+  SYNC = 'sync',                 // 同步执行
+  ASYNC = 'async',               // 异步执行
+}
+
+/**
+ * 任务实体
+ */
+export interface Task {
+  // 基础字段
+  id: string;                    // 任务 ID (UUID) - 别名为 taskId
+  taskId: string;                // 任务 ID (UUID) - 与 id 相同
+  type?: TaskType;               // 任务类型（可选，向后兼容）
+  status: TaskStatus;            // 任务状态
+  priority: TaskPriority;        // 优先级
+
+  // 执行模式和需求
+  mode: ExecutionMode;           // 执行模式（同步/异步）
+  topic: string;                 // 选题/主题
+  requirements: string;          // 需求描述
+  targetAudience: string;        // 目标受众
+  keywords?: string[];           // 关键词
+  tone?: string;                 // 语气风格
+
+  // 硬性约束
+  hardConstraints?: {            // 硬性指标
+    minWords?: number;           // 最小字数
+    maxWords?: number;           // 最大字数
+    keywords?: string[];         // 必须包含的关键词
+  };
+
+  // 元数据
+  userId?: string;               // 用户 ID（外键，可选）
+  estimatedTokens?: number;      // 预估 Token 数量
+  maxTokens?: number;            // 最大 Token 限制
+
+  // 状态字段
+  currentStep?: string;          // 当前步骤
+  workerId?: string;             // Worker ID（多 Worker 抢占）
+  assignedWorkerId?: string;     // 分配的 Worker ID（别名，向后兼容）
+
+  // 重试计数（持久化）
+  textRetryCount: number;        // 文本质检重试次数
+  imageRetryCount: number;       // 配图质检重试次数
+
+  // 乐观锁（并发控制）
+  version: number;               // 版本号
+
+  // 时间戳
+  createdAt: Date;               // 创建时间
+  updatedAt: Date;               // 更新时间
+  startedAt?: Date;              // 开始时间
+  completedAt?: Date;            // 完成时间
+  deletedAt?: Date;              // 软删除时间
+
+  // 抢占时间（向后兼容）
+  claimedAt?: Date;              // 抢占时间
+
+  // 错误信息
+  errorMessage?: string;         // 错误信息
+
+  // State 快照（崩溃恢复）
+  stateSnapshot?: object;        // LangGraph State 序列化
+
+  // 幂等性
+  idempotencyKey?: string;       // 幂等键（防止重复提交）
+
+  // 结果引用（一对多）
+  steps?: TaskStep[];            // 任务步骤（可选，用于关联查询）
+  results?: Result[];            // 生成结果（可选，用于关联查询）
+}
+
+/**
+ * 任务创建参数
+ */
+export interface CreateTaskParams {
+  mode: ExecutionMode;           // 执行模式
+  topic: string;
+  requirements: string;
+  targetAudience: string;
+  type?: TaskType;               // 任务类型（可选）
+  keywords?: string[];
+  tone?: string;
+  priority?: TaskPriority;
+  maxTokens?: number;
+  userId?: string;               // 用户 ID（可选）
+  hardConstraints?: {            // 硬性约束
+    minWords?: number;
+    maxWords?: number;
+    keywords?: string[];
+  };
+  idempotencyKey?: string;       // 幂等键（可选）
+}
+
+/**
+ * 任务更新参数
+ */
+export interface UpdateTaskParams {
+  status?: TaskStatus;
+  currentStep?: string;
+  workerId?: string;
+  assignedWorkerId?: string;     // 别名，向后兼容
+  claimedAt?: Date;
+  startedAt?: Date;
+  completedAt?: Date;
+  estimatedTokens?: number;
+  errorMessage?: string;
+  stateSnapshot?: object;
+  version: number;               // 必需：用于乐观锁
+}
+
+import type { TaskStep } from './TaskStep.js';
+import type { Result } from './Result.js';
+
+/**
+ * 任务状态转换
+ */
+export const TaskStatusTransitions: Record<TaskStatus, TaskStatus[]> = {
+  [TaskStatus.PENDING]: [TaskStatus.RUNNING, TaskStatus.CANCELLED],
+  [TaskStatus.RUNNING]: [TaskStatus.WAITING, TaskStatus.COMPLETED, TaskStatus.FAILED, TaskStatus.CANCELLED],
+  [TaskStatus.WAITING]: [TaskStatus.RUNNING, TaskStatus.COMPLETED, TaskStatus.FAILED, TaskStatus.CANCELLED],
+  [TaskStatus.COMPLETED]: [], // 终态，不可转换
+  [TaskStatus.FAILED]: [],    // 终态，不可转换
+  [TaskStatus.CANCELLED]: [], // 终态，不可转换
+};
+
+/**
+ * 检查状态转换是否合法
+ */
+export function isValidStatusTransition(
+  currentStatus: TaskStatus,
+  newStatus: TaskStatus
+): boolean {
+  const allowedTransitions = TaskStatusTransitions[currentStatus];
+  return allowedTransitions.includes(newStatus);
+}
+

package/src/domain/entities/TaskStep.ts

@@ -0,0 +1,188 @@
+/**
+ * TaskStep 领域模型
+ *
+ * 表示任务的一个执行步骤
+ */
+
+import { TaskStatus } from './Task.js';
+
+/**
+ * 步骤类型
+ */
+export enum StepType {
+  SEARCH = 'search',             // 搜索选题
+  ORGANIZE = 'organize',         // 整理资料
+  WRITE = 'write',               // 撰写内容
+  CHECK_TEXT = 'check_text',     // 文本质量检查
+  GEN_IMAGE = 'gen_image',       // 生成配图
+  CHECK_IMAGE = 'check_image',   // 图片质量检查
+}
+
+/**
+ * 步骤输入定义
+ */
+export interface StepInput {
+  [key: string]: any;
+}
+
+/**
+ * 搜索步骤输入
+ */
+export interface SearchStepInput extends StepInput {
+  topic: string;
+  requirements: string;
+  targetAudience: string;
+  keywords?: string[];
+}
+
+/**
+ * 整理步骤输入
+ */
+export interface OrganizeStepInput extends StepInput {
+  searchResults: any[];
+  requirements: string;
+}
+
+/**
+ * 撰写步骤输入
+ */
+export interface WriteStepInput extends StepInput {
+  organizedContent: any;
+  topic: string;
+  targetAudience: string;
+  tone?: string;
+  maxTokens?: number;
+}
+
+/**
+ * 文本检查步骤输入
+ */
+export interface CheckTextStepInput extends StepInput {
+  text: string;
+  requirements: string;
+}
+
+/**
+ * 生成图片步骤输入
+ */
+export interface GenImageStepInput extends StepInput {
+  text: string;
+  requirements: string;
+}
+
+/**
+ * 图片检查步骤输入
+ */
+export interface CheckImageStepInput extends StepInput {
+  imageUrl: string;
+  requirements: string;
+}
+
+/**
+ * 步骤输出定义
+ */
+export interface StepOutput {
+  [key: string]: any;
+}
+
+/**
+ * 搜索步骤输出
+ */
+export interface SearchStepOutput extends StepOutput {
+  results: any[];
+  summary: string;
+}
+
+/**
+ * 整理步骤输出
+ */
+export interface OrganizeStepOutput extends StepOutput {
+  organized: any;
+  outline: string;
+}
+
+/**
+ * 撰写步骤输出
+ */
+export interface WriteStepOutput extends StepOutput {
+  text: string;
+  tokenCount: number;
+}
+
+/**
+ * 检查步骤输出
+ */
+export interface CheckStepOutput extends StepOutput {
+  passed: boolean;
+  score?: number;
+  reason?: string;
+  details?: any;
+}
+
+/**
+ * 步骤状态
+ */
+export enum StepStatus {
+  PENDING = 'pending',
+  RUNNING = 'running',
+  COMPLETED = 'completed',
+  FAILED = 'failed',
+  SKIPPED = 'skipped',
+}
+
+/**
+ * 任务步骤实体
+ */
+export interface TaskStep {
+  // 基础字段
+  id: number;                    // 自增 ID
+  taskId: string;                // 所属任务 ID
+  stepType: StepType;            // 步骤类型
+  status: StepStatus;            // 步骤状态
+  attempt: number;               // 尝试次数（用于重试）
+
+  // 输入输出
+  inputData?: object;            // 输入数据 (JSON)
+  outputData?: object;           // 输出数据 (JSON，可选)
+
+  // 兼容旧接口
+  input?: StepInput;             // 输入参数（别名）
+  output?: StepOutput;           // 输出结果（别名）
+
+  // 错误处理
+  errorMessage?: string;         // 错误信息（可选）
+  error?: string;                // 错误信息（别名，向后兼容）
+  retryCount?: number;           // 重试次数（别名，向后兼容）
+
+  // 时间戳
+  createdAt: Date;               // 创建时间
+  startedAt?: Date;              // 开始时间
+  completedAt?: Date;            // 完成时间
+  durationMs?: number;           // 执行耗时（毫秒）
+
+  // 执行信息（向后兼容）
+  duration?: number;             // 执行时长（毫秒，别名）
+  workerId?: string;             // 执行 Worker ID
+}
+
+/**
+ * 创建步骤参数
+ */
+export interface CreateStepParams {
+  taskId: string;
+  stepType: StepType;
+  input: StepInput;
+  workerId?: string;
+}
+
+/**
+ * 更新步骤参数
+ */
+export interface UpdateStepParams {
+  status?: TaskStatus;
+  output?: StepOutput;
+  error?: string;
+  startedAt?: Date;
+  completedAt?: Date;
+  duration?: number;
+}

package/src/domain/entities/TokenUsage.ts

@@ -0,0 +1,119 @@
+/**
+ * TokenUsage 领域模型
+ *
+ * 表示 API Token 使用记录和成本统计
+ */
+
+/**
+ * Token 使用记录实体
+ */
+export interface TokenUsage {
+  // 基础字段
+  id: number;                    // 自增 ID
+  taskId: string;                // 任务 ID
+  traceId: string;               // 链路追踪 ID
+  stepName: string;              // 步骤名称
+  apiName: string;               // API 名称 (deepseek, doubao)
+  modelName: string;             // 模型名称
+
+  // Token 统计
+  tokensIn: number;              // 输入 token
+  tokensOut: number;             // 输出 token
+  totalTokens: number;           // 总 token
+
+  // 成本计算
+  costPer1kTokensIn: number;     // 每 1k 输入 token 成本
+  costPer1kTokensOut: number;    // 每 1k 输出 token 成本
+  totalCost: number;             // 总成本
+
+  // 元数据
+  metadata?: TokenUsageMetadata; // 元数据（可选）
+
+  // 时间戳
+  createdAt: Date;               // 创建时间
+}
+
+/**
+ * Token 使用元数据
+ */
+export interface TokenUsageMetadata {
+  temperature?: number;          // 温度参数
+  maxTokens?: number;            // 最大 token 限制
+  duration?: number;             // API 调用耗时（毫秒）
+
+  // 其他元数据
+  [key: string]: any;
+}
+
+/**
+ * 创建 Token 使用记录参数
+ */
+export interface CreateTokenUsageParams {
+  taskId: string;
+  traceId: string;
+  stepName: string;
+  apiName: string;
+  modelName: string;
+  tokensIn: number;
+  tokensOut: number;
+  costPer1kTokensIn: number;
+  costPer1kTokensOut: number;
+  metadata?: TokenUsageMetadata;
+}
+
+/**
+ * Token 成本配置
+ */
+export interface TokenCostConfig {
+  apiName: string;
+  modelName: string;
+  costPer1kTokensIn: number;
+  costPer1kTokensOut: number;
+}
+
+/**
+ * 常用 API 成本配置（单位：元/1k tokens）
+ */
+export const TOKEN_COST_CONFIGS: Record<string, TokenCostConfig[]> = {
+  deepseek: [
+    {
+      apiName: 'deepseek',
+      modelName: 'deepseek-chat',
+      costPer1kTokensIn: 0.001,  // ¥0.001/1k tokens
+      costPer1kTokensOut: 0.002, // ¥0.002/1k tokens
+    },
+  ],
+  doubao: [
+    {
+      apiName: 'doubao',
+      modelName: 'doubao-v1',
+      costPer1kTokensIn: 0.008,
+      costPer1kTokensOut: 0.008,
+    },
+  ],
+};
+
+/**
+ * 计算 Token 成本
+ */
+export function calculateTokenCost(
+  tokensIn: number,
+  tokensOut: number,
+  costPer1kTokensIn: number,
+  costPer1kTokensOut: number
+): number {
+  const costIn = (tokensIn / 1000) * costPer1kTokensIn;
+  const costOut = (tokensOut / 1000) * costPer1kTokensOut;
+  return Number((costIn + costOut).toFixed(6));
+}
+
+/**
+ * 获取 API 成本配置
+ */
+export function getTokenCostConfig(
+  apiName: string,
+  modelName: string
+): TokenCostConfig | undefined {
+  const configs = TOKEN_COST_CONFIGS[apiName];
+  return configs?.find(config => config.modelName === modelName);
+}

package/src/domain/entities/index.ts

@@ -0,0 +1,20 @@
+/**
+ * 领域实体导出
+ *
+ * 统一导出所有领域模型
+ */
+
+// Task 实体 - 包含基础的 Task、TaskStep、Result、QualityCheck 接口
+export * from './Task.js';
+
+// TaskStep 实体 - 包含详细的步骤类型和输入输出定义
+export * from './TaskStep.js';
+
+// Result 实体 - 包含详细的结果类型和元数据定义
+export * from './Result.js';
+
+// QualityCheck 实体 - 包含详细的质量检查类型和检查结果定义
+export * from './QualityCheck.js';
+
+// TokenUsage 实体
+export * from './TokenUsage.js';

package/src/domain/repositories/QualityCheckRepository.ts

@@ -0,0 +1,52 @@
+/**
+ * Quality Check Repository 接口
+ */
+
+/**
+ * 创建质量检查记录参数
+ */
+export interface CreateQualityCheckParams {
+  taskId: string;
+  checkType: 'text' | 'image';
+  score: number;
+  passed: boolean;
+  hardConstraintsPassed: boolean;
+  details: Record<string, any>;
+  fixSuggestions?: string[];
+  rubricVersion?: string;
+  modelName?: string;
+  promptHash?: string;
+}
+
+/**
+ * 质量检查记录
+ */
+export interface QualityCheckRecord {
+  id: number;
+  taskId: string;
+  checkType: 'text' | 'image';
+  score: number;
+  passed: boolean;
+  hardConstraintsPassed: boolean;
+  details: Record<string, any>;
+  fixSuggestions: string[] | null;
+  rubricVersion: string | null;
+  modelName: string | null;
+  promptHash: string | null;
+  createdAt: Date;
+}
+
+/**
+ * Quality Check Repository 接口
+ */
+export interface IQualityCheckRepository {
+  /**
+   * 创建质量检查记录
+   */
+  create(params: CreateQualityCheckParams): Promise<void>;
+
+  /**
+   * 根据任务 ID 查询质量检查
+   */
+  findByTaskId(taskId: string): Promise<QualityCheckRecord[]>;
+}

package/src/domain/repositories/ResultRepository.ts

@@ -0,0 +1,47 @@
+/**
+ * Result Repository 接口
+ */
+
+/**
+ * 创建结果参数
+ */
+export interface CreateResultParams {
+  taskId: string;
+  resultType: 'article' | 'image' | 'text';
+  content?: string;
+  filePath?: string;
+  metadata?: Record<string, any>;
+}
+
+/**
+ * 结果记录
+ */
+export interface ResultRecord {
+  id: number;
+  taskId: string;
+  resultType: 'article' | 'image' | 'text';
+  content: string | null;
+  filePath: string | null;
+  metadata: Record<string, any> | null;
+  createdAt: Date;
+}
+
+/**
+ * Result Repository 接口
+ */
+export interface IResultRepository {
+  /**
+   * 创建结果记录
+   */
+  create(params: CreateResultParams): Promise<void>;
+
+  /**
+   * 根据任务 ID 查询结果
+   */
+  findByTaskId(taskId: string): Promise<ResultRecord[]>;
+
+  /**
+   * 删除任务的所有结果
+   */
+  deleteByTaskId(taskId: string): Promise<void>;
+}