llama-stack-client 0.3.5 → 0.4.0-rc2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +153 -0
- package/README.md +48 -39
- package/error.d.ts.map +1 -1
- package/error.js +1 -0
- package/error.js.map +1 -1
- package/error.mjs.map +1 -1
- package/index.d.mts +33 -54
- package/index.d.ts +33 -54
- package/index.d.ts.map +1 -1
- package/index.js +20 -29
- package/index.js.map +1 -1
- package/index.mjs +18 -28
- package/index.mjs.map +1 -1
- package/lib/response-helpers.d.ts +23 -0
- package/lib/response-helpers.d.ts.map +1 -0
- package/lib/response-helpers.js +53 -0
- package/lib/response-helpers.js.map +1 -0
- package/lib/response-helpers.mjs +49 -0
- package/lib/response-helpers.mjs.map +1 -0
- package/package.json +7 -1
- package/pagination.d.ts.map +1 -1
- package/pagination.js +2 -4
- package/pagination.js.map +1 -1
- package/pagination.mjs +1 -4
- package/pagination.mjs.map +1 -1
- package/resource.d.ts.map +1 -1
- package/resource.js.map +1 -1
- package/resource.mjs.map +1 -1
- package/resources/alpha/admin.d.ts +41 -0
- package/resources/alpha/admin.d.ts.map +1 -0
- package/resources/alpha/admin.js +46 -0
- package/resources/alpha/admin.js.map +1 -0
- package/resources/alpha/admin.mjs +42 -0
- package/resources/alpha/admin.mjs.map +1 -0
- package/resources/alpha/alpha.d.ts +26 -0
- package/resources/alpha/alpha.d.ts.map +1 -0
- package/resources/alpha/alpha.js +61 -0
- package/resources/alpha/alpha.js.map +1 -0
- package/resources/alpha/alpha.mjs +34 -0
- package/resources/alpha/alpha.mjs.map +1 -0
- package/resources/{benchmarks.d.ts → alpha/benchmarks.d.ts} +41 -15
- package/resources/alpha/benchmarks.d.ts.map +1 -0
- package/resources/{benchmarks.js → alpha/benchmarks.js} +18 -4
- package/resources/alpha/benchmarks.js.map +1 -0
- package/resources/alpha/benchmarks.mjs +44 -0
- package/resources/alpha/benchmarks.mjs.map +1 -0
- package/resources/alpha/eval/eval.d.ts +160 -0
- package/resources/alpha/eval/eval.d.ts.map +1 -0
- package/resources/{eval → alpha/eval}/eval.js +7 -6
- package/resources/alpha/eval/eval.js.map +1 -0
- package/resources/{eval → alpha/eval}/eval.mjs +7 -5
- package/resources/alpha/eval/eval.mjs.map +1 -0
- package/resources/alpha/eval/index.d.ts +3 -0
- package/resources/alpha/eval/index.d.ts.map +1 -0
- package/resources/{eval → alpha/eval}/index.js +2 -1
- package/resources/alpha/eval/index.js.map +1 -0
- package/resources/{eval → alpha/eval}/index.mjs +2 -0
- package/resources/alpha/eval/index.mjs.map +1 -0
- package/resources/{eval → alpha/eval}/jobs.d.ts +2 -2
- package/resources/alpha/eval/jobs.d.ts.map +1 -0
- package/resources/{eval → alpha/eval}/jobs.js +6 -5
- package/resources/alpha/eval/jobs.js.map +1 -0
- package/resources/{eval → alpha/eval}/jobs.mjs +6 -4
- package/resources/alpha/eval/jobs.mjs.map +1 -0
- package/resources/alpha/eval.d.ts.map +1 -0
- package/resources/{eval.js → alpha/eval.js} +1 -0
- package/resources/alpha/eval.js.map +1 -0
- package/resources/alpha/eval.mjs.map +1 -0
- package/resources/alpha/index.d.ts +7 -0
- package/resources/alpha/index.d.ts.map +1 -0
- package/resources/alpha/index.js +23 -0
- package/resources/alpha/index.js.map +1 -0
- package/resources/alpha/index.mjs +14 -0
- package/resources/alpha/index.mjs.map +1 -0
- package/resources/alpha/inference.d.ts +85 -0
- package/resources/alpha/inference.d.ts.map +1 -0
- package/resources/{agents/steps.js → alpha/inference.js} +9 -8
- package/resources/alpha/inference.js.map +1 -0
- package/resources/alpha/inference.mjs +17 -0
- package/resources/alpha/inference.mjs.map +1 -0
- package/resources/alpha/post-training/index.d.ts.map +1 -0
- package/resources/{post-training → alpha/post-training}/index.js +2 -1
- package/resources/alpha/post-training/index.js.map +1 -0
- package/resources/{post-training → alpha/post-training}/index.mjs +2 -0
- package/resources/alpha/post-training/index.mjs.map +1 -0
- package/resources/alpha/post-training/job.d.ts +113 -0
- package/resources/alpha/post-training/job.d.ts.map +1 -0
- package/resources/{post-training → alpha/post-training}/job.js +7 -6
- package/resources/alpha/post-training/job.js.map +1 -0
- package/resources/{post-training → alpha/post-training}/job.mjs +7 -5
- package/resources/alpha/post-training/job.mjs.map +1 -0
- package/resources/alpha/post-training/post-training.d.ts +245 -0
- package/resources/alpha/post-training/post-training.d.ts.map +1 -0
- package/resources/{post-training → alpha/post-training}/post-training.js +5 -4
- package/resources/alpha/post-training/post-training.js.map +1 -0
- package/resources/{post-training → alpha/post-training}/post-training.mjs +5 -3
- package/resources/alpha/post-training/post-training.mjs.map +1 -0
- package/resources/alpha/post-training.d.ts.map +1 -0
- package/resources/{post-training.js → alpha/post-training.js} +2 -1
- package/resources/alpha/post-training.js.map +1 -0
- package/resources/{post-training.mjs → alpha/post-training.mjs} +2 -0
- package/resources/alpha/post-training.mjs.map +1 -0
- package/resources/alpha.d.ts +2 -0
- package/resources/alpha.d.ts.map +1 -0
- package/resources/alpha.js +25 -0
- package/resources/alpha.js.map +1 -0
- package/resources/alpha.mjs +9 -0
- package/resources/alpha.mjs.map +1 -0
- package/resources/batches.d.ts +322 -0
- package/resources/batches.d.ts.map +1 -0
- package/resources/batches.js +45 -0
- package/resources/batches.js.map +1 -0
- package/resources/batches.mjs +40 -0
- package/resources/batches.mjs.map +1 -0
- package/resources/beta/beta.d.ts +10 -0
- package/resources/beta/beta.d.ts.map +1 -0
- package/resources/{tool-runtime/tool-runtime.js → beta/beta.js} +10 -22
- package/resources/beta/beta.js.map +1 -0
- package/resources/beta/beta.mjs +18 -0
- package/resources/beta/beta.mjs.map +1 -0
- package/resources/{datasets.d.ts → beta/datasets.d.ts} +117 -99
- package/resources/beta/datasets.d.ts.map +1 -0
- package/resources/{datasets.js → beta/datasets.js} +14 -9
- package/resources/beta/datasets.js.map +1 -0
- package/resources/{datasets.mjs → beta/datasets.mjs} +14 -8
- package/resources/beta/datasets.mjs.map +1 -0
- package/resources/beta/index.d.ts +3 -0
- package/resources/beta/index.d.ts.map +1 -0
- package/resources/beta/index.js +15 -0
- package/resources/beta/index.js.map +1 -0
- package/resources/beta/index.mjs +10 -0
- package/resources/beta/index.mjs.map +1 -0
- package/resources/beta.d.ts +2 -0
- package/resources/beta.d.ts.map +1 -0
- package/resources/{agents.js → beta.js} +4 -3
- package/resources/beta.js.map +1 -0
- package/resources/beta.mjs +9 -0
- package/resources/beta.mjs.map +1 -0
- package/resources/chat/chat.d.ts +75 -80
- package/resources/chat/chat.d.ts.map +1 -1
- package/resources/chat/chat.js +2 -2
- package/resources/chat/chat.js.map +1 -1
- package/resources/chat/chat.mjs +3 -2
- package/resources/chat/chat.mjs.map +1 -1
- package/resources/chat/completions.d.ts +760 -1406
- package/resources/chat/completions.d.ts.map +1 -1
- package/resources/chat/completions.js +8 -14
- package/resources/chat/completions.js.map +1 -1
- package/resources/chat/completions.mjs +7 -11
- package/resources/chat/completions.mjs.map +1 -1
- package/resources/chat/index.d.ts +1 -1
- package/resources/chat/index.d.ts.map +1 -1
- package/resources/chat/index.js +3 -3
- package/resources/chat/index.js.map +1 -1
- package/resources/chat/index.mjs +3 -1
- package/resources/chat/index.mjs.map +1 -1
- package/resources/chat.js +2 -1
- package/resources/chat.js.map +1 -1
- package/resources/chat.mjs +2 -0
- package/resources/chat.mjs.map +1 -1
- package/resources/completions.d.ts +54 -97
- package/resources/completions.d.ts.map +1 -1
- package/resources/completions.js +3 -6
- package/resources/completions.js.map +1 -1
- package/resources/completions.mjs +3 -5
- package/resources/completions.mjs.map +1 -1
- package/resources/conversations/conversations.d.ts +320 -0
- package/resources/conversations/conversations.d.ts.map +1 -0
- package/resources/conversations/conversations.js +78 -0
- package/resources/conversations/conversations.js.map +1 -0
- package/resources/conversations/conversations.mjs +51 -0
- package/resources/conversations/conversations.mjs.map +1 -0
- package/resources/conversations/index.d.ts +3 -0
- package/resources/conversations/index.d.ts.map +1 -0
- package/resources/conversations/index.js +16 -0
- package/resources/conversations/index.js.map +1 -0
- package/resources/conversations/index.mjs +10 -0
- package/resources/conversations/index.mjs.map +1 -0
- package/resources/conversations/items.d.ts +874 -0
- package/resources/conversations/items.d.ts.map +1 -0
- package/resources/conversations/items.js +51 -0
- package/resources/conversations/items.js.map +1 -0
- package/resources/conversations/items.mjs +46 -0
- package/resources/conversations/items.mjs.map +1 -0
- package/resources/conversations.d.ts +2 -0
- package/resources/conversations.d.ts.map +1 -0
- package/resources/conversations.js +25 -0
- package/resources/conversations.js.map +1 -0
- package/resources/conversations.mjs +9 -0
- package/resources/conversations.mjs.map +1 -0
- package/resources/embeddings.d.ts +10 -52
- package/resources/embeddings.d.ts.map +1 -1
- package/resources/embeddings.js +5 -2
- package/resources/embeddings.js.map +1 -1
- package/resources/embeddings.mjs +5 -1
- package/resources/embeddings.mjs.map +1 -1
- package/resources/files.d.ts +44 -54
- package/resources/files.d.ts.map +1 -1
- package/resources/files.js +18 -9
- package/resources/files.js.map +1 -1
- package/resources/files.mjs +18 -8
- package/resources/files.mjs.map +1 -1
- package/resources/index.d.ts +11 -15
- package/resources/index.d.ts.map +1 -1
- package/resources/index.js +15 -21
- package/resources/index.js.map +1 -1
- package/resources/index.mjs +9 -11
- package/resources/index.mjs.map +1 -1
- package/resources/inspect.d.ts +3 -71
- package/resources/inspect.d.ts.map +1 -1
- package/resources/inspect.js +2 -1
- package/resources/inspect.js.map +1 -1
- package/resources/inspect.mjs +2 -0
- package/resources/inspect.mjs.map +1 -1
- package/resources/models/index.d.ts +2 -2
- package/resources/models/index.d.ts.map +1 -1
- package/resources/models/index.js +2 -1
- package/resources/models/index.js.map +1 -1
- package/resources/models/index.mjs +2 -0
- package/resources/models/index.mjs.map +1 -1
- package/resources/models/models.d.ts +71 -31
- package/resources/models/models.d.ts.map +1 -1
- package/resources/models/models.js +13 -2
- package/resources/models/models.js.map +1 -1
- package/resources/models/models.mjs +13 -1
- package/resources/models/models.mjs.map +1 -1
- package/resources/models/openai.d.ts +2 -16
- package/resources/models/openai.d.ts.map +1 -1
- package/resources/models/openai.js +3 -2
- package/resources/models/openai.js.map +1 -1
- package/resources/models/openai.mjs +3 -1
- package/resources/models/openai.mjs.map +1 -1
- package/resources/models.js +2 -1
- package/resources/models.js.map +1 -1
- package/resources/models.mjs +2 -0
- package/resources/models.mjs.map +1 -1
- package/resources/moderations.d.ts +9 -35
- package/resources/moderations.d.ts.map +1 -1
- package/resources/moderations.js +5 -2
- package/resources/moderations.js.map +1 -1
- package/resources/moderations.mjs +5 -1
- package/resources/moderations.mjs.map +1 -1
- package/resources/prompts/index.d.ts +3 -0
- package/resources/prompts/index.d.ts.map +1 -0
- package/resources/prompts/index.js +15 -0
- package/resources/prompts/index.js.map +1 -0
- package/resources/prompts/index.mjs +10 -0
- package/resources/prompts/index.mjs.map +1 -0
- package/resources/prompts/prompts.d.ts +96 -0
- package/resources/prompts/prompts.d.ts.map +1 -0
- package/resources/{agents/agents.js → prompts/prompts.js} +43 -30
- package/resources/prompts/prompts.js.map +1 -0
- package/resources/prompts/prompts.mjs +66 -0
- package/resources/prompts/prompts.mjs.map +1 -0
- package/resources/prompts/versions.d.ts +12 -0
- package/resources/prompts/versions.d.ts.map +1 -0
- package/resources/prompts/versions.js +23 -0
- package/resources/prompts/versions.js.map +1 -0
- package/resources/prompts/versions.mjs +19 -0
- package/resources/prompts/versions.mjs.map +1 -0
- package/resources/prompts.d.ts +2 -0
- package/resources/prompts.d.ts.map +1 -0
- package/resources/prompts.js +25 -0
- package/resources/prompts.js.map +1 -0
- package/resources/prompts.mjs +9 -0
- package/resources/prompts.mjs.map +1 -0
- package/resources/providers.d.ts +4 -13
- package/resources/providers.d.ts.map +1 -1
- package/resources/providers.js +2 -1
- package/resources/providers.js.map +1 -1
- package/resources/providers.mjs +2 -0
- package/resources/providers.mjs.map +1 -1
- package/resources/responses/index.js +2 -1
- package/resources/responses/index.js.map +1 -1
- package/resources/responses/index.mjs +2 -0
- package/resources/responses/index.mjs.map +1 -1
- package/resources/responses/input-items.d.ts +246 -151
- package/resources/responses/input-items.d.ts.map +1 -1
- package/resources/responses/input-items.js +3 -2
- package/resources/responses/input-items.js.map +1 -1
- package/resources/responses/input-items.mjs +3 -1
- package/resources/responses/input-items.mjs.map +1 -1
- package/resources/responses/responses.d.ts +1841 -1455
- package/resources/responses/responses.d.ts.map +1 -1
- package/resources/responses/responses.js +8 -11
- package/resources/responses/responses.js.map +1 -1
- package/resources/responses/responses.mjs +8 -10
- package/resources/responses/responses.mjs.map +1 -1
- package/resources/responses.js +2 -1
- package/resources/responses.js.map +1 -1
- package/resources/responses.mjs +2 -0
- package/resources/responses.mjs.map +1 -1
- package/resources/routes.d.ts +11 -10
- package/resources/routes.d.ts.map +1 -1
- package/resources/routes.js +8 -6
- package/resources/routes.js.map +1 -1
- package/resources/routes.mjs +8 -5
- package/resources/routes.mjs.map +1 -1
- package/resources/safety.d.ts +144 -10
- package/resources/safety.d.ts.map +1 -1
- package/resources/safety.js +4 -1
- package/resources/safety.js.map +1 -1
- package/resources/safety.mjs +4 -0
- package/resources/safety.mjs.map +1 -1
- package/resources/scoring-functions.d.ts +120 -50
- package/resources/scoring-functions.d.ts.map +1 -1
- package/resources/scoring-functions.js +15 -1
- package/resources/scoring-functions.js.map +1 -1
- package/resources/scoring-functions.mjs +15 -0
- package/resources/scoring-functions.mjs.map +1 -1
- package/resources/scoring.d.ts +85 -26
- package/resources/scoring.d.ts.map +1 -1
- package/resources/scoring.js +2 -1
- package/resources/scoring.js.map +1 -1
- package/resources/scoring.mjs +2 -0
- package/resources/scoring.mjs.map +1 -1
- package/resources/shared.d.ts +173 -591
- package/resources/shared.d.ts.map +1 -1
- package/resources/shared.js +2 -0
- package/resources/shared.js.map +1 -1
- package/resources/shared.mjs +2 -0
- package/resources/shared.mjs.map +1 -1
- package/resources/shields.d.ts +19 -24
- package/resources/shields.d.ts.map +1 -1
- package/resources/shields.js +6 -1
- package/resources/shields.js.map +1 -1
- package/resources/shields.mjs +6 -0
- package/resources/shields.mjs.map +1 -1
- package/resources/tool-runtime.d.ts +151 -1
- package/resources/tool-runtime.d.ts.map +1 -1
- package/resources/tool-runtime.js +22 -16
- package/resources/tool-runtime.js.map +1 -1
- package/resources/tool-runtime.mjs +20 -1
- package/resources/tool-runtime.mjs.map +1 -1
- package/resources/toolgroups.d.ts +27 -37
- package/resources/toolgroups.d.ts.map +1 -1
- package/resources/toolgroups.js +10 -1
- package/resources/toolgroups.js.map +1 -1
- package/resources/toolgroups.mjs +10 -0
- package/resources/toolgroups.mjs.map +1 -1
- package/resources/tools.d.ts +9 -76
- package/resources/tools.d.ts.map +1 -1
- package/resources/tools.js +4 -1
- package/resources/tools.js.map +1 -1
- package/resources/tools.mjs +4 -0
- package/resources/tools.mjs.map +1 -1
- package/resources/vector-io.d.ts +272 -148
- package/resources/vector-io.d.ts.map +1 -1
- package/resources/vector-io.js +2 -1
- package/resources/vector-io.js.map +1 -1
- package/resources/vector-io.mjs +2 -0
- package/resources/vector-io.mjs.map +1 -1
- package/resources/vector-stores/file-batches.d.ts +111 -0
- package/resources/vector-stores/file-batches.d.ts.map +1 -0
- package/resources/vector-stores/file-batches.js +45 -0
- package/resources/vector-stores/file-batches.js.map +1 -0
- package/resources/vector-stores/file-batches.mjs +41 -0
- package/resources/vector-stores/file-batches.mjs.map +1 -0
- package/resources/vector-stores/files.d.ts +83 -135
- package/resources/vector-stores/files.d.ts.map +1 -1
- package/resources/vector-stores/files.js +15 -14
- package/resources/vector-stores/files.js.map +1 -1
- package/resources/vector-stores/files.mjs +15 -13
- package/resources/vector-stores/files.mjs.map +1 -1
- package/resources/vector-stores/index.d.ts +2 -1
- package/resources/vector-stores/index.d.ts.map +1 -1
- package/resources/vector-stores/index.js +5 -2
- package/resources/vector-stores/index.js.map +1 -1
- package/resources/vector-stores/index.mjs +3 -0
- package/resources/vector-stores/index.mjs.map +1 -1
- package/resources/vector-stores/vector-stores.d.ts +118 -217
- package/resources/vector-stores/vector-stores.d.ts.map +1 -1
- package/resources/vector-stores/vector-stores.js +18 -12
- package/resources/vector-stores/vector-stores.js.map +1 -1
- package/resources/vector-stores/vector-stores.mjs +18 -11
- package/resources/vector-stores/vector-stores.mjs.map +1 -1
- package/resources/vector-stores.js +2 -1
- package/resources/vector-stores.js.map +1 -1
- package/resources/vector-stores.mjs +2 -0
- package/resources/vector-stores.mjs.map +1 -1
- package/src/error.ts +1 -0
- package/src/index.ts +92 -267
- package/src/lib/response-helpers.ts +60 -0
- package/src/pagination.ts +2 -5
- package/src/resource.ts +1 -0
- package/src/resources/alpha/admin.ts +84 -0
- package/src/resources/alpha/alpha.ts +93 -0
- package/src/resources/{benchmarks.ts → alpha/benchmarks.ts} +51 -16
- package/src/resources/{eval → alpha/eval}/eval.ts +83 -65
- package/src/resources/{eval → alpha/eval}/index.ts +1 -2
- package/src/resources/{eval → alpha/eval}/jobs.ts +6 -6
- package/src/resources/{eval.ts → alpha/eval.ts} +1 -0
- package/src/resources/alpha/index.ts +36 -0
- package/src/resources/alpha/inference.ts +132 -0
- package/src/resources/{post-training → alpha/post-training}/index.ts +1 -1
- package/src/resources/alpha/post-training/job.ts +177 -0
- package/src/resources/alpha/post-training/post-training.ts +372 -0
- package/src/resources/{post-training.ts → alpha/post-training.ts} +1 -1
- package/src/resources/alpha.ts +9 -0
- package/src/resources/batches.ts +572 -0
- package/src/resources/beta/beta.ts +41 -0
- package/src/resources/{datasets.ts → beta/datasets.ts} +128 -110
- package/src/resources/beta/index.ts +20 -0
- package/src/resources/{agents.ts → beta.ts} +2 -2
- package/src/resources/chat/chat.ts +84 -83
- package/src/resources/chat/completions.ts +876 -1486
- package/src/resources/chat/index.ts +1 -2
- package/src/resources/chat.ts +1 -1
- package/src/resources/completions.ts +59 -103
- package/src/resources/conversations/conversations.ts +491 -0
- package/src/resources/conversations/index.ts +25 -0
- package/src/resources/conversations/items.ts +1365 -0
- package/src/resources/conversations.ts +9 -0
- package/src/resources/embeddings.ts +14 -55
- package/src/resources/files.ts +54 -61
- package/src/resources/index.ts +35 -111
- package/src/resources/inspect.ts +4 -84
- package/src/resources/models/index.ts +4 -2
- package/src/resources/models/models.ts +79 -30
- package/src/resources/models/openai.ts +4 -24
- package/src/resources/models.ts +1 -1
- package/src/resources/moderations.ts +10 -36
- package/src/resources/prompts/index.ts +19 -0
- package/src/resources/prompts/prompts.ts +171 -0
- package/src/resources/prompts/versions.ts +26 -0
- package/src/resources/prompts.ts +9 -0
- package/src/resources/providers.ts +5 -18
- package/src/resources/responses/index.ts +1 -1
- package/src/resources/responses/input-items.ts +365 -161
- package/src/resources/responses/responses.ts +2959 -1870
- package/src/resources/responses.ts +1 -1
- package/src/resources/routes.ts +25 -14
- package/src/resources/safety.ts +199 -9
- package/src/resources/scoring-functions.ts +168 -52
- package/src/resources/scoring.ts +125 -22
- package/src/resources/shared.ts +187 -665
- package/src/resources/shields.ts +18 -23
- package/src/resources/tool-runtime.ts +210 -2
- package/src/resources/toolgroups.ts +26 -36
- package/src/resources/tools.ts +10 -94
- package/src/resources/vector-io.ts +323 -145
- package/src/resources/vector-stores/file-batches.ts +201 -0
- package/src/resources/vector-stores/files.ts +123 -143
- package/src/resources/vector-stores/index.ts +9 -1
- package/src/resources/vector-stores/vector-stores.ts +150 -216
- package/src/resources/vector-stores.ts +1 -1
- package/src/version.ts +1 -1
- package/version.d.ts +1 -1
- package/version.d.ts.map +1 -1
- package/version.js +1 -1
- package/version.js.map +1 -1
- package/version.mjs +1 -1
- package/version.mjs.map +1 -1
- package/resources/agents/agents.d.ts +0 -246
- package/resources/agents/agents.d.ts.map +0 -1
- package/resources/agents/agents.js.map +0 -1
- package/resources/agents/agents.mjs +0 -52
- package/resources/agents/agents.mjs.map +0 -1
- package/resources/agents/index.d.ts +0 -5
- package/resources/agents/index.d.ts.map +0 -1
- package/resources/agents/index.js +0 -18
- package/resources/agents/index.js.map +0 -1
- package/resources/agents/index.mjs +0 -10
- package/resources/agents/index.mjs.map +0 -1
- package/resources/agents/session.d.ts +0 -98
- package/resources/agents/session.d.ts.map +0 -1
- package/resources/agents/session.js +0 -42
- package/resources/agents/session.js.map +0 -1
- package/resources/agents/session.mjs +0 -37
- package/resources/agents/session.mjs.map +0 -1
- package/resources/agents/steps.d.ts +0 -22
- package/resources/agents/steps.d.ts.map +0 -1
- package/resources/agents/steps.js.map +0 -1
- package/resources/agents/steps.mjs +0 -15
- package/resources/agents/steps.mjs.map +0 -1
- package/resources/agents/turn.d.ts +0 -461
- package/resources/agents/turn.d.ts.map +0 -1
- package/resources/agents/turn.js +0 -34
- package/resources/agents/turn.js.map +0 -1
- package/resources/agents/turn.mjs +0 -29
- package/resources/agents/turn.mjs.map +0 -1
- package/resources/agents.d.ts +0 -2
- package/resources/agents.d.ts.map +0 -1
- package/resources/agents.js.map +0 -1
- package/resources/agents.mjs +0 -7
- package/resources/agents.mjs.map +0 -1
- package/resources/benchmarks.d.ts.map +0 -1
- package/resources/benchmarks.js.map +0 -1
- package/resources/benchmarks.mjs +0 -31
- package/resources/benchmarks.mjs.map +0 -1
- package/resources/datasets.d.ts.map +0 -1
- package/resources/datasets.js.map +0 -1
- package/resources/datasets.mjs.map +0 -1
- package/resources/eval/eval.d.ts +0 -159
- package/resources/eval/eval.d.ts.map +0 -1
- package/resources/eval/eval.js.map +0 -1
- package/resources/eval/eval.mjs.map +0 -1
- package/resources/eval/index.d.ts +0 -3
- package/resources/eval/index.d.ts.map +0 -1
- package/resources/eval/index.js.map +0 -1
- package/resources/eval/index.mjs.map +0 -1
- package/resources/eval/jobs.d.ts.map +0 -1
- package/resources/eval/jobs.js.map +0 -1
- package/resources/eval/jobs.mjs.map +0 -1
- package/resources/eval.d.ts.map +0 -1
- package/resources/eval.js.map +0 -1
- package/resources/eval.mjs.map +0 -1
- package/resources/inference.d.ts +0 -580
- package/resources/inference.d.ts.map +0 -1
- package/resources/inference.js +0 -54
- package/resources/inference.js.map +0 -1
- package/resources/inference.mjs +0 -49
- package/resources/inference.mjs.map +0 -1
- package/resources/post-training/index.d.ts.map +0 -1
- package/resources/post-training/index.js.map +0 -1
- package/resources/post-training/index.mjs.map +0 -1
- package/resources/post-training/job.d.ts +0 -205
- package/resources/post-training/job.d.ts.map +0 -1
- package/resources/post-training/job.js.map +0 -1
- package/resources/post-training/job.mjs.map +0 -1
- package/resources/post-training/post-training.d.ts +0 -402
- package/resources/post-training/post-training.d.ts.map +0 -1
- package/resources/post-training/post-training.js.map +0 -1
- package/resources/post-training/post-training.mjs.map +0 -1
- package/resources/post-training.d.ts.map +0 -1
- package/resources/post-training.js.map +0 -1
- package/resources/post-training.mjs.map +0 -1
- package/resources/synthetic-data-generation.d.ts +0 -47
- package/resources/synthetic-data-generation.d.ts.map +0 -1
- package/resources/synthetic-data-generation.js +0 -20
- package/resources/synthetic-data-generation.js.map +0 -1
- package/resources/synthetic-data-generation.mjs +0 -15
- package/resources/synthetic-data-generation.mjs.map +0 -1
- package/resources/telemetry.d.ts +0 -519
- package/resources/telemetry.d.ts.map +0 -1
- package/resources/telemetry.js +0 -70
- package/resources/telemetry.js.map +0 -1
- package/resources/telemetry.mjs +0 -65
- package/resources/telemetry.mjs.map +0 -1
- package/resources/tool-runtime/index.d.ts +0 -3
- package/resources/tool-runtime/index.d.ts.map +0 -1
- package/resources/tool-runtime/index.js +0 -14
- package/resources/tool-runtime/index.js.map +0 -1
- package/resources/tool-runtime/index.mjs +0 -8
- package/resources/tool-runtime/index.mjs.map +0 -1
- package/resources/tool-runtime/rag-tool.d.ts +0 -45
- package/resources/tool-runtime/rag-tool.d.ts.map +0 -1
- package/resources/tool-runtime/rag-tool.js +0 -30
- package/resources/tool-runtime/rag-tool.js.map +0 -1
- package/resources/tool-runtime/rag-tool.mjs +0 -25
- package/resources/tool-runtime/rag-tool.mjs.map +0 -1
- package/resources/tool-runtime/tool-runtime.d.ts +0 -132
- package/resources/tool-runtime/tool-runtime.d.ts.map +0 -1
- package/resources/tool-runtime/tool-runtime.js.map +0 -1
- package/resources/tool-runtime/tool-runtime.mjs +0 -29
- package/resources/tool-runtime/tool-runtime.mjs.map +0 -1
- package/resources/vector-dbs.d.ts +0 -128
- package/resources/vector-dbs.d.ts.map +0 -1
- package/resources/vector-dbs.js +0 -41
- package/resources/vector-dbs.js.map +0 -1
- package/resources/vector-dbs.mjs +0 -36
- package/resources/vector-dbs.mjs.map +0 -1
- package/src/resources/agents/agents.ts +0 -372
- package/src/resources/agents/index.ts +0 -44
- package/src/resources/agents/session.ts +0 -169
- package/src/resources/agents/steps.ts +0 -47
- package/src/resources/agents/turn.ts +0 -638
- package/src/resources/inference.ts +0 -768
- package/src/resources/post-training/job.ts +0 -274
- package/src/resources/post-training/post-training.ts +0 -516
- package/src/resources/synthetic-data-generation.ts +0 -65
- package/src/resources/telemetry.ts +0 -692
- package/src/resources/tool-runtime/index.ts +0 -17
- package/src/resources/tool-runtime/rag-tool.ts +0 -69
- package/src/resources/tool-runtime/tool-runtime.ts +0 -191
- package/src/resources/vector-dbs.ts +0 -191
- /package/resources/{eval.d.ts → alpha/eval.d.ts} +0 -0
- /package/resources/{eval.mjs → alpha/eval.mjs} +0 -0
- /package/resources/{post-training → alpha/post-training}/index.d.ts +0 -0
- /package/resources/{post-training.d.ts → alpha/post-training.d.ts} +0 -0
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
import { APIResource } from "../../resource.js";
|
|
2
|
+
import * as Core from "../../core.js";
|
|
3
|
+
export declare class Inference extends APIResource {
|
|
4
|
+
/**
|
|
5
|
+
* Rerank a list of documents based on their relevance to a query.
|
|
6
|
+
*/
|
|
7
|
+
rerank(body: InferenceRerankParams, options?: Core.RequestOptions): Core.APIPromise<InferenceRerankResponse>;
|
|
8
|
+
}
|
|
9
|
+
export type InferenceRerankResponse = Array<InferenceRerankResponse.InferenceRerankResponseItem>;
|
|
10
|
+
export declare namespace InferenceRerankResponse {
|
|
11
|
+
/**
|
|
12
|
+
* A single rerank result from a reranking response.
|
|
13
|
+
*/
|
|
14
|
+
interface InferenceRerankResponseItem {
|
|
15
|
+
index: number;
|
|
16
|
+
relevance_score: number;
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
export interface InferenceRerankParams {
|
|
20
|
+
items: Array<string | InferenceRerankParams.OpenAIChatCompletionContentPartTextParam | InferenceRerankParams.OpenAIChatCompletionContentPartImageParam>;
|
|
21
|
+
model: string;
|
|
22
|
+
/**
|
|
23
|
+
* Text content part for OpenAI-compatible chat completion messages.
|
|
24
|
+
*/
|
|
25
|
+
query: string | InferenceRerankParams.OpenAIChatCompletionContentPartTextParam | InferenceRerankParams.OpenAIChatCompletionContentPartImageParam;
|
|
26
|
+
max_num_results?: number | null;
|
|
27
|
+
}
|
|
28
|
+
export declare namespace InferenceRerankParams {
|
|
29
|
+
/**
|
|
30
|
+
* Text content part for OpenAI-compatible chat completion messages.
|
|
31
|
+
*/
|
|
32
|
+
interface OpenAIChatCompletionContentPartTextParam {
|
|
33
|
+
text: string;
|
|
34
|
+
type?: 'text';
|
|
35
|
+
}
|
|
36
|
+
/**
|
|
37
|
+
* Image content part for OpenAI-compatible chat completion messages.
|
|
38
|
+
*/
|
|
39
|
+
interface OpenAIChatCompletionContentPartImageParam {
|
|
40
|
+
/**
|
|
41
|
+
* Image URL specification for OpenAI-compatible chat completion messages.
|
|
42
|
+
*/
|
|
43
|
+
image_url: OpenAIChatCompletionContentPartImageParam.ImageURL;
|
|
44
|
+
type?: 'image_url';
|
|
45
|
+
}
|
|
46
|
+
namespace OpenAIChatCompletionContentPartImageParam {
|
|
47
|
+
/**
|
|
48
|
+
* Image URL specification for OpenAI-compatible chat completion messages.
|
|
49
|
+
*/
|
|
50
|
+
interface ImageURL {
|
|
51
|
+
url: string;
|
|
52
|
+
detail?: string | null;
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
/**
|
|
56
|
+
* Text content part for OpenAI-compatible chat completion messages.
|
|
57
|
+
*/
|
|
58
|
+
interface OpenAIChatCompletionContentPartTextParam {
|
|
59
|
+
text: string;
|
|
60
|
+
type?: 'text';
|
|
61
|
+
}
|
|
62
|
+
/**
|
|
63
|
+
* Image content part for OpenAI-compatible chat completion messages.
|
|
64
|
+
*/
|
|
65
|
+
interface OpenAIChatCompletionContentPartImageParam {
|
|
66
|
+
/**
|
|
67
|
+
* Image URL specification for OpenAI-compatible chat completion messages.
|
|
68
|
+
*/
|
|
69
|
+
image_url: OpenAIChatCompletionContentPartImageParam.ImageURL;
|
|
70
|
+
type?: 'image_url';
|
|
71
|
+
}
|
|
72
|
+
namespace OpenAIChatCompletionContentPartImageParam {
|
|
73
|
+
/**
|
|
74
|
+
* Image URL specification for OpenAI-compatible chat completion messages.
|
|
75
|
+
*/
|
|
76
|
+
interface ImageURL {
|
|
77
|
+
url: string;
|
|
78
|
+
detail?: string | null;
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
export declare namespace Inference {
|
|
83
|
+
export { type InferenceRerankResponse as InferenceRerankResponse, type InferenceRerankParams as InferenceRerankParams, };
|
|
84
|
+
}
|
|
85
|
+
//# sourceMappingURL=inference.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../src/resources/alpha/inference.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,IAAI,MAAM,YAAY,CAAC;AAEnC,qBAAa,SAAU,SAAQ,WAAW;IACxC;;OAEG;IACH,MAAM,CACJ,IAAI,EAAE,qBAAqB,EAC3B,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,uBAAuB,CAAC;CAO5C;AAED,MAAM,MAAM,uBAAuB,GAAG,KAAK,CAAC,uBAAuB,CAAC,2BAA2B,CAAC,CAAC;AAEjG,yBAAiB,uBAAuB,CAAC;IACvC;;OAEG;IACH,UAAiB,2BAA2B;QAC1C,KAAK,EAAE,MAAM,CAAC;QAEd,eAAe,EAAE,MAAM,CAAC;KACzB;CACF;AAED,MAAM,WAAW,qBAAqB;IACpC,KAAK,EAAE,KAAK,CACR,MAAM,GACN,qBAAqB,CAAC,wCAAwC,GAC9D,qBAAqB,CAAC,yCAAyC,CAClE,CAAC;IAEF,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EACD,MAAM,GACN,qBAAqB,CAAC,wCAAwC,GAC9D,qBAAqB,CAAC,yCAAyC,CAAC;IAEpE,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED,yBAAiB,qBAAqB,CAAC;IACrC;;OAEG;IACH,UAAiB,wCAAwC;QACvD,IAAI,EAAE,MAAM,CAAC;QAEb,IAAI,CAAC,EAAE,MAAM,CAAC;KACf;IAED;;OAEG;IACH,UAAiB,yCAAyC;QACxD;;WAEG;QACH,SAAS,EAAE,yCAAyC,CAAC,QAAQ,CAAC;QAE9D,IAAI,CAAC,EAAE,WAAW,CAAC;KACpB;IAED,UAAiB,yCAAyC,CAAC;QACzD;;WAEG;QACH,UAAiB,QAAQ;YACvB,GAAG,EAAE,MAAM,CAAC;YAEZ,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;SACxB;KACF;IAED;;OAEG;IACH,UAAiB,wCAAwC;QACvD,IAAI,EAAE,MAAM,CAAC;QAEb,IAAI,CAAC,EAAE,MAAM,CAAC;KACf;IAED;;OAEG;IACH,UAAiB,yCAAyC;QACxD;;WAEG;QACH,SAAS,EAAE,yCAAyC,CAAC,QAAQ,CAAC;QAE9D,IAAI,CAAC,EAAE,WAAW,CAAC;KACpB;IAED,UAAiB,yCAAyC,CAAC;QACzD;;WAEG;QACH,UAAiB,QAAQ;YACvB,GAAG,EAAE,MAAM,CAAC;YAEZ,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;SACxB;KACF;CACF;AAED,MAAM,CAAC,OAAO,WAAW,SAAS,CAAC;IACjC,OAAO,EACL,KAAK,uBAAuB,IAAI,uBAAuB,EACvD,KAAK,qBAAqB,IAAI,qBAAqB,GACpD,CAAC;CACH"}
|
|
@@ -4,17 +4,18 @@
|
|
|
4
4
|
//
|
|
5
5
|
// This source code is licensed under the terms described in the LICENSE file in
|
|
6
6
|
// the root directory of this source tree.
|
|
7
|
-
|
|
8
|
-
exports.Steps = void 0;
|
|
7
|
+
//
|
|
9
8
|
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
9
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
10
|
+
exports.Inference = void 0;
|
|
10
11
|
const resource_1 = require("../../resource.js");
|
|
11
|
-
class
|
|
12
|
+
class Inference extends resource_1.APIResource {
|
|
12
13
|
/**
|
|
13
|
-
*
|
|
14
|
+
* Rerank a list of documents based on their relevance to a query.
|
|
14
15
|
*/
|
|
15
|
-
|
|
16
|
-
return this._client.
|
|
16
|
+
rerank(body, options) {
|
|
17
|
+
return this._client.post('/v1alpha/inference/rerank', { body, ...options })._thenUnwrap((obj) => obj.data);
|
|
17
18
|
}
|
|
18
19
|
}
|
|
19
|
-
exports.
|
|
20
|
-
//# sourceMappingURL=
|
|
20
|
+
exports.Inference = Inference;
|
|
21
|
+
//# sourceMappingURL=inference.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"inference.js","sourceRoot":"","sources":["../../src/resources/alpha/inference.ts"],"names":[],"mappings":";AAAA,qDAAqD;AACrD,uBAAuB;AACvB,EAAE;AACF,gFAAgF;AAChF,0CAA0C;AAC1C,EAAE;AACF,sFAAsF;;;AAEtF,gDAA6C;AAG7C,MAAa,SAAU,SAAQ,sBAAW;IACxC;;OAEG;IACH,MAAM,CACJ,IAA2B,EAC3B,OAA6B;QAE7B,OACE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAGpE,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;CACF;AAdD,8BAcC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
// Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
2
|
+
// All rights reserved.
|
|
3
|
+
//
|
|
4
|
+
// This source code is licensed under the terms described in the LICENSE file in
|
|
5
|
+
// the root directory of this source tree.
|
|
6
|
+
//
|
|
7
|
+
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
8
|
+
import { APIResource } from "../../resource.mjs";
|
|
9
|
+
export class Inference extends APIResource {
|
|
10
|
+
/**
|
|
11
|
+
* Rerank a list of documents based on their relevance to a query.
|
|
12
|
+
*/
|
|
13
|
+
rerank(body, options) {
|
|
14
|
+
return this._client.post('/v1alpha/inference/rerank', { body, ...options })._thenUnwrap((obj) => obj.data);
|
|
15
|
+
}
|
|
16
|
+
}
|
|
17
|
+
//# sourceMappingURL=inference.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"inference.mjs","sourceRoot":"","sources":["../../src/resources/alpha/inference.ts"],"names":[],"mappings":"AAAA,qDAAqD;AACrD,uBAAuB;AACvB,EAAE;AACF,gFAAgF;AAChF,0CAA0C;AAC1C,EAAE;AACF,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAGtB,MAAM,OAAO,SAAU,SAAQ,WAAW;IACxC;;OAEG;IACH,MAAM,CACJ,IAA2B,EAC3B,OAA6B;QAE7B,OACE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAGpE,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/resources/alpha/post-training/index.ts"],"names":[],"mappings":"AAQA,OAAO,EACL,GAAG,EACH,KAAK,eAAe,EACpB,KAAK,oBAAoB,EACzB,KAAK,iBAAiB,EACtB,KAAK,kBAAkB,EACvB,KAAK,eAAe,EACpB,KAAK,eAAe,GACrB,MAAM,OAAO,CAAC;AACf,OAAO,EACL,YAAY,EACZ,KAAK,eAAe,EACpB,KAAK,4BAA4B,EACjC,KAAK,eAAe,EACpB,KAAK,oCAAoC,EACzC,KAAK,oCAAoC,GAC1C,MAAM,iBAAiB,CAAC"}
|
|
@@ -4,9 +4,10 @@
|
|
|
4
4
|
//
|
|
5
5
|
// This source code is licensed under the terms described in the LICENSE file in
|
|
6
6
|
// the root directory of this source tree.
|
|
7
|
+
//
|
|
8
|
+
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
7
9
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
8
10
|
exports.PostTraining = exports.Job = void 0;
|
|
9
|
-
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
10
11
|
var job_1 = require("./job.js");
|
|
11
12
|
Object.defineProperty(exports, "Job", { enumerable: true, get: function () { return job_1.Job; } });
|
|
12
13
|
var post_training_1 = require("./post-training.js");
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/resources/alpha/post-training/index.ts"],"names":[],"mappings":";AAAA,qDAAqD;AACrD,uBAAuB;AACvB,EAAE;AACF,gFAAgF;AAChF,0CAA0C;AAC1C,EAAE;AACF,sFAAsF;;;AAEtF,gCAQe;AAPb,0FAAA,GAAG,OAAA;AAQL,oDAOyB;AANvB,6GAAA,YAAY,OAAA"}
|
|
@@ -3,6 +3,8 @@
|
|
|
3
3
|
//
|
|
4
4
|
// This source code is licensed under the terms described in the LICENSE file in
|
|
5
5
|
// the root directory of this source tree.
|
|
6
|
+
//
|
|
7
|
+
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
6
8
|
export { Job, } from "./job.mjs";
|
|
7
9
|
export { PostTraining, } from "./post-training.mjs";
|
|
8
10
|
//# sourceMappingURL=index.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../../src/resources/alpha/post-training/index.ts"],"names":[],"mappings":"AAAA,qDAAqD;AACrD,uBAAuB;AACvB,EAAE;AACF,gFAAgF;AAChF,0CAA0C;AAC1C,EAAE;AACF,sFAAsF;OAE/E,EACL,GAAG,GAOJ;OACM,EACL,YAAY,GAMb"}
|
|
@@ -0,0 +1,113 @@
|
|
|
1
|
+
import { APIResource } from "../../../resource.js";
|
|
2
|
+
import * as Core from "../../../core.js";
|
|
3
|
+
import * as PostTrainingAPI from "./post-training.js";
|
|
4
|
+
export declare class Job extends APIResource {
|
|
5
|
+
/**
|
|
6
|
+
* Get all training jobs.
|
|
7
|
+
*/
|
|
8
|
+
list(options?: Core.RequestOptions): Core.APIPromise<JobListResponse>;
|
|
9
|
+
/**
|
|
10
|
+
* Get the artifacts of a training job.
|
|
11
|
+
*/
|
|
12
|
+
artifacts(query: JobArtifactsParams, options?: Core.RequestOptions): Core.APIPromise<JobArtifactsResponse>;
|
|
13
|
+
/**
|
|
14
|
+
* Cancel a training job.
|
|
15
|
+
*/
|
|
16
|
+
cancel(body: JobCancelParams, options?: Core.RequestOptions): Core.APIPromise<void>;
|
|
17
|
+
/**
|
|
18
|
+
* Get the status of a training job.
|
|
19
|
+
*/
|
|
20
|
+
status(query: JobStatusParams, options?: Core.RequestOptions): Core.APIPromise<JobStatusResponse>;
|
|
21
|
+
}
|
|
22
|
+
export type JobListResponse = Array<PostTrainingAPI.PostTrainingJob>;
|
|
23
|
+
/**
|
|
24
|
+
* Artifacts of a finetuning job.
|
|
25
|
+
*/
|
|
26
|
+
export interface JobArtifactsResponse {
|
|
27
|
+
job_uuid: string;
|
|
28
|
+
checkpoints?: Array<JobArtifactsResponse.Checkpoint>;
|
|
29
|
+
}
|
|
30
|
+
export declare namespace JobArtifactsResponse {
|
|
31
|
+
/**
|
|
32
|
+
* Checkpoint created during training runs.
|
|
33
|
+
*/
|
|
34
|
+
interface Checkpoint {
|
|
35
|
+
created_at: string;
|
|
36
|
+
epoch: number;
|
|
37
|
+
identifier: string;
|
|
38
|
+
path: string;
|
|
39
|
+
post_training_job_id: string;
|
|
40
|
+
/**
|
|
41
|
+
* Training metrics captured during post-training jobs.
|
|
42
|
+
*/
|
|
43
|
+
training_metrics?: Checkpoint.TrainingMetrics | null;
|
|
44
|
+
}
|
|
45
|
+
namespace Checkpoint {
|
|
46
|
+
/**
|
|
47
|
+
* Training metrics captured during post-training jobs.
|
|
48
|
+
*/
|
|
49
|
+
interface TrainingMetrics {
|
|
50
|
+
epoch: number;
|
|
51
|
+
perplexity: number;
|
|
52
|
+
train_loss: number;
|
|
53
|
+
validation_loss: number;
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
/**
|
|
58
|
+
* Status of a finetuning job.
|
|
59
|
+
*/
|
|
60
|
+
export interface JobStatusResponse {
|
|
61
|
+
job_uuid: string;
|
|
62
|
+
/**
|
|
63
|
+
* Status of a job execution.
|
|
64
|
+
*/
|
|
65
|
+
status: 'completed' | 'in_progress' | 'failed' | 'scheduled' | 'cancelled';
|
|
66
|
+
checkpoints?: Array<JobStatusResponse.Checkpoint>;
|
|
67
|
+
completed_at?: string | null;
|
|
68
|
+
resources_allocated?: {
|
|
69
|
+
[key: string]: unknown;
|
|
70
|
+
} | null;
|
|
71
|
+
scheduled_at?: string | null;
|
|
72
|
+
started_at?: string | null;
|
|
73
|
+
}
|
|
74
|
+
export declare namespace JobStatusResponse {
|
|
75
|
+
/**
|
|
76
|
+
* Checkpoint created during training runs.
|
|
77
|
+
*/
|
|
78
|
+
interface Checkpoint {
|
|
79
|
+
created_at: string;
|
|
80
|
+
epoch: number;
|
|
81
|
+
identifier: string;
|
|
82
|
+
path: string;
|
|
83
|
+
post_training_job_id: string;
|
|
84
|
+
/**
|
|
85
|
+
* Training metrics captured during post-training jobs.
|
|
86
|
+
*/
|
|
87
|
+
training_metrics?: Checkpoint.TrainingMetrics | null;
|
|
88
|
+
}
|
|
89
|
+
namespace Checkpoint {
|
|
90
|
+
/**
|
|
91
|
+
* Training metrics captured during post-training jobs.
|
|
92
|
+
*/
|
|
93
|
+
interface TrainingMetrics {
|
|
94
|
+
epoch: number;
|
|
95
|
+
perplexity: number;
|
|
96
|
+
train_loss: number;
|
|
97
|
+
validation_loss: number;
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
export interface JobArtifactsParams {
|
|
102
|
+
job_uuid: string;
|
|
103
|
+
}
|
|
104
|
+
export interface JobCancelParams {
|
|
105
|
+
job_uuid: string;
|
|
106
|
+
}
|
|
107
|
+
export interface JobStatusParams {
|
|
108
|
+
job_uuid: string;
|
|
109
|
+
}
|
|
110
|
+
export declare namespace Job {
|
|
111
|
+
export { type JobListResponse as JobListResponse, type JobArtifactsResponse as JobArtifactsResponse, type JobStatusResponse as JobStatusResponse, type JobArtifactsParams as JobArtifactsParams, type JobCancelParams as JobCancelParams, type JobStatusParams as JobStatusParams, };
|
|
112
|
+
}
|
|
113
|
+
//# sourceMappingURL=job.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"job.d.ts","sourceRoot":"","sources":["../../../src/resources/alpha/post-training/job.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,IAAI,MAAM,eAAe,CAAC;AACtC,OAAO,KAAK,eAAe,MAAM,iBAAiB,CAAC;AAEnD,qBAAa,GAAI,SAAQ,WAAW;IAClC;;OAEG;IACH,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;IAMrE;;OAEG;IACH,SAAS,CAAC,KAAK,EAAE,kBAAkB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAAC;IAI1G;;OAEG;IACH,MAAM,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAQnF;;OAEG;IACH,MAAM,CAAC,KAAK,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC;CAGlG;AAED,MAAM,MAAM,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;AAErE;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,MAAM,CAAC;IAEjB,WAAW,CAAC,EAAE,KAAK,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;CACtD;AAED,yBAAiB,oBAAoB,CAAC;IACpC;;OAEG;IACH,UAAiB,UAAU;QACzB,UAAU,EAAE,MAAM,CAAC;QAEnB,KAAK,EAAE,MAAM,CAAC;QAEd,UAAU,EAAE,MAAM,CAAC;QAEnB,IAAI,EAAE,MAAM,CAAC;QAEb,oBAAoB,EAAE,MAAM,CAAC;QAE7B;;WAEG;QACH,gBAAgB,CAAC,EAAE,UAAU,CAAC,eAAe,GAAG,IAAI,CAAC;KACtD;IAED,UAAiB,UAAU,CAAC;QAC1B;;WAEG;QACH,UAAiB,eAAe;YAC9B,KAAK,EAAE,MAAM,CAAC;YAEd,UAAU,EAAE,MAAM,CAAC;YAEnB,UAAU,EAAE,MAAM,CAAC;YAEnB,eAAe,EAAE,MAAM,CAAC;SACzB;KACF;CACF;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,MAAM,CAAC;IAEjB;;OAEG;IACH,MAAM,EAAE,WAAW,GAAG,aAAa,GAAG,QAAQ,GAAG,WAAW,GAAG,WAAW,CAAC;IAE3E,WAAW,CAAC,EAAE,KAAK,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;IAElD,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAE7B,mBAAmB,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,GAAG,IAAI,CAAC;IAExD,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAE7B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC5B;AAED,yBAAiB,iBAAiB,CAAC;IACjC;;OAEG;IACH,UAAiB,UAAU;QACzB,UAAU,EAAE,MAAM,CAAC;QAEnB,KAAK,EAAE,MAAM,CAAC;QAEd,UAAU,EAAE,MAAM,CAAC;QAEnB,IAAI,EAAE,MAAM,CAAC;QAEb,oBAAoB,EAAE,MAAM,CAAC;QAE7B;;WAEG;QACH,gBAAgB,CAAC,EAAE,UAAU,CAAC,eAAe,GAAG,IAAI,CAAC;KACtD;IAED,UAAiB,UAAU,CAAC;QAC1B;;WAEG;QACH,UAAiB,eAAe;YAC9B,KAAK,EAAE,MAAM,CAAC;YAEd,UAAU,EAAE,MAAM,CAAC;YAEnB,UAAU,EAAE,MAAM,CAAC;YAEnB,eAAe,EAAE,MAAM,CAAC;SACzB;KACF;CACF;AAED,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,CAAC,OAAO,WAAW,GAAG,CAAC;IAC3B,OAAO,EACL,KAAK,eAAe,IAAI,eAAe,EACvC,KAAK,oBAAoB,IAAI,oBAAoB,EACjD,KAAK,iBAAiB,IAAI,iBAAiB,EAC3C,KAAK,kBAAkB,IAAI,kBAAkB,EAC7C,KAAK,eAAe,IAAI,eAAe,EACvC,KAAK,eAAe,IAAI,eAAe,GACxC,CAAC;CACH"}
|
|
@@ -4,28 +4,29 @@
|
|
|
4
4
|
//
|
|
5
5
|
// This source code is licensed under the terms described in the LICENSE file in
|
|
6
6
|
// the root directory of this source tree.
|
|
7
|
+
//
|
|
8
|
+
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
7
9
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
8
10
|
exports.Job = void 0;
|
|
9
|
-
|
|
10
|
-
const resource_1 = require("../../resource.js");
|
|
11
|
+
const resource_1 = require("../../../resource.js");
|
|
11
12
|
class Job extends resource_1.APIResource {
|
|
12
13
|
/**
|
|
13
14
|
* Get all training jobs.
|
|
14
15
|
*/
|
|
15
16
|
list(options) {
|
|
16
|
-
return this._client.get('/
|
|
17
|
+
return this._client.get('/v1alpha/post-training/jobs', options)._thenUnwrap((obj) => obj.data);
|
|
17
18
|
}
|
|
18
19
|
/**
|
|
19
20
|
* Get the artifacts of a training job.
|
|
20
21
|
*/
|
|
21
22
|
artifacts(query, options) {
|
|
22
|
-
return this._client.get('/
|
|
23
|
+
return this._client.get('/v1alpha/post-training/job/artifacts', { query, ...options });
|
|
23
24
|
}
|
|
24
25
|
/**
|
|
25
26
|
* Cancel a training job.
|
|
26
27
|
*/
|
|
27
28
|
cancel(body, options) {
|
|
28
|
-
return this._client.post('/
|
|
29
|
+
return this._client.post('/v1alpha/post-training/job/cancel', {
|
|
29
30
|
body,
|
|
30
31
|
...options,
|
|
31
32
|
headers: { Accept: '*/*', ...options?.headers },
|
|
@@ -35,7 +36,7 @@ class Job extends resource_1.APIResource {
|
|
|
35
36
|
* Get the status of a training job.
|
|
36
37
|
*/
|
|
37
38
|
status(query, options) {
|
|
38
|
-
return this._client.get('/
|
|
39
|
+
return this._client.get('/v1alpha/post-training/job/status', { query, ...options });
|
|
39
40
|
}
|
|
40
41
|
}
|
|
41
42
|
exports.Job = Job;
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"job.js","sourceRoot":"","sources":["../../../src/resources/alpha/post-training/job.ts"],"names":[],"mappings":";AAAA,qDAAqD;AACrD,uBAAuB;AACvB,EAAE;AACF,gFAAgF;AAChF,0CAA0C;AAC1C,EAAE;AACF,sFAAsF;;;AAEtF,mDAAgD;AAIhD,MAAa,GAAI,SAAQ,sBAAW;IAClC;;OAEG;IACH,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,OAAO,CACxD,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,KAAyB,EAAE,OAA6B;QAChE,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACzF,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,IAAqB,EAAE,OAA6B;QACzD,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,mCAAmC,EAAE;YAC5D,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAsB,EAAE,OAA6B;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,mCAAmC,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACtF,CAAC;CACF;AAlCD,kBAkCC"}
|
|
@@ -3,25 +3,27 @@
|
|
|
3
3
|
//
|
|
4
4
|
// This source code is licensed under the terms described in the LICENSE file in
|
|
5
5
|
// the root directory of this source tree.
|
|
6
|
-
|
|
6
|
+
//
|
|
7
|
+
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
8
|
+
import { APIResource } from "../../../resource.mjs";
|
|
7
9
|
export class Job extends APIResource {
|
|
8
10
|
/**
|
|
9
11
|
* Get all training jobs.
|
|
10
12
|
*/
|
|
11
13
|
list(options) {
|
|
12
|
-
return this._client.get('/
|
|
14
|
+
return this._client.get('/v1alpha/post-training/jobs', options)._thenUnwrap((obj) => obj.data);
|
|
13
15
|
}
|
|
14
16
|
/**
|
|
15
17
|
* Get the artifacts of a training job.
|
|
16
18
|
*/
|
|
17
19
|
artifacts(query, options) {
|
|
18
|
-
return this._client.get('/
|
|
20
|
+
return this._client.get('/v1alpha/post-training/job/artifacts', { query, ...options });
|
|
19
21
|
}
|
|
20
22
|
/**
|
|
21
23
|
* Cancel a training job.
|
|
22
24
|
*/
|
|
23
25
|
cancel(body, options) {
|
|
24
|
-
return this._client.post('/
|
|
26
|
+
return this._client.post('/v1alpha/post-training/job/cancel', {
|
|
25
27
|
body,
|
|
26
28
|
...options,
|
|
27
29
|
headers: { Accept: '*/*', ...options?.headers },
|
|
@@ -31,7 +33,7 @@ export class Job extends APIResource {
|
|
|
31
33
|
* Get the status of a training job.
|
|
32
34
|
*/
|
|
33
35
|
status(query, options) {
|
|
34
|
-
return this._client.get('/
|
|
36
|
+
return this._client.get('/v1alpha/post-training/job/status', { query, ...options });
|
|
35
37
|
}
|
|
36
38
|
}
|
|
37
39
|
//# sourceMappingURL=job.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"job.mjs","sourceRoot":"","sources":["../../../src/resources/alpha/post-training/job.ts"],"names":[],"mappings":"AAAA,qDAAqD;AACrD,uBAAuB;AACvB,EAAE;AACF,gFAAgF;AAChF,0CAA0C;AAC1C,EAAE;AACF,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAItB,MAAM,OAAO,GAAI,SAAQ,WAAW;IAClC;;OAEG;IACH,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,OAAO,CACxD,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,KAAyB,EAAE,OAA6B;QAChE,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACzF,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,IAAqB,EAAE,OAA6B;QACzD,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,mCAAmC,EAAE;YAC5D,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAsB,EAAE,OAA6B;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,mCAAmC,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACtF,CAAC;CACF"}
|
|
@@ -0,0 +1,245 @@
|
|
|
1
|
+
import { APIResource } from "../../../resource.js";
|
|
2
|
+
import * as Core from "../../../core.js";
|
|
3
|
+
import * as JobAPI from "./job.js";
|
|
4
|
+
import { Job, JobArtifactsParams, JobArtifactsResponse, JobCancelParams, JobListResponse, JobStatusParams, JobStatusResponse } from "./job.js";
|
|
5
|
+
export declare class PostTraining extends APIResource {
|
|
6
|
+
job: JobAPI.Job;
|
|
7
|
+
/**
|
|
8
|
+
* Run preference optimization of a model.
|
|
9
|
+
*/
|
|
10
|
+
preferenceOptimize(body: PostTrainingPreferenceOptimizeParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJob>;
|
|
11
|
+
/**
|
|
12
|
+
* Run supervised fine-tuning of a model.
|
|
13
|
+
*/
|
|
14
|
+
supervisedFineTune(body: PostTrainingSupervisedFineTuneParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJob>;
|
|
15
|
+
}
|
|
16
|
+
/**
|
|
17
|
+
* Configuration for Low-Rank Adaptation (LoRA) fine-tuning.
|
|
18
|
+
*/
|
|
19
|
+
export type AlgorithmConfig = AlgorithmConfig.LoraFinetuningConfig | AlgorithmConfig.QatFinetuningConfig;
|
|
20
|
+
export declare namespace AlgorithmConfig {
|
|
21
|
+
/**
|
|
22
|
+
* Configuration for Low-Rank Adaptation (LoRA) fine-tuning.
|
|
23
|
+
*/
|
|
24
|
+
interface LoraFinetuningConfig {
|
|
25
|
+
alpha: number;
|
|
26
|
+
apply_lora_to_mlp: boolean;
|
|
27
|
+
apply_lora_to_output: boolean;
|
|
28
|
+
lora_attn_modules: Array<string>;
|
|
29
|
+
rank: number;
|
|
30
|
+
quantize_base?: boolean | null;
|
|
31
|
+
type?: 'LoRA';
|
|
32
|
+
use_dora?: boolean | null;
|
|
33
|
+
}
|
|
34
|
+
/**
|
|
35
|
+
* Configuration for Quantization-Aware Training (QAT) fine-tuning.
|
|
36
|
+
*/
|
|
37
|
+
interface QatFinetuningConfig {
|
|
38
|
+
group_size: number;
|
|
39
|
+
quantizer_name: string;
|
|
40
|
+
type?: 'QAT';
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
export interface ListPostTrainingJobsResponse {
|
|
44
|
+
data: JobAPI.JobListResponse;
|
|
45
|
+
}
|
|
46
|
+
export interface PostTrainingJob {
|
|
47
|
+
job_uuid: string;
|
|
48
|
+
}
|
|
49
|
+
export interface PostTrainingPreferenceOptimizeParams {
|
|
50
|
+
/**
|
|
51
|
+
* Configuration for Direct Preference Optimization (DPO) alignment.
|
|
52
|
+
*/
|
|
53
|
+
algorithm_config: PostTrainingPreferenceOptimizeParams.AlgorithmConfig;
|
|
54
|
+
finetuned_model: string;
|
|
55
|
+
hyperparam_search_config: {
|
|
56
|
+
[key: string]: unknown;
|
|
57
|
+
};
|
|
58
|
+
job_uuid: string;
|
|
59
|
+
logger_config: {
|
|
60
|
+
[key: string]: unknown;
|
|
61
|
+
};
|
|
62
|
+
/**
|
|
63
|
+
* Comprehensive configuration for the training process.
|
|
64
|
+
*/
|
|
65
|
+
training_config: PostTrainingPreferenceOptimizeParams.TrainingConfig;
|
|
66
|
+
}
|
|
67
|
+
export declare namespace PostTrainingPreferenceOptimizeParams {
|
|
68
|
+
/**
|
|
69
|
+
* Configuration for Direct Preference Optimization (DPO) alignment.
|
|
70
|
+
*/
|
|
71
|
+
interface AlgorithmConfig {
|
|
72
|
+
beta: number;
|
|
73
|
+
loss_type?: 'sigmoid' | 'hinge' | 'ipo' | 'kto_pair';
|
|
74
|
+
}
|
|
75
|
+
/**
|
|
76
|
+
* Comprehensive configuration for the training process.
|
|
77
|
+
*/
|
|
78
|
+
interface TrainingConfig {
|
|
79
|
+
n_epochs: number;
|
|
80
|
+
/**
|
|
81
|
+
* Configuration for training data and data loading.
|
|
82
|
+
*/
|
|
83
|
+
data_config?: TrainingConfig.DataConfig | null;
|
|
84
|
+
dtype?: string | null;
|
|
85
|
+
/**
|
|
86
|
+
* Configuration for memory and compute efficiency optimizations.
|
|
87
|
+
*/
|
|
88
|
+
efficiency_config?: TrainingConfig.EfficiencyConfig | null;
|
|
89
|
+
gradient_accumulation_steps?: number;
|
|
90
|
+
max_steps_per_epoch?: number;
|
|
91
|
+
max_validation_steps?: number | null;
|
|
92
|
+
/**
|
|
93
|
+
* Configuration parameters for the optimization algorithm.
|
|
94
|
+
*/
|
|
95
|
+
optimizer_config?: TrainingConfig.OptimizerConfig | null;
|
|
96
|
+
}
|
|
97
|
+
namespace TrainingConfig {
|
|
98
|
+
/**
|
|
99
|
+
* Configuration for training data and data loading.
|
|
100
|
+
*/
|
|
101
|
+
interface DataConfig {
|
|
102
|
+
batch_size: number;
|
|
103
|
+
/**
|
|
104
|
+
* Format of the training dataset.
|
|
105
|
+
*/
|
|
106
|
+
data_format: 'instruct' | 'dialog';
|
|
107
|
+
dataset_id: string;
|
|
108
|
+
shuffle: boolean;
|
|
109
|
+
packed?: boolean | null;
|
|
110
|
+
train_on_input?: boolean | null;
|
|
111
|
+
validation_dataset_id?: string | null;
|
|
112
|
+
}
|
|
113
|
+
/**
|
|
114
|
+
* Configuration for memory and compute efficiency optimizations.
|
|
115
|
+
*/
|
|
116
|
+
interface EfficiencyConfig {
|
|
117
|
+
enable_activation_checkpointing?: boolean | null;
|
|
118
|
+
enable_activation_offloading?: boolean | null;
|
|
119
|
+
fsdp_cpu_offload?: boolean | null;
|
|
120
|
+
memory_efficient_fsdp_wrap?: boolean | null;
|
|
121
|
+
}
|
|
122
|
+
/**
|
|
123
|
+
* Configuration parameters for the optimization algorithm.
|
|
124
|
+
*/
|
|
125
|
+
interface OptimizerConfig {
|
|
126
|
+
lr: number;
|
|
127
|
+
num_warmup_steps: number;
|
|
128
|
+
/**
|
|
129
|
+
* Available optimizer algorithms for training.
|
|
130
|
+
*/
|
|
131
|
+
optimizer_type: 'adam' | 'adamw' | 'sgd';
|
|
132
|
+
weight_decay: number;
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
export interface PostTrainingSupervisedFineTuneParams {
|
|
137
|
+
hyperparam_search_config: {
|
|
138
|
+
[key: string]: unknown;
|
|
139
|
+
};
|
|
140
|
+
job_uuid: string;
|
|
141
|
+
logger_config: {
|
|
142
|
+
[key: string]: unknown;
|
|
143
|
+
};
|
|
144
|
+
/**
|
|
145
|
+
* Comprehensive configuration for the training process.
|
|
146
|
+
*/
|
|
147
|
+
training_config: PostTrainingSupervisedFineTuneParams.TrainingConfig;
|
|
148
|
+
/**
|
|
149
|
+
* Configuration for Low-Rank Adaptation (LoRA) fine-tuning.
|
|
150
|
+
*/
|
|
151
|
+
algorithm_config?: PostTrainingSupervisedFineTuneParams.LoraFinetuningConfig | PostTrainingSupervisedFineTuneParams.QatFinetuningConfig | null;
|
|
152
|
+
checkpoint_dir?: string | null;
|
|
153
|
+
/**
|
|
154
|
+
* Model descriptor for training if not in provider config`
|
|
155
|
+
*/
|
|
156
|
+
model?: string | null;
|
|
157
|
+
}
|
|
158
|
+
export declare namespace PostTrainingSupervisedFineTuneParams {
|
|
159
|
+
/**
|
|
160
|
+
* Comprehensive configuration for the training process.
|
|
161
|
+
*/
|
|
162
|
+
interface TrainingConfig {
|
|
163
|
+
n_epochs: number;
|
|
164
|
+
/**
|
|
165
|
+
* Configuration for training data and data loading.
|
|
166
|
+
*/
|
|
167
|
+
data_config?: TrainingConfig.DataConfig | null;
|
|
168
|
+
dtype?: string | null;
|
|
169
|
+
/**
|
|
170
|
+
* Configuration for memory and compute efficiency optimizations.
|
|
171
|
+
*/
|
|
172
|
+
efficiency_config?: TrainingConfig.EfficiencyConfig | null;
|
|
173
|
+
gradient_accumulation_steps?: number;
|
|
174
|
+
max_steps_per_epoch?: number;
|
|
175
|
+
max_validation_steps?: number | null;
|
|
176
|
+
/**
|
|
177
|
+
* Configuration parameters for the optimization algorithm.
|
|
178
|
+
*/
|
|
179
|
+
optimizer_config?: TrainingConfig.OptimizerConfig | null;
|
|
180
|
+
}
|
|
181
|
+
namespace TrainingConfig {
|
|
182
|
+
/**
|
|
183
|
+
* Configuration for training data and data loading.
|
|
184
|
+
*/
|
|
185
|
+
interface DataConfig {
|
|
186
|
+
batch_size: number;
|
|
187
|
+
/**
|
|
188
|
+
* Format of the training dataset.
|
|
189
|
+
*/
|
|
190
|
+
data_format: 'instruct' | 'dialog';
|
|
191
|
+
dataset_id: string;
|
|
192
|
+
shuffle: boolean;
|
|
193
|
+
packed?: boolean | null;
|
|
194
|
+
train_on_input?: boolean | null;
|
|
195
|
+
validation_dataset_id?: string | null;
|
|
196
|
+
}
|
|
197
|
+
/**
|
|
198
|
+
* Configuration for memory and compute efficiency optimizations.
|
|
199
|
+
*/
|
|
200
|
+
interface EfficiencyConfig {
|
|
201
|
+
enable_activation_checkpointing?: boolean | null;
|
|
202
|
+
enable_activation_offloading?: boolean | null;
|
|
203
|
+
fsdp_cpu_offload?: boolean | null;
|
|
204
|
+
memory_efficient_fsdp_wrap?: boolean | null;
|
|
205
|
+
}
|
|
206
|
+
/**
|
|
207
|
+
* Configuration parameters for the optimization algorithm.
|
|
208
|
+
*/
|
|
209
|
+
interface OptimizerConfig {
|
|
210
|
+
lr: number;
|
|
211
|
+
num_warmup_steps: number;
|
|
212
|
+
/**
|
|
213
|
+
* Available optimizer algorithms for training.
|
|
214
|
+
*/
|
|
215
|
+
optimizer_type: 'adam' | 'adamw' | 'sgd';
|
|
216
|
+
weight_decay: number;
|
|
217
|
+
}
|
|
218
|
+
}
|
|
219
|
+
/**
|
|
220
|
+
* Configuration for Low-Rank Adaptation (LoRA) fine-tuning.
|
|
221
|
+
*/
|
|
222
|
+
interface LoraFinetuningConfig {
|
|
223
|
+
alpha: number;
|
|
224
|
+
apply_lora_to_mlp: boolean;
|
|
225
|
+
apply_lora_to_output: boolean;
|
|
226
|
+
lora_attn_modules: Array<string>;
|
|
227
|
+
rank: number;
|
|
228
|
+
quantize_base?: boolean | null;
|
|
229
|
+
type?: 'LoRA';
|
|
230
|
+
use_dora?: boolean | null;
|
|
231
|
+
}
|
|
232
|
+
/**
|
|
233
|
+
* Configuration for Quantization-Aware Training (QAT) fine-tuning.
|
|
234
|
+
*/
|
|
235
|
+
interface QatFinetuningConfig {
|
|
236
|
+
group_size: number;
|
|
237
|
+
quantizer_name: string;
|
|
238
|
+
type?: 'QAT';
|
|
239
|
+
}
|
|
240
|
+
}
|
|
241
|
+
export declare namespace PostTraining {
|
|
242
|
+
export { type AlgorithmConfig as AlgorithmConfig, type ListPostTrainingJobsResponse as ListPostTrainingJobsResponse, type PostTrainingJob as PostTrainingJob, type PostTrainingPreferenceOptimizeParams as PostTrainingPreferenceOptimizeParams, type PostTrainingSupervisedFineTuneParams as PostTrainingSupervisedFineTuneParams, };
|
|
243
|
+
export { Job as Job, type JobListResponse as JobListResponse, type JobArtifactsResponse as JobArtifactsResponse, type JobStatusResponse as JobStatusResponse, type JobArtifactsParams as JobArtifactsParams, type JobCancelParams as JobCancelParams, type JobStatusParams as JobStatusParams, };
|
|
244
|
+
}
|
|
245
|
+
//# sourceMappingURL=post-training.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"post-training.d.ts","sourceRoot":"","sources":["../../../src/resources/alpha/post-training/post-training.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,IAAI,MAAM,eAAe,CAAC;AACtC,OAAO,KAAK,MAAM,MAAM,OAAO,CAAC;AAChC,OAAO,EACL,GAAG,EACH,kBAAkB,EAClB,oBAAoB,EACpB,eAAe,EACf,eAAe,EACf,eAAe,EACf,iBAAiB,EAClB,MAAM,OAAO,CAAC;AAEf,qBAAa,YAAa,SAAQ,WAAW;IAC3C,GAAG,EAAE,MAAM,CAAC,GAAG,CAAgC;IAE/C;;OAEG;IACH,kBAAkB,CAChB,IAAI,EAAE,oCAAoC,EAC1C,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;IAInC;;OAEG;IACH,kBAAkB,CAChB,IAAI,EAAE,oCAAoC,EAC1C,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;CAGpC;AAED;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,eAAe,CAAC,oBAAoB,GAAG,eAAe,CAAC,mBAAmB,CAAC;AAEzG,yBAAiB,eAAe,CAAC;IAC/B;;OAEG;IACH,UAAiB,oBAAoB;QACnC,KAAK,EAAE,MAAM,CAAC;QAEd,iBAAiB,EAAE,OAAO,CAAC;QAE3B,oBAAoB,EAAE,OAAO,CAAC;QAE9B,iBAAiB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAEjC,IAAI,EAAE,MAAM,CAAC;QAEb,aAAa,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;QAE/B,IAAI,CAAC,EAAE,MAAM,CAAC;QAEd,QAAQ,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;KAC3B;IAED;;OAEG;IACH,UAAiB,mBAAmB;QAClC,UAAU,EAAE,MAAM,CAAC;QAEnB,cAAc,EAAE,MAAM,CAAC;QAEvB,IAAI,CAAC,EAAE,KAAK,CAAC;KACd;CACF;AAED,MAAM,WAAW,4BAA4B;IAC3C,IAAI,EAAE,MAAM,CAAC,eAAe,CAAC;CAC9B;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,oCAAoC;IACnD;;OAEG;IACH,gBAAgB,EAAE,oCAAoC,CAAC,eAAe,CAAC;IAEvE,eAAe,EAAE,MAAM,CAAC;IAExB,wBAAwB,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IAErD,QAAQ,EAAE,MAAM,CAAC;IAEjB,aAAa,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IAE1C;;OAEG;IACH,eAAe,EAAE,oCAAoC,CAAC,cAAc,CAAC;CACtE;AAED,yBAAiB,oCAAoC,CAAC;IACpD;;OAEG;IACH,UAAiB,eAAe;QAC9B,IAAI,EAAE,MAAM,CAAC;QAEb,SAAS,CAAC,EAAE,SAAS,GAAG,OAAO,GAAG,KAAK,GAAG,UAAU,CAAC;KACtD;IAED;;OAEG;IACH,UAAiB,cAAc;QAC7B,QAAQ,EAAE,MAAM,CAAC;QAEjB;;WAEG;QACH,WAAW,CAAC,EAAE,cAAc,CAAC,UAAU,GAAG,IAAI,CAAC;QAE/C,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAEtB;;WAEG;QACH,iBAAiB,CAAC,EAAE,cAAc,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAE3D,2BAA2B,CAAC,EAAE,MAAM,CAAC;QAErC,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAE7B,oBAAoB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAErC;;WAEG;QACH,gBAAgB,CAAC,EAAE,cAAc,CAAC,eAAe,GAAG,IAAI,CAAC;KAC1D;IAED,UAAiB,cAAc,CAAC;QAC9B;;WAEG;QACH,UAAiB,UAAU;YACzB,UAAU,EAAE,MAAM,CAAC;YAEnB;;eAEG;YACH,WAAW,EAAE,UAAU,GAAG,QAAQ,CAAC;YAEnC,UAAU,EAAE,MAAM,CAAC;YAEnB,OAAO,EAAE,OAAO,CAAC;YAEjB,MAAM,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAExB,cAAc,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAEhC,qBAAqB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;SACvC;QAED;;WAEG;QACH,UAAiB,gBAAgB;YAC/B,+BAA+B,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAEjD,4BAA4B,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAE9C,gBAAgB,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAElC,0BAA0B,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;SAC7C;QAED;;WAEG;QACH,UAAiB,eAAe;YAC9B,EAAE,EAAE,MAAM,CAAC;YAEX,gBAAgB,EAAE,MAAM,CAAC;YAEzB;;eAEG;YACH,cAAc,EAAE,MAAM,GAAG,OAAO,GAAG,KAAK,CAAC;YAEzC,YAAY,EAAE,MAAM,CAAC;SACtB;KACF;CACF;AAED,MAAM,WAAW,oCAAoC;IACnD,wBAAwB,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IAErD,QAAQ,EAAE,MAAM,CAAC;IAEjB,aAAa,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IAE1C;;OAEG;IACH,eAAe,EAAE,oCAAoC,CAAC,cAAc,CAAC;IAErE;;OAEG;IACH,gBAAgB,CAAC,EACb,oCAAoC,CAAC,oBAAoB,GACzD,oCAAoC,CAAC,mBAAmB,GACxD,IAAI,CAAC;IAET,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAE/B;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACvB;AAED,yBAAiB,oCAAoC,CAAC;IACpD;;OAEG;IACH,UAAiB,cAAc;QAC7B,QAAQ,EAAE,MAAM,CAAC;QAEjB;;WAEG;QACH,WAAW,CAAC,EAAE,cAAc,CAAC,UAAU,GAAG,IAAI,CAAC;QAE/C,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAEtB;;WAEG;QACH,iBAAiB,CAAC,EAAE,cAAc,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAE3D,2BAA2B,CAAC,EAAE,MAAM,CAAC;QAErC,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAE7B,oBAAoB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAErC;;WAEG;QACH,gBAAgB,CAAC,EAAE,cAAc,CAAC,eAAe,GAAG,IAAI,CAAC;KAC1D;IAED,UAAiB,cAAc,CAAC;QAC9B;;WAEG;QACH,UAAiB,UAAU;YACzB,UAAU,EAAE,MAAM,CAAC;YAEnB;;eAEG;YACH,WAAW,EAAE,UAAU,GAAG,QAAQ,CAAC;YAEnC,UAAU,EAAE,MAAM,CAAC;YAEnB,OAAO,EAAE,OAAO,CAAC;YAEjB,MAAM,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAExB,cAAc,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAEhC,qBAAqB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;SACvC;QAED;;WAEG;QACH,UAAiB,gBAAgB;YAC/B,+BAA+B,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAEjD,4BAA4B,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAE9C,gBAAgB,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;YAElC,0BAA0B,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;SAC7C;QAED;;WAEG;QACH,UAAiB,eAAe;YAC9B,EAAE,EAAE,MAAM,CAAC;YAEX,gBAAgB,EAAE,MAAM,CAAC;YAEzB;;eAEG;YACH,cAAc,EAAE,MAAM,GAAG,OAAO,GAAG,KAAK,CAAC;YAEzC,YAAY,EAAE,MAAM,CAAC;SACtB;KACF;IAED;;OAEG;IACH,UAAiB,oBAAoB;QACnC,KAAK,EAAE,MAAM,CAAC;QAEd,iBAAiB,EAAE,OAAO,CAAC;QAE3B,oBAAoB,EAAE,OAAO,CAAC;QAE9B,iBAAiB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAEjC,IAAI,EAAE,MAAM,CAAC;QAEb,aAAa,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;QAE/B,IAAI,CAAC,EAAE,MAAM,CAAC;QAEd,QAAQ,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;KAC3B;IAED;;OAEG;IACH,UAAiB,mBAAmB;QAClC,UAAU,EAAE,MAAM,CAAC;QAEnB,cAAc,EAAE,MAAM,CAAC;QAEvB,IAAI,CAAC,EAAE,KAAK,CAAC;KACd;CACF;AAID,MAAM,CAAC,OAAO,WAAW,YAAY,CAAC;IACpC,OAAO,EACL,KAAK,eAAe,IAAI,eAAe,EACvC,KAAK,4BAA4B,IAAI,4BAA4B,EACjE,KAAK,eAAe,IAAI,eAAe,EACvC,KAAK,oCAAoC,IAAI,oCAAoC,EACjF,KAAK,oCAAoC,IAAI,oCAAoC,GAClF,CAAC;IAEF,OAAO,EACL,GAAG,IAAI,GAAG,EACV,KAAK,eAAe,IAAI,eAAe,EACvC,KAAK,oBAAoB,IAAI,oBAAoB,EACjD,KAAK,iBAAiB,IAAI,iBAAiB,EAC3C,KAAK,kBAAkB,IAAI,kBAAkB,EAC7C,KAAK,eAAe,IAAI,eAAe,EACvC,KAAK,eAAe,IAAI,eAAe,GACxC,CAAC;CACH"}
|