Merge branch 'pr1573' into api_2

docs/_static/llama-stack-spec.yaml

@@ -1698,7 +1698,7 @@ paths:
           $ref: '#/components/responses/DefaultError'
       tags:
         - Datasets
-      description: Register a new dataset through a file or
+      description: Register a new dataset.
       parameters: []
       requestBody:
         content:
@@ -4974,17 +4974,17 @@ components:
         - scoring_functions
         - metadata
       title: Benchmark
-    DataReference:
+    DataSource:
       oneOf:
-        - $ref: '#/components/schemas/URIDataReference'
-        - $ref: '#/components/schemas/HuggingfaceDataReference'
-        - $ref: '#/components/schemas/RowsDataReference'
+        - $ref: '#/components/schemas/URIDataSource'
+        - $ref: '#/components/schemas/HuggingfaceDataSource'
+        - $ref: '#/components/schemas/RowsDataSource'
       discriminator:
         propertyName: type
         mapping:
-          uri: '#/components/schemas/URIDataReference'
-          huggingface: '#/components/schemas/HuggingfaceDataReference'
-          rows: '#/components/schemas/RowsDataReference'
+          uri: '#/components/schemas/URIDataSource'
+          huggingface: '#/components/schemas/HuggingfaceDataSource'
+          rows: '#/components/schemas/RowsDataSource'
     Dataset:
       type: object
       properties:
@@ -5001,12 +5001,12 @@ components:
         schema:
           type: string
           enum:
-            - jsonl_messages
+            - messages
           title: Schema
           description: >-
             Schema of the dataset. Each type has a different column format.
-        data_reference:
-          $ref: '#/components/schemas/DataReference'
+        data_source:
+          $ref: '#/components/schemas/DataSource'
         metadata:
           type: object
           additionalProperties:
@@ -5024,10 +5024,10 @@ components:
         - provider_id
         - type
         - schema
-        - data_reference
+        - data_source
         - metadata
       title: Dataset
-    HuggingfaceDataReference:
+    HuggingfaceDataSource:
       type: object
       properties:
         type:
@@ -5051,8 +5051,8 @@ components:
         - type
         - dataset_path
         - params
-      title: HuggingfaceDataReference
-    RowsDataReference:
+      title: HuggingfaceDataSource
+    RowsDataSource:
       type: object
       properties:
         type:
@@ -5075,8 +5075,8 @@ components:
       required:
         - type
         - rows
-      title: RowsDataReference
-    URIDataReference:
+      title: RowsDataSource
+    URIDataSource:
       type: object
       properties:
         type:
@@ -5089,7 +5089,7 @@ components:
       required:
         - type
         - uri
-      title: URIDataReference
+      title: URIDataSource
     Model:
       type: object
       properties:
@@ -6472,12 +6472,12 @@ components:
         schema:
           type: string
           enum:
-            - jsonl_messages
+            - messages
           description: >-
-            The schema format of the dataset. One of - jsonl_messages: The dataset
-            is a JSONL file with messages in column format
-        data_reference:
-          $ref: '#/components/schemas/DataReference'
+            The schema format of the dataset. One of - messages: The dataset contains
+            a messages column with list of messages for post-training.
+        data_source:
+          $ref: '#/components/schemas/DataSource'
           description: >-
             The data reference of the dataset. Examples: - { "type": "uri", "uri":
             "https://mywebsite.com/mydata.jsonl" } - { "type": "uri", "uri": "lsfs://mydata.jsonl"
@@ -6504,7 +6504,7 @@ components:
       additionalProperties: false
       required:
         - schema
-        - data_reference
+        - data_source
       title: RegisterDatasetRequest
     RegisterModelRequest:
       type: object