diff --git a/docs/_static/llama-stack-spec.html b/docs/_static/llama-stack-spec.html
index 856c6e715..47f48df3c 100644
--- a/docs/_static/llama-stack-spec.html
+++ b/docs/_static/llama-stack-spec.html
@@ -6855,7 +6855,7 @@
                         "title": "DatasetPurpose",
                         "description": "Purpose of the dataset. Each type has a different column format."
                     },
-                    "data_source": {
+                    "source": {
                         "$ref": "#/components/schemas/DataSource"
                     },
                     "metadata": {
@@ -6891,7 +6891,7 @@
                     "provider_id",
                     "type",
                     "purpose",
-                    "data_source",
+                    "source",
                     "metadata"
                 ],
                 "title": "Dataset"
diff --git a/docs/_static/llama-stack-spec.yaml b/docs/_static/llama-stack-spec.yaml
index 93ba4ba30..16ef6fed4 100644
--- a/docs/_static/llama-stack-spec.yaml
+++ b/docs/_static/llama-stack-spec.yaml
@@ -4746,7 +4746,7 @@ components:
           title: DatasetPurpose
           description: >-
             Purpose of the dataset. Each type has a different column format.
-        data_source:
+        source:
           $ref: '#/components/schemas/DataSource'
         metadata:
           type: object
@@ -4765,7 +4765,7 @@ components:
         - provider_id
         - type
         - purpose
-        - data_source
+        - source
         - metadata
       title: Dataset
     HuggingfaceDataSource: