llama-stack-client 0.1.6 → 0.1.8

package/resources/tool-runtime.js

@@ -0,0 +1,19 @@
+"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 __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./tool-runtime/index.js"), exports);
+//# sourceMappingURL=tool-runtime.js.map

package/resources/tool-runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-runtime.js","sourceRoot":"","sources":["../src/resources/tool-runtime.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;;;;;;;;;;;AAEtF,0DAAqC"}

package/resources/tool-runtime.mjs

@@ -0,0 +1,3 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+export * from "./tool-runtime/index.mjs";
+//# sourceMappingURL=tool-runtime.mjs.map

package/resources/tool-runtime.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-runtime.mjs","sourceRoot":"","sources":["../src/resources/tool-runtime.ts"],"names":[],"mappings":"AAAA,sFAAsF"}

package/resources/vector-dbs.d.ts

@@ -1,7 +1,7 @@
 import { APIResource } from "../resource.js";
 import * as Core from "../core.js";
 export declare class VectorDBs extends APIResource {
-    retrieve(vectorDBId: string, options?: Core.RequestOptions): Core.APIPromise<VectorDBRetrieveResponse | null>;
+    retrieve(vectorDBId: string, options?: Core.RequestOptions): Core.APIPromise<VectorDBRetrieveResponse>;
     list(options?: Core.RequestOptions): Core.APIPromise<VectorDBListResponse>;
     register(body: VectorDBRegisterParams, options?: Core.RequestOptions): Core.APIPromise<VectorDBRegisterResponse>;
     unregister(vectorDBId: string, options?: Core.RequestOptions): Core.APIPromise<void>;

package/resources/vector-dbs.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vector-dbs.d.ts","sourceRoot":"","sources":["../src/resources/vector-dbs.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAEhC,qBAAa,SAAU,SAAQ,WAAW;IACxC,QAAQ,CACN,UAAU,EAAE,MAAM,EAClB,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,wBAAwB,GAAG,IAAI,CAAC;IAInD,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAAC;IAM1E,QAAQ,CACN,IAAI,EAAE,sBAAsB,EAC5B,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,wBAAwB,CAAC;IAI5C,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;CAMrF;AAED,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,oBAAoB,CAAC;CAC5B;AAED,MAAM,WAAW,wBAAwB;IACvC,mBAAmB,EAAE,MAAM,CAAC;IAE5B,eAAe,EAAE,MAAM,CAAC;IAExB,UAAU,EAAE,MAAM,CAAC;IAEnB,WAAW,EAAE,MAAM,CAAC;IAEpB,oBAAoB,EAAE,MAAM,CAAC;IAE7B,IAAI,EAAE,WAAW,CAAC;CACnB;AAED,MAAM,MAAM,oBAAoB,GAAG,KAAK,CAAC,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;AAExF,yBAAiB,oBAAoB,CAAC;IACpC,UAAiB,wBAAwB;QACvC,mBAAmB,EAAE,MAAM,CAAC;QAE5B,eAAe,EAAE,MAAM,CAAC;QAExB,UAAU,EAAE,MAAM,CAAC;QAEnB,WAAW,EAAE,MAAM,CAAC;QAEpB,oBAAoB,EAAE,MAAM,CAAC;QAE7B,IAAI,EAAE,WAAW,CAAC;KACnB;CACF;AAED,MAAM,WAAW,wBAAwB;IACvC,mBAAmB,EAAE,MAAM,CAAC;IAE5B,eAAe,EAAE,MAAM,CAAC;IAExB,UAAU,EAAE,MAAM,CAAC;IAEnB,WAAW,EAAE,MAAM,CAAC;IAEpB,oBAAoB,EAAE,MAAM,CAAC;IAE7B,IAAI,EAAE,WAAW,CAAC;CACnB;AAED,MAAM,WAAW,sBAAsB;IACrC,eAAe,EAAE,MAAM,CAAC;IAExB,YAAY,EAAE,MAAM,CAAC;IAErB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAE7B,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAChC;AAED,MAAM,CAAC,OAAO,WAAW,SAAS,CAAC;IACjC,OAAO,EACL,KAAK,qBAAqB,IAAI,qBAAqB,EACnD,KAAK,wBAAwB,IAAI,wBAAwB,EACzD,KAAK,oBAAoB,IAAI,oBAAoB,EACjD,KAAK,wBAAwB,IAAI,wBAAwB,EACzD,KAAK,sBAAsB,IAAI,sBAAsB,GACtD,CAAC;CACH"}
+{"version":3,"file":"vector-dbs.d.ts","sourceRoot":"","sources":["../src/resources/vector-dbs.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAEhC,qBAAa,SAAU,SAAQ,WAAW;IACxC,QAAQ,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,wBAAwB,CAAC;IAItG,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAAC;IAM1E,QAAQ,CACN,IAAI,EAAE,sBAAsB,EAC5B,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,wBAAwB,CAAC;IAI5C,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;CAMrF;AAED,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,oBAAoB,CAAC;CAC5B;AAED,MAAM,WAAW,wBAAwB;IACvC,mBAAmB,EAAE,MAAM,CAAC;IAE5B,eAAe,EAAE,MAAM,CAAC;IAExB,UAAU,EAAE,MAAM,CAAC;IAEnB,WAAW,EAAE,MAAM,CAAC;IAEpB,oBAAoB,EAAE,MAAM,CAAC;IAE7B,IAAI,EAAE,WAAW,CAAC;CACnB;AAED,MAAM,MAAM,oBAAoB,GAAG,KAAK,CAAC,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;AAExF,yBAAiB,oBAAoB,CAAC;IACpC,UAAiB,wBAAwB;QACvC,mBAAmB,EAAE,MAAM,CAAC;QAE5B,eAAe,EAAE,MAAM,CAAC;QAExB,UAAU,EAAE,MAAM,CAAC;QAEnB,WAAW,EAAE,MAAM,CAAC;QAEpB,oBAAoB,EAAE,MAAM,CAAC;QAE7B,IAAI,EAAE,WAAW,CAAC;KACnB;CACF;AAED,MAAM,WAAW,wBAAwB;IACvC,mBAAmB,EAAE,MAAM,CAAC;IAE5B,eAAe,EAAE,MAAM,CAAC;IAExB,UAAU,EAAE,MAAM,CAAC;IAEnB,WAAW,EAAE,MAAM,CAAC;IAEpB,oBAAoB,EAAE,MAAM,CAAC;IAE7B,IAAI,EAAE,WAAW,CAAC;CACnB;AAED,MAAM,WAAW,sBAAsB;IACrC,eAAe,EAAE,MAAM,CAAC;IAExB,YAAY,EAAE,MAAM,CAAC;IAErB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAE7B,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAChC;AAED,MAAM,CAAC,OAAO,WAAW,SAAS,CAAC;IACjC,OAAO,EACL,KAAK,qBAAqB,IAAI,qBAAqB,EACnD,KAAK,wBAAwB,IAAI,wBAAwB,EACzD,KAAK,oBAAoB,IAAI,oBAAoB,EACjD,KAAK,wBAAwB,IAAI,wBAAwB,EACzD,KAAK,sBAAsB,IAAI,sBAAsB,GACtD,CAAC;CACH"}

package/resources/vector-dbs.js.map

@@ -1 +1 @@
-{"version":3,"file":"vector-dbs.js","sourceRoot":"","sources":["../src/resources/vector-dbs.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAG1C,MAAa,SAAU,SAAQ,sBAAW;IACxC,QAAQ,CACN,UAAkB,EAClB,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,OAAO,CAC3C,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ,CACN,IAA4B,EAC5B,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,UAAU,CAAC,UAAkB,EAAE,OAA6B;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,kBAAkB,UAAU,EAAE,EAAE;YACzD,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;CACF;AA3BD,8BA2BC"}
+{"version":3,"file":"vector-dbs.js","sourceRoot":"","sources":["../src/resources/vector-dbs.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAG1C,MAAa,SAAU,SAAQ,sBAAW;IACxC,QAAQ,CAAC,UAAkB,EAAE,OAA6B;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,OAAO,CAC3C,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ,CACN,IAA4B,EAC5B,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,UAAU,CAAC,UAAkB,EAAE,OAA6B;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,kBAAkB,UAAU,EAAE,EAAE;YACzD,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;CACF;AAxBD,8BAwBC"}

package/resources/vector-dbs.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"vector-dbs.mjs","sourceRoot":"","sources":["../src/resources/vector-dbs.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAGtB,MAAM,OAAO,SAAU,SAAQ,WAAW;IACxC,QAAQ,CACN,UAAkB,EAClB,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,OAAO,CAC3C,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ,CACN,IAA4B,EAC5B,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,UAAU,CAAC,UAAkB,EAAE,OAA6B;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,kBAAkB,UAAU,EAAE,EAAE;YACzD,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;CACF"}
+{"version":3,"file":"vector-dbs.mjs","sourceRoot":"","sources":["../src/resources/vector-dbs.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAGtB,MAAM,OAAO,SAAU,SAAQ,WAAW;IACxC,QAAQ,CAAC,UAAkB,EAAE,OAA6B;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,OAAO,CAC3C,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ,CACN,IAA4B,EAC5B,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,UAAU,CAAC,UAAkB,EAAE,OAA6B;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,kBAAkB,UAAU,EAAE,EAAE;YACzD,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;CACF"}

package/resources.d.ts

@@ -0,0 +1,2 @@
+export * from "./resources/index.js";
+//# sourceMappingURL=resources.d.ts.map

package/resources.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"resources.d.ts","sourceRoot":"","sources":["src/resources.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC"}

package/resources.js

@@ -0,0 +1,18 @@
+"use strict";
+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 __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./resources/index.js"), exports);
+//# sourceMappingURL=resources.js.map

package/resources.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resources.js","sourceRoot":"","sources":["src/resources.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,uDAAkC"}

package/resources.mjs

@@ -0,0 +1,2 @@
+export * from "./resources/index.mjs";
+//# sourceMappingURL=resources.mjs.map

package/resources.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"resources.mjs","sourceRoot":"","sources":["src/resources.ts"],"names":[],"mappings":""}

package/src/core.ts

@@ -62,8 +62,8 @@ async function defaultParseResponse<T>(props: APIResponseProps): Promise<T> {
   }
 
   const contentType = response.headers.get('content-type');
-  const isJSON =
-    contentType?.includes('application/json') || contentType?.includes('application/vnd.api+json');
+  const mediaType = contentType?.split(';')[0]?.trim();
+  const isJSON = mediaType?.includes('application/json') || mediaType?.endsWith('+json');
   if (isJSON) {
     const json = await response.json();
 
@@ -409,7 +409,7 @@ export abstract class APIClient {
       !headers ? {}
       : Symbol.iterator in headers ?
         Object.fromEntries(Array.from(headers as Iterable<string[]>).map((header) => [...header]))
-      : { ...headers }
+      : { ...(headers as any as Record<string, string>) }
     );
   }
 

package/src/index.ts

@@ -4,6 +4,8 @@ import { type Agent } from './_shims/index';
 import * as qs from './internal/qs';
 import * as Core from './core';
 import * as Errors from './error';
+import * as Pagination from './pagination';
+import { type DatasetsIterrowsParams, DatasetsIterrowsResponse } from './pagination';
 import * as Uploads from './uploads';
 import * as API from './resources/index';
 import {
@@ -20,14 +22,11 @@ import {
   ListBenchmarksResponse,
 } from './resources/benchmarks';
 import {
-  Datasetio,
-  DatasetioAppendRowsParams,
-  DatasetioGetRowsPaginatedParams,
-  PaginatedRowsResult,
-} from './resources/datasetio';
-import {
+  DatasetIterrowsParams,
+  DatasetIterrowsResponse,
   DatasetListResponse,
   DatasetRegisterParams,
+  DatasetRegisterResponse,
   DatasetRetrieveResponse,
   Datasets,
   ListDatasetsResponse,
@@ -286,7 +285,6 @@ export class LlamaStackClient extends Core.APIClient {
   shields: API.Shields = new API.Shields(this);
   syntheticDataGeneration: API.SyntheticDataGeneration = new API.SyntheticDataGeneration(this);
   telemetry: API.Telemetry = new API.Telemetry(this);
-  datasetio: API.Datasetio = new API.Datasetio(this);
   scoring: API.Scoring = new API.Scoring(this);
   scoringFunctions: API.ScoringFunctions = new API.ScoringFunctions(this);
   benchmarks: API.Benchmarks = new API.Benchmarks(this);
@@ -353,13 +351,18 @@ LlamaStackClient.Safety = Safety;
 LlamaStackClient.Shields = Shields;
 LlamaStackClient.SyntheticDataGeneration = SyntheticDataGeneration;
 LlamaStackClient.Telemetry = Telemetry;
-LlamaStackClient.Datasetio = Datasetio;
 LlamaStackClient.Scoring = Scoring;
 LlamaStackClient.ScoringFunctions = ScoringFunctions;
 LlamaStackClient.Benchmarks = Benchmarks;
 export declare namespace LlamaStackClient {
   export type RequestOptions = Core.RequestOptions;
 
+  export import DatasetsIterrows = Pagination.DatasetsIterrows;
+  export {
+    type DatasetsIterrowsParams as DatasetsIterrowsParams,
+    type DatasetsIterrowsResponse as DatasetsIterrowsResponse,
+  };
+
   export {
     Toolgroups as Toolgroups,
     type ListToolGroupsResponse as ListToolGroupsResponse,
@@ -407,6 +410,9 @@ export declare namespace LlamaStackClient {
     type ListDatasetsResponse as ListDatasetsResponse,
     type DatasetRetrieveResponse as DatasetRetrieveResponse,
     type DatasetListResponse as DatasetListResponse,
+    type DatasetIterrowsResponse as DatasetIterrowsResponse,
+    type DatasetRegisterResponse as DatasetRegisterResponse,
+    type DatasetIterrowsParams as DatasetIterrowsParams,
     type DatasetRegisterParams as DatasetRegisterParams,
   };
 
@@ -528,13 +534,6 @@ export declare namespace LlamaStackClient {
     type TelemetrySaveSpansToDatasetParams as TelemetrySaveSpansToDatasetParams,
   };
 
-  export {
-    Datasetio as Datasetio,
-    type PaginatedRowsResult as PaginatedRowsResult,
-    type DatasetioAppendRowsParams as DatasetioAppendRowsParams,
-    type DatasetioGetRowsPaginatedParams as DatasetioGetRowsPaginatedParams,
-  };
-
   export {
     Scoring as Scoring,
     type ScoringScoreResponse as ScoringScoreResponse,

package/src/pagination.ts

@@ -0,0 +1,62 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { AbstractPage, Response, APIClient, FinalRequestOptions, PageInfo } from './core';
+
+export interface DatasetsIterrowsResponse<Item> {
+  data: Array<Item>;
+
+  next_index: number;
+}
+
+export interface DatasetsIterrowsParams {
+  dataset_id?: string;
+
+  start_index?: number;
+
+  limit?: number;
+}
+
+export class DatasetsIterrows<Item> extends AbstractPage<Item> implements DatasetsIterrowsResponse<Item> {
+  data: Array<Item>;
+
+  next_index: number;
+
+  constructor(
+    client: APIClient,
+    response: Response,
+    body: DatasetsIterrowsResponse<Item>,
+    options: FinalRequestOptions,
+  ) {
+    super(client, response, body, options);
+
+    this.data = body.data || [];
+    this.next_index = body.next_index || 0;
+  }
+
+  getPaginatedItems(): Item[] {
+    return this.data ?? [];
+  }
+
+  // @deprecated Please use `nextPageInfo()` instead
+  nextPageParams(): Partial<DatasetsIterrowsParams> | null {
+    const info = this.nextPageInfo();
+    if (!info) return null;
+    if ('params' in info) return info.params;
+    const params = Object.fromEntries(info.url.searchParams);
+    if (!Object.keys(params).length) return null;
+    return params;
+  }
+
+  nextPageInfo(): PageInfo | null {
+    const cursor = this.next_index;
+    if (!cursor) {
+      return null;
+    }
+
+    return {
+      params: {
+        start_index: cursor,
+      },
+    };
+  }
+}

package/src/resources/agents/turn.ts

@@ -482,10 +482,9 @@ export type TurnResumeParams = TurnResumeParamsNonStreaming | TurnResumeParamsSt
 
 export interface TurnResumeParamsBase {
   /**
-   * The tool call responses to resume the turn with. NOTE: ToolResponseMessage will
-   * be deprecated. Use ToolResponse.
+   * The tool call responses to resume the turn with.
    */
-  tool_responses: Array<AgentsAPI.ToolResponse> | Array<Shared.ToolResponseMessage>;
+  tool_responses: Array<AgentsAPI.ToolResponse>;
 
   /**
    * Whether to stream the response.

package/src/resources/agents.ts

@@ -0,0 +1,3 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export * from './agents/index';

package/src/resources/benchmarks.ts

@@ -4,7 +4,7 @@ import { APIResource } from '../resource';
 import * as Core from '../core';
 
 export class Benchmarks extends APIResource {
-  retrieve(benchmarkId: string, options?: Core.RequestOptions): Core.APIPromise<Benchmark | null> {
+  retrieve(benchmarkId: string, options?: Core.RequestOptions): Core.APIPromise<Benchmark> {
     return this._client.get(`/v1/eval/benchmarks/${benchmarkId}`, options);
   }
 

package/src/resources/datasets.ts

@@ -1,14 +1,11 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 import { APIResource } from '../resource';
+import { isRequestOptions } from '../core';
 import * as Core from '../core';
-import * as Shared from './shared';
 
 export class Datasets extends APIResource {
-  retrieve(
-    datasetId: string,
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<DatasetRetrieveResponse | null> {
+  retrieve(datasetId: string, options?: Core.RequestOptions): Core.APIPromise<DatasetRetrieveResponse> {
     return this._client.get(`/v1/datasets/${datasetId}`, options);
   }
 
@@ -18,12 +15,34 @@ export class Datasets extends APIResource {
     )._thenUnwrap((obj) => obj.data);
   }
 
-  register(body: DatasetRegisterParams, options?: Core.RequestOptions): Core.APIPromise<void> {
-    return this._client.post('/v1/datasets', {
-      body,
-      ...options,
-      headers: { Accept: '*/*', ...options?.headers },
-    });
+  /**
+   * Get a paginated list of rows from a dataset. Uses cursor-based pagination.
+   */
+  iterrows(
+    datasetId: string,
+    query?: DatasetIterrowsParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<DatasetIterrowsResponse>;
+  iterrows(datasetId: string, options?: Core.RequestOptions): Core.APIPromise<DatasetIterrowsResponse>;
+  iterrows(
+    datasetId: string,
+    query: DatasetIterrowsParams | Core.RequestOptions = {},
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<DatasetIterrowsResponse> {
+    if (isRequestOptions(query)) {
+      return this.iterrows(datasetId, {}, query);
+    }
+    return this._client.get(`/v1/datasetio/iterrows/${datasetId}`, { query, ...options });
+  }
+
+  /**
+   * Register a new dataset.
+   */
+  register(
+    body: DatasetRegisterParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<DatasetRegisterResponse> {
+    return this._client.post('/v1/datasets', { body, ...options });
   }
 
   unregister(datasetId: string, options?: Core.RequestOptions): Core.APIPromise<void> {
@@ -39,8 +58,6 @@ export interface ListDatasetsResponse {
 }
 
 export interface DatasetRetrieveResponse {
-  dataset_schema: Record<string, Shared.ParamType>;
-
   identifier: string;
 
   metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
@@ -49,23 +66,53 @@ export interface DatasetRetrieveResponse {
 
   provider_resource_id: string;
 
-  type: 'dataset';
+  /**
+   * Purpose of the dataset. Each purpose has a required input data schema.
+   */
+  purpose: 'post-training/messages' | 'eval/question-answer' | 'eval/messages-answer';
+
+  /**
+   * A dataset that can be obtained from a URI.
+   */
+  source: DatasetRetrieveResponse.UriDataSource | DatasetRetrieveResponse.RowsDataSource;
 
-  url: DatasetRetrieveResponse.URL;
+  type: 'dataset';
 }
 
 export namespace DatasetRetrieveResponse {
-  export interface URL {
+  /**
+   * A dataset that can be obtained from a URI.
+   */
+  export interface UriDataSource {
+    type: 'uri';
+
+    /**
+     * The dataset can be obtained from a URI. E.g. -
+     * "https://mywebsite.com/mydata.jsonl" - "lsfs://mydata.jsonl" -
+     * "data:csv;base64,{base64_content}"
+     */
     uri: string;
   }
+
+  /**
+   * A dataset stored in rows.
+   */
+  export interface RowsDataSource {
+    /**
+     * The dataset is stored in rows. E.g. - [ {"messages": [{"role": "user",
+     * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
+     * ]
+     */
+    rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+
+    type: 'rows';
+  }
 }
 
 export type DatasetListResponse = Array<DatasetListResponse.DatasetListResponseItem>;
 
 export namespace DatasetListResponse {
   export interface DatasetListResponseItem {
-    dataset_schema: Record<string, Shared.ParamType>;
-
     identifier: string;
 
     metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
@@ -74,36 +121,197 @@ export namespace DatasetListResponse {
 
     provider_resource_id: string;
 
-    type: 'dataset';
+    /**
+     * Purpose of the dataset. Each purpose has a required input data schema.
+     */
+    purpose: 'post-training/messages' | 'eval/question-answer' | 'eval/messages-answer';
+
+    /**
+     * A dataset that can be obtained from a URI.
+     */
+    source: DatasetListResponseItem.UriDataSource | DatasetListResponseItem.RowsDataSource;
 
-    url: DatasetListResponseItem.URL;
+    type: 'dataset';
   }
 
   export namespace DatasetListResponseItem {
-    export interface URL {
+    /**
+     * A dataset that can be obtained from a URI.
+     */
+    export interface UriDataSource {
+      type: 'uri';
+
+      /**
+       * The dataset can be obtained from a URI. E.g. -
+       * "https://mywebsite.com/mydata.jsonl" - "lsfs://mydata.jsonl" -
+       * "data:csv;base64,{base64_content}"
+       */
       uri: string;
     }
+
+    /**
+     * A dataset stored in rows.
+     */
+    export interface RowsDataSource {
+      /**
+       * The dataset is stored in rows. E.g. - [ {"messages": [{"role": "user",
+       * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
+       * ]
+       */
+      rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+
+      type: 'rows';
+    }
   }
 }
 
-export interface DatasetRegisterParams {
-  dataset_id: string;
+/**
+ * A paginated list of rows from a dataset.
+ */
+export interface DatasetIterrowsResponse {
+  /**
+   * The rows in the current page.
+   */
+  data: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+
+  /**
+   * Index into dataset for the first row in the next page. None if there are no more
+   * rows.
+   */
+  next_start_index?: number;
+}
 
-  dataset_schema: Record<string, Shared.ParamType>;
+export interface DatasetRegisterResponse {
+  identifier: string;
 
-  url: DatasetRegisterParams.URL;
+  metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
 
-  metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  provider_id: string;
+
+  provider_resource_id: string;
+
+  /**
+   * Purpose of the dataset. Each purpose has a required input data schema.
+   */
+  purpose: 'post-training/messages' | 'eval/question-answer' | 'eval/messages-answer';
 
-  provider_dataset_id?: string;
+  /**
+   * A dataset that can be obtained from a URI.
+   */
+  source: DatasetRegisterResponse.UriDataSource | DatasetRegisterResponse.RowsDataSource;
 
-  provider_id?: string;
+  type: 'dataset';
+}
+
+export namespace DatasetRegisterResponse {
+  /**
+   * A dataset that can be obtained from a URI.
+   */
+  export interface UriDataSource {
+    type: 'uri';
+
+    /**
+     * The dataset can be obtained from a URI. E.g. -
+     * "https://mywebsite.com/mydata.jsonl" - "lsfs://mydata.jsonl" -
+     * "data:csv;base64,{base64_content}"
+     */
+    uri: string;
+  }
+
+  /**
+   * A dataset stored in rows.
+   */
+  export interface RowsDataSource {
+    /**
+     * The dataset is stored in rows. E.g. - [ {"messages": [{"role": "user",
+     * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
+     * ]
+     */
+    rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+
+    type: 'rows';
+  }
+}
+
+export interface DatasetIterrowsParams {
+  /**
+   * The number of rows to get.
+   */
+  limit?: number;
+
+  /**
+   * Index into dataset for the first row to get. Get all rows if None.
+   */
+  start_index?: number;
+}
+
+export interface DatasetRegisterParams {
+  /**
+   * The purpose of the dataset. One of - "post-training/messages": The dataset
+   * contains a messages column with list of messages for post-training. {
+   * "messages": [ {"role": "user", "content": "Hello, world!"}, {"role":
+   * "assistant", "content": "Hello, world!"}, ] } - "eval/question-answer": The
+   * dataset contains a question column and an answer column for evaluation. {
+   * "question": "What is the capital of France?", "answer": "Paris" } -
+   * "eval/messages-answer": The dataset contains a messages column with list of
+   * messages and an answer column for evaluation. { "messages": [ {"role": "user",
+   * "content": "Hello, my name is John Doe."}, {"role": "assistant", "content":
+   * "Hello, John Doe. How can I help you today?"}, {"role": "user", "content":
+   * "What's my name?"}, ], "answer": "John Doe" }
+   */
+  purpose: 'post-training/messages' | 'eval/question-answer' | 'eval/messages-answer';
+
+  /**
+   * The data source of the dataset. Ensure that the data source schema is compatible
+   * with the purpose of the dataset. Examples: - { "type": "uri", "uri":
+   * "https://mywebsite.com/mydata.jsonl" } - { "type": "uri", "uri":
+   * "lsfs://mydata.jsonl" } - { "type": "uri", "uri":
+   * "data:csv;base64,{base64_content}" } - { "type": "uri", "uri":
+   * "huggingface://llamastack/simpleqa?split=train" } - { "type": "rows", "rows": [
+   * { "messages": [ {"role": "user", "content": "Hello, world!"}, {"role":
+   * "assistant", "content": "Hello, world!"}, ] } ] }
+   */
+  source: DatasetRegisterParams.UriDataSource | DatasetRegisterParams.RowsDataSource;
+
+  /**
+   * The ID of the dataset. If not provided, an ID will be generated.
+   */
+  dataset_id?: string;
+
+  /**
+   * The metadata for the dataset. - E.g. {"description": "My dataset"}
+   */
+  metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
 }
 
 export namespace DatasetRegisterParams {
-  export interface URL {
+  /**
+   * A dataset that can be obtained from a URI.
+   */
+  export interface UriDataSource {
+    type: 'uri';
+
+    /**
+     * The dataset can be obtained from a URI. E.g. -
+     * "https://mywebsite.com/mydata.jsonl" - "lsfs://mydata.jsonl" -
+     * "data:csv;base64,{base64_content}"
+     */
     uri: string;
   }
+
+  /**
+   * A dataset stored in rows.
+   */
+  export interface RowsDataSource {
+    /**
+     * The dataset is stored in rows. E.g. - [ {"messages": [{"role": "user",
+     * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
+     * ]
+     */
+    rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+
+    type: 'rows';
+  }
 }
 
 export declare namespace Datasets {
@@ -111,6 +319,9 @@ export declare namespace Datasets {
     type ListDatasetsResponse as ListDatasetsResponse,
     type DatasetRetrieveResponse as DatasetRetrieveResponse,
     type DatasetListResponse as DatasetListResponse,
+    type DatasetIterrowsResponse as DatasetIterrowsResponse,
+    type DatasetRegisterResponse as DatasetRegisterResponse,
+    type DatasetIterrowsParams as DatasetIterrowsParams,
     type DatasetRegisterParams as DatasetRegisterParams,
   };
 }

package/src/resources/eval/eval.ts

@@ -5,7 +5,7 @@ import * as Core from '../../core';
 import * as ScoringFunctionsAPI from '../scoring-functions';
 import * as Shared from '../shared';
 import * as JobsAPI from './jobs';
-import { JobStatusResponse, Jobs } from './jobs';
+import { Jobs } from './jobs';
 
 export class Eval extends APIResource {
   jobs: JobsAPI.Jobs = new JobsAPI.Jobs(this._client);
@@ -131,6 +131,8 @@ export interface EvaluateResponse {
 
 export interface Job {
   job_id: string;
+
+  status: 'completed' | 'in_progress' | 'failed' | 'scheduled';
 }
 
 export interface EvalEvaluateRowsParams {
@@ -195,5 +197,5 @@ export declare namespace Eval {
     type EvalRunEvalAlphaParams as EvalRunEvalAlphaParams,
   };
 
-  export { Jobs as Jobs, type JobStatusResponse as JobStatusResponse };
+  export { Jobs as Jobs };
 }