move jobs, fix errors

This commit is contained in:
Xi Yan 2024-10-22 09:30:08 -07:00
parent 5836c09c57
commit a4f5f1f890
4 changed files with 23 additions and 17 deletions

View file

@ -11,6 +11,9 @@ from typing_extensions import Annotated
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_models.schema_utils import json_schema_type, webmethod
from llama_stack.apis.scoring_functions import * # noqa: F403
from llama_stack.apis.agents import AgentConfig
from llama_stack.apis.common.job_types import Job
from llama_stack.apis.scoring import * # noqa: F403
@json_schema_type
@ -32,11 +35,6 @@ EvalCandidate = Annotated[
]
@json_schema_type
class Job(BaseModel):
job_id: str
@json_schema_type
class EvaluateResponse(BaseModel):
generations: List[Dict[str, Any]]