Function processModel

providers/jiekou/scripts/generate-models.ts:185–231

(apiModel: z.infer<typeof JiekouModel>)

}

function processModel(apiModel: z.infer<typeof JiekouModel>): ProcessedModel {
  const features = apiModel.features ?? [];
  const inputModalities = apiModel.input_modalities ?? ["text"];
  const outputModalities = apiModel.output_modalities ?? ["text"];

  // Convert price: divide by 10000 (from 0.0001 USD to USD)
  const inputCost = apiModel.input_token_price_per_m / 10000;
  const outputCost = apiModel.output_token_price_per_m / 10000;

  // Features mapping (ignore "serverless")
  const hasToolCall = features.includes("function-calling");
  const hasStructuredOutput = features.includes("structured-outputs");
  const hasReasoning = features.includes("reasoning");

  // Attachment: true if image/video/audio in input modalities
  const hasAttachment =
    inputModalities.includes("image") ||
    inputModalities.includes("video") ||
    inputModalities.includes("audio");

  const yearMonth = getYearMonth();

  return {
    name: apiModel.id,
    family: inferFamily(apiModel.id),
    release_date: yearMonth,
    last_updated: yearMonth,
    attachment: hasAttachment,
    reasoning: hasReasoning,
    temperature: true,
    tool_call: hasToolCall,
    structured_output: hasStructuredOutput,
    open_weights: isOpenWeights(apiModel.id),
    cost: {
      input: inputCost,
      output: outputCost,
    },
    limit: {
      context: apiModel.context_size,
      output: apiModel.max_output_tokens,
    },
    modalities: {
      input: inputModalities,
      output: outputModalities,
    },
  };
}

function formatToml(model: ProcessedModel): string {
  const lines: string[] = [];