Added compare script and fixed key name issue in fetch models

Fixes #310

Author: tombeckenham

scripts/compare-openrouter-models.ts

@@ -0,0 +1,173 @@
+/**
+ * Compare openrouter.models.ts between the current branch and main.
+ * Outputs a markdown summary of added/removed/updated models suitable for a PR description.
+ *
+ * Usage: npx tsx scripts/compare-openrouter-models.ts [--updated]
+ */
+
+import { execSync } from 'node:child_process'
+import { readFileSync } from 'node:fs'
+import { resolve } from 'node:path'
+
+interface ModelSnapshot {
+  id: string
+  name: string
+  context_length: string
+  pricing_prompt: string
+  pricing_completion: string
+  modality: string
+}
+
+/**
+ * Extract model snapshots from source using regex.
+ * Captures the key fields that reviewers care about for detecting updates.
+ */
+function extractModels(source: string): Map<string, ModelSnapshot> {
+  const models = new Map<string, ModelSnapshot>()
+
+  // Split into model blocks (each starts with `  {` and ends with `  },` or `  }\n]`)
+  const blocks = source.split(/\n {2}\{/)
+
+  for (const block of blocks) {
+    const id = block.match(/id:\s*'([^']+)'/)?.[1]
+    if (!id) continue
+
+    const name = block.match(/name:\s*'([^']+)'/)?.[1] ?? ''
+    const context_length =
+      block.match(/^\s*context_length:\s*(.+),?\s*$/m)?.[1] ?? ''
+    const pricing_prompt = block.match(/prompt:\s*'([^']+)'/)?.[1] ?? ''
+    const pricing_completion = block.match(/completion:\s*'([^']+)'/)?.[1] ?? ''
+    const modality = block.match(/modality:\s*'([^']+)'/)?.[1] ?? ''
+
+    models.set(id, {
+      id,
+      name,
+      context_length,
+      pricing_prompt,
+      pricing_completion,
+      modality,
+    })
+  }
+
+  return models
+}
+
+function describeChanges(
+  oldModel: ModelSnapshot,
+  newModel: ModelSnapshot,
+): Array<string> {
+  const changes: Array<string> = []
+
+  if (oldModel.name !== newModel.name) {
+    changes.push(`name: "${oldModel.name}" → "${newModel.name}"`)
+  }
+  if (oldModel.context_length !== newModel.context_length) {
+    changes.push(
+      `context_length: ${oldModel.context_length} → ${newModel.context_length}`,
+    )
+  }
+  if (oldModel.pricing_prompt !== newModel.pricing_prompt) {
+    changes.push(
+      `prompt price: ${oldModel.pricing_prompt} → ${newModel.pricing_prompt}`,
+    )
+  }
+  if (oldModel.pricing_completion !== newModel.pricing_completion) {
+    changes.push(
+      `completion price: ${oldModel.pricing_completion} → ${newModel.pricing_completion}`,
+    )
+  }
+  if (oldModel.modality !== newModel.modality) {
+    changes.push(`modality: "${oldModel.modality}" → "${newModel.modality}"`)
+  }
+
+  return changes
+}
+
+const showUpdated = process.argv.includes('--updated')
+
+const modelsPath = 'scripts/openrouter.models.ts'
+
+// Current branch version
+const currentSource = readFileSync(
+  resolve(import.meta.dirname, '..', modelsPath),
+  'utf-8',
+)
+
+// Main branch version — hardcoded git command, no user input (safe from injection)
+let mainSource: string
+try {
+  mainSource = execSync(`git show main:${modelsPath}`, {
+    encoding: 'utf-8',
+  })
+} catch {
+  console.error('Could not read main branch version. Are you in a git repo?')
+  process.exit(1)
+}
+
+const currentModels = extractModels(currentSource)
+const mainModels = extractModels(mainSource)
+
+const currentIds = [...currentModels.keys()]
+const mainIds = [...mainModels.keys()]
+const currentSet = new Set(currentIds)
+const mainSet = new Set(mainIds)
+
+const added = currentIds.filter((id) => !mainSet.has(id)).sort()
+const removed = mainIds.filter((id) => !currentSet.has(id)).sort()
+
+// Find updated models (present in both, but with changed fields)
+const updated: Array<{ id: string; changes: Array<string> }> = []
+if (showUpdated) {
+  for (const id of currentIds) {
+    if (!mainSet.has(id)) continue
+    const changes = describeChanges(mainModels.get(id)!, currentModels.get(id)!)
+    if (changes.length > 0) {
+      updated.push({ id, changes })
+    }
+  }
+  updated.sort((a, b) => a.id.localeCompare(b.id))
+}
+
+// Output markdown
+const lines: Array<string> = []
+
+lines.push('## OpenRouter Models Update Summary')
+lines.push('')
+lines.push(
+  `**Model count**: ${mainIds.length} (main) → ${currentIds.length} (this branch)`,
+)
+lines.push('')
+
+if (added.length > 0) {
+  lines.push(`### Added models (${added.length})`)
+  lines.push('')
+  for (const id of added) {
+    lines.push(`- \`${id}\``)
+  }
+  lines.push('')
+}
+
+if (removed.length > 0) {
+  lines.push(`### Removed models (${removed.length})`)
+  lines.push('')
+  for (const id of removed) {
+    lines.push(`- \`${id}\``)
+  }
+  lines.push('')
+}
+
+if (updated.length > 0) {
+  lines.push(`### Updated models (${updated.length})`)
+  lines.push('')
+  for (const { id, changes } of updated) {
+    lines.push(`- \`${id}\`: ${changes.join(', ')}`)
+  }
+  lines.push('')
+}
+
+if (added.length === 0 && removed.length === 0 && updated.length === 0) {
+  lines.push('No changes detected.')
+  lines.push('')
+}
+
+console.log(lines.join('\n'))

scripts/fetch-openrouter-models.ts

@@ -5,6 +5,16 @@ const OUTPUT_PATH = new URL('./openrouter.models.ts', import.meta.url)
 
 const ARRAY_START = 'export const models: Array<OpenRouterModel> = ['
 
+const VALID_IDENTIFIER = /^[A-Za-z_$][A-Za-z0-9_$]*$/
+
+function isRecord(v: unknown): v is Record<string, unknown> {
+  return typeof v === 'object' && v !== null && !Array.isArray(v)
+}
+
+function serializeKey(k: string): string {
+  return VALID_IDENTIFIER.test(k) ? k : JSON.stringify(k)
+}
+
 function serializeValue(value: unknown, indent: number): string {
   const pad = '  '.repeat(indent)
   const childPad = '  '.repeat(indent + 1)
@@ -33,11 +43,12 @@ function serializeValue(value: unknown, indent: number): string {
     )
     return `[\n${items.join('\n')}\n${pad}]`
   }
-  if (typeof value === 'object') {
-    const entries = Object.entries(value as Record<string, unknown>)
+  if (isRecord(value)) {
+    const entries = Object.entries(value)
     if (entries.length === 0) return '{}'
     const lines = entries.map(
-      ([k, v]) => `${childPad}${k}: ${serializeValue(v, indent + 1)},`,
+      ([k, v]) =>
+        `${childPad}${serializeKey(k)}: ${serializeValue(v, indent + 1)},`,
     )
     return `{\n${lines.join('\n')}\n${pad}}`
   }