Function createAIChatManager

apps/cli/src/ai/chat-command.test.ts:15–57

(
  existing: Array<{ id: string; sessionId: string; assistantId?: string }> = []
)

}

function createAIChatManager(
  existing: Array<{ id: string; sessionId: string; assistantId?: string }> = []
): AIChatManager {
  const chats = new Map<string, { id: string; sessionId: string; title: string | null; assistantId: string }>(
    existing.map((chat) => [chat.id, { ...chat, title: null, assistantId: chat.assistantId ?? 'general_cn' }])
  )
  const messages: Array<{
    aiChatId: string
    role: string
    content: string
    contentBlocks?: ContentBlock[]
    tokenUsage?: TokenUsageData
  }> = []

  return {
    getAIChat: (aiChatId: string) => chats.get(aiChatId) ?? null,
    createAIChat: (sessionId: string, title: string | undefined, assistantId: string) => {
      const id = `ai_chat_${chats.size + 1}`
      const chat = { id, sessionId, title: title ?? null, assistantId }
      chats.set(id, chat)
      return chat
    },
    addMessage: (
      aiChatId: string,
      role: string,
      content: string,
      _dataKeywords?: string[],
      _dataMessageCount?: number,
      contentBlocks?: ContentBlock[],
      tokenUsage?: TokenUsageData
    ) => {
      messages.push({
        aiChatId,
        role,
        content,
        ...(contentBlocks ? { contentBlocks } : {}),
        ...(tokenUsage ? { tokenUsage } : {}),
      })
      return { id: `msg_${messages.length}`, aiChatId, role, content, timestamp: 1 }
    },
    __messages: messages,
  } as unknown as AIChatManager
}

class MemoryWritable extends Writable {
  chunks: string[] = []