llama-stack-client 0.3.5 → 0.4.0-rc2

package/src/resources/{datasets.ts → beta/datasets.ts}

@@ -3,19 +3,19 @@
 //
 // This source code is licensed under the terms described in the LICENSE file in
 // the root directory of this source tree.
-
+//
 // 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 { APIResource } from '../../resource';
+import { isRequestOptions } from '../../core';
+import * as Core from '../../core';
 
 export class Datasets extends APIResource {
   /**
    * Get a dataset by its ID.
    */
   retrieve(datasetId: string, options?: Core.RequestOptions): Core.APIPromise<DatasetRetrieveResponse> {
-    return this._client.get(`/v1/datasets/${datasetId}`, options);
+    return this._client.get(`/v1beta/datasets/${datasetId}`, options);
   }
 
   /**
@@ -23,7 +23,7 @@ export class Datasets extends APIResource {
    */
   list(options?: Core.RequestOptions): Core.APIPromise<DatasetListResponse> {
     return (
-      this._client.get('/v1/datasets', options) as Core.APIPromise<{ data: DatasetListResponse }>
+      this._client.get('/v1beta/datasets', options) as Core.APIPromise<{ data: DatasetListResponse }>
     )._thenUnwrap((obj) => obj.data);
   }
 
@@ -35,7 +35,7 @@ export class Datasets extends APIResource {
     body: DatasetAppendrowsParams,
     options?: Core.RequestOptions,
   ): Core.APIPromise<void> {
-    return this._client.post(`/v1/datasetio/append-rows/${datasetId}`, {
+    return this._client.post(`/v1beta/datasetio/append-rows/${datasetId}`, {
       body,
       ...options,
       headers: { Accept: '*/*', ...options?.headers },
@@ -43,7 +43,9 @@ export class Datasets extends APIResource {
   }
 
   /**
-   * Get a paginated list of rows from a dataset. Uses offset-based pagination where:
+   * Get a paginated list of rows from a dataset.
+   *
+   * Uses offset-based pagination where:
    *
    * - start_index: The starting index (0-based). If None, starts from beginning.
    * - limit: Number of items to return. If None or -1, returns all items.
@@ -67,24 +69,28 @@ export class Datasets extends APIResource {
     if (isRequestOptions(query)) {
       return this.iterrows(datasetId, {}, query);
     }
-    return this._client.get(`/v1/datasetio/iterrows/${datasetId}`, { query, ...options });
+    return this._client.get(`/v1beta/datasetio/iterrows/${datasetId}`, { query, ...options });
   }
 
   /**
    * Register a new dataset.
+   *
+   * @deprecated
    */
   register(
     body: DatasetRegisterParams,
     options?: Core.RequestOptions,
   ): Core.APIPromise<DatasetRegisterResponse> {
-    return this._client.post('/v1/datasets', { body, ...options });
+    return this._client.post('/v1beta/datasets', { body, ...options });
   }
 
   /**
    * Unregister a dataset by its ID.
+   *
+   * @deprecated
    */
   unregister(datasetId: string, options?: Core.RequestOptions): Core.APIPromise<void> {
-    return this._client.delete(`/v1/datasets/${datasetId}`, {
+    return this._client.delete(`/v1beta/datasets/${datasetId}`, {
       ...options,
       headers: { Accept: '*/*', ...options?.headers },
     });
@@ -105,13 +111,14 @@ export interface ListDatasetsResponse {
  * Dataset resource for storing and accessing training or evaluation data.
  */
 export interface DatasetRetrieveResponse {
+  /**
+   * Unique identifier for this resource in llama stack
+   */
   identifier: string;
 
   /**
-   * Additional metadata for the dataset
+   * ID of the provider that owns this resource
    */
-  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
   provider_id: string;
 
   /**
@@ -125,11 +132,19 @@ export interface DatasetRetrieveResponse {
   source: DatasetRetrieveResponse.UriDataSource | DatasetRetrieveResponse.RowsDataSource;
 
   /**
-   * Type of resource, always 'dataset' for datasets
+   * Any additional metadata for this dataset
    */
-  type: 'dataset';
+  metadata?: { [key: string]: unknown };
 
-  provider_resource_id?: string;
+  /**
+   * Unique identifier for this resource in the provider
+   */
+  provider_resource_id?: string | null;
+
+  /**
+   * Type of resource, always 'dataset' for datasets
+   */
+  type?: 'dataset';
 }
 
 export namespace DatasetRetrieveResponse {
@@ -137,14 +152,17 @@ export namespace DatasetRetrieveResponse {
    * 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" -
+     * 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;
+
+    /**
+     * The type of data source.
+     */
+    type?: 'uri';
   }
 
   /**
@@ -152,13 +170,15 @@ export namespace DatasetRetrieveResponse {
    */
   export interface RowsDataSource {
     /**
-     * The dataset is stored in rows. E.g. - [ {"messages": [{"role": "user",
-     * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
-     * ]
+     * The dataset is stored in rows. E.g. [{"messages": [{"role": "user", "content":
+     * "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}]
      */
-    rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+    rows: Array<{ [key: string]: unknown }>;
 
-    type: 'rows';
+    /**
+     * The type of data source.
+     */
+    type?: 'rows';
   }
 }
 
@@ -172,13 +192,14 @@ export namespace DatasetListResponse {
    * Dataset resource for storing and accessing training or evaluation data.
    */
   export interface DatasetListResponseItem {
+    /**
+     * Unique identifier for this resource in llama stack
+     */
     identifier: string;
 
     /**
-     * Additional metadata for the dataset
+     * ID of the provider that owns this resource
      */
-    metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
     provider_id: string;
 
     /**
@@ -192,11 +213,19 @@ export namespace DatasetListResponse {
     source: DatasetListResponseItem.UriDataSource | DatasetListResponseItem.RowsDataSource;
 
     /**
-     * Type of resource, always 'dataset' for datasets
+     * Any additional metadata for this dataset
+     */
+    metadata?: { [key: string]: unknown };
+
+    /**
+     * Unique identifier for this resource in the provider
      */
-    type: 'dataset';
+    provider_resource_id?: string | null;
 
-    provider_resource_id?: string;
+    /**
+     * Type of resource, always 'dataset' for datasets
+     */
+    type?: 'dataset';
   }
 
   export namespace DatasetListResponseItem {
@@ -204,14 +233,17 @@ export namespace DatasetListResponse {
      * 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" -
+       * 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;
+
+      /**
+       * The type of data source.
+       */
+      type?: 'uri';
     }
 
     /**
@@ -219,13 +251,15 @@ export namespace DatasetListResponse {
      */
     export interface RowsDataSource {
       /**
-       * The dataset is stored in rows. E.g. - [ {"messages": [{"role": "user",
-       * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
-       * ]
+       * The dataset is stored in rows. E.g. [{"messages": [{"role": "user", "content":
+       * "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}]
        */
-      rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+      rows: Array<{ [key: string]: unknown }>;
 
-      type: 'rows';
+      /**
+       * The type of data source.
+       */
+      type?: 'rows';
     }
   }
 }
@@ -234,33 +268,25 @@ export namespace DatasetListResponse {
  * A generic paginated response that follows a simple format.
  */
 export interface DatasetIterrowsResponse {
-  /**
-   * The list of items for the current page
-   */
-  data: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+  data: Array<{ [key: string]: unknown }>;
 
-  /**
-   * Whether there are more items available after this set
-   */
   has_more: boolean;
 
-  /**
-   * The URL for accessing this list
-   */
-  url?: string;
+  url?: string | null;
 }
 
 /**
  * Dataset resource for storing and accessing training or evaluation data.
  */
 export interface DatasetRegisterResponse {
+  /**
+   * Unique identifier for this resource in llama stack
+   */
   identifier: string;
 
   /**
-   * Additional metadata for the dataset
+   * ID of the provider that owns this resource
    */
-  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
   provider_id: string;
 
   /**
@@ -274,11 +300,19 @@ export interface DatasetRegisterResponse {
   source: DatasetRegisterResponse.UriDataSource | DatasetRegisterResponse.RowsDataSource;
 
   /**
-   * Type of resource, always 'dataset' for datasets
+   * Any additional metadata for this dataset
+   */
+  metadata?: { [key: string]: unknown };
+
+  /**
+   * Unique identifier for this resource in the provider
    */
-  type: 'dataset';
+  provider_resource_id?: string | null;
 
-  provider_resource_id?: string;
+  /**
+   * Type of resource, always 'dataset' for datasets
+   */
+  type?: 'dataset';
 }
 
 export namespace DatasetRegisterResponse {
@@ -286,14 +320,17 @@ 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" -
+     * 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;
+
+    /**
+     * The type of data source.
+     */
+    type?: 'uri';
   }
 
   /**
@@ -301,72 +338,48 @@ export namespace DatasetRegisterResponse {
    */
   export interface RowsDataSource {
     /**
-     * The dataset is stored in rows. E.g. - [ {"messages": [{"role": "user",
-     * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
-     * ]
+     * The dataset is stored in rows. E.g. [{"messages": [{"role": "user", "content":
+     * "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}]
      */
-    rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+    rows: Array<{ [key: string]: unknown }>;
 
-    type: 'rows';
+    /**
+     * The type of data source.
+     */
+    type?: 'rows';
   }
 }
 
 export interface DatasetAppendrowsParams {
-  /**
-   * The rows to append to the dataset.
-   */
-  rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+  rows: Array<{ [key: string]: unknown }>;
 }
 
 export interface DatasetIterrowsParams {
-  /**
-   * The number of rows to get.
-   */
-  limit?: number;
+  limit?: number | null;
 
-  /**
-   * Index into dataset for the first row to get. Get all rows if None.
-   */
-  start_index?: number;
+  start_index?: number | null;
 }
 
 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" }
+   * The purpose of the dataset.
    */
   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!"}, ] } ] }
+   * The data source of the dataset.
    */
   source: DatasetRegisterParams.UriDataSource | DatasetRegisterParams.RowsDataSource;
 
   /**
    * The ID of the dataset. If not provided, an ID will be generated.
    */
-  dataset_id?: string;
+  dataset_id?: string | null;
 
   /**
-   * The metadata for the dataset. - E.g. {"description": "My dataset"}.
+   * The metadata for the dataset.
    */
-  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+  metadata?: { [key: string]: unknown } | null;
 }
 
 export namespace DatasetRegisterParams {
@@ -374,14 +387,17 @@ export namespace DatasetRegisterParams {
    * 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" -
+     * 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;
+
+    /**
+     * The type of data source.
+     */
+    type?: 'uri';
   }
 
   /**
@@ -389,13 +405,15 @@ export namespace DatasetRegisterParams {
    */
   export interface RowsDataSource {
     /**
-     * The dataset is stored in rows. E.g. - [ {"messages": [{"role": "user",
-     * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
-     * ]
+     * The dataset is stored in rows. E.g. [{"messages": [{"role": "user", "content":
+     * "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}]
      */
-    rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+    rows: Array<{ [key: string]: unknown }>;
 
-    type: 'rows';
+    /**
+     * The type of data source.
+     */
+    type?: 'rows';
   }
 }
 

package/src/resources/beta/index.ts

@@ -0,0 +1,20 @@
+// Copyright (c) Meta Platforms, Inc. and affiliates.
+// All rights reserved.
+//
+// This source code is licensed under the terms described in the LICENSE file in
+// the root directory of this source tree.
+//
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export { Beta } from './beta';
+export {
+  Datasets,
+  type ListDatasetsResponse,
+  type DatasetRetrieveResponse,
+  type DatasetListResponse,
+  type DatasetIterrowsResponse,
+  type DatasetRegisterResponse,
+  type DatasetAppendrowsParams,
+  type DatasetIterrowsParams,
+  type DatasetRegisterParams,
+} from './datasets';

package/src/resources/{agents.ts → beta.ts}

@@ -3,7 +3,7 @@
 //
 // This source code is licensed under the terms described in the LICENSE file in
 // the root directory of this source tree.
-
+//
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-export * from './agents/index';
+export * from './beta/index';