llama-stack-client 0.0.35 → 0.1.0

package/resources/models.d.ts

@@ -1,46 +1,31 @@
 import { APIResource } from "../resource.js";
 import * as Core from "../core.js";
-import * as ModelsAPI from "./models.js";
 export declare class Models extends APIResource {
-    list(params?: ModelListParams, options?: Core.RequestOptions): Core.APIPromise<ModelServingSpec>;
-    list(options?: Core.RequestOptions): Core.APIPromise<ModelServingSpec>;
-    get(params: ModelGetParams, options?: Core.RequestOptions): Core.APIPromise<ModelServingSpec | null>;
+    retrieve(modelId: string, options?: Core.RequestOptions): Core.APIPromise<Model | null>;
+    list(options?: Core.RequestOptions): Core.APIPromise<ModelListResponse>;
+    register(body: ModelRegisterParams, options?: Core.RequestOptions): Core.APIPromise<Model>;
+    unregister(modelId: string, options?: Core.RequestOptions): Core.APIPromise<void>;
 }
-export interface ModelServingSpec {
-    /**
-     * The model family and SKU of the model along with other parameters corresponding
-     * to the model.
-     */
-    llama_model: unknown;
-    provider_config: ModelServingSpec.ProviderConfig;
+export interface ListModelsResponse {
+    data: Array<Model>;
 }
-export declare namespace ModelServingSpec {
-    interface ProviderConfig {
-        config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-        provider_id: string;
-    }
+export interface Model {
+    identifier: string;
+    metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    model_type: 'llm' | 'embedding';
+    provider_id: string;
+    provider_resource_id: string;
+    type: 'model';
 }
-export interface ModelListParams {
-    /**
-     * JSON-encoded provider data which will be made available to the adapter servicing
-     * the API
-     */
-    'X-LlamaStack-ProviderData'?: string;
-}
-export interface ModelGetParams {
-    /**
-     * Query param:
-     */
-    core_model_id: string;
-    /**
-     * Header param: JSON-encoded provider data which will be made available to the
-     * adapter servicing the API
-     */
-    'X-LlamaStack-ProviderData'?: string;
+export type ModelListResponse = Array<Model>;
+export interface ModelRegisterParams {
+    model_id: string;
+    metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    model_type?: 'llm' | 'embedding';
+    provider_id?: string;
+    provider_model_id?: string;
 }
 export declare namespace Models {
-    export import ModelServingSpec = ModelsAPI.ModelServingSpec;
-    export import ModelListParams = ModelsAPI.ModelListParams;
-    export import ModelGetParams = ModelsAPI.ModelGetParams;
+    export { type ListModelsResponse as ListModelsResponse, type Model as Model, type ModelListResponse as ModelListResponse, type ModelRegisterParams as ModelRegisterParams, };
 }
 //# sourceMappingURL=models.d.ts.map

package/resources/models.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/resources/models.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAChC,OAAO,KAAK,SAAS,MAAM,UAAU,CAAC;AAEtC,qBAAa,MAAO,SAAQ,WAAW;IACrC,IAAI,CAAC,MAAM,CAAC,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,gBAAgB,CAAC;IAChG,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,gBAAgB,CAAC;IAoBtE,GAAG,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,gBAAgB,GAAG,IAAI,CAAC;CAarG;AAED,MAAM,WAAW,gBAAgB;IAC/B;;;OAGG;IACH,WAAW,EAAE,OAAO,CAAC;IAErB,eAAe,EAAE,gBAAgB,CAAC,cAAc,CAAC;CAClD;AAED,yBAAiB,gBAAgB,CAAC;IAChC,UAAiB,cAAc;QAC7B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;QAEpF,WAAW,EAAE,MAAM,CAAC;KACrB;CACF;AAED,MAAM,WAAW,eAAe;IAC9B;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,aAAa,EAAE,MAAM,CAAC;IAEtB;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,yBAAiB,MAAM,CAAC;IACtB,MAAM,QAAQ,gBAAgB,GAAG,SAAS,CAAC,gBAAgB,CAAC;IAC5D,MAAM,QAAQ,eAAe,GAAG,SAAS,CAAC,eAAe,CAAC;IAC1D,MAAM,QAAQ,cAAc,GAAG,SAAS,CAAC,cAAc,CAAC;CACzD"}
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/resources/models.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAEhC,qBAAa,MAAO,SAAQ,WAAW;IACrC,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,GAAG,IAAI,CAAC;IAIvF,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC;IAMvE,QAAQ,CAAC,IAAI,EAAE,mBAAmB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;IAI1F,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;CAMlF;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;CACpB;AAED,MAAM,WAAW,KAAK;IACpB,UAAU,EAAE,MAAM,CAAC;IAEnB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAEtF,UAAU,EAAE,KAAK,GAAG,WAAW,CAAC;IAEhC,WAAW,EAAE,MAAM,CAAC;IAEpB,oBAAoB,EAAE,MAAM,CAAC;IAE7B,IAAI,EAAE,OAAO,CAAC;CACf;AAED,MAAM,MAAM,iBAAiB,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;AAE7C,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,MAAM,CAAC;IAEjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAEvF,UAAU,CAAC,EAAE,KAAK,GAAG,WAAW,CAAC;IAEjC,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,CAAC,OAAO,WAAW,MAAM,CAAC;IAC9B,OAAO,EACL,KAAK,kBAAkB,IAAI,kBAAkB,EAC7C,KAAK,KAAK,IAAI,KAAK,EACnB,KAAK,iBAAiB,IAAI,iBAAiB,EAC3C,KAAK,mBAAmB,IAAI,mBAAmB,GAChD,CAAC;CACH"}

package/resources/models.js

@@ -3,38 +3,22 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Models = void 0;
 const resource_1 = require("../resource.js");
-const core_1 = require("../core.js");
 class Models extends resource_1.APIResource {
-    list(params = {}, options) {
-        if ((0, core_1.isRequestOptions)(params)) {
-            return this.list({}, params);
-        }
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData } = params;
-        return this._client.get('/models/list', {
-            ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
-        });
+    retrieve(modelId, options) {
+        return this._client.get(`/v1/models/${modelId}`, options);
+    }
+    list(options) {
+        return this._client.get('/v1/models', options)._thenUnwrap((obj) => obj.data);
+    }
+    register(body, options) {
+        return this._client.post('/v1/models', { body, ...options });
     }
-    get(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-        return this._client.get('/models/get', {
-            query,
+    unregister(modelId, options) {
+        return this._client.delete(`/v1/models/${modelId}`, {
             ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
+            headers: { Accept: '*/*', ...options?.headers },
         });
     }
 }
 exports.Models = Models;
-(function (Models) {
-})(Models = exports.Models || (exports.Models = {}));
 //# sourceMappingURL=models.js.map

package/resources/models.js.map

@@ -1 +1 @@
-{"version":3,"file":"models.js","sourceRoot":"","sources":["../src/resources/models.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAC1C,qCAA2C;AAI3C,MAAa,MAAO,SAAQ,sBAAW;IAGrC,IAAI,CACF,SAAgD,EAAE,EAClD,OAA6B;QAE7B,IAAI,IAAA,uBAAgB,EAAC,MAAM,CAAC,EAAE;YAC5B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;SAC9B;QACD,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,MAAM,CAAC;QACxE,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE;YACtC,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;IAED,GAAG,CAAC,MAAsB,EAAE,OAA6B;QACvD,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE;YACrC,KAAK;YACL,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAnCD,wBAmCC;AAyCD,WAAiB,MAAM;AAIvB,CAAC,EAJgB,MAAM,GAAN,cAAM,KAAN,cAAM,QAItB"}
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../src/resources/models.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAG1C,MAAa,MAAO,SAAQ,sBAAW;IACrC,QAAQ,CAAC,OAAe,EAAE,OAA6B;QACrD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,OAAO,CACvC,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ,CAAC,IAAyB,EAAE,OAA6B;QAC/D,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,UAAU,CAAC,OAAe,EAAE,OAA6B;QACvD,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc,OAAO,EAAE,EAAE;YAClD,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;CACF;AArBD,wBAqBC"}

package/resources/models.mjs

@@ -1,36 +1,20 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 import { APIResource } from "../resource.mjs";
-import { isRequestOptions } from "../core.mjs";
 export class Models extends APIResource {
-    list(params = {}, options) {
-        if (isRequestOptions(params)) {
-            return this.list({}, params);
-        }
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData } = params;
-        return this._client.get('/models/list', {
-            ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
-        });
+    retrieve(modelId, options) {
+        return this._client.get(`/v1/models/${modelId}`, options);
+    }
+    list(options) {
+        return this._client.get('/v1/models', options)._thenUnwrap((obj) => obj.data);
+    }
+    register(body, options) {
+        return this._client.post('/v1/models', { body, ...options });
     }
-    get(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-        return this._client.get('/models/get', {
-            query,
+    unregister(modelId, options) {
+        return this._client.delete(`/v1/models/${modelId}`, {
             ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
+            headers: { Accept: '*/*', ...options?.headers },
         });
     }
 }
-(function (Models) {
-})(Models || (Models = {}));
 //# sourceMappingURL=models.mjs.map

package/resources/models.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"models.mjs","sourceRoot":"","sources":["../src/resources/models.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;OACf,EAAE,gBAAgB,EAAE;AAI3B,MAAM,OAAO,MAAO,SAAQ,WAAW;IAGrC,IAAI,CACF,SAAgD,EAAE,EAClD,OAA6B;QAE7B,IAAI,gBAAgB,CAAC,MAAM,CAAC,EAAE;YAC5B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;SAC9B;QACD,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,MAAM,CAAC;QACxE,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE;YACtC,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;IAED,GAAG,CAAC,MAAsB,EAAE,OAA6B;QACvD,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE;YACrC,KAAK;YACL,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAyCD,WAAiB,MAAM;AAIvB,CAAC,EAJgB,MAAM,KAAN,MAAM,QAItB"}
+{"version":3,"file":"models.mjs","sourceRoot":"","sources":["../src/resources/models.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAGtB,MAAM,OAAO,MAAO,SAAQ,WAAW;IACrC,QAAQ,CAAC,OAAe,EAAE,OAA6B;QACrD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,OAAO,CACvC,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ,CAAC,IAAyB,EAAE,OAA6B;QAC/D,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,UAAU,CAAC,OAAe,EAAE,OAA6B;QACvD,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc,OAAO,EAAE,EAAE;YAClD,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;CACF"}

package/resources/post-training/index.d.ts

@@ -1,3 +1,3 @@
-export { PostTrainingJob, PostTrainingPreferenceOptimizeParams, PostTrainingSupervisedFineTuneParams, PostTraining, } from "./post-training.js";
-export { PostTrainingJobArtifacts, PostTrainingJobLogStream, PostTrainingJobStatus, JobListParams, JobArtifactsParams, JobCancelParams, JobLogsParams, JobStatusParams, Jobs, } from "./jobs.js";
+export { Job, type JobListResponse, type JobArtifactsResponse, type JobStatusResponse, type JobArtifactsParams, type JobCancelParams, type JobStatusParams, } from "./job.js";
+export { PostTraining, type AlgorithmConfig, type ListPostTrainingJobsResponse, type PostTrainingJob, type PostTrainingPreferenceOptimizeParams, type PostTrainingSupervisedFineTuneParams, } from "./post-training.js";
 //# sourceMappingURL=index.d.ts.map

package/resources/post-training/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/resources/post-training/index.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,eAAe,EACf,oCAAoC,EACpC,oCAAoC,EACpC,YAAY,GACb,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,wBAAwB,EACxB,wBAAwB,EACxB,qBAAqB,EACrB,aAAa,EACb,kBAAkB,EAClB,eAAe,EACf,aAAa,EACb,eAAe,EACf,IAAI,GACL,MAAM,QAAQ,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/resources/post-training/index.ts"],"names":[],"mappings":"AAEA,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"}

package/resources/post-training/index.js

@@ -1,9 +1,9 @@
 "use strict";
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Jobs = exports.PostTraining = void 0;
+exports.PostTraining = exports.Job = void 0;
+var job_1 = require("./job.js");
+Object.defineProperty(exports, "Job", { enumerable: true, get: function () { return job_1.Job; } });
 var post_training_1 = require("./post-training.js");
 Object.defineProperty(exports, "PostTraining", { enumerable: true, get: function () { return post_training_1.PostTraining; } });
-var jobs_1 = require("./jobs.js");
-Object.defineProperty(exports, "Jobs", { enumerable: true, get: function () { return jobs_1.Jobs; } });
 //# sourceMappingURL=index.js.map

package/resources/post-training/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/resources/post-training/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,oDAKyB;AADvB,6GAAA,YAAY,OAAA;AAEd,kCAUgB;AADd,4FAAA,IAAI,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/resources/post-training/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gCAQe;AAPb,0FAAA,GAAG,OAAA;AAQL,oDAOyB;AANvB,6GAAA,YAAY,OAAA"}

package/resources/post-training/index.mjs

@@ -1,4 +1,4 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+export { Job, } from "./job.mjs";
 export { PostTraining, } from "./post-training.mjs";
-export { Jobs, } from "./jobs.mjs";
 //# sourceMappingURL=index.mjs.map

package/resources/post-training/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../src/resources/post-training/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAIL,YAAY,GACb;OACM,EASL,IAAI,GACL"}
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../src/resources/post-training/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EACL,GAAG,GAOJ;OACM,EACL,YAAY,GAMb"}

package/resources/post-training/job.d.ts

@@ -0,0 +1,40 @@
+import { APIResource } from "../../resource.js";
+import * as Core from "../../core.js";
+export declare class Job extends APIResource {
+    list(options?: Core.RequestOptions): Core.APIPromise<JobListResponse>;
+    artifacts(query: JobArtifactsParams, options?: Core.RequestOptions): Core.APIPromise<JobArtifactsResponse | null>;
+    cancel(body: JobCancelParams, options?: Core.RequestOptions): Core.APIPromise<void>;
+    status(query: JobStatusParams, options?: Core.RequestOptions): Core.APIPromise<JobStatusResponse | null>;
+}
+export type JobListResponse = Array<JobListResponse.JobListResponseItem>;
+export declare namespace JobListResponse {
+    interface JobListResponseItem {
+        job_uuid: string;
+    }
+}
+export interface JobArtifactsResponse {
+    checkpoints: Array<unknown>;
+    job_uuid: string;
+}
+export interface JobStatusResponse {
+    checkpoints: Array<unknown>;
+    job_uuid: string;
+    status: 'completed' | 'in_progress' | 'failed' | 'scheduled';
+    completed_at?: string;
+    resources_allocated?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    scheduled_at?: string;
+    started_at?: string;
+}
+export interface JobArtifactsParams {
+    job_uuid: string;
+}
+export interface JobCancelParams {
+    job_uuid: string;
+}
+export interface JobStatusParams {
+    job_uuid: string;
+}
+export declare namespace Job {
+    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, };
+}
+//# sourceMappingURL=job.d.ts.map

package/resources/post-training/job.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"job.d.ts","sourceRoot":"","sources":["../../src/resources/post-training/job.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,IAAI,MAAM,YAAY,CAAC;AAEnC,qBAAa,GAAI,SAAQ,WAAW;IAClC,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;IAMrE,SAAS,CACP,KAAK,EAAE,kBAAkB,EACzB,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,oBAAoB,GAAG,IAAI,CAAC;IAI/C,MAAM,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAQnF,MAAM,CAAC,KAAK,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,iBAAiB,GAAG,IAAI,CAAC;CAGzG;AAED,MAAM,MAAM,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;AAEzE,yBAAiB,eAAe,CAAC;IAC/B,UAAiB,mBAAmB;QAClC,QAAQ,EAAE,MAAM,CAAC;KAClB;CACF;AAED,MAAM,WAAW,oBAAoB;IACnC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAE5B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,iBAAiB;IAChC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAE5B,QAAQ,EAAE,MAAM,CAAC;IAEjB,MAAM,EAAE,WAAW,GAAG,aAAa,GAAG,QAAQ,GAAG,WAAW,CAAC;IAE7D,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB,mBAAmB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAElG,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;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"}

package/resources/post-training/job.js

@@ -0,0 +1,25 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Job = void 0;
+const resource_1 = require("../../resource.js");
+class Job extends resource_1.APIResource {
+    list(options) {
+        return this._client.get('/v1/post-training/jobs', options)._thenUnwrap((obj) => obj.data);
+    }
+    artifacts(query, options) {
+        return this._client.get('/v1/post-training/job/artifacts', { query, ...options });
+    }
+    cancel(body, options) {
+        return this._client.post('/v1/post-training/job/cancel', {
+            body,
+            ...options,
+            headers: { Accept: '*/*', ...options?.headers },
+        });
+    }
+    status(query, options) {
+        return this._client.get('/v1/post-training/job/status', { query, ...options });
+    }
+}
+exports.Job = Job;
+//# sourceMappingURL=job.js.map

package/resources/post-training/job.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"job.js","sourceRoot":"","sources":["../../src/resources/post-training/job.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gDAA6C;AAG7C,MAAa,GAAI,SAAQ,sBAAW;IAClC,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,OAAO,CACnD,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,SAAS,CACP,KAAyB,EACzB,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACpF,CAAC;IAED,MAAM,CAAC,IAAqB,EAAE,OAA6B;QACzD,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,8BAA8B,EAAE;YACvD,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,KAAsB,EAAE,OAA6B;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,8BAA8B,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACjF,CAAC;CACF;AAzBD,kBAyBC"}

package/resources/post-training/job.mjs

@@ -0,0 +1,21 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+import { APIResource } from "../../resource.mjs";
+export class Job extends APIResource {
+    list(options) {
+        return this._client.get('/v1/post-training/jobs', options)._thenUnwrap((obj) => obj.data);
+    }
+    artifacts(query, options) {
+        return this._client.get('/v1/post-training/job/artifacts', { query, ...options });
+    }
+    cancel(body, options) {
+        return this._client.post('/v1/post-training/job/cancel', {
+            body,
+            ...options,
+            headers: { Accept: '*/*', ...options?.headers },
+        });
+    }
+    status(query, options) {
+        return this._client.get('/v1/post-training/job/status', { query, ...options });
+    }
+}
+//# sourceMappingURL=job.mjs.map

package/resources/post-training/job.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"job.mjs","sourceRoot":"","sources":["../../src/resources/post-training/job.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAGtB,MAAM,OAAO,GAAI,SAAQ,WAAW;IAClC,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,OAAO,CACnD,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,SAAS,CACP,KAAyB,EACzB,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACpF,CAAC;IAED,MAAM,CAAC,IAAqB,EAAE,OAA6B;QACzD,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,8BAA8B,EAAE;YACvD,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,KAAsB,EAAE,OAA6B;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,8BAA8B,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACjF,CAAC;CACF"}

package/resources/post-training/post-training.d.ts

@@ -1,62 +1,48 @@
 import { APIResource } from "../../resource.js";
 import * as Core from "../../core.js";
-import * as PostTrainingAPI from "./post-training.js";
-import * as DatasetsAPI from "../datasets.js";
-import * as JobsAPI from "./jobs.js";
+import * as JobAPI from "./job.js";
+import { Job, JobArtifactsParams, JobArtifactsResponse, JobCancelParams, JobListResponse, JobStatusParams, JobStatusResponse } from "./job.js";
 export declare class PostTraining extends APIResource {
-    jobs: JobsAPI.Jobs;
-    preferenceOptimize(params: PostTrainingPreferenceOptimizeParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJob>;
-    supervisedFineTune(params: PostTrainingSupervisedFineTuneParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJob>;
+    job: JobAPI.Job;
+    preferenceOptimize(body: PostTrainingPreferenceOptimizeParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJob>;
+    supervisedFineTune(body: PostTrainingSupervisedFineTuneParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJob>;
+}
+export type AlgorithmConfig = AlgorithmConfig.LoraFinetuningConfig | AlgorithmConfig.QatFinetuningConfig;
+export declare namespace AlgorithmConfig {
+    interface LoraFinetuningConfig {
+        alpha: number;
+        apply_lora_to_mlp: boolean;
+        apply_lora_to_output: boolean;
+        lora_attn_modules: Array<string>;
+        rank: number;
+        type: 'LoRA';
+        quantize_base?: boolean;
+        use_dora?: boolean;
+    }
+    interface QatFinetuningConfig {
+        group_size: number;
+        quantizer_name: string;
+        type: 'QAT';
+    }
+}
+export interface ListPostTrainingJobsResponse {
+    data: Array<ListPostTrainingJobsResponse.Data>;
+}
+export declare namespace ListPostTrainingJobsResponse {
+    interface Data {
+        job_uuid: string;
+    }
 }
 export interface PostTrainingJob {
     job_uuid: string;
 }
 export interface PostTrainingPreferenceOptimizeParams {
-    /**
-     * Body param:
-     */
-    algorithm: 'dpo';
-    /**
-     * Body param:
-     */
     algorithm_config: PostTrainingPreferenceOptimizeParams.AlgorithmConfig;
-    /**
-     * Body param:
-     */
-    dataset: DatasetsAPI.TrainEvalDataset;
-    /**
-     * Body param:
-     */
     finetuned_model: string;
-    /**
-     * Body param:
-     */
     hyperparam_search_config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-    /**
-     * Body param:
-     */
     job_uuid: string;
-    /**
-     * Body param:
-     */
     logger_config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-    /**
-     * Body param:
-     */
-    optimizer_config: PostTrainingPreferenceOptimizeParams.OptimizerConfig;
-    /**
-     * Body param:
-     */
     training_config: PostTrainingPreferenceOptimizeParams.TrainingConfig;
-    /**
-     * Body param:
-     */
-    validation_dataset: DatasetsAPI.TrainEvalDataset;
-    /**
-     * Header param: JSON-encoded provider data which will be made available to the
-     * adapter servicing the API
-     */
-    'X-LlamaStack-ProviderData'?: string;
 }
 export declare namespace PostTrainingPreferenceOptimizeParams {
     interface AlgorithmConfig {
@@ -65,119 +51,86 @@ export declare namespace PostTrainingPreferenceOptimizeParams {
         reward_clip: number;
         reward_scale: number;
     }
-    interface OptimizerConfig {
-        lr: number;
-        lr_min: number;
-        optimizer_type: 'adam' | 'adamw' | 'sgd';
-        weight_decay: number;
-    }
     interface TrainingConfig {
-        batch_size: number;
-        enable_activation_checkpointing: boolean;
-        fsdp_cpu_offload: boolean;
-        memory_efficient_fsdp_wrap: boolean;
+        data_config: TrainingConfig.DataConfig;
+        gradient_accumulation_steps: number;
+        max_steps_per_epoch: number;
+        max_validation_steps: number;
         n_epochs: number;
-        n_iters: number;
-        shuffle: boolean;
+        optimizer_config: TrainingConfig.OptimizerConfig;
+        dtype?: string;
+        efficiency_config?: TrainingConfig.EfficiencyConfig;
+    }
+    namespace TrainingConfig {
+        interface DataConfig {
+            batch_size: number;
+            data_format: 'instruct' | 'dialog';
+            dataset_id: string;
+            shuffle: boolean;
+            packed?: boolean;
+            train_on_input?: boolean;
+            validation_dataset_id?: string;
+        }
+        interface OptimizerConfig {
+            lr: number;
+            num_warmup_steps: number;
+            optimizer_type: 'adam' | 'adamw' | 'sgd';
+            weight_decay: number;
+        }
+        interface EfficiencyConfig {
+            enable_activation_checkpointing?: boolean;
+            enable_activation_offloading?: boolean;
+            fsdp_cpu_offload?: boolean;
+            memory_efficient_fsdp_wrap?: boolean;
+        }
     }
 }
 export interface PostTrainingSupervisedFineTuneParams {
-    /**
-     * Body param:
-     */
-    algorithm: 'full' | 'lora' | 'qlora' | 'dora';
-    /**
-     * Body param:
-     */
-    algorithm_config: PostTrainingSupervisedFineTuneParams.LoraFinetuningConfig | PostTrainingSupervisedFineTuneParams.QLoraFinetuningConfig | PostTrainingSupervisedFineTuneParams.DoraFinetuningConfig;
-    /**
-     * Body param:
-     */
-    dataset: DatasetsAPI.TrainEvalDataset;
-    /**
-     * Body param:
-     */
     hyperparam_search_config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-    /**
-     * Body param:
-     */
     job_uuid: string;
-    /**
-     * Body param:
-     */
     logger_config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-    /**
-     * Body param:
-     */
     model: string;
-    /**
-     * Body param:
-     */
-    optimizer_config: PostTrainingSupervisedFineTuneParams.OptimizerConfig;
-    /**
-     * Body param:
-     */
     training_config: PostTrainingSupervisedFineTuneParams.TrainingConfig;
-    /**
-     * Body param:
-     */
-    validation_dataset: DatasetsAPI.TrainEvalDataset;
-    /**
-     * Header param: JSON-encoded provider data which will be made available to the
-     * adapter servicing the API
-     */
-    'X-LlamaStack-ProviderData'?: string;
+    algorithm_config?: AlgorithmConfig;
+    checkpoint_dir?: string;
 }
 export declare namespace PostTrainingSupervisedFineTuneParams {
-    interface LoraFinetuningConfig {
-        alpha: number;
-        apply_lora_to_mlp: boolean;
-        apply_lora_to_output: boolean;
-        lora_attn_modules: Array<string>;
-        rank: number;
-    }
-    interface QLoraFinetuningConfig {
-        alpha: number;
-        apply_lora_to_mlp: boolean;
-        apply_lora_to_output: boolean;
-        lora_attn_modules: Array<string>;
-        rank: number;
-    }
-    interface DoraFinetuningConfig {
-        alpha: number;
-        apply_lora_to_mlp: boolean;
-        apply_lora_to_output: boolean;
-        lora_attn_modules: Array<string>;
-        rank: number;
-    }
-    interface OptimizerConfig {
-        lr: number;
-        lr_min: number;
-        optimizer_type: 'adam' | 'adamw' | 'sgd';
-        weight_decay: number;
-    }
     interface TrainingConfig {
-        batch_size: number;
-        enable_activation_checkpointing: boolean;
-        fsdp_cpu_offload: boolean;
-        memory_efficient_fsdp_wrap: boolean;
+        data_config: TrainingConfig.DataConfig;
+        gradient_accumulation_steps: number;
+        max_steps_per_epoch: number;
+        max_validation_steps: number;
         n_epochs: number;
-        n_iters: number;
-        shuffle: boolean;
+        optimizer_config: TrainingConfig.OptimizerConfig;
+        dtype?: string;
+        efficiency_config?: TrainingConfig.EfficiencyConfig;
+    }
+    namespace TrainingConfig {
+        interface DataConfig {
+            batch_size: number;
+            data_format: 'instruct' | 'dialog';
+            dataset_id: string;
+            shuffle: boolean;
+            packed?: boolean;
+            train_on_input?: boolean;
+            validation_dataset_id?: string;
+        }
+        interface OptimizerConfig {
+            lr: number;
+            num_warmup_steps: number;
+            optimizer_type: 'adam' | 'adamw' | 'sgd';
+            weight_decay: number;
+        }
+        interface EfficiencyConfig {
+            enable_activation_checkpointing?: boolean;
+            enable_activation_offloading?: boolean;
+            fsdp_cpu_offload?: boolean;
+            memory_efficient_fsdp_wrap?: boolean;
+        }
     }
 }
 export declare namespace PostTraining {
-    export import PostTrainingJob = PostTrainingAPI.PostTrainingJob;
-    export import PostTrainingPreferenceOptimizeParams = PostTrainingAPI.PostTrainingPreferenceOptimizeParams;
-    export import PostTrainingSupervisedFineTuneParams = PostTrainingAPI.PostTrainingSupervisedFineTuneParams;
-    export import Jobs = JobsAPI.Jobs;
-    export import PostTrainingJobArtifacts = JobsAPI.PostTrainingJobArtifacts;
-    export import PostTrainingJobLogStream = JobsAPI.PostTrainingJobLogStream;
-    export import PostTrainingJobStatus = JobsAPI.PostTrainingJobStatus;
-    export import JobListParams = JobsAPI.JobListParams;
-    export import JobArtifactsParams = JobsAPI.JobArtifactsParams;
-    export import JobCancelParams = JobsAPI.JobCancelParams;
-    export import JobLogsParams = JobsAPI.JobLogsParams;
-    export import JobStatusParams = JobsAPI.JobStatusParams;
+    export { type AlgorithmConfig as AlgorithmConfig, type ListPostTrainingJobsResponse as ListPostTrainingJobsResponse, type PostTrainingJob as PostTrainingJob, type PostTrainingPreferenceOptimizeParams as PostTrainingPreferenceOptimizeParams, type PostTrainingSupervisedFineTuneParams as PostTrainingSupervisedFineTuneParams, };
+    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, };
 }
 //# sourceMappingURL=post-training.d.ts.map