phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

a

Browse source
This commit is contained in:
Sai Soundararaj 2025-07-01 16:36:35 -07:00
parent 441259f2e3
commit b1b93088c5
5 changed files with 56 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api_update_plan.md
View file

@ -227,8 +227,8 @@ Before finalizing documentation, verify:
[x] 6. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/tools/rag_tool.py` - RAG tool runtime
[x] 7. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/vector_io/vector_io.py` - Vector database operations
[x] 8. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/vector_dbs/vector_dbs.py` - Vector database management
9. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/files/files.py` - File management
10. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/datasets/datasets.py` - Dataset management
[x] 9. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/files/files.py` - File management
[x] 10. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/datasets/datasets.py` - Dataset management
11. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/datasetio/datasetio.py` - Dataset I/O operations
12. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/post_training/post_training.py` - Training and fine-tuning
13. `/Users/saip/Documents/GitHub/llama-stack/llama_stack/apis/eval/eval.py` - Evaluation framework

31
docs/_static/llama-stack-spec.html vendored
View file

@ -10366,9 +10366,9 @@
"tool",
"tool_group"
],
"title": "ResourceType",
"const": "dataset",
"default": "dataset"
"default": "dataset",
"description": "Type of resource, always 'dataset' for datasets"
},
"purpose": {
"type": "string",
@ -10377,11 +10377,11 @@
"eval/question-answer",
"eval/messages-answer"
],
"title": "DatasetPurpose",
"description": "Purpose of the dataset. Each purpose has a required input data schema."
"description": "Purpose of the dataset indicating its intended use"
},
"source": {
"$ref": "#/components/schemas/DataSource"
"$ref": "#/components/schemas/DataSource",
"description": "Data source configuration for the dataset"
},
"metadata": {
"type": "object",
@ -10406,7 +10406,8 @@
"type": "object"
}
]
}
},
"description": "Additional metadata for the dataset"
}
},
"additionalProperties": false,
@ -10418,7 +10419,8 @@
"source",
"metadata"
],
"title": "Dataset"
"title": "Dataset",
"description": "Dataset resource for storing and accessing training or evaluation data."
},
"RowsDataSource": {
"type": "object",
@ -12007,14 +12009,16 @@
"type": "array",
"items": {
"$ref": "#/components/schemas/Dataset"
}
},
"description": "List of datasets"
}
},
"additionalProperties": false,
"required": [
"data"
],
"title": "ListDatasetsResponse"
"title": "ListDatasetsResponse",
"description": "Response from listing datasets."
},
"ListModelsResponse": {
"type": "object",
@ -14091,13 +14095,16 @@
"description": "List of file objects"
},
"has_more": {
"type": "boolean"
"type": "boolean",
"description": "Whether there are more files available beyond this page"
},
"first_id": {
"type": "string"
"type": "string",
"description": "ID of the first file in the list for pagination"
},
"last_id": {
"type": "string"
"type": "string",
"description": "ID of the last file in the list for pagination"
},
"object": {
"type": "string",

19
docs/_static/llama-stack-spec.yaml vendored
View file

@ -7459,20 +7459,22 @@ components:
- benchmark
- tool
- tool_group
title: ResourceType
const: dataset
default: dataset
description: >-
Type of resource, always 'dataset' for datasets
purpose:
type: string
enum:
- post-training/messages
- eval/question-answer
- eval/messages-answer
title: DatasetPurpose
description: >-
Purpose of the dataset. Each purpose has a required input data schema.
Purpose of the dataset indicating its intended use
source:
$ref: '#/components/schemas/DataSource'
description: >-
Data source configuration for the dataset
metadata:
type: object
additionalProperties:
@ -7483,6 +7485,7 @@ components:
- type: string
- type: array
- type: object
description: Additional metadata for the dataset
additionalProperties: false
required:
- identifier
@ -7492,6 +7495,8 @@ components:
- source
- metadata
title: Dataset
description: >-
Dataset resource for storing and accessing training or evaluation data.
RowsDataSource:
type: object
properties:
@ -8606,10 +8611,12 @@ components:
type: array
items:
$ref: '#/components/schemas/Dataset'
description: List of datasets
additionalProperties: false
required:
- data
title: ListDatasetsResponse
description: Response from listing datasets.
ListModelsResponse:
type: object
properties:
@ -10081,10 +10088,16 @@ components:
description: List of file objects
has_more:
type: boolean
description: >-
Whether there are more files available beyond this page
first_id:
type: string
description: >-
ID of the first file in the list for pagination
last_id:
type: string
description: >-
ID of the last file in the list for pagination
object:
type: string
const: list

16
llama_stack/apis/datasets/datasets.py
View file

@ -94,6 +94,10 @@ register_schema(DataSource, name="DataSource")
class CommonDatasetFields(BaseModel):
"""
Common fields for a dataset.
:param purpose: Purpose of the dataset indicating its intended use
:param source: Data source configuration for the dataset
:param metadata: Additional metadata for the dataset
"""
purpose: DatasetPurpose
@ -106,6 +110,10 @@ class CommonDatasetFields(BaseModel):
@json_schema_type
class Dataset(CommonDatasetFields, Resource):
"""Dataset resource for storing and accessing training or evaluation data.
:param type: Type of resource, always 'dataset' for datasets
"""
type: Literal[ResourceType.dataset] = ResourceType.dataset
@property
@ -118,10 +126,18 @@ class Dataset(CommonDatasetFields, Resource):
class DatasetInput(CommonDatasetFields, BaseModel):
"""Input parameters for dataset operations.
:param dataset_id: Unique identifier for the dataset
"""
dataset_id: str
class ListDatasetsResponse(BaseModel):
"""Response from listing datasets.
:param data: List of datasets
"""
data: list[Dataset]

3
llama_stack/apis/files/files.py
View file

@ -54,6 +54,9 @@ class ListOpenAIFileResponse(BaseModel):
Response for listing files in OpenAI Files API.
:param data: List of file objects
:param has_more: Whether there are more files available beyond this page
:param first_id: ID of the first file in the list for pagination
:param last_id: ID of the last file in the list for pagination
:param object: The object type, which is always "list"
"""