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

Merge 71caa271ad into 76dcf47320

Browse source
This commit is contained in:
Charlie Doern 2025-06-02 17:32:30 -04:00 • committed by GitHub
commit 4a7bdf1b87
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 393 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

1
llama_stack/apis/datatypes.py
View file

@ -27,6 +27,7 @@ class Api(Enum):
telemetry = "telemetry"
models = "models"
post_training_models = "post_training_models"
shields = "shields"
vector_dbs = "vector_dbs"
datasets = "datasets"

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

@ -13,6 +13,7 @@ from pydantic import BaseModel, Field
from llama_stack.apis.common.content_types import URL
from llama_stack.apis.common.job_types import JobStatus
from llama_stack.apis.common.training_types import Checkpoint
from llama_stack.apis.models import Model
from llama_stack.schema_utils import json_schema_type, register_schema, webmethod
@ -168,7 +169,13 @@ class PostTrainingJobArtifactsResponse(BaseModel):
# TODO(ashwin): metrics, evals
class ModelStore(Protocol):
async def get_model(self, identifier: str) -> Model: ...
class PostTraining(Protocol):
model_store: ModelStore | None = None
@webmethod(route="/post-training/supervised-fine-tune", method="POST")
async def supervised_fine_tune(
self,