import { RouterBrain } from '@router-brain/sdk';

const rb = new RouterBrain('sk-your-api-key');

// List all models
const all = await rb.models();
console.log(all.data.length, 'models available');

// Search by keyword
const models = await rb.models({ q: 'gpt-4' });
for (const m of models.data) {
  console.log(m.id, m.pricing.prompt, '/', m.pricing.completion);
}

// Filter by modality + paginate
const result = await rb.models({
  outputModalities: ['image'],
  inputModalities: ['text'],
  limit: 5,
  sort: 'newest',
});
console.log(result.has_more ? 'more available' : 'last page');

const detail = await rb.modelEndpoint('gpt-4o');

console.log(detail.name);           // Display name
console.log(detail.pricing.prompt); // Prompt token price

// Per-endpoint details
for (const ep of detail.endpoints) {
  console.log(ep.provider_name);     // Provider
  console.log(ep.latency_30m);       // Avg latency in ms
  console.log(ep.uptime_5m);         // Recent availability
}

const result = await rb.rerank({
  model: 'rerank-model-name',
  query: 'Applications of quantum computing',
  documents: [
    'Quantum computing has important applications in cryptography',
    'Classical computers use binary bits (0 or 1)',
    'Qubits use superposition to represent 0 and 1 simultaneously',
    "Shor's algorithm efficiently factors large integers on quantum computers",
  ],
  top_n: 2,
});

for (const item of result.results) {
  console.log(`#${item.index}  score: ${item.relevance_score}`);
}

// Return original documents alongside scores
const result2 = await rb.rerank({
  model: 'rerank-model-name',
  query: 'RAG retrieval-augmented generation',
  documents: docs,
  return_documents: true,
});