phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-11 11:50:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

openapi gen

Browse source
This commit is contained in:
Xi Yan 2024-10-15 00:44:54 -07:00
parent 9cc0a54f0b
commit d2b62157a3
5 changed files with 1126 additions and 777 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -15,6 +15,26 @@ from pydantic import BaseModel, Field
from typing_extensions import Annotated
@json_schema_type
class TrainEvalDatasetColumnType(Enum):
dialog = "dialog"
text = "text"
media = "media"
number = "number"
json = "json"
@json_schema_type
class TrainEvalDataset(BaseModel):
"""Dataset to be used for training or evaluating language models."""
# TODO(ashwin): figure out if we need to add an enum for a "dataset type"
columns: Dict[str, TrainEvalDatasetColumnType]
content_url: URL
metadata: Optional[Dict[str, Any]] = None
@json_schema_type
class GenerationInput(BaseModel):
messages: List[Message]

2
llama_stack/apis/post_training/post_training.py
View file

@ -14,7 +14,7 @@ from llama_models.schema_utils import json_schema_type, webmethod
from pydantic import BaseModel, Field
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.apis.dataset import * # noqa: F403
from llama_stack.apis.datasets import * # noqa: F403
from llama_stack.apis.common.training_types import * # noqa: F403