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

// 给函数参数加类型
function handleModelResult(model: ModelItem) {
  console.log(model.id, model.pricing.prompt);
}

// 标注 API 响应类型
const result: ModelsResponse = await rb.models({ q: 'gpt' });

// 标注图片生成结果
const img: ImageResponseJson = await rb.image({
  model: 'dall-e-3',
  prompt: '...',
}).json();

// 访问字段时有自动补全
img.data.forEach((d: ImageData) => {
  console.log(d.url);
});