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.
Files changed (580) hide show
  1. package/CHANGELOG.md +153 -0
  2. package/README.md +48 -39
  3. package/error.d.ts.map +1 -1
  4. package/error.js +1 -0
  5. package/error.js.map +1 -1
  6. package/error.mjs.map +1 -1
  7. package/index.d.mts +33 -54
  8. package/index.d.ts +33 -54
  9. package/index.d.ts.map +1 -1
  10. package/index.js +20 -29
  11. package/index.js.map +1 -1
  12. package/index.mjs +18 -28
  13. package/index.mjs.map +1 -1
  14. package/lib/response-helpers.d.ts +23 -0
  15. package/lib/response-helpers.d.ts.map +1 -0
  16. package/lib/response-helpers.js +53 -0
  17. package/lib/response-helpers.js.map +1 -0
  18. package/lib/response-helpers.mjs +49 -0
  19. package/lib/response-helpers.mjs.map +1 -0
  20. package/package.json +7 -1
  21. package/pagination.d.ts.map +1 -1
  22. package/pagination.js +2 -4
  23. package/pagination.js.map +1 -1
  24. package/pagination.mjs +1 -4
  25. package/pagination.mjs.map +1 -1
  26. package/resource.d.ts.map +1 -1
  27. package/resource.js.map +1 -1
  28. package/resource.mjs.map +1 -1
  29. package/resources/alpha/admin.d.ts +41 -0
  30. package/resources/alpha/admin.d.ts.map +1 -0
  31. package/resources/alpha/admin.js +46 -0
  32. package/resources/alpha/admin.js.map +1 -0
  33. package/resources/alpha/admin.mjs +42 -0
  34. package/resources/alpha/admin.mjs.map +1 -0
  35. package/resources/alpha/alpha.d.ts +26 -0
  36. package/resources/alpha/alpha.d.ts.map +1 -0
  37. package/resources/alpha/alpha.js +61 -0
  38. package/resources/alpha/alpha.js.map +1 -0
  39. package/resources/alpha/alpha.mjs +34 -0
  40. package/resources/alpha/alpha.mjs.map +1 -0
  41. package/resources/{benchmarks.d.ts → alpha/benchmarks.d.ts} +41 -15
  42. package/resources/alpha/benchmarks.d.ts.map +1 -0
  43. package/resources/{benchmarks.js → alpha/benchmarks.js} +18 -4
  44. package/resources/alpha/benchmarks.js.map +1 -0
  45. package/resources/alpha/benchmarks.mjs +44 -0
  46. package/resources/alpha/benchmarks.mjs.map +1 -0
  47. package/resources/alpha/eval/eval.d.ts +160 -0
  48. package/resources/alpha/eval/eval.d.ts.map +1 -0
  49. package/resources/{eval → alpha/eval}/eval.js +7 -6
  50. package/resources/alpha/eval/eval.js.map +1 -0
  51. package/resources/{eval → alpha/eval}/eval.mjs +7 -5
  52. package/resources/alpha/eval/eval.mjs.map +1 -0
  53. package/resources/alpha/eval/index.d.ts +3 -0
  54. package/resources/alpha/eval/index.d.ts.map +1 -0
  55. package/resources/{eval → alpha/eval}/index.js +2 -1
  56. package/resources/alpha/eval/index.js.map +1 -0
  57. package/resources/{eval → alpha/eval}/index.mjs +2 -0
  58. package/resources/alpha/eval/index.mjs.map +1 -0
  59. package/resources/{eval → alpha/eval}/jobs.d.ts +2 -2
  60. package/resources/alpha/eval/jobs.d.ts.map +1 -0
  61. package/resources/{eval → alpha/eval}/jobs.js +6 -5
  62. package/resources/alpha/eval/jobs.js.map +1 -0
  63. package/resources/{eval → alpha/eval}/jobs.mjs +6 -4
  64. package/resources/alpha/eval/jobs.mjs.map +1 -0
  65. package/resources/alpha/eval.d.ts.map +1 -0
  66. package/resources/{eval.js → alpha/eval.js} +1 -0
  67. package/resources/alpha/eval.js.map +1 -0
  68. package/resources/alpha/eval.mjs.map +1 -0
  69. package/resources/alpha/index.d.ts +7 -0
  70. package/resources/alpha/index.d.ts.map +1 -0
  71. package/resources/alpha/index.js +23 -0
  72. package/resources/alpha/index.js.map +1 -0
  73. package/resources/alpha/index.mjs +14 -0
  74. package/resources/alpha/index.mjs.map +1 -0
  75. package/resources/alpha/inference.d.ts +85 -0
  76. package/resources/alpha/inference.d.ts.map +1 -0
  77. package/resources/{agents/steps.js → alpha/inference.js} +9 -8
  78. package/resources/alpha/inference.js.map +1 -0
  79. package/resources/alpha/inference.mjs +17 -0
  80. package/resources/alpha/inference.mjs.map +1 -0
  81. package/resources/alpha/post-training/index.d.ts.map +1 -0
  82. package/resources/{post-training → alpha/post-training}/index.js +2 -1
  83. package/resources/alpha/post-training/index.js.map +1 -0
  84. package/resources/{post-training → alpha/post-training}/index.mjs +2 -0
  85. package/resources/alpha/post-training/index.mjs.map +1 -0
  86. package/resources/alpha/post-training/job.d.ts +113 -0
  87. package/resources/alpha/post-training/job.d.ts.map +1 -0
  88. package/resources/{post-training → alpha/post-training}/job.js +7 -6
  89. package/resources/alpha/post-training/job.js.map +1 -0
  90. package/resources/{post-training → alpha/post-training}/job.mjs +7 -5
  91. package/resources/alpha/post-training/job.mjs.map +1 -0
  92. package/resources/alpha/post-training/post-training.d.ts +245 -0
  93. package/resources/alpha/post-training/post-training.d.ts.map +1 -0
  94. package/resources/{post-training → alpha/post-training}/post-training.js +5 -4
  95. package/resources/alpha/post-training/post-training.js.map +1 -0
  96. package/resources/{post-training → alpha/post-training}/post-training.mjs +5 -3
  97. package/resources/alpha/post-training/post-training.mjs.map +1 -0
  98. package/resources/alpha/post-training.d.ts.map +1 -0
  99. package/resources/{post-training.js → alpha/post-training.js} +2 -1
  100. package/resources/alpha/post-training.js.map +1 -0
  101. package/resources/{post-training.mjs → alpha/post-training.mjs} +2 -0
  102. package/resources/alpha/post-training.mjs.map +1 -0
  103. package/resources/alpha.d.ts +2 -0
  104. package/resources/alpha.d.ts.map +1 -0
  105. package/resources/alpha.js +25 -0
  106. package/resources/alpha.js.map +1 -0
  107. package/resources/alpha.mjs +9 -0
  108. package/resources/alpha.mjs.map +1 -0
  109. package/resources/batches.d.ts +322 -0
  110. package/resources/batches.d.ts.map +1 -0
  111. package/resources/batches.js +45 -0
  112. package/resources/batches.js.map +1 -0
  113. package/resources/batches.mjs +40 -0
  114. package/resources/batches.mjs.map +1 -0
  115. package/resources/beta/beta.d.ts +10 -0
  116. package/resources/beta/beta.d.ts.map +1 -0
  117. package/resources/{tool-runtime/tool-runtime.js → beta/beta.js} +10 -22
  118. package/resources/beta/beta.js.map +1 -0
  119. package/resources/beta/beta.mjs +18 -0
  120. package/resources/beta/beta.mjs.map +1 -0
  121. package/resources/{datasets.d.ts → beta/datasets.d.ts} +117 -99
  122. package/resources/beta/datasets.d.ts.map +1 -0
  123. package/resources/{datasets.js → beta/datasets.js} +14 -9
  124. package/resources/beta/datasets.js.map +1 -0
  125. package/resources/{datasets.mjs → beta/datasets.mjs} +14 -8
  126. package/resources/beta/datasets.mjs.map +1 -0
  127. package/resources/beta/index.d.ts +3 -0
  128. package/resources/beta/index.d.ts.map +1 -0
  129. package/resources/beta/index.js +15 -0
  130. package/resources/beta/index.js.map +1 -0
  131. package/resources/beta/index.mjs +10 -0
  132. package/resources/beta/index.mjs.map +1 -0
  133. package/resources/beta.d.ts +2 -0
  134. package/resources/beta.d.ts.map +1 -0
  135. package/resources/{agents.js → beta.js} +4 -3
  136. package/resources/beta.js.map +1 -0
  137. package/resources/beta.mjs +9 -0
  138. package/resources/beta.mjs.map +1 -0
  139. package/resources/chat/chat.d.ts +75 -80
  140. package/resources/chat/chat.d.ts.map +1 -1
  141. package/resources/chat/chat.js +2 -2
  142. package/resources/chat/chat.js.map +1 -1
  143. package/resources/chat/chat.mjs +3 -2
  144. package/resources/chat/chat.mjs.map +1 -1
  145. package/resources/chat/completions.d.ts +760 -1406
  146. package/resources/chat/completions.d.ts.map +1 -1
  147. package/resources/chat/completions.js +8 -14
  148. package/resources/chat/completions.js.map +1 -1
  149. package/resources/chat/completions.mjs +7 -11
  150. package/resources/chat/completions.mjs.map +1 -1
  151. package/resources/chat/index.d.ts +1 -1
  152. package/resources/chat/index.d.ts.map +1 -1
  153. package/resources/chat/index.js +3 -3
  154. package/resources/chat/index.js.map +1 -1
  155. package/resources/chat/index.mjs +3 -1
  156. package/resources/chat/index.mjs.map +1 -1
  157. package/resources/chat.js +2 -1
  158. package/resources/chat.js.map +1 -1
  159. package/resources/chat.mjs +2 -0
  160. package/resources/chat.mjs.map +1 -1
  161. package/resources/completions.d.ts +54 -97
  162. package/resources/completions.d.ts.map +1 -1
  163. package/resources/completions.js +3 -6
  164. package/resources/completions.js.map +1 -1
  165. package/resources/completions.mjs +3 -5
  166. package/resources/completions.mjs.map +1 -1
  167. package/resources/conversations/conversations.d.ts +320 -0
  168. package/resources/conversations/conversations.d.ts.map +1 -0
  169. package/resources/conversations/conversations.js +78 -0
  170. package/resources/conversations/conversations.js.map +1 -0
  171. package/resources/conversations/conversations.mjs +51 -0
  172. package/resources/conversations/conversations.mjs.map +1 -0
  173. package/resources/conversations/index.d.ts +3 -0
  174. package/resources/conversations/index.d.ts.map +1 -0
  175. package/resources/conversations/index.js +16 -0
  176. package/resources/conversations/index.js.map +1 -0
  177. package/resources/conversations/index.mjs +10 -0
  178. package/resources/conversations/index.mjs.map +1 -0
  179. package/resources/conversations/items.d.ts +874 -0
  180. package/resources/conversations/items.d.ts.map +1 -0
  181. package/resources/conversations/items.js +51 -0
  182. package/resources/conversations/items.js.map +1 -0
  183. package/resources/conversations/items.mjs +46 -0
  184. package/resources/conversations/items.mjs.map +1 -0
  185. package/resources/conversations.d.ts +2 -0
  186. package/resources/conversations.d.ts.map +1 -0
  187. package/resources/conversations.js +25 -0
  188. package/resources/conversations.js.map +1 -0
  189. package/resources/conversations.mjs +9 -0
  190. package/resources/conversations.mjs.map +1 -0
  191. package/resources/embeddings.d.ts +10 -52
  192. package/resources/embeddings.d.ts.map +1 -1
  193. package/resources/embeddings.js +5 -2
  194. package/resources/embeddings.js.map +1 -1
  195. package/resources/embeddings.mjs +5 -1
  196. package/resources/embeddings.mjs.map +1 -1
  197. package/resources/files.d.ts +44 -54
  198. package/resources/files.d.ts.map +1 -1
  199. package/resources/files.js +18 -9
  200. package/resources/files.js.map +1 -1
  201. package/resources/files.mjs +18 -8
  202. package/resources/files.mjs.map +1 -1
  203. package/resources/index.d.ts +11 -15
  204. package/resources/index.d.ts.map +1 -1
  205. package/resources/index.js +15 -21
  206. package/resources/index.js.map +1 -1
  207. package/resources/index.mjs +9 -11
  208. package/resources/index.mjs.map +1 -1
  209. package/resources/inspect.d.ts +3 -71
  210. package/resources/inspect.d.ts.map +1 -1
  211. package/resources/inspect.js +2 -1
  212. package/resources/inspect.js.map +1 -1
  213. package/resources/inspect.mjs +2 -0
  214. package/resources/inspect.mjs.map +1 -1
  215. package/resources/models/index.d.ts +2 -2
  216. package/resources/models/index.d.ts.map +1 -1
  217. package/resources/models/index.js +2 -1
  218. package/resources/models/index.js.map +1 -1
  219. package/resources/models/index.mjs +2 -0
  220. package/resources/models/index.mjs.map +1 -1
  221. package/resources/models/models.d.ts +71 -31
  222. package/resources/models/models.d.ts.map +1 -1
  223. package/resources/models/models.js +13 -2
  224. package/resources/models/models.js.map +1 -1
  225. package/resources/models/models.mjs +13 -1
  226. package/resources/models/models.mjs.map +1 -1
  227. package/resources/models/openai.d.ts +2 -16
  228. package/resources/models/openai.d.ts.map +1 -1
  229. package/resources/models/openai.js +3 -2
  230. package/resources/models/openai.js.map +1 -1
  231. package/resources/models/openai.mjs +3 -1
  232. package/resources/models/openai.mjs.map +1 -1
  233. package/resources/models.js +2 -1
  234. package/resources/models.js.map +1 -1
  235. package/resources/models.mjs +2 -0
  236. package/resources/models.mjs.map +1 -1
  237. package/resources/moderations.d.ts +9 -35
  238. package/resources/moderations.d.ts.map +1 -1
  239. package/resources/moderations.js +5 -2
  240. package/resources/moderations.js.map +1 -1
  241. package/resources/moderations.mjs +5 -1
  242. package/resources/moderations.mjs.map +1 -1
  243. package/resources/prompts/index.d.ts +3 -0
  244. package/resources/prompts/index.d.ts.map +1 -0
  245. package/resources/prompts/index.js +15 -0
  246. package/resources/prompts/index.js.map +1 -0
  247. package/resources/prompts/index.mjs +10 -0
  248. package/resources/prompts/index.mjs.map +1 -0
  249. package/resources/prompts/prompts.d.ts +96 -0
  250. package/resources/prompts/prompts.d.ts.map +1 -0
  251. package/resources/{agents/agents.js → prompts/prompts.js} +43 -30
  252. package/resources/prompts/prompts.js.map +1 -0
  253. package/resources/prompts/prompts.mjs +66 -0
  254. package/resources/prompts/prompts.mjs.map +1 -0
  255. package/resources/prompts/versions.d.ts +12 -0
  256. package/resources/prompts/versions.d.ts.map +1 -0
  257. package/resources/prompts/versions.js +23 -0
  258. package/resources/prompts/versions.js.map +1 -0
  259. package/resources/prompts/versions.mjs +19 -0
  260. package/resources/prompts/versions.mjs.map +1 -0
  261. package/resources/prompts.d.ts +2 -0
  262. package/resources/prompts.d.ts.map +1 -0
  263. package/resources/prompts.js +25 -0
  264. package/resources/prompts.js.map +1 -0
  265. package/resources/prompts.mjs +9 -0
  266. package/resources/prompts.mjs.map +1 -0
  267. package/resources/providers.d.ts +4 -13
  268. package/resources/providers.d.ts.map +1 -1
  269. package/resources/providers.js +2 -1
  270. package/resources/providers.js.map +1 -1
  271. package/resources/providers.mjs +2 -0
  272. package/resources/providers.mjs.map +1 -1
  273. package/resources/responses/index.js +2 -1
  274. package/resources/responses/index.js.map +1 -1
  275. package/resources/responses/index.mjs +2 -0
  276. package/resources/responses/index.mjs.map +1 -1
  277. package/resources/responses/input-items.d.ts +246 -151
  278. package/resources/responses/input-items.d.ts.map +1 -1
  279. package/resources/responses/input-items.js +3 -2
  280. package/resources/responses/input-items.js.map +1 -1
  281. package/resources/responses/input-items.mjs +3 -1
  282. package/resources/responses/input-items.mjs.map +1 -1
  283. package/resources/responses/responses.d.ts +1841 -1455
  284. package/resources/responses/responses.d.ts.map +1 -1
  285. package/resources/responses/responses.js +8 -11
  286. package/resources/responses/responses.js.map +1 -1
  287. package/resources/responses/responses.mjs +8 -10
  288. package/resources/responses/responses.mjs.map +1 -1
  289. package/resources/responses.js +2 -1
  290. package/resources/responses.js.map +1 -1
  291. package/resources/responses.mjs +2 -0
  292. package/resources/responses.mjs.map +1 -1
  293. package/resources/routes.d.ts +11 -10
  294. package/resources/routes.d.ts.map +1 -1
  295. package/resources/routes.js +8 -6
  296. package/resources/routes.js.map +1 -1
  297. package/resources/routes.mjs +8 -5
  298. package/resources/routes.mjs.map +1 -1
  299. package/resources/safety.d.ts +144 -10
  300. package/resources/safety.d.ts.map +1 -1
  301. package/resources/safety.js +4 -1
  302. package/resources/safety.js.map +1 -1
  303. package/resources/safety.mjs +4 -0
  304. package/resources/safety.mjs.map +1 -1
  305. package/resources/scoring-functions.d.ts +120 -50
  306. package/resources/scoring-functions.d.ts.map +1 -1
  307. package/resources/scoring-functions.js +15 -1
  308. package/resources/scoring-functions.js.map +1 -1
  309. package/resources/scoring-functions.mjs +15 -0
  310. package/resources/scoring-functions.mjs.map +1 -1
  311. package/resources/scoring.d.ts +85 -26
  312. package/resources/scoring.d.ts.map +1 -1
  313. package/resources/scoring.js +2 -1
  314. package/resources/scoring.js.map +1 -1
  315. package/resources/scoring.mjs +2 -0
  316. package/resources/scoring.mjs.map +1 -1
  317. package/resources/shared.d.ts +173 -591
  318. package/resources/shared.d.ts.map +1 -1
  319. package/resources/shared.js +2 -0
  320. package/resources/shared.js.map +1 -1
  321. package/resources/shared.mjs +2 -0
  322. package/resources/shared.mjs.map +1 -1
  323. package/resources/shields.d.ts +19 -24
  324. package/resources/shields.d.ts.map +1 -1
  325. package/resources/shields.js +6 -1
  326. package/resources/shields.js.map +1 -1
  327. package/resources/shields.mjs +6 -0
  328. package/resources/shields.mjs.map +1 -1
  329. package/resources/tool-runtime.d.ts +151 -1
  330. package/resources/tool-runtime.d.ts.map +1 -1
  331. package/resources/tool-runtime.js +22 -16
  332. package/resources/tool-runtime.js.map +1 -1
  333. package/resources/tool-runtime.mjs +20 -1
  334. package/resources/tool-runtime.mjs.map +1 -1
  335. package/resources/toolgroups.d.ts +27 -37
  336. package/resources/toolgroups.d.ts.map +1 -1
  337. package/resources/toolgroups.js +10 -1
  338. package/resources/toolgroups.js.map +1 -1
  339. package/resources/toolgroups.mjs +10 -0
  340. package/resources/toolgroups.mjs.map +1 -1
  341. package/resources/tools.d.ts +9 -76
  342. package/resources/tools.d.ts.map +1 -1
  343. package/resources/tools.js +4 -1
  344. package/resources/tools.js.map +1 -1
  345. package/resources/tools.mjs +4 -0
  346. package/resources/tools.mjs.map +1 -1
  347. package/resources/vector-io.d.ts +272 -148
  348. package/resources/vector-io.d.ts.map +1 -1
  349. package/resources/vector-io.js +2 -1
  350. package/resources/vector-io.js.map +1 -1
  351. package/resources/vector-io.mjs +2 -0
  352. package/resources/vector-io.mjs.map +1 -1
  353. package/resources/vector-stores/file-batches.d.ts +111 -0
  354. package/resources/vector-stores/file-batches.d.ts.map +1 -0
  355. package/resources/vector-stores/file-batches.js +45 -0
  356. package/resources/vector-stores/file-batches.js.map +1 -0
  357. package/resources/vector-stores/file-batches.mjs +41 -0
  358. package/resources/vector-stores/file-batches.mjs.map +1 -0
  359. package/resources/vector-stores/files.d.ts +83 -135
  360. package/resources/vector-stores/files.d.ts.map +1 -1
  361. package/resources/vector-stores/files.js +15 -14
  362. package/resources/vector-stores/files.js.map +1 -1
  363. package/resources/vector-stores/files.mjs +15 -13
  364. package/resources/vector-stores/files.mjs.map +1 -1
  365. package/resources/vector-stores/index.d.ts +2 -1
  366. package/resources/vector-stores/index.d.ts.map +1 -1
  367. package/resources/vector-stores/index.js +5 -2
  368. package/resources/vector-stores/index.js.map +1 -1
  369. package/resources/vector-stores/index.mjs +3 -0
  370. package/resources/vector-stores/index.mjs.map +1 -1
  371. package/resources/vector-stores/vector-stores.d.ts +118 -217
  372. package/resources/vector-stores/vector-stores.d.ts.map +1 -1
  373. package/resources/vector-stores/vector-stores.js +18 -12
  374. package/resources/vector-stores/vector-stores.js.map +1 -1
  375. package/resources/vector-stores/vector-stores.mjs +18 -11
  376. package/resources/vector-stores/vector-stores.mjs.map +1 -1
  377. package/resources/vector-stores.js +2 -1
  378. package/resources/vector-stores.js.map +1 -1
  379. package/resources/vector-stores.mjs +2 -0
  380. package/resources/vector-stores.mjs.map +1 -1
  381. package/src/error.ts +1 -0
  382. package/src/index.ts +92 -267
  383. package/src/lib/response-helpers.ts +60 -0
  384. package/src/pagination.ts +2 -5
  385. package/src/resource.ts +1 -0
  386. package/src/resources/alpha/admin.ts +84 -0
  387. package/src/resources/alpha/alpha.ts +93 -0
  388. package/src/resources/{benchmarks.ts → alpha/benchmarks.ts} +51 -16
  389. package/src/resources/{eval → alpha/eval}/eval.ts +83 -65
  390. package/src/resources/{eval → alpha/eval}/index.ts +1 -2
  391. package/src/resources/{eval → alpha/eval}/jobs.ts +6 -6
  392. package/src/resources/{eval.ts → alpha/eval.ts} +1 -0
  393. package/src/resources/alpha/index.ts +36 -0
  394. package/src/resources/alpha/inference.ts +132 -0
  395. package/src/resources/{post-training → alpha/post-training}/index.ts +1 -1
  396. package/src/resources/alpha/post-training/job.ts +177 -0
  397. package/src/resources/alpha/post-training/post-training.ts +372 -0
  398. package/src/resources/{post-training.ts → alpha/post-training.ts} +1 -1
  399. package/src/resources/alpha.ts +9 -0
  400. package/src/resources/batches.ts +572 -0
  401. package/src/resources/beta/beta.ts +41 -0
  402. package/src/resources/{datasets.ts → beta/datasets.ts} +128 -110
  403. package/src/resources/beta/index.ts +20 -0
  404. package/src/resources/{agents.ts → beta.ts} +2 -2
  405. package/src/resources/chat/chat.ts +84 -83
  406. package/src/resources/chat/completions.ts +876 -1486
  407. package/src/resources/chat/index.ts +1 -2
  408. package/src/resources/chat.ts +1 -1
  409. package/src/resources/completions.ts +59 -103
  410. package/src/resources/conversations/conversations.ts +491 -0
  411. package/src/resources/conversations/index.ts +25 -0
  412. package/src/resources/conversations/items.ts +1365 -0
  413. package/src/resources/conversations.ts +9 -0
  414. package/src/resources/embeddings.ts +14 -55
  415. package/src/resources/files.ts +54 -61
  416. package/src/resources/index.ts +35 -111
  417. package/src/resources/inspect.ts +4 -84
  418. package/src/resources/models/index.ts +4 -2
  419. package/src/resources/models/models.ts +79 -30
  420. package/src/resources/models/openai.ts +4 -24
  421. package/src/resources/models.ts +1 -1
  422. package/src/resources/moderations.ts +10 -36
  423. package/src/resources/prompts/index.ts +19 -0
  424. package/src/resources/prompts/prompts.ts +171 -0
  425. package/src/resources/prompts/versions.ts +26 -0
  426. package/src/resources/prompts.ts +9 -0
  427. package/src/resources/providers.ts +5 -18
  428. package/src/resources/responses/index.ts +1 -1
  429. package/src/resources/responses/input-items.ts +365 -161
  430. package/src/resources/responses/responses.ts +2959 -1870
  431. package/src/resources/responses.ts +1 -1
  432. package/src/resources/routes.ts +25 -14
  433. package/src/resources/safety.ts +199 -9
  434. package/src/resources/scoring-functions.ts +168 -52
  435. package/src/resources/scoring.ts +125 -22
  436. package/src/resources/shared.ts +187 -665
  437. package/src/resources/shields.ts +18 -23
  438. package/src/resources/tool-runtime.ts +210 -2
  439. package/src/resources/toolgroups.ts +26 -36
  440. package/src/resources/tools.ts +10 -94
  441. package/src/resources/vector-io.ts +323 -145
  442. package/src/resources/vector-stores/file-batches.ts +201 -0
  443. package/src/resources/vector-stores/files.ts +123 -143
  444. package/src/resources/vector-stores/index.ts +9 -1
  445. package/src/resources/vector-stores/vector-stores.ts +150 -216
  446. package/src/resources/vector-stores.ts +1 -1
  447. package/src/version.ts +1 -1
  448. package/version.d.ts +1 -1
  449. package/version.d.ts.map +1 -1
  450. package/version.js +1 -1
  451. package/version.js.map +1 -1
  452. package/version.mjs +1 -1
  453. package/version.mjs.map +1 -1
  454. package/resources/agents/agents.d.ts +0 -246
  455. package/resources/agents/agents.d.ts.map +0 -1
  456. package/resources/agents/agents.js.map +0 -1
  457. package/resources/agents/agents.mjs +0 -52
  458. package/resources/agents/agents.mjs.map +0 -1
  459. package/resources/agents/index.d.ts +0 -5
  460. package/resources/agents/index.d.ts.map +0 -1
  461. package/resources/agents/index.js +0 -18
  462. package/resources/agents/index.js.map +0 -1
  463. package/resources/agents/index.mjs +0 -10
  464. package/resources/agents/index.mjs.map +0 -1
  465. package/resources/agents/session.d.ts +0 -98
  466. package/resources/agents/session.d.ts.map +0 -1
  467. package/resources/agents/session.js +0 -42
  468. package/resources/agents/session.js.map +0 -1
  469. package/resources/agents/session.mjs +0 -37
  470. package/resources/agents/session.mjs.map +0 -1
  471. package/resources/agents/steps.d.ts +0 -22
  472. package/resources/agents/steps.d.ts.map +0 -1
  473. package/resources/agents/steps.js.map +0 -1
  474. package/resources/agents/steps.mjs +0 -15
  475. package/resources/agents/steps.mjs.map +0 -1
  476. package/resources/agents/turn.d.ts +0 -461
  477. package/resources/agents/turn.d.ts.map +0 -1
  478. package/resources/agents/turn.js +0 -34
  479. package/resources/agents/turn.js.map +0 -1
  480. package/resources/agents/turn.mjs +0 -29
  481. package/resources/agents/turn.mjs.map +0 -1
  482. package/resources/agents.d.ts +0 -2
  483. package/resources/agents.d.ts.map +0 -1
  484. package/resources/agents.js.map +0 -1
  485. package/resources/agents.mjs +0 -7
  486. package/resources/agents.mjs.map +0 -1
  487. package/resources/benchmarks.d.ts.map +0 -1
  488. package/resources/benchmarks.js.map +0 -1
  489. package/resources/benchmarks.mjs +0 -31
  490. package/resources/benchmarks.mjs.map +0 -1
  491. package/resources/datasets.d.ts.map +0 -1
  492. package/resources/datasets.js.map +0 -1
  493. package/resources/datasets.mjs.map +0 -1
  494. package/resources/eval/eval.d.ts +0 -159
  495. package/resources/eval/eval.d.ts.map +0 -1
  496. package/resources/eval/eval.js.map +0 -1
  497. package/resources/eval/eval.mjs.map +0 -1
  498. package/resources/eval/index.d.ts +0 -3
  499. package/resources/eval/index.d.ts.map +0 -1
  500. package/resources/eval/index.js.map +0 -1
  501. package/resources/eval/index.mjs.map +0 -1
  502. package/resources/eval/jobs.d.ts.map +0 -1
  503. package/resources/eval/jobs.js.map +0 -1
  504. package/resources/eval/jobs.mjs.map +0 -1
  505. package/resources/eval.d.ts.map +0 -1
  506. package/resources/eval.js.map +0 -1
  507. package/resources/eval.mjs.map +0 -1
  508. package/resources/inference.d.ts +0 -580
  509. package/resources/inference.d.ts.map +0 -1
  510. package/resources/inference.js +0 -54
  511. package/resources/inference.js.map +0 -1
  512. package/resources/inference.mjs +0 -49
  513. package/resources/inference.mjs.map +0 -1
  514. package/resources/post-training/index.d.ts.map +0 -1
  515. package/resources/post-training/index.js.map +0 -1
  516. package/resources/post-training/index.mjs.map +0 -1
  517. package/resources/post-training/job.d.ts +0 -205
  518. package/resources/post-training/job.d.ts.map +0 -1
  519. package/resources/post-training/job.js.map +0 -1
  520. package/resources/post-training/job.mjs.map +0 -1
  521. package/resources/post-training/post-training.d.ts +0 -402
  522. package/resources/post-training/post-training.d.ts.map +0 -1
  523. package/resources/post-training/post-training.js.map +0 -1
  524. package/resources/post-training/post-training.mjs.map +0 -1
  525. package/resources/post-training.d.ts.map +0 -1
  526. package/resources/post-training.js.map +0 -1
  527. package/resources/post-training.mjs.map +0 -1
  528. package/resources/synthetic-data-generation.d.ts +0 -47
  529. package/resources/synthetic-data-generation.d.ts.map +0 -1
  530. package/resources/synthetic-data-generation.js +0 -20
  531. package/resources/synthetic-data-generation.js.map +0 -1
  532. package/resources/synthetic-data-generation.mjs +0 -15
  533. package/resources/synthetic-data-generation.mjs.map +0 -1
  534. package/resources/telemetry.d.ts +0 -519
  535. package/resources/telemetry.d.ts.map +0 -1
  536. package/resources/telemetry.js +0 -70
  537. package/resources/telemetry.js.map +0 -1
  538. package/resources/telemetry.mjs +0 -65
  539. package/resources/telemetry.mjs.map +0 -1
  540. package/resources/tool-runtime/index.d.ts +0 -3
  541. package/resources/tool-runtime/index.d.ts.map +0 -1
  542. package/resources/tool-runtime/index.js +0 -14
  543. package/resources/tool-runtime/index.js.map +0 -1
  544. package/resources/tool-runtime/index.mjs +0 -8
  545. package/resources/tool-runtime/index.mjs.map +0 -1
  546. package/resources/tool-runtime/rag-tool.d.ts +0 -45
  547. package/resources/tool-runtime/rag-tool.d.ts.map +0 -1
  548. package/resources/tool-runtime/rag-tool.js +0 -30
  549. package/resources/tool-runtime/rag-tool.js.map +0 -1
  550. package/resources/tool-runtime/rag-tool.mjs +0 -25
  551. package/resources/tool-runtime/rag-tool.mjs.map +0 -1
  552. package/resources/tool-runtime/tool-runtime.d.ts +0 -132
  553. package/resources/tool-runtime/tool-runtime.d.ts.map +0 -1
  554. package/resources/tool-runtime/tool-runtime.js.map +0 -1
  555. package/resources/tool-runtime/tool-runtime.mjs +0 -29
  556. package/resources/tool-runtime/tool-runtime.mjs.map +0 -1
  557. package/resources/vector-dbs.d.ts +0 -128
  558. package/resources/vector-dbs.d.ts.map +0 -1
  559. package/resources/vector-dbs.js +0 -41
  560. package/resources/vector-dbs.js.map +0 -1
  561. package/resources/vector-dbs.mjs +0 -36
  562. package/resources/vector-dbs.mjs.map +0 -1
  563. package/src/resources/agents/agents.ts +0 -372
  564. package/src/resources/agents/index.ts +0 -44
  565. package/src/resources/agents/session.ts +0 -169
  566. package/src/resources/agents/steps.ts +0 -47
  567. package/src/resources/agents/turn.ts +0 -638
  568. package/src/resources/inference.ts +0 -768
  569. package/src/resources/post-training/job.ts +0 -274
  570. package/src/resources/post-training/post-training.ts +0 -516
  571. package/src/resources/synthetic-data-generation.ts +0 -65
  572. package/src/resources/telemetry.ts +0 -692
  573. package/src/resources/tool-runtime/index.ts +0 -17
  574. package/src/resources/tool-runtime/rag-tool.ts +0 -69
  575. package/src/resources/tool-runtime/tool-runtime.ts +0 -191
  576. package/src/resources/vector-dbs.ts +0 -191
  577. /package/resources/{eval.d.ts → alpha/eval.d.ts} +0 -0
  578. /package/resources/{eval.mjs → alpha/eval.mjs} +0 -0
  579. /package/resources/{post-training → alpha/post-training}/index.d.ts +0 -0
  580. /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
- Object.defineProperty(exports, "__esModule", { value: true });
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 Steps extends resource_1.APIResource {
12
+ class Inference extends resource_1.APIResource {
12
13
  /**
13
- * Retrieve an agent step by its ID.
14
+ * Rerank a list of documents based on their relevance to a query.
14
15
  */
15
- retrieve(agentId, sessionId, turnId, stepId, options) {
16
- return this._client.get(`/v1/agents/${agentId}/session/${sessionId}/turn/${turnId}/step/${stepId}`, options);
16
+ rerank(body, options) {
17
+ return this._client.post('/v1alpha/inference/rerank', { body, ...options })._thenUnwrap((obj) => obj.data);
17
18
  }
18
19
  }
19
- exports.Steps = Steps;
20
- //# sourceMappingURL=steps.js.map
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
- // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
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('/v1/post-training/jobs', options)._thenUnwrap((obj) => obj.data);
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('/v1/post-training/job/artifacts', { query, ...options });
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('/v1/post-training/job/cancel', {
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('/v1/post-training/job/status', { query, ...options });
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
- import { APIResource } from "../../resource.mjs";
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('/v1/post-training/jobs', options)._thenUnwrap((obj) => obj.data);
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('/v1/post-training/job/artifacts', { query, ...options });
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('/v1/post-training/job/cancel', {
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('/v1/post-training/job/status', { query, ...options });
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"}