1 | |
2 | type Cohere = { |
3 | apiKey: string; |
4 | }; |
5 | |
6 | * Retrieves metrics measured during the training of a fine-tuned model. |
7 | * |
8 | */ |
9 | export async function main( |
10 | auth: Cohere, |
11 | finetuned_model_id: string, |
12 | page_size: string | undefined, |
13 | page_token: string | undefined, |
14 | ) { |
15 | const url = new URL( |
16 | `https://api.cohere.com/v1/finetuning/finetuned-models/${finetuned_model_id}/training-step-metrics`, |
17 | ); |
18 | for (const [k, v] of [ |
19 | ["page_size", page_size], |
20 | ["page_token", page_token], |
21 | ]) { |
22 | if (v !== undefined && v !== "" && k !== undefined) { |
23 | url.searchParams.append(k, v); |
24 | } |
25 | } |
26 | const response = await fetch(url, { |
27 | method: "GET", |
28 | headers: { |
29 | Authorization: "Bearer " + auth.apiKey, |
30 | }, |
31 | body: undefined, |
32 | }); |
33 | if (!response.ok) { |
34 | const text = await response.text(); |
35 | throw new Error(`${response.status} ${text}`); |
36 | } |
37 | return await response.json(); |
38 | } |
39 |
|