phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

revert job related

Browse source
This commit is contained in:
Xi Yan 2025-03-17 17:12:28 -07:00
parent 452b2b1284
commit ade3391170
2 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/apis/common/job_types.py
View file

@ -12,6 +12,12 @@ from pydantic import BaseModel
from llama_stack.schema_utils import json_schema_type from llama_stack.schema_utils import json_schema_type
@json_schema_type
class Job(BaseModel):
# NOTE: this will be DEPRECATED in favour of CommonJobFields
job_id: str
class JobType(Enum): class JobType(Enum):
batch_inference = "batch_inference" batch_inference = "batch_inference"
evaluation = "evaluation" evaluation = "evaluation"
@ -26,14 +32,6 @@ class JobStatus(Enum):
cancelled = "cancelled" cancelled = "cancelled"
class JobArtifact(BaseModel):
"""
A job artifact is a file or directory that is produced by a job.
"""
path: str
@json_schema_type @json_schema_type
class CommonJobFields(BaseModel): class CommonJobFields(BaseModel):
"""Common fields for all jobs. """Common fields for all jobs.

8
llama_stack/apis/eval/eval.py
View file

@ -10,7 +10,7 @@ from pydantic import BaseModel, Field
from typing_extensions import Annotated from typing_extensions import Annotated
from llama_stack.apis.agents import AgentConfig from llama_stack.apis.agents import AgentConfig
from llama_stack.apis.common.job_types import CommonJobFields, JobStatus from llama_stack.apis.common.job_types import Job, JobStatus
from llama_stack.apis.inference import SamplingParams, SystemMessage from llama_stack.apis.inference import SamplingParams, SystemMessage
from llama_stack.apis.scoring import ScoringResult from llama_stack.apis.scoring import ScoringResult
from llama_stack.apis.scoring_functions import ScoringFnParams from llama_stack.apis.scoring_functions import ScoringFnParams
@ -91,7 +91,7 @@ class Eval(Protocol):
self, self,
benchmark_id: str, benchmark_id: str,
benchmark_config: BenchmarkConfig, benchmark_config: BenchmarkConfig,
) -> CommonJobFields: ) -> Job:
"""Run an evaluation on a benchmark. """Run an evaluation on a benchmark.
:param benchmark_id: The ID of the benchmark to run the evaluation on. :param benchmark_id: The ID of the benchmark to run the evaluation on.
@ -135,7 +135,9 @@ class Eval(Protocol):
""" """
... ...
@webmethod(route="/eval/benchmarks/{benchmark_id}/jobs/{job_id}/result", method="GET") @webmethod(
route="/eval/benchmarks/{benchmark_id}/jobs/{job_id}/result", method="GET"
)
async def job_result(self, benchmark_id: str, job_id: str) -> EvaluateResponse: async def job_result(self, benchmark_id: str, job_id: str) -> EvaluateResponse:
"""Get the result of a job. """Get the result of a job.