Method getDb

packages/node-runtime/src/ai/chats.ts:128–168

()

  }

  private getDb(): Database.Database {
    if (this.db) return this.db

    this.ensureDir(this.aiDataDir)
    const dbPath = path.join(this.aiDataDir, 'conversations.db')
    this.db = this.nativeBinding ? new Database(dbPath, { nativeBinding: this.nativeBinding }) : new Database(dbPath)
    this.db.pragma('journal_mode = WAL')

    this.db.exec(`
      CREATE TABLE IF NOT EXISTS ai_chat (
        id TEXT PRIMARY KEY,
        session_id TEXT NOT NULL,
        title TEXT,
        assistant_id TEXT DEFAULT '${DEFAULT_GENERAL_ID}',
        active_message_id TEXT,
        created_at INTEGER NOT NULL,
        updated_at INTEGER NOT NULL
      );

      CREATE TABLE IF NOT EXISTS ai_message (
        id TEXT PRIMARY KEY,
        ai_chat_id TEXT NOT NULL,
        role TEXT NOT NULL,
        content TEXT NOT NULL,
        timestamp INTEGER NOT NULL,
        data_keywords TEXT,
        data_message_count INTEGER,
        content_blocks TEXT,
        parent_id TEXT,
        sibling_group_id TEXT,
        branch_index INTEGER DEFAULT 0,
        debug_context TEXT,
        token_usage TEXT,
        FOREIGN KEY(ai_chat_id) REFERENCES ai_chat(id) ON DELETE CASCADE
      );

    `)

    this.migrateDatabase(this.db)
    return this.db
  }

  private tableExists(db: Database.Database, tableName: string): boolean {
    const row = db.prepare("SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = ?").get(tableName) as