Method addMessage

packages/node-runtime/src/ai/chats.ts:587–644

(
    aiChatId: string,
    role: AIMessageRole,
    content: string,
    dataKeywords?: string[],
    dataMessageCount?: number,
    contentBlocks?: ContentBlock[],
    tokenUsage?: TokenUsageData
  )

  // ==================== 消息管理 ====================

  addMessage(
    aiChatId: string,
    role: AIMessageRole,
    content: string,
    dataKeywords?: string[],
    dataMessageCount?: number,
    contentBlocks?: ContentBlock[],
    tokenUsage?: TokenUsageData
  ): AIMessage {
    const db = this.getDb()
    const now = Math.floor(Date.now() / 1000)
    const id = this.generateId('msg')
    const parentId = this.getActiveMessageId(aiChatId)
    const siblingGroupId = id
    const branchIndex = 0

    const pendingDebug = role === 'assistant' ? this.pendingDebugContextMap.get(aiChatId) : undefined
    if (pendingDebug) {
      this.pendingDebugContextMap.delete(aiChatId)
    }

    db.prepare(
      `INSERT INTO ai_message (
         id, ai_chat_id, role, content, timestamp, data_keywords, data_message_count,
         content_blocks, token_usage, debug_context, parent_id, sibling_group_id, branch_index
       )
       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
    ).run(
      id,
      aiChatId,
      role,
      content,
      now,
      dataKeywords ? JSON.stringify(dataKeywords) : null,
      dataMessageCount ?? null,
      contentBlocks ? JSON.stringify(contentBlocks) : null,
      tokenUsage ? JSON.stringify(tokenUsage) : null,
      pendingDebug ?? null,
      parentId,
      siblingGroupId,
      branchIndex
    )

    db.prepare('UPDATE ai_chat SET active_message_id = ?, updated_at = ? WHERE id = ?').run(id, now, aiChatId)

    return {
      id,
      aiChatId,
      role,
      content,
      timestamp: now,
      parentId,
      dataKeywords,
      dataMessageCount,
      contentBlocks,
      tokenUsage,
    }
  }