diff --git a/docs/_static/llama-stack-spec.html b/docs/_static/llama-stack-spec.html
index 5044594f3..740bff6e4 100644
--- a/docs/_static/llama-stack-spec.html
+++ b/docs/_static/llama-stack-spec.html
@@ -9385,7 +9385,7 @@
                         "enum": [
                             "messages"
                         ],
-                        "description": "The schema format of the dataset. One of - jsonl_messages: The dataset is a JSONL file with messages in column format"
+                        "description": "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",
diff --git a/docs/_static/llama-stack-spec.yaml b/docs/_static/llama-stack-spec.yaml
index 692d537c4..c6f3c2327 100644
--- a/docs/_static/llama-stack-spec.yaml
+++ b/docs/_static/llama-stack-spec.yaml
@@ -6348,8 +6348,8 @@ components:
           enum:
             - messages
           description: >-
-            The schema format of the dataset. One of - jsonl_messages: The dataset
-            is a JSONL file with messages in column format
+            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: >-
diff --git a/llama_stack/apis/datasets/datasets.py b/llama_stack/apis/datasets/datasets.py
index 5693c86ed..4b3ce3e6f 100644
--- a/llama_stack/apis/datasets/datasets.py
+++ b/llama_stack/apis/datasets/datasets.py
@@ -108,7 +108,7 @@ class Datasets(Protocol):
         Register a new dataset.
 
         :param schema: The schema format of the dataset. One of
-            - jsonl_messages: The dataset is a JSONL file with messages in column format
+            - messages: The dataset contains a messages column with list of messages for post-training.
         :param data_source: The data source of the dataset. Examples:
            - {
                "type": "uri",