Function main

src/data/embed.ts:34–179

()

}

async function main() {
  console.log('Loading items from database...');
  const db = new Database(DB_PATH, { readonly: true });

  // Get all issues and PRs
  const issues = db
    .prepare(
      `
    SELECT i.id, i.title, i.body, i.number, r.full_name as repo
    FROM issues i
    JOIN repos r ON i.repo_id = r.id
    WHERE i.title IS NOT NULL
  `,
    )
    .all() as Array<{
    id: number;
    title: string;
    body: string | null;
    number: number;
    repo: string;
  }>;

  const pulls = db
    .prepare(
      `
    SELECT p.id, p.title, p.body, p.number, r.full_name as repo
    FROM pulls p
    JOIN repos r ON p.repo_id = r.id
    WHERE p.title IS NOT NULL
  `,
    )
    .all() as Array<{
    id: number;
    title: string;
    body: string | null;
    number: number;
    repo: string;
  }>;

  console.log(`Found ${issues.length} issues and ${pulls.length} PRs`);

  const allItems = [
    ...issues.map((i) => ({ ...i, type: 'issue' as const })),
    ...pulls.map((p) => ({ ...p, type: 'pull' as const })),
  ];

  console.log(`Total: ${allItems.length} items to embed`);

  // Check for existing embeddings
  let existingIndex: EmbeddingIndex | null = null;
  let existingEmbeddings: Float32Array | null = null;

  if (existsSync(INDEX_PATH) && existsSync(EMBEDDINGS_PATH)) {
    console.log('Found existing embeddings, checking for updates...');
    existingIndex = JSON.parse(readFileSync(INDEX_PATH, 'utf-8'));
    const buffer = readFileSync(EMBEDDINGS_PATH);
    existingEmbeddings = new Float32Array(buffer.buffer, buffer.byteOffset, buffer.byteLength / 4);