|
18 | 18 |
|
19 | 19 | import argparse |
20 | 20 | import sys |
21 | | -import time |
22 | | - |
23 | | -from openadapt_evals.telemetry import track_agent_run, track_agent_run_completed |
24 | 21 |
|
25 | 22 |
|
26 | 23 | def main(argv: list[str] | None = None) -> int: |
@@ -252,43 +249,13 @@ def _cmd_mock(args: argparse.Namespace) -> int: |
252 | 249 | """Run mock evaluation.""" |
253 | 250 | # Delegate to existing CLI implementation |
254 | 251 | from openadapt_evals.benchmarks.cli import cmd_mock |
255 | | - start = time.perf_counter() |
256 | | - track_agent_run( |
257 | | - phase="start", |
258 | | - entrypoint="oa evals mock", |
259 | | - mode="mock", |
260 | | - agent_class=getattr(args, "agent", "mock"), |
261 | | - ) |
262 | | - rc = cmd_mock(args) |
263 | | - track_agent_run_completed( |
264 | | - entrypoint="oa evals mock", |
265 | | - mode="mock", |
266 | | - agent_class=getattr(args, "agent", "mock"), |
267 | | - return_code=rc, |
268 | | - duration_seconds=round(time.perf_counter() - start, 3), |
269 | | - ) |
270 | | - return rc |
| 252 | + return cmd_mock(args) |
271 | 253 |
|
272 | 254 |
|
273 | 255 | def _cmd_run(args: argparse.Namespace) -> int: |
274 | 256 | """Run live evaluation.""" |
275 | 257 | from openadapt_evals.benchmarks.cli import cmd_live |
276 | | - start = time.perf_counter() |
277 | | - track_agent_run( |
278 | | - phase="start", |
279 | | - entrypoint="oa evals run", |
280 | | - mode="live", |
281 | | - agent_class=getattr(args, "agent", "unknown"), |
282 | | - ) |
283 | | - rc = cmd_live(args) |
284 | | - track_agent_run_completed( |
285 | | - entrypoint="oa evals run", |
286 | | - mode="live", |
287 | | - agent_class=getattr(args, "agent", "unknown"), |
288 | | - return_code=rc, |
289 | | - duration_seconds=round(time.perf_counter() - start, 3), |
290 | | - ) |
291 | | - return rc |
| 258 | + return cmd_live(args) |
292 | 259 |
|
293 | 260 |
|
294 | 261 | def _cmd_probe(args: argparse.Namespace) -> int: |
|
0 commit comments