|
14 | 14 | Eval, |
15 | 15 | EvalCandidate, |
16 | 16 | EvaluateResponse, |
| 17 | + EvaluateRowsRequest, |
17 | 18 | Job, |
| 19 | + JobCancelRequest, |
| 20 | + JobResultRequest, |
18 | 21 | JobStatus, |
| 22 | + JobStatusRequest, |
19 | 23 | ListBenchmarksResponse, |
20 | 24 | ProviderSpec, |
21 | 25 | RemoteProviderSpec, |
| 26 | + RunEvalRequest, |
22 | 27 | ScoringResult, |
23 | 28 | json_schema_type, |
24 | 29 | ) |
25 | | -except ModuleNotFoundError: # Legacy llama_stack layout |
| 30 | + from llama_stack_api.eval.compat import ( |
| 31 | + resolve_evaluate_rows_request, |
| 32 | + resolve_job_cancel_request, |
| 33 | + resolve_job_result_request, |
| 34 | + resolve_job_status_request, |
| 35 | + resolve_run_eval_request, |
| 36 | + ) |
| 37 | +except ModuleNotFoundError: # Legacy llama_stack layout (pre-0.5.x) |
| 38 | + # Note: Request objects (RunEvalRequest, etc.) were introduced in 0.5.x |
| 39 | + # Legacy versions only support individual parameter style |
26 | 40 | from llama_stack.apis.benchmarks import Benchmark, ListBenchmarksResponse |
27 | 41 | from llama_stack.apis.common.job_types import Job, JobStatus |
28 | 42 | from llama_stack.apis.datatypes import Api |
|
40 | 54 | ) |
41 | 55 | from llama_stack.schema_utils import json_schema_type |
42 | 56 |
|
| 57 | + # Request objects don't exist in pre-0.5.x, set to None |
| 58 | + RunEvalRequest = None # type: ignore |
| 59 | + EvaluateRowsRequest = None # type: ignore |
| 60 | + JobStatusRequest = None # type: ignore |
| 61 | + JobCancelRequest = None # type: ignore |
| 62 | + JobResultRequest = None # type: ignore |
| 63 | + |
| 64 | + # Compat resolve functions don't exist in pre-0.5.x, set to None |
| 65 | + resolve_run_eval_request = None # type: ignore |
| 66 | + resolve_evaluate_rows_request = None # type: ignore |
| 67 | + resolve_job_status_request = None # type: ignore |
| 68 | + resolve_job_cancel_request = None # type: ignore |
| 69 | + resolve_job_result_request = None # type: ignore |
| 70 | + |
43 | 71 | __all__ = [ |
44 | 72 | "Api", |
45 | 73 | "Benchmark", |
|
48 | 76 | "Eval", |
49 | 77 | "EvalCandidate", |
50 | 78 | "EvaluateResponse", |
| 79 | + "EvaluateRowsRequest", |
51 | 80 | "Job", |
| 81 | + "JobCancelRequest", |
| 82 | + "JobResultRequest", |
52 | 83 | "JobStatus", |
| 84 | + "JobStatusRequest", |
53 | 85 | "ListBenchmarksResponse", |
54 | 86 | "ProviderSpec", |
55 | 87 | "RemoteProviderSpec", |
| 88 | + "RunEvalRequest", |
56 | 89 | "ScoringResult", |
57 | 90 | "json_schema_type", |
| 91 | + "resolve_evaluate_rows_request", |
| 92 | + "resolve_job_cancel_request", |
| 93 | + "resolve_job_result_request", |
| 94 | + "resolve_job_status_request", |
| 95 | + "resolve_run_eval_request", |
58 | 96 | ] |
0 commit comments