diff --git a/docs/_static/llama-stack-spec.html b/docs/_static/llama-stack-spec.html
index 083678699..462034c3d 100644
--- a/docs/_static/llama-stack-spec.html
+++ b/docs/_static/llama-stack-spec.html
@@ -60,7 +60,7 @@
}
},
"tags": [
- "DatasetIO"
+ "Datasets"
],
"description": "",
"parameters": [
@@ -524,7 +524,7 @@
}
},
"tags": [
- "Files (Coming Soon)"
+ "Files"
],
"description": "List all buckets.",
"parameters": [
@@ -564,7 +564,7 @@
}
},
"tags": [
- "Files (Coming Soon)"
+ "Files"
],
"description": "Create a new upload session for a file identified by a bucket and key.",
"parameters": [],
@@ -791,7 +791,7 @@
}
},
"tags": [
- "Files (Coming Soon)"
+ "Files"
],
"description": "Get a file info identified by a bucket and key.",
"parameters": [
@@ -841,7 +841,7 @@
}
},
"tags": [
- "Files (Coming Soon)"
+ "Files"
],
"description": "Delete a file identified by a bucket and key.",
"parameters": [
@@ -1830,7 +1830,7 @@
}
},
"tags": [
- "Files (Coming Soon)"
+ "Files"
],
"description": "Returns information about an existsing upload session",
"parameters": [
@@ -1878,7 +1878,7 @@
}
},
"tags": [
- "Files (Coming Soon)"
+ "Files"
],
"description": "Upload file content to an existing upload session. On the server, request body will have the raw bytes that are uploaded.",
"parameters": [
@@ -2204,7 +2204,7 @@
}
},
"tags": [
- "DatasetIO"
+ "Datasets"
],
"description": "Get a paginated list of rows from a dataset. Uses cursor-based pagination.",
"parameters": [
@@ -2603,7 +2603,7 @@
}
},
"tags": [
- "Files (Coming Soon)"
+ "Files"
],
"description": "List all files in a bucket.",
"parameters": [
@@ -10274,7 +10274,7 @@
"name": "Benchmarks"
},
{
- "name": "DatasetIO"
+ "name": "Datasets"
},
{
"name": "Datasets"
@@ -10284,7 +10284,7 @@
"x-displayName": "Llama Stack Evaluation API for running evaluations on model and agent candidates."
},
{
- "name": "Files (Coming Soon)"
+ "name": "Files"
},
{
"name": "Inference",
@@ -10342,10 +10342,10 @@
"Agents",
"BatchInference (Coming Soon)",
"Benchmarks",
- "DatasetIO",
+ "Datasets",
"Datasets",
"Eval",
- "Files (Coming Soon)",
+ "Files",
"Inference",
"Inspect",
"Models",
diff --git a/docs/_static/llama-stack-spec.yaml b/docs/_static/llama-stack-spec.yaml
index 43880dac8..79adc221a 100644
--- a/docs/_static/llama-stack-spec.yaml
+++ b/docs/_static/llama-stack-spec.yaml
@@ -26,7 +26,7 @@ paths:
default:
$ref: '#/components/responses/DefaultError'
tags:
- - DatasetIO
+ - Datasets
description: ''
parameters:
- name: dataset_id
@@ -350,7 +350,7 @@ paths:
default:
$ref: '#/components/responses/DefaultError'
tags:
- - Files (Coming Soon)
+ - Files
description: List all buckets.
parameters:
- name: bucket
@@ -377,7 +377,7 @@ paths:
default:
$ref: '#/components/responses/DefaultError'
tags:
- - Files (Coming Soon)
+ - Files
description: >-
Create a new upload session for a file identified by a bucket and key.
parameters: []
@@ -536,7 +536,7 @@ paths:
default:
$ref: '#/components/responses/DefaultError'
tags:
- - Files (Coming Soon)
+ - Files
description: >-
Get a file info identified by a bucket and key.
parameters:
@@ -572,7 +572,7 @@ paths:
default:
$ref: '#/components/responses/DefaultError'
tags:
- - Files (Coming Soon)
+ - Files
description: >-
Delete a file identified by a bucket and key.
parameters:
@@ -1224,7 +1224,7 @@ paths:
default:
$ref: '#/components/responses/DefaultError'
tags:
- - Files (Coming Soon)
+ - Files
description: >-
Returns information about an existsing upload session
parameters:
@@ -1255,7 +1255,7 @@ paths:
default:
$ref: '#/components/responses/DefaultError'
tags:
- - Files (Coming Soon)
+ - Files
description: >-
Upload file content to an existing upload session. On the server, request
body will have the raw bytes that are uploaded.
@@ -1477,7 +1477,7 @@ paths:
default:
$ref: '#/components/responses/DefaultError'
tags:
- - DatasetIO
+ - Datasets
description: >-
Get a paginated list of rows from a dataset. Uses cursor-based pagination.
parameters:
@@ -1754,7 +1754,7 @@ paths:
default:
$ref: '#/components/responses/DefaultError'
tags:
- - Files (Coming Soon)
+ - Files
description: List all files in a bucket.
parameters:
- name: bucket
@@ -6931,12 +6931,12 @@ tags:
Agents API for creating and interacting with agentic systems.
- name: BatchInference (Coming Soon)
- name: Benchmarks
- - name: DatasetIO
+ - name: Datasets
- name: Datasets
- name: Eval
x-displayName: >-
Llama Stack Evaluation API for running evaluations on model and agent candidates.
- - name: Files (Coming Soon)
+ - name: Files
- name: Inference
description: >-
This API provides the raw interface to the underlying models. Two kinds of models
@@ -6971,10 +6971,10 @@ x-tagGroups:
- Agents
- BatchInference (Coming Soon)
- Benchmarks
- - DatasetIO
+ - Datasets
- Datasets
- Eval
- - Files (Coming Soon)
+ - Files
- Inference
- Inspect
- Models
diff --git a/docs/openapi_generator/pyopenapi/generator.py b/docs/openapi_generator/pyopenapi/generator.py
index 91f32e6c8..a7ee87125 100644
--- a/docs/openapi_generator/pyopenapi/generator.py
+++ b/docs/openapi_generator/pyopenapi/generator.py
@@ -435,7 +435,7 @@ class Generator:
)
self.schema_builder = SchemaBuilder(schema_generator)
self.responses = {}
-
+
# Create standard error responses
self._create_standard_error_responses()
@@ -446,7 +446,7 @@ class Generator:
"""
# Get the Error schema
error_schema = self.schema_builder.classdef_to_ref(Error)
-
+
# Create standard error responses
self.responses["BadRequest400"] = Response(
description="The request was invalid or malformed",
@@ -457,11 +457,11 @@ class Generator:
"status": 400,
"title": "Bad Request",
"detail": "The request was invalid or malformed",
- }
+ },
)
- }
+ },
)
-
+
self.responses["TooManyRequests429"] = Response(
description="The client has sent too many requests in a given amount of time",
content={
@@ -471,11 +471,11 @@ class Generator:
"status": 429,
"title": "Too Many Requests",
"detail": "You have exceeded the rate limit. Please try again later.",
- }
+ },
)
- }
+ },
)
-
+
self.responses["InternalServerError500"] = Response(
description="The server encountered an unexpected error",
content={
@@ -485,11 +485,11 @@ class Generator:
"status": 500,
"title": "Internal Server Error",
"detail": "An unexpected error occurred. Our team has been notified.",
- }
+ },
)
- }
+ },
)
-
+
# Add a default error response for any unhandled error cases
self.responses["DefaultError"] = Response(
description="An unexpected error occurred",
@@ -500,9 +500,9 @@ class Generator:
"status": 0,
"title": "Error",
"detail": "An unexpected error occurred",
- }
+ },
)
- }
+ },
)
def _build_type_tag(self, ref: str, schema: Schema) -> Tag:
@@ -547,11 +547,14 @@ class Generator:
"SyntheticDataGeneration",
"PostTraining",
"BatchInference",
- "Files",
]:
op.defining_class.__name__ = f"{op.defining_class.__name__} (Coming Soon)"
print(op.defining_class.__name__)
+ # TODO (xiyan): temporary fix for datasetio inner impl + datasets api
+ if op.defining_class.__name__ in ["DatasetIO"]:
+ op.defining_class.__name__ = "Datasets"
+
doc_string = parse_type(op.func_ref)
doc_params = dict(
(param.name, param.description) for param in doc_string.params.values()
@@ -598,7 +601,9 @@ class Generator:
# data passed in request body as raw bytes cannot have request parameters
if raw_bytes_request_body and op.request_params:
- raise ValueError("Cannot have both raw bytes request body and request parameters")
+ raise ValueError(
+ "Cannot have both raw bytes request body and request parameters"
+ )
# data passed in request body as raw bytes
if raw_bytes_request_body:
@@ -719,7 +724,7 @@ class Generator:
responses.update(response_builder.build_response(response_options))
assert len(responses.keys()) > 0, f"No responses found for {op.name}"
-
+
# Add standard error response references
if self.options.include_standard_error_responses:
if "400" not in responses:
@@ -730,7 +735,7 @@ class Generator:
responses["500"] = ResponseRef("InternalServerError500")
if "default" not in responses:
responses["default"] = ResponseRef("DefaultError")
-
+
if op.event_type is not None:
builder = ContentBuilder(self.schema_builder)
callbacks = {