import {
  RouterBrain,
  RouterBrainConfigs,
  ModelItem,
  ModelsQueryParams,
  ModelsResponse,
  ModelEndpointItem,
  ModelEndpointResponse,
  RerankCreateParams,
  RerankResponse,
  RerankResultItem,
  ImageResponse,
  ImageResponseJson,
  ImageTaskResponseJson,
  ImageResponseUsage,
  ImageAdapterProps,
  ImageAdapterImage,
  ImageData,
} from '@router-brain/sdk';

// Type your function parameters
function handleModelResult(model: ModelItem) {
  console.log(model.id, model.pricing.prompt);
}

// Type API responses
const result: ModelsResponse = await rb.models({ q: 'gpt' });

// Type image generation results
const img: ImageResponseJson = await rb.image({
  model: 'dall-e-3',
  prompt: '...',
}).json();

// Access individual fields with autocomplete
img.data.forEach((d: ImageData) => {
  console.log(d.url);
});