llama-stack-client 0.0.1-alpha.0

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

@@ -0,0 +1,91 @@
+import { APIResource } from "../../resource.js";
+import * as Core from "../../core.js";
+import * as JobsAPI from "./jobs.js";
+import * as PostTrainingAPI from "./post-training.js";
+export declare class Jobs extends APIResource {
+    list(params?: JobListParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingAPI.PostTrainingJob>;
+    list(options?: Core.RequestOptions): Core.APIPromise<PostTrainingAPI.PostTrainingJob>;
+    artifacts(params: JobArtifactsParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJobArtifacts>;
+    cancel(params: JobCancelParams, options?: Core.RequestOptions): Core.APIPromise<void>;
+    logs(params: JobLogsParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJobLogStream>;
+    status(params: JobStatusParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJobStatus>;
+}
+export interface PostTrainingJobArtifacts {
+    checkpoints: Array<unknown>;
+    job_uuid: string;
+}
+export interface PostTrainingJobLogStream {
+    job_uuid: string;
+    log_lines: Array<string>;
+}
+export interface PostTrainingJobStatus {
+    checkpoints: Array<unknown>;
+    job_uuid: string;
+    status: 'running' | 'completed' | 'failed' | 'scheduled';
+    completed_at?: string;
+    resources_allocated?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    scheduled_at?: string;
+    started_at?: string;
+}
+export interface JobListParams {
+    /**
+     * JSON-encoded provider data which will be made available to the adapter servicing
+     * the API
+     */
+    'X-LlamaStack-ProviderData'?: string;
+}
+export interface JobArtifactsParams {
+    /**
+     * Query param:
+     */
+    job_uuid: string;
+    /**
+     * Header param: JSON-encoded provider data which will be made available to the
+     * adapter servicing the API
+     */
+    'X-LlamaStack-ProviderData'?: string;
+}
+export interface JobCancelParams {
+    /**
+     * Body param:
+     */
+    job_uuid: string;
+    /**
+     * Header param: JSON-encoded provider data which will be made available to the
+     * adapter servicing the API
+     */
+    'X-LlamaStack-ProviderData'?: string;
+}
+export interface JobLogsParams {
+    /**
+     * Query param:
+     */
+    job_uuid: string;
+    /**
+     * Header param: JSON-encoded provider data which will be made available to the
+     * adapter servicing the API
+     */
+    'X-LlamaStack-ProviderData'?: string;
+}
+export interface JobStatusParams {
+    /**
+     * Query param:
+     */
+    job_uuid: string;
+    /**
+     * Header param: JSON-encoded provider data which will be made available to the
+     * adapter servicing the API
+     */
+    'X-LlamaStack-ProviderData'?: string;
+}
+export declare namespace 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;
+}
+//# sourceMappingURL=jobs.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"jobs.d.ts","sourceRoot":"","sources":["../../src/resources/post-training/jobs.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C,OAAO,KAAK,IAAI,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,OAAO,MAAM,QAAQ,CAAC;AAClC,OAAO,KAAK,eAAe,MAAM,iBAAiB,CAAC;AAEnD,qBAAa,IAAK,SAAQ,WAAW;IACnC,IAAI,CACF,MAAM,CAAC,EAAE,aAAa,EACtB,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,eAAe,CAAC;IACnD,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,eAAe,CAAC;IAoBrF,SAAS,CACP,MAAM,EAAE,kBAAkB,EAC1B,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,wBAAwB,CAAC;IAc5C,MAAM,CAAC,MAAM,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAerF,IAAI,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,wBAAwB,CAAC;IAcrG,MAAM,CAAC,MAAM,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC;CAavG;AAED,MAAM,WAAW,wBAAwB;IACvC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAE5B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,wBAAwB;IACvC,QAAQ,EAAE,MAAM,CAAC;IAEjB,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;CAC1B;AAED,MAAM,WAAW,qBAAqB;IACpC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAE5B,QAAQ,EAAE,MAAM,CAAC;IAEjB,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,WAAW,CAAC;IAEzD,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,aAAa;IAC5B;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,MAAM,WAAW,eAAe;IAC9B;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,MAAM,WAAW,eAAe;IAC9B;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,yBAAiB,IAAI,CAAC;IACpB,MAAM,QAAQ,wBAAwB,GAAG,OAAO,CAAC,wBAAwB,CAAC;IAC1E,MAAM,QAAQ,wBAAwB,GAAG,OAAO,CAAC,wBAAwB,CAAC;IAC1E,MAAM,QAAQ,qBAAqB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IACpE,MAAM,QAAQ,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;IACpD,MAAM,QAAQ,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAC9D,MAAM,QAAQ,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;IACxD,MAAM,QAAQ,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;IACpD,MAAM,QAAQ,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;CACzD"}

package/resources/post-training/jobs.js

@@ -0,0 +1,80 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Jobs = void 0;
+const resource_1 = require("../../resource.js");
+const core_1 = require("../../core.js");
+class Jobs 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('/post_training/jobs', {
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    artifacts(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
+        return this._client.get('/post_training/job/artifacts', {
+            query,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    cancel(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
+        return this._client.post('/post_training/job/cancel', {
+            body,
+            ...options,
+            headers: {
+                Accept: '*/*',
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    logs(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
+        return this._client.get('/post_training/job/logs', {
+            query,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    status(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
+        return this._client.get('/post_training/job/status', {
+            query,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+}
+exports.Jobs = Jobs;
+(function (Jobs) {
+})(Jobs = exports.Jobs || (exports.Jobs = {}));
+//# sourceMappingURL=jobs.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"jobs.js","sourceRoot":"","sources":["../../src/resources/post-training/jobs.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gDAA6C;AAC7C,wCAA8C;AAK9C,MAAa,IAAK,SAAQ,sBAAW;IAMnC,IAAI,CACF,SAA8C,EAAE,EAChD,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,qBAAqB,EAAE;YAC7C,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,SAAS,CACP,MAA0B,EAC1B,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,8BAA8B,EAAE;YACtD,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;IAED,MAAM,CAAC,MAAuB,EAAE,OAA6B;QAC3D,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,2BAA2B,EAAE;YACpD,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,MAAM,EAAE,KAAK;gBACb,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,IAAI,CAAC,MAAqB,EAAE,OAA6B;QACvD,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,yBAAyB,EAAE;YACjD,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;IAED,MAAM,CAAC,MAAuB,EAAE,OAA6B;QAC3D,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE;YACnD,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;AApFD,oBAoFC;AA0FD,WAAiB,IAAI;AASrB,CAAC,EATgB,IAAI,GAAJ,YAAI,KAAJ,YAAI,QASpB"}

package/resources/post-training/jobs.mjs

@@ -0,0 +1,76 @@
+// 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 Jobs extends APIResource {
+    list(params = {}, options) {
+        if (isRequestOptions(params)) {
+            return this.list({}, params);
+        }
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData } = params;
+        return this._client.get('/post_training/jobs', {
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    artifacts(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
+        return this._client.get('/post_training/job/artifacts', {
+            query,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    cancel(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
+        return this._client.post('/post_training/job/cancel', {
+            body,
+            ...options,
+            headers: {
+                Accept: '*/*',
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    logs(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
+        return this._client.get('/post_training/job/logs', {
+            query,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    status(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
+        return this._client.get('/post_training/job/status', {
+            query,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+}
+(function (Jobs) {
+})(Jobs || (Jobs = {}));
+//# sourceMappingURL=jobs.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"jobs.mjs","sourceRoot":"","sources":["../../src/resources/post-training/jobs.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;OACf,EAAE,gBAAgB,EAAE;AAK3B,MAAM,OAAO,IAAK,SAAQ,WAAW;IAMnC,IAAI,CACF,SAA8C,EAAE,EAChD,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,qBAAqB,EAAE;YAC7C,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,SAAS,CACP,MAA0B,EAC1B,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,8BAA8B,EAAE;YACtD,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;IAED,MAAM,CAAC,MAAuB,EAAE,OAA6B;QAC3D,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,2BAA2B,EAAE;YACpD,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,MAAM,EAAE,KAAK;gBACb,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,IAAI,CAAC,MAAqB,EAAE,OAA6B;QACvD,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,yBAAyB,EAAE;YACjD,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;IAED,MAAM,CAAC,MAAuB,EAAE,OAA6B;QAC3D,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE;YACnD,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;AA0FD,WAAiB,IAAI;AASrB,CAAC,EATgB,IAAI,KAAJ,IAAI,QASpB"}

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

@@ -0,0 +1,183 @@
+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";
+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>;
+}
+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 {
+        epsilon: number;
+        gamma: number;
+        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;
+        n_epochs: number;
+        n_iters: number;
+        shuffle: 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;
+}
+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;
+        n_epochs: number;
+        n_iters: number;
+        shuffle: 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;
+}
+//# sourceMappingURL=post-training.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"post-training.d.ts","sourceRoot":"","sources":["../../src/resources/post-training/post-training.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,IAAI,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,eAAe,MAAM,iBAAiB,CAAC;AACnD,OAAO,KAAK,WAAW,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,OAAO,MAAM,QAAQ,CAAC;AAElC,qBAAa,YAAa,SAAQ,WAAW;IAC3C,IAAI,EAAE,OAAO,CAAC,IAAI,CAAkC;IAEpD,kBAAkB,CAChB,MAAM,EAAE,oCAAoC,EAC5C,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;IAcnC,kBAAkB,CAChB,MAAM,EAAE,oCAAoC,EAC5C,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;CAapC;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,oCAAoC;IACnD;;OAEG;IACH,SAAS,EAAE,KAAK,CAAC;IAEjB;;OAEG;IACH,gBAAgB,EAAE,oCAAoC,CAAC,eAAe,CAAC;IAEvE;;OAEG;IACH,OAAO,EAAE,WAAW,CAAC,gBAAgB,CAAC;IAEtC;;OAEG;IACH,eAAe,EAAE,MAAM,CAAC;IAExB;;OAEG;IACH,wBAAwB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAEtG;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;OAEG;IACH,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAE3F;;OAEG;IACH,gBAAgB,EAAE,oCAAoC,CAAC,eAAe,CAAC;IAEvE;;OAEG;IACH,eAAe,EAAE,oCAAoC,CAAC,cAAc,CAAC;IAErE;;OAEG;IACH,kBAAkB,EAAE,WAAW,CAAC,gBAAgB,CAAC;IAEjD;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,yBAAiB,oCAAoC,CAAC;IACpD,UAAiB,eAAe;QAC9B,OAAO,EAAE,MAAM,CAAC;QAEhB,KAAK,EAAE,MAAM,CAAC;QAEd,WAAW,EAAE,MAAM,CAAC;QAEpB,YAAY,EAAE,MAAM,CAAC;KACtB;IAED,UAAiB,eAAe;QAC9B,EAAE,EAAE,MAAM,CAAC;QAEX,MAAM,EAAE,MAAM,CAAC;QAEf,cAAc,EAAE,MAAM,GAAG,OAAO,GAAG,KAAK,CAAC;QAEzC,YAAY,EAAE,MAAM,CAAC;KACtB;IAED,UAAiB,cAAc;QAC7B,UAAU,EAAE,MAAM,CAAC;QAEnB,+BAA+B,EAAE,OAAO,CAAC;QAEzC,gBAAgB,EAAE,OAAO,CAAC;QAE1B,0BAA0B,EAAE,OAAO,CAAC;QAEpC,QAAQ,EAAE,MAAM,CAAC;QAEjB,OAAO,EAAE,MAAM,CAAC;QAEhB,OAAO,EAAE,OAAO,CAAC;KAClB;CACF;AAED,MAAM,WAAW,oCAAoC;IACnD;;OAEG;IACH,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,CAAC;IAE9C;;OAEG;IACH,gBAAgB,EACZ,oCAAoC,CAAC,oBAAoB,GACzD,oCAAoC,CAAC,qBAAqB,GAC1D,oCAAoC,CAAC,oBAAoB,CAAC;IAE9D;;OAEG;IACH,OAAO,EAAE,WAAW,CAAC,gBAAgB,CAAC;IAEtC;;OAEG;IACH,wBAAwB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAEtG;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;OAEG;IACH,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAE3F;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,gBAAgB,EAAE,oCAAoC,CAAC,eAAe,CAAC;IAEvE;;OAEG;IACH,eAAe,EAAE,oCAAoC,CAAC,cAAc,CAAC;IAErE;;OAEG;IACH,kBAAkB,EAAE,WAAW,CAAC,gBAAgB,CAAC;IAEjD;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,yBAAiB,oCAAoC,CAAC;IACpD,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;KACd;IAED,UAAiB,qBAAqB;QACpC,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;KACd;IAED,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;KACd;IAED,UAAiB,eAAe;QAC9B,EAAE,EAAE,MAAM,CAAC;QAEX,MAAM,EAAE,MAAM,CAAC;QAEf,cAAc,EAAE,MAAM,GAAG,OAAO,GAAG,KAAK,CAAC;QAEzC,YAAY,EAAE,MAAM,CAAC;KACtB;IAED,UAAiB,cAAc;QAC7B,UAAU,EAAE,MAAM,CAAC;QAEnB,+BAA+B,EAAE,OAAO,CAAC;QAEzC,gBAAgB,EAAE,OAAO,CAAC;QAE1B,0BAA0B,EAAE,OAAO,CAAC;QAEpC,QAAQ,EAAE,MAAM,CAAC;QAEjB,OAAO,EAAE,MAAM,CAAC;QAEhB,OAAO,EAAE,OAAO,CAAC;KAClB;CACF;AAED,yBAAiB,YAAY,CAAC;IAC5B,MAAM,QAAQ,eAAe,GAAG,eAAe,CAAC,eAAe,CAAC;IAChE,MAAM,QAAQ,oCAAoC,GAAG,eAAe,CAAC,oCAAoC,CAAC;IAC1G,MAAM,QAAQ,oCAAoC,GAAG,eAAe,CAAC,oCAAoC,CAAC;IAC1G,MAAM,QAAQ,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAClC,MAAM,QAAQ,wBAAwB,GAAG,OAAO,CAAC,wBAAwB,CAAC;IAC1E,MAAM,QAAQ,wBAAwB,GAAG,OAAO,CAAC,wBAAwB,CAAC;IAC1E,MAAM,QAAQ,qBAAqB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IACpE,MAAM,QAAQ,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;IACpD,MAAM,QAAQ,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAC9D,MAAM,QAAQ,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;IACxD,MAAM,QAAQ,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;IACpD,MAAM,QAAQ,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;CACzD"}

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

@@ -0,0 +1,66 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PostTraining = void 0;
+const resource_1 = require("../../resource.js");
+const JobsAPI = __importStar(require("./jobs.js"));
+class PostTraining extends resource_1.APIResource {
+    constructor() {
+        super(...arguments);
+        this.jobs = new JobsAPI.Jobs(this._client);
+    }
+    preferenceOptimize(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
+        return this._client.post('/post_training/preference_optimize', {
+            body,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    supervisedFineTune(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
+        return this._client.post('/post_training/supervised_fine_tune', {
+            body,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+}
+exports.PostTraining = PostTraining;
+(function (PostTraining) {
+    PostTraining.Jobs = JobsAPI.Jobs;
+})(PostTraining = exports.PostTraining || (exports.PostTraining = {}));
+//# sourceMappingURL=post-training.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"post-training.js","sourceRoot":"","sources":["../../src/resources/post-training/post-training.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;;;;;;;;;;;;;;;;;;;;;AAEtF,gDAA6C;AAI7C,mDAAkC;AAElC,MAAa,YAAa,SAAQ,sBAAW;IAA7C;;QACE,SAAI,GAAiB,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAmCtD,CAAC;IAjCC,kBAAkB,CAChB,MAA4C,EAC5C,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE;YAC7D,IAAI;YACJ,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,kBAAkB,CAChB,MAA4C,EAC5C,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE;YAC9D,IAAI;YACJ,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;AApCD,oCAoCC;AAmOD,WAAiB,YAAY;IAIb,iBAAI,GAAG,OAAO,CAAC,IAAI,CAAC;AASpC,CAAC,EAbgB,YAAY,GAAZ,oBAAY,KAAZ,oBAAY,QAa5B"}

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

@@ -0,0 +1,39 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+import { APIResource } from "../../resource.mjs";
+import * as JobsAPI from "./jobs.mjs";
+export class PostTraining extends APIResource {
+    constructor() {
+        super(...arguments);
+        this.jobs = new JobsAPI.Jobs(this._client);
+    }
+    preferenceOptimize(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
+        return this._client.post('/post_training/preference_optimize', {
+            body,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+    supervisedFineTune(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
+        return this._client.post('/post_training/supervised_fine_tune', {
+            body,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+}
+(function (PostTraining) {
+    PostTraining.Jobs = JobsAPI.Jobs;
+})(PostTraining || (PostTraining = {}));
+//# sourceMappingURL=post-training.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"post-training.mjs","sourceRoot":"","sources":["../../src/resources/post-training/post-training.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;OAIf,KAAK,OAAO;AAEnB,MAAM,OAAO,YAAa,SAAQ,WAAW;IAA7C;;QACE,SAAI,GAAiB,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAmCtD,CAAC;IAjCC,kBAAkB,CAChB,MAA4C,EAC5C,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE;YAC7D,IAAI;YACJ,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,kBAAkB,CAChB,MAA4C,EAC5C,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE;YAC9D,IAAI;YACJ,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;AAmOD,WAAiB,YAAY;IAIb,iBAAI,GAAG,OAAO,CAAC,IAAI,CAAC;AASpC,CAAC,EAbgB,YAAY,KAAZ,YAAY,QAa5B"}

package/resources/reward-scoring.d.ts

@@ -0,0 +1,47 @@
+import { APIResource } from "../resource.js";
+import * as Core from "../core.js";
+import * as RewardScoringAPI from "./reward-scoring.js";
+import * as Shared from "./shared.js";
+export declare class RewardScoringResource extends APIResource {
+    score(params: RewardScoringScoreParams, options?: Core.RequestOptions): Core.APIPromise<RewardScoring>;
+}
+export interface RewardScoring {
+    scored_generations: Array<ScoredDialogGenerations>;
+}
+export interface ScoredDialogGenerations {
+    dialog: Array<Shared.UserMessage | Shared.SystemMessage | Shared.ToolResponseMessage | Shared.CompletionMessage>;
+    scored_generations: Array<ScoredDialogGenerations.ScoredGeneration>;
+}
+export declare namespace ScoredDialogGenerations {
+    interface ScoredGeneration {
+        message: Shared.UserMessage | Shared.SystemMessage | Shared.ToolResponseMessage | Shared.CompletionMessage;
+        score: number;
+    }
+}
+export interface RewardScoringScoreParams {
+    /**
+     * Body param:
+     */
+    dialog_generations: Array<RewardScoringScoreParams.DialogGeneration>;
+    /**
+     * Body param:
+     */
+    model: string;
+    /**
+     * Header param: JSON-encoded provider data which will be made available to the
+     * adapter servicing the API
+     */
+    'X-LlamaStack-ProviderData'?: string;
+}
+export declare namespace RewardScoringScoreParams {
+    interface DialogGeneration {
+        dialog: Array<Shared.UserMessage | Shared.SystemMessage | Shared.ToolResponseMessage | Shared.CompletionMessage>;
+        sampled_generations: Array<Shared.UserMessage | Shared.SystemMessage | Shared.ToolResponseMessage | Shared.CompletionMessage>;
+    }
+}
+export declare namespace RewardScoringResource {
+    export import RewardScoring = RewardScoringAPI.RewardScoring;
+    export import ScoredDialogGenerations = RewardScoringAPI.ScoredDialogGenerations;
+    export import RewardScoringScoreParams = RewardScoringAPI.RewardScoringScoreParams;
+}
+//# sourceMappingURL=reward-scoring.d.ts.map

package/resources/reward-scoring.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reward-scoring.d.ts","sourceRoot":"","sources":["../src/resources/reward-scoring.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAChC,OAAO,KAAK,gBAAgB,MAAM,kBAAkB,CAAC;AACrD,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC;AAEnC,qBAAa,qBAAsB,SAAQ,WAAW;IACpD,KAAK,CAAC,MAAM,EAAE,wBAAwB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC;CAavG;AAED,MAAM,WAAW,aAAa;IAC5B,kBAAkB,EAAE,KAAK,CAAC,uBAAuB,CAAC,CAAC;CACpD;AAED,MAAM,WAAW,uBAAuB;IACtC,MAAM,EAAE,KAAK,CACX,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,aAAa,GAAG,MAAM,CAAC,mBAAmB,GAAG,MAAM,CAAC,iBAAiB,CAClG,CAAC;IAEF,kBAAkB,EAAE,KAAK,CAAC,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;CACrE;AAED,yBAAiB,uBAAuB,CAAC;IACvC,UAAiB,gBAAgB;QAC/B,OAAO,EACH,MAAM,CAAC,WAAW,GAClB,MAAM,CAAC,aAAa,GACpB,MAAM,CAAC,mBAAmB,GAC1B,MAAM,CAAC,iBAAiB,CAAC;QAE7B,KAAK,EAAE,MAAM,CAAC;KACf;CACF;AAED,MAAM,WAAW,wBAAwB;IACvC;;OAEG;IACH,kBAAkB,EAAE,KAAK,CAAC,wBAAwB,CAAC,gBAAgB,CAAC,CAAC;IAErE;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,yBAAiB,wBAAwB,CAAC;IACxC,UAAiB,gBAAgB;QAC/B,MAAM,EAAE,KAAK,CACX,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,aAAa,GAAG,MAAM,CAAC,mBAAmB,GAAG,MAAM,CAAC,iBAAiB,CAClG,CAAC;QAEF,mBAAmB,EAAE,KAAK,CACxB,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,aAAa,GAAG,MAAM,CAAC,mBAAmB,GAAG,MAAM,CAAC,iBAAiB,CAClG,CAAC;KACH;CACF;AAED,yBAAiB,qBAAqB,CAAC;IACrC,MAAM,QAAQ,aAAa,GAAG,gBAAgB,CAAC,aAAa,CAAC;IAC7D,MAAM,QAAQ,uBAAuB,GAAG,gBAAgB,CAAC,uBAAuB,CAAC;IACjF,MAAM,QAAQ,wBAAwB,GAAG,gBAAgB,CAAC,wBAAwB,CAAC;CACpF"}

package/resources/reward-scoring.js

@@ -0,0 +1,24 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RewardScoringResource = void 0;
+const resource_1 = require("../resource.js");
+class RewardScoringResource extends resource_1.APIResource {
+    score(params, options) {
+        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
+        return this._client.post('/reward_scoring/score', {
+            body,
+            ...options,
+            headers: {
+                ...(xLlamaStackProviderData != null ?
+                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+                    : undefined),
+                ...options?.headers,
+            },
+        });
+    }
+}
+exports.RewardScoringResource = RewardScoringResource;
+(function (RewardScoringResource) {
+})(RewardScoringResource = exports.RewardScoringResource || (exports.RewardScoringResource = {}));
+//# sourceMappingURL=reward-scoring.js.map

package/resources/reward-scoring.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reward-scoring.js","sourceRoot":"","sources":["../src/resources/reward-scoring.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAK1C,MAAa,qBAAsB,SAAQ,sBAAW;IACpD,KAAK,CAAC,MAAgC,EAAE,OAA6B;QACnE,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,uBAAuB,EAAE;YAChD,IAAI;YACJ,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;AAdD,sDAcC;AAwDD,WAAiB,qBAAqB;AAItC,CAAC,EAJgB,qBAAqB,GAArB,6BAAqB,KAArB,6BAAqB,QAIrC"}