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

Add OpenAPI generation utility, update SPEC to reflect latest types

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-15 13:45:45 -07:00
parent 417ba2aea0
commit 1f5eb9ff96
10 changed files with 770 additions and 656 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_toolchain/post_training/api/endpoints.py
View file

@ -95,13 +95,13 @@ class PostTrainingJobArtifactsResponse(BaseModel):
class PostTraining(Protocol):
@webmethod(route="/post_training/supervised_fine_tune")
def post_supervised_fine_tune(
def supervised_fine_tune(
self,
request: PostTrainingSFTRequest,
) -> PostTrainingJob: ...
@webmethod(route="/post_training/preference_optimize")
def post_preference_optimize(
def preference_optimize(
self,
request: PostTrainingRLHFRequest,
) -> PostTrainingJob: ...