synth-ai 0.2.14__py3-none-any.whl → 0.4.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of synth-ai might be problematic. Click here for more details.

Files changed (1091) hide show
  1. synth_ai/__init__.py +19 -40
  2. synth_ai/__main__.py +30 -3
  3. synth_ai/cli/__init__.py +105 -70
  4. synth_ai/cli/__main__.py +42 -0
  5. synth_ai/cli/_internal/__init__.py +5 -0
  6. synth_ai/cli/_internal/modal_wrapper.py +31 -0
  7. synth_ai/cli/_internal/storage.py +20 -0
  8. synth_ai/cli/_internal/typer_patch.py +47 -0
  9. synth_ai/cli/_internal/validate_task_app.py +29 -0
  10. synth_ai/cli/agents/__init__.py +17 -0
  11. synth_ai/cli/agents/claude.py +77 -0
  12. synth_ai/cli/agents/codex.py +265 -0
  13. synth_ai/cli/agents/opencode.py +253 -0
  14. synth_ai/cli/commands/__init__.py +18 -0
  15. synth_ai/cli/commands/artifacts/__init__.py +13 -0
  16. synth_ai/cli/commands/artifacts/client.py +119 -0
  17. synth_ai/cli/commands/artifacts/config.py +57 -0
  18. synth_ai/cli/commands/artifacts/core.py +24 -0
  19. synth_ai/cli/commands/artifacts/download.py +188 -0
  20. synth_ai/cli/commands/artifacts/export.py +186 -0
  21. synth_ai/cli/commands/artifacts/list.py +156 -0
  22. synth_ai/cli/commands/artifacts/parsing.py +250 -0
  23. synth_ai/cli/commands/artifacts/show.py +336 -0
  24. synth_ai/cli/commands/baseline/__init__.py +12 -0
  25. synth_ai/cli/commands/baseline/core.py +636 -0
  26. synth_ai/cli/commands/baseline/list.py +94 -0
  27. synth_ai/cli/commands/demo/__init__.py +3 -0
  28. synth_ai/cli/commands/demo/core.py +153 -0
  29. synth_ai/cli/commands/eval/__init__.py +19 -0
  30. synth_ai/cli/commands/eval/core.py +1113 -0
  31. synth_ai/cli/commands/eval/errors.py +81 -0
  32. synth_ai/cli/commands/eval/validation.py +133 -0
  33. synth_ai/cli/commands/filter/__init__.py +12 -0
  34. synth_ai/cli/commands/filter/core.py +424 -0
  35. synth_ai/cli/commands/filter/errors.py +55 -0
  36. synth_ai/cli/commands/filter/validation.py +77 -0
  37. synth_ai/cli/commands/help/__init__.py +185 -0
  38. synth_ai/cli/commands/help/core.py +72 -0
  39. synth_ai/cli/commands/scan/__init__.py +19 -0
  40. synth_ai/cli/commands/scan/cloudflare_scanner.py +403 -0
  41. synth_ai/cli/commands/scan/core.py +344 -0
  42. synth_ai/cli/commands/scan/health_checker.py +242 -0
  43. synth_ai/cli/commands/scan/local_scanner.py +278 -0
  44. synth_ai/cli/commands/scan/models.py +83 -0
  45. synth_ai/cli/commands/smoke/__init__.py +7 -0
  46. synth_ai/cli/commands/smoke/core.py +1438 -0
  47. synth_ai/cli/commands/status/__init__.py +66 -0
  48. synth_ai/cli/commands/status/client.py +192 -0
  49. synth_ai/cli/commands/status/config.py +92 -0
  50. synth_ai/cli/commands/status/errors.py +20 -0
  51. synth_ai/cli/commands/status/formatters.py +164 -0
  52. synth_ai/cli/commands/status/subcommands/__init__.py +9 -0
  53. synth_ai/cli/commands/status/subcommands/files.py +79 -0
  54. synth_ai/cli/commands/status/subcommands/jobs.py +334 -0
  55. synth_ai/cli/commands/status/subcommands/models.py +79 -0
  56. synth_ai/cli/commands/status/subcommands/pricing.py +23 -0
  57. synth_ai/cli/commands/status/subcommands/runs.py +81 -0
  58. synth_ai/cli/commands/status/subcommands/session.py +182 -0
  59. synth_ai/cli/commands/status/subcommands/summary.py +47 -0
  60. synth_ai/cli/commands/status/subcommands/usage.py +203 -0
  61. synth_ai/cli/commands/status/utils.py +114 -0
  62. synth_ai/cli/commands/train/__init__.py +53 -0
  63. synth_ai/cli/commands/train/core.py +22 -0
  64. synth_ai/cli/commands/train/errors.py +117 -0
  65. synth_ai/cli/commands/train/judge_schemas.py +201 -0
  66. synth_ai/cli/commands/train/judge_validation.py +305 -0
  67. synth_ai/cli/commands/train/prompt_learning_validation.py +633 -0
  68. synth_ai/cli/commands/train/validation.py +392 -0
  69. synth_ai/cli/demo_apps/__init__.py +10 -0
  70. synth_ai/cli/demo_apps/core/__init__.py +28 -0
  71. synth_ai/cli/demo_apps/core/cli.py +1735 -0
  72. synth_ai/cli/demo_apps/crafter/crafter_fft_4b.toml +55 -0
  73. synth_ai/cli/demo_apps/crafter/grpo_crafter_task_app.py +186 -0
  74. synth_ai/cli/demo_apps/crafter/rl_from_base_qwen4b.toml +74 -0
  75. synth_ai/cli/demo_apps/demo_registry.py +176 -0
  76. synth_ai/cli/demo_apps/demo_task_apps/core.py +440 -0
  77. synth_ai/cli/demo_apps/demo_task_apps/crafter/__init__.py +1 -0
  78. synth_ai/cli/demo_apps/demo_task_apps/crafter/grpo_crafter_task_app.py +185 -0
  79. synth_ai/cli/demo_apps/demo_task_apps/math/modal_task_app.py +742 -0
  80. synth_ai/cli/demo_apps/demo_task_apps/math/task_app_entry.py +39 -0
  81. synth_ai/cli/demo_apps/math/__init__.py +1 -0
  82. synth_ai/cli/demo_apps/math/_common.py +16 -0
  83. synth_ai/cli/demo_apps/math/app.py +38 -0
  84. synth_ai/cli/demo_apps/math/config.toml +76 -0
  85. synth_ai/cli/demo_apps/math/deploy_modal.py +54 -0
  86. synth_ai/cli/demo_apps/math/modal_task_app.py +702 -0
  87. synth_ai/cli/demo_apps/math/task_app_entry.py +53 -0
  88. synth_ai/cli/demo_apps/mipro/main.py +271 -0
  89. synth_ai/cli/demo_apps/mipro/task_app.py +933 -0
  90. synth_ai/cli/demo_apps/mipro/train_cfg.toml +92 -0
  91. synth_ai/cli/demos/__init__.py +12 -0
  92. synth_ai/cli/demos/demo.py +32 -0
  93. synth_ai/cli/demos/rl_demo.py +254 -0
  94. synth_ai/cli/deploy.py +216 -0
  95. synth_ai/cli/infra/__init__.py +14 -0
  96. synth_ai/cli/infra/balance.py +216 -0
  97. synth_ai/cli/infra/mcp.py +35 -0
  98. synth_ai/cli/infra/modal_app.py +36 -0
  99. synth_ai/cli/infra/setup.py +69 -0
  100. synth_ai/cli/infra/status.py +16 -0
  101. synth_ai/cli/infra/turso.py +77 -0
  102. synth_ai/cli/lib/__init__.py +10 -0
  103. synth_ai/cli/lib/agents.py +76 -0
  104. synth_ai/cli/lib/apps/modal_app.py +101 -0
  105. synth_ai/cli/lib/apps/task_app.py +643 -0
  106. synth_ai/cli/lib/bin.py +39 -0
  107. synth_ai/cli/lib/env.py +375 -0
  108. synth_ai/cli/lib/errors.py +85 -0
  109. synth_ai/cli/lib/modal.py +315 -0
  110. synth_ai/cli/lib/plotting.py +126 -0
  111. synth_ai/cli/lib/prompt_args.py +39 -0
  112. synth_ai/cli/lib/prompts.py +284 -0
  113. synth_ai/cli/lib/sqld.py +122 -0
  114. synth_ai/cli/lib/task_app_discovery.py +884 -0
  115. synth_ai/cli/lib/task_app_env.py +295 -0
  116. synth_ai/cli/lib/train_cfgs.py +300 -0
  117. synth_ai/cli/lib/tunnel_records.py +207 -0
  118. synth_ai/cli/local/__init__.py +14 -0
  119. synth_ai/cli/local/experiment_queue/__init__.py +72 -0
  120. synth_ai/cli/local/experiment_queue/api_schemas.py +221 -0
  121. synth_ai/cli/local/experiment_queue/celery_app.py +208 -0
  122. synth_ai/cli/local/experiment_queue/config.py +128 -0
  123. synth_ai/cli/local/experiment_queue/config_utils.py +272 -0
  124. synth_ai/cli/local/experiment_queue/database.py +175 -0
  125. synth_ai/cli/local/experiment_queue/dispatcher.py +119 -0
  126. synth_ai/cli/local/experiment_queue/models.py +231 -0
  127. synth_ai/cli/local/experiment_queue/progress_info.py +160 -0
  128. synth_ai/cli/local/experiment_queue/results.py +373 -0
  129. synth_ai/cli/local/experiment_queue/schemas.py +131 -0
  130. synth_ai/cli/local/experiment_queue/service.py +344 -0
  131. synth_ai/cli/local/experiment_queue/status.py +372 -0
  132. synth_ai/cli/local/experiment_queue/status_tracker.py +360 -0
  133. synth_ai/cli/local/experiment_queue/tasks.py +1984 -0
  134. synth_ai/cli/local/experiment_queue/trace_storage.py +65 -0
  135. synth_ai/cli/local/experiment_queue/validation.py +157 -0
  136. synth_ai/cli/local/session/__init__.py +92 -0
  137. synth_ai/cli/local/session/client.py +383 -0
  138. synth_ai/cli/local/session/constants.py +63 -0
  139. synth_ai/cli/local/session/exceptions.py +105 -0
  140. synth_ai/cli/local/session/manager.py +139 -0
  141. synth_ai/cli/local/session/models.py +89 -0
  142. synth_ai/cli/local/session/query.py +110 -0
  143. synth_ai/cli/root.py +30 -6
  144. synth_ai/cli/task_apps/__init__.py +26 -0
  145. synth_ai/cli/task_apps/commands.py +3153 -0
  146. synth_ai/cli/task_apps/deploy.py +7 -0
  147. synth_ai/cli/task_apps/list.py +26 -0
  148. synth_ai/cli/task_apps/main.py +36 -0
  149. synth_ai/cli/task_apps/modal_serve.py +11 -0
  150. synth_ai/cli/task_apps/serve.py +11 -0
  151. synth_ai/cli/training/__init__.py +8 -0
  152. synth_ai/cli/training/train.py +5 -0
  153. synth_ai/cli/training/train_cfg.py +34 -0
  154. synth_ai/cli/training/watch.py +506 -0
  155. synth_ai/cli/turso.py +34 -55
  156. synth_ai/cli/usage.py +159 -0
  157. synth_ai/cli/utils/__init__.py +8 -0
  158. synth_ai/cli/utils/experiments.py +235 -0
  159. synth_ai/cli/utils/queue.py +504 -0
  160. synth_ai/cli/utils/recent.py +133 -0
  161. synth_ai/cli/utils/traces.py +164 -0
  162. synth_ai/contracts/__init__.py +67 -0
  163. synth_ai/core/__init__.py +100 -0
  164. synth_ai/core/_utils/__init__.py +54 -0
  165. synth_ai/core/_utils/base_url.py +10 -0
  166. synth_ai/core/_utils/http.py +10 -0
  167. synth_ai/core/_utils/prompts.py +14 -0
  168. synth_ai/core/_utils/task_app_state.py +12 -0
  169. synth_ai/core/_utils/user_config.py +10 -0
  170. synth_ai/core/apps/common.py +116 -0
  171. synth_ai/core/auth.py +95 -0
  172. synth_ai/core/cfgs.py +240 -0
  173. synth_ai/core/config/__init__.py +16 -0
  174. synth_ai/core/config/base.py +168 -0
  175. synth_ai/core/config/resolver.py +89 -0
  176. synth_ai/core/env.py +220 -0
  177. synth_ai/core/errors.py +126 -0
  178. synth_ai/core/http.py +230 -0
  179. synth_ai/core/integrations/__init__.py +11 -0
  180. synth_ai/core/integrations/cloudflare.py +1710 -0
  181. synth_ai/core/integrations/mcp/__init__.py +6 -0
  182. synth_ai/core/integrations/mcp/__main__.py +8 -0
  183. synth_ai/core/integrations/mcp/claude.py +36 -0
  184. synth_ai/core/integrations/mcp/main.py +254 -0
  185. synth_ai/core/integrations/mcp/setup.py +100 -0
  186. synth_ai/core/integrations/modal.py +277 -0
  187. synth_ai/core/json.py +72 -0
  188. synth_ai/core/log_filter.py +99 -0
  189. synth_ai/core/logging.py +82 -0
  190. synth_ai/core/paths.py +107 -0
  191. synth_ai/core/pricing.py +109 -0
  192. synth_ai/core/process.py +233 -0
  193. synth_ai/core/ssl.py +25 -0
  194. synth_ai/core/storage/__init__.py +71 -0
  195. synth_ai/core/task_app_state.py +318 -0
  196. synth_ai/core/telemetry.py +282 -0
  197. synth_ai/core/tracing_v3/__init__.py +99 -0
  198. synth_ai/core/tracing_v3/abstractions.py +302 -0
  199. synth_ai/core/tracing_v3/config.py +229 -0
  200. synth_ai/core/tracing_v3/constants.py +21 -0
  201. synth_ai/core/tracing_v3/db_config.py +182 -0
  202. synth_ai/core/tracing_v3/decorators.py +401 -0
  203. synth_ai/core/tracing_v3/llm_call_record_helpers.py +437 -0
  204. synth_ai/core/tracing_v3/migration_helper.py +119 -0
  205. synth_ai/core/tracing_v3/session_tracer.py +542 -0
  206. synth_ai/core/tracing_v3/storage/base.py +211 -0
  207. synth_ai/core/tracing_v3/storage/config.py +109 -0
  208. synth_ai/core/tracing_v3/storage/factory.py +39 -0
  209. synth_ai/core/tracing_v3/trace_utils.py +326 -0
  210. synth_ai/core/tracing_v3/turso/daemon.py +278 -0
  211. synth_ai/core/tracing_v3/turso/models.py +470 -0
  212. synth_ai/core/tracing_v3/turso/native_manager.py +1385 -0
  213. synth_ai/core/tracing_v3/utils.py +108 -0
  214. synth_ai/core/urls.py +18 -0
  215. synth_ai/core/user_config.py +137 -0
  216. synth_ai/core/uvicorn.py +222 -0
  217. synth_ai/data/__init__.py +110 -0
  218. synth_ai/data/enums.py +141 -0
  219. synth_ai/data/rewards.py +152 -0
  220. synth_ai/data/specs.py +36 -0
  221. synth_ai/data/traces.py +35 -0
  222. synth_ai/products/__init__.py +6 -0
  223. synth_ai/products/graph_evolve/__init__.py +46 -0
  224. synth_ai/products/graph_evolve/client.py +226 -0
  225. synth_ai/products/graph_evolve/config.py +591 -0
  226. synth_ai/products/graph_evolve/converters/__init__.py +42 -0
  227. synth_ai/products/graph_evolve/converters/openai_sft.py +484 -0
  228. synth_ai/products/graph_evolve/examples/hotpotqa/config.toml +109 -0
  229. synth_ai/products/graph_evolve/run.py +222 -0
  230. synth_ai/sdk/__init__.py +119 -0
  231. synth_ai/sdk/api/__init__.py +1 -0
  232. synth_ai/sdk/api/models/supported.py +514 -0
  233. synth_ai/sdk/api/research_agent/__init__.py +86 -0
  234. synth_ai/sdk/api/research_agent/cli.py +428 -0
  235. synth_ai/sdk/api/research_agent/config.py +357 -0
  236. synth_ai/sdk/api/research_agent/job.py +717 -0
  237. synth_ai/sdk/api/train/__init__.py +85 -0
  238. synth_ai/sdk/api/train/builders.py +895 -0
  239. synth_ai/sdk/api/train/cli.py +2188 -0
  240. synth_ai/sdk/api/train/config_finder.py +267 -0
  241. synth_ai/sdk/api/train/configs/__init__.py +65 -0
  242. synth_ai/sdk/api/train/configs/prompt_learning.py +1706 -0
  243. synth_ai/sdk/api/train/configs/rl.py +188 -0
  244. synth_ai/sdk/api/train/configs/sft.py +99 -0
  245. synth_ai/sdk/api/train/configs/shared.py +81 -0
  246. synth_ai/sdk/api/train/context_learning.py +312 -0
  247. synth_ai/sdk/api/train/env_resolver.py +418 -0
  248. synth_ai/sdk/api/train/graph_validators.py +216 -0
  249. synth_ai/sdk/api/train/graphgen.py +984 -0
  250. synth_ai/sdk/api/train/graphgen_models.py +823 -0
  251. synth_ai/sdk/api/train/graphgen_validators.py +109 -0
  252. synth_ai/sdk/api/train/pollers.py +124 -0
  253. synth_ai/sdk/api/train/progress/__init__.py +97 -0
  254. synth_ai/sdk/api/train/progress/dataclasses.py +569 -0
  255. synth_ai/sdk/api/train/progress/events.py +326 -0
  256. synth_ai/sdk/api/train/progress/results.py +428 -0
  257. synth_ai/sdk/api/train/progress/tracker.py +641 -0
  258. synth_ai/sdk/api/train/prompt_learning.py +470 -0
  259. synth_ai/sdk/api/train/rl.py +442 -0
  260. synth_ai/sdk/api/train/sft.py +396 -0
  261. synth_ai/sdk/api/train/summary.py +522 -0
  262. synth_ai/sdk/api/train/supported_algos.py +147 -0
  263. synth_ai/sdk/api/train/task_app.py +331 -0
  264. synth_ai/sdk/api/train/utils.py +279 -0
  265. synth_ai/sdk/api/train/validators.py +2424 -0
  266. synth_ai/sdk/baseline/__init__.py +25 -0
  267. synth_ai/sdk/baseline/config.py +209 -0
  268. synth_ai/sdk/baseline/discovery.py +216 -0
  269. synth_ai/sdk/baseline/execution.py +154 -0
  270. synth_ai/sdk/graphs/__init__.py +15 -0
  271. synth_ai/sdk/graphs/completions.py +570 -0
  272. synth_ai/sdk/inference/__init__.py +6 -0
  273. synth_ai/sdk/inference/client.py +128 -0
  274. synth_ai/sdk/jobs/__init__.py +16 -0
  275. synth_ai/sdk/jobs/client.py +371 -0
  276. synth_ai/sdk/judging/__init__.py +15 -0
  277. synth_ai/sdk/judging/base.py +24 -0
  278. synth_ai/sdk/judging/client.py +191 -0
  279. synth_ai/sdk/judging/schemas.py +222 -0
  280. synth_ai/sdk/learning/__init__.py +69 -0
  281. synth_ai/sdk/learning/client.py +240 -0
  282. synth_ai/sdk/learning/ft_client.py +7 -0
  283. synth_ai/sdk/learning/health.py +49 -0
  284. synth_ai/sdk/learning/jobs.py +202 -0
  285. synth_ai/sdk/learning/prompt_extraction.py +334 -0
  286. synth_ai/sdk/learning/prompt_learning_client.py +455 -0
  287. synth_ai/sdk/learning/prompt_learning_types.py +185 -0
  288. synth_ai/sdk/learning/rl/client.py +268 -0
  289. synth_ai/sdk/learning/rl/contracts.py +27 -0
  290. synth_ai/sdk/learning/rl/env_keys.py +166 -0
  291. synth_ai/sdk/learning/rl/secrets.py +13 -0
  292. synth_ai/sdk/learning/sft/client.py +95 -0
  293. synth_ai/sdk/learning/sft/config.py +270 -0
  294. synth_ai/sdk/learning/sft/data.py +698 -0
  295. synth_ai/sdk/learning/validators.py +52 -0
  296. synth_ai/sdk/research_agent/__init__.py +34 -0
  297. synth_ai/sdk/research_agent/container_builder.py +328 -0
  298. synth_ai/sdk/research_agent/container_spec.py +198 -0
  299. synth_ai/sdk/research_agent/defaults.py +34 -0
  300. synth_ai/sdk/research_agent/results_collector.py +69 -0
  301. synth_ai/sdk/specs/__init__.py +46 -0
  302. synth_ai/sdk/specs/dataclasses.py +149 -0
  303. synth_ai/sdk/specs/loader.py +144 -0
  304. synth_ai/sdk/specs/serializer.py +199 -0
  305. synth_ai/sdk/specs/validation.py +250 -0
  306. synth_ai/sdk/streaming/__init__.py +35 -0
  307. synth_ai/sdk/streaming/config.py +94 -0
  308. synth_ai/sdk/streaming/handlers.py +1997 -0
  309. synth_ai/sdk/streaming/streamer.py +704 -0
  310. synth_ai/sdk/streaming/types.py +112 -0
  311. synth_ai/sdk/task/__init__.py +151 -0
  312. synth_ai/sdk/task/apps/__init__.py +133 -0
  313. synth_ai/sdk/task/config.py +261 -0
  314. synth_ai/sdk/task/contracts.py +298 -0
  315. synth_ai/sdk/task/datasets.py +108 -0
  316. synth_ai/sdk/task/in_process.py +1190 -0
  317. synth_ai/sdk/task/in_process_runner.py +309 -0
  318. synth_ai/sdk/task/inference_api.py +299 -0
  319. synth_ai/sdk/task/proxy.py +287 -0
  320. synth_ai/sdk/task/rubrics/__init__.py +55 -0
  321. synth_ai/sdk/task/rubrics/loaders.py +156 -0
  322. synth_ai/sdk/task/rubrics.py +219 -0
  323. synth_ai/sdk/task/server.py +580 -0
  324. synth_ai/sdk/task/trace_correlation_helpers.py +506 -0
  325. synth_ai/sdk/task/tracing_utils.py +95 -0
  326. synth_ai/sdk/task/validators.py +456 -0
  327. synth_ai/sdk/tracing/__init__.py +39 -0
  328. synth_ai/sdk/training/__init__.py +102 -0
  329. synth_ai/sdk/usage/__init__.py +37 -0
  330. synth_ai/sdk/usage/client.py +171 -0
  331. synth_ai/sdk/usage/models.py +261 -0
  332. synth_ai/utils/__init__.py +213 -0
  333. synth_ai-0.4.1.dist-info/METADATA +195 -0
  334. synth_ai-0.4.1.dist-info/RECORD +379 -0
  335. synth_ai-0.4.1.dist-info/top_level.txt +1 -0
  336. examples/__init__.py +0 -16
  337. examples/analyze_semantic_words.sh +0 -17
  338. examples/crafter_debug_render.py +0 -186
  339. examples/dev/qwen3_32b_qlora_4xh100.toml +0 -40
  340. examples/multi_step/configs/README_verilog_rl.md +0 -77
  341. examples/multi_step/configs/VERILOG_REWARDS.md +0 -90
  342. examples/multi_step/configs/VERILOG_RL_CHECKLIST.md +0 -183
  343. examples/multi_step/configs/crafter_eval_synth_qwen4b.toml +0 -35
  344. examples/multi_step/configs/crafter_eval_text_only_groq_qwen32b.toml +0 -36
  345. examples/multi_step/configs/crafter_rl_outcome.toml +0 -74
  346. examples/multi_step/configs/crafter_rl_stepwise_hosted_judge.toml +0 -187
  347. examples/multi_step/configs/crafter_rl_stepwise_shaped.toml +0 -83
  348. examples/multi_step/configs/crafter_rl_stepwise_simple.toml +0 -78
  349. examples/multi_step/configs/crafter_synth_backend.md +0 -40
  350. examples/multi_step/configs/verilog_eval_groq_qwen32b.toml +0 -31
  351. examples/multi_step/configs/verilog_eval_synth_qwen8b.toml +0 -33
  352. examples/multi_step/configs/verilog_rl_lora.toml +0 -190
  353. examples/multi_step/crafter_rl_lora.md +0 -70
  354. examples/multi_step/judges/crafter_backend_judge.py +0 -220
  355. examples/multi_step/judges/verilog_backend_judge.py +0 -234
  356. examples/multi_step/readme.md +0 -48
  357. examples/multi_step/sse_metrics_streaming_notes.md +0 -357
  358. examples/multi_step/task_app_config_notes.md +0 -494
  359. examples/multi_step/verilog_rl_lora.md +0 -218
  360. examples/qwen_coder/README.md +0 -102
  361. examples/qwen_coder/_shared.py +0 -113
  362. examples/qwen_coder/configs/coder_lora_30b.toml +0 -61
  363. examples/qwen_coder/configs/coder_lora_4b.toml +0 -57
  364. examples/qwen_coder/configs/coder_lora_small.toml +0 -58
  365. examples/qwen_coder/generate_dataset.py +0 -98
  366. examples/qwen_coder/infer_ft_smoke.py +0 -65
  367. examples/qwen_coder/infer_prod_proxy.py +0 -73
  368. examples/qwen_coder/infer_via_synth.py +0 -87
  369. examples/qwen_coder/scripts/infer_coder.sh +0 -19
  370. examples/qwen_coder/scripts/train_coder_30b.sh +0 -22
  371. examples/qwen_coder/sft_full_17b.py +0 -103
  372. examples/qwen_coder/sft_lora_30b.py +0 -110
  373. examples/qwen_coder/subset_jsonl.py +0 -39
  374. examples/qwen_coder/todos.md +0 -38
  375. examples/qwen_coder/validate_jsonl.py +0 -60
  376. examples/rl/README.md +0 -169
  377. examples/rl/download_dataset.py +0 -80
  378. examples/run_crafter_demo.sh +0 -10
  379. examples/sft/README.md +0 -139
  380. examples/sft/configs/crafter_fft_qwen0p6b.toml +0 -44
  381. examples/sft/configs/crafter_lora_qwen0p6b.toml +0 -45
  382. examples/sft/evaluate.py +0 -119
  383. examples/sft/export_dataset.py +0 -117
  384. examples/sft/generate_traces.py +0 -164
  385. examples/swe/__init__.py +0 -12
  386. examples/swe/task_app/README.md +0 -105
  387. examples/swe/task_app/__init__.py +0 -2
  388. examples/swe/task_app/grpo_swe_mini.py +0 -601
  389. examples/swe/task_app/grpo_swe_mini_task_app.py +0 -136
  390. examples/swe/task_app/hosted/README.md +0 -173
  391. examples/swe/task_app/hosted/__init__.py +0 -5
  392. examples/swe/task_app/hosted/branching.py +0 -143
  393. examples/swe/task_app/hosted/environment_routes.py +0 -1289
  394. examples/swe/task_app/hosted/envs/__init__.py +0 -1
  395. examples/swe/task_app/hosted/envs/crafter/__init__.py +0 -6
  396. examples/swe/task_app/hosted/envs/crafter/app.py +0 -1
  397. examples/swe/task_app/hosted/envs/crafter/environment.py +0 -522
  398. examples/swe/task_app/hosted/envs/crafter/policy.py +0 -478
  399. examples/swe/task_app/hosted/envs/crafter/react_agent.py +0 -108
  400. examples/swe/task_app/hosted/envs/crafter/shared.py +0 -305
  401. examples/swe/task_app/hosted/envs/crafter/tools.py +0 -47
  402. examples/swe/task_app/hosted/envs/mini_swe/__init__.py +0 -8
  403. examples/swe/task_app/hosted/envs/mini_swe/environment.py +0 -1164
  404. examples/swe/task_app/hosted/envs/mini_swe/policy.py +0 -355
  405. examples/swe/task_app/hosted/envs/mini_swe/shared.py +0 -83
  406. examples/swe/task_app/hosted/envs/mini_swe/tools.py +0 -96
  407. examples/swe/task_app/hosted/hosted_app.py +0 -204
  408. examples/swe/task_app/hosted/inference/__init__.py +0 -5
  409. examples/swe/task_app/hosted/inference/openai_client.py +0 -618
  410. examples/swe/task_app/hosted/main.py +0 -100
  411. examples/swe/task_app/hosted/policy_routes.py +0 -1079
  412. examples/swe/task_app/hosted/registry.py +0 -195
  413. examples/swe/task_app/hosted/rollout.py +0 -1911
  414. examples/swe/task_app/hosted/storage/__init__.py +0 -5
  415. examples/swe/task_app/hosted/storage/volume.py +0 -211
  416. examples/swe/task_app/hosted/test_agents.py +0 -161
  417. examples/swe/task_app/hosted/test_service.py +0 -136
  418. examples/swe/task_app/hosted/utils.py +0 -62
  419. examples/task_apps/IMAGE_ONLY_EVAL_QUICKSTART.md +0 -258
  420. examples/task_apps/TESTING.md +0 -275
  421. examples/task_apps/crafter/CREATE_SFT_DATASET.md +0 -273
  422. examples/task_apps/crafter/EVAL_IMAGE_ONLY_RESULTS.md +0 -152
  423. examples/task_apps/crafter/FILTER_COMMAND_STATUS.md +0 -174
  424. examples/task_apps/crafter/FILTER_COMMAND_SUCCESS.md +0 -268
  425. examples/task_apps/crafter/QUERY_EXAMPLES.md +0 -203
  426. examples/task_apps/crafter/README_IMAGE_ONLY_EVAL.md +0 -316
  427. examples/task_apps/crafter/__init__.py +0 -0
  428. examples/task_apps/crafter/eval_image_only_gpt4o.toml +0 -28
  429. examples/task_apps/crafter/eval_text_only_groq_llama.toml +0 -36
  430. examples/task_apps/crafter/filter_sft_dataset.toml +0 -16
  431. examples/task_apps/crafter/task_app/README.md +0 -42
  432. examples/task_apps/crafter/task_app/__init__.py +0 -5
  433. examples/task_apps/crafter/task_app/grpo_crafter.py +0 -973
  434. examples/task_apps/crafter/task_app/grpo_crafter_task_app.py +0 -146
  435. examples/task_apps/crafter/task_app/synth_envs_hosted/README.md +0 -173
  436. examples/task_apps/crafter/task_app/synth_envs_hosted/__init__.py +0 -5
  437. examples/task_apps/crafter/task_app/synth_envs_hosted/branching.py +0 -143
  438. examples/task_apps/crafter/task_app/synth_envs_hosted/environment_routes.py +0 -1226
  439. examples/task_apps/crafter/task_app/synth_envs_hosted/envs/__init__.py +0 -1
  440. examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/__init__.py +0 -6
  441. examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/app.py +0 -1
  442. examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/environment.py +0 -532
  443. examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/policy.py +0 -547
  444. examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/react_agent.py +0 -123
  445. examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/shared.py +0 -305
  446. examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/tools.py +0 -47
  447. examples/task_apps/crafter/task_app/synth_envs_hosted/hosted_app.py +0 -204
  448. examples/task_apps/crafter/task_app/synth_envs_hosted/inference/__init__.py +0 -5
  449. examples/task_apps/crafter/task_app/synth_envs_hosted/inference/openai_client.py +0 -704
  450. examples/task_apps/crafter/task_app/synth_envs_hosted/main.py +0 -100
  451. examples/task_apps/crafter/task_app/synth_envs_hosted/policy_routes.py +0 -1152
  452. examples/task_apps/crafter/task_app/synth_envs_hosted/registry.py +0 -195
  453. examples/task_apps/crafter/task_app/synth_envs_hosted/rollout.py +0 -2160
  454. examples/task_apps/crafter/task_app/synth_envs_hosted/storage/__init__.py +0 -5
  455. examples/task_apps/crafter/task_app/synth_envs_hosted/storage/volume.py +0 -211
  456. examples/task_apps/crafter/task_app/synth_envs_hosted/test_agents.py +0 -161
  457. examples/task_apps/crafter/task_app/synth_envs_hosted/test_service.py +0 -136
  458. examples/task_apps/crafter/task_app/synth_envs_hosted/utils.py +0 -218
  459. examples/task_apps/dev/pokemon_emerald/__init__.py +0 -2
  460. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/README.md +0 -811
  461. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/__init__.py +0 -120
  462. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/action.py +0 -160
  463. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/memory.py +0 -155
  464. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/perception.py +0 -69
  465. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/planning.py +0 -96
  466. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/simple.py +0 -1502
  467. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/system_prompt.py +0 -4
  468. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/grab_map.py +0 -68
  469. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/manual.py +0 -216
  470. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/__init__.py +0 -35
  471. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/emerald_utils.py +0 -631
  472. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/emulator.py +0 -1544
  473. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/enums.py +0 -1428
  474. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/memory_reader.py +0 -4848
  475. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/types.py +0 -41
  476. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/utils.py +0 -298
  477. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pyproject.toml +0 -95
  478. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/run.py +0 -204
  479. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/__init__.py +0 -0
  480. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/app.py +0 -2152
  481. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/client.py +0 -429
  482. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/frame_server.py +0 -155
  483. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/README.md +0 -78
  484. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/__init__.py +0 -0
  485. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/run_tests.py +0 -122
  486. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_agent_direct.py +0 -76
  487. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_agent_prompts.py +0 -413
  488. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_battle_state_formatting.py +0 -204
  489. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_dialogue_detection.py +0 -133
  490. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_dialogue_detection_comprehensive.py +0 -229
  491. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_direct_agent_emulator.py +0 -300
  492. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_fps_adjustment_pytest.py +0 -205
  493. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_house_to_outside_direct.py +0 -200
  494. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_house_to_outside_transition.py +0 -284
  495. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_map_ground_truth_comparison.py +0 -468
  496. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_memory_map.py +0 -575
  497. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_server_map_validation.py +0 -311
  498. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_torchic_state.py +0 -259
  499. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/__init__.py +0 -0
  500. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/anticheat.py +0 -372
  501. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/checkpoint.py +0 -296
  502. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/error_handler.py +0 -275
  503. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/get_local_ip.py +0 -22
  504. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/helpers.py +0 -44
  505. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/llm_logger.py +0 -514
  506. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_formatter.py +0 -415
  507. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_stitcher.py +0 -1763
  508. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_stitcher_singleton.py +0 -33
  509. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_trimmer.py +0 -106
  510. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_visualizer.py +0 -334
  511. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/ocr_dialogue.py +0 -1020
  512. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/recording.py +0 -188
  513. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/state_formatter.py +0 -1481
  514. examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/vlm.py +0 -862
  515. examples/task_apps/dev/pokemon_emerald/modal_app.py +0 -114
  516. examples/task_apps/dev/pokemon_emerald/task_app/README.md +0 -81
  517. examples/task_apps/dev/pokemon_emerald/task_app/__init__.py +0 -6
  518. examples/task_apps/dev/pokemon_emerald/task_app/pokemon_emerald.py +0 -685
  519. examples/task_apps/enron/__init__.py +0 -1
  520. examples/task_apps/enron/eval_groq_qwen32.toml +0 -16
  521. examples/task_apps/enron/filter_sft.toml +0 -5
  522. examples/task_apps/enron/task_app/README.md +0 -14
  523. examples/task_apps/enron/task_app/__init__.py +0 -1
  524. examples/task_apps/enron/task_app/grpo_enron.py +0 -906
  525. examples/task_apps/enron/task_app/grpo_enron_task_app.py +0 -146
  526. examples/task_apps/enron/tests/__init__.py +0 -4
  527. examples/task_apps/enron/tests/conftest.py +0 -115
  528. examples/task_apps/enron/tests/integration/__init__.py +0 -4
  529. examples/task_apps/enron/tests/integration/test_enron_eval.py +0 -179
  530. examples/task_apps/enron/tests/integration/test_enron_rollout.py +0 -135
  531. examples/task_apps/enron/tests/unit/__init__.py +0 -4
  532. examples/task_apps/enron/tests/unit/test_enron_environment.py +0 -126
  533. examples/task_apps/math/README.md +0 -22
  534. examples/task_apps/math/__init__.py +0 -0
  535. examples/task_apps/math/math_single_step.py +0 -1000
  536. examples/task_apps/math/math_task_app.py +0 -115
  537. examples/task_apps/pokemon_battle/__init__.py +0 -2
  538. examples/task_apps/pokemon_battle/modal_app.py +0 -104
  539. examples/task_apps/pokemon_battle/task_app/README.md +0 -68
  540. examples/task_apps/pokemon_battle/task_app/__init__.py +0 -6
  541. examples/task_apps/pokemon_battle/task_app/pokemon_showdown.py +0 -932
  542. examples/task_apps/pokemon_red/EVAL_IMAGE_ONLY_COMPLETE.md +0 -283
  543. examples/task_apps/pokemon_red/EVAL_IMAGE_ONLY_STATUS.md +0 -155
  544. examples/task_apps/pokemon_red/README.md +0 -357
  545. examples/task_apps/pokemon_red/README_IMAGE_ONLY_EVAL.md +0 -415
  546. examples/task_apps/pokemon_red/__init__.py +0 -3
  547. examples/task_apps/pokemon_red/eval_image_only_gpt4o.toml +0 -29
  548. examples/task_apps/pokemon_red/eval_pokemon_red_policy.py +0 -225
  549. examples/task_apps/pokemon_red/pallet_town_rl_config.toml +0 -75
  550. examples/task_apps/pokemon_red/task_app.py +0 -799
  551. examples/task_apps/pokemon_red/test_pallet_town_rewards.py +0 -193
  552. examples/task_apps/sokoban/README.md +0 -307
  553. examples/task_apps/sokoban/__init__.py +0 -3
  554. examples/task_apps/sokoban/eval_groq_qwen32.toml +0 -16
  555. examples/task_apps/sokoban/eval_openai_gpt5.toml +0 -16
  556. examples/task_apps/sokoban/filter_sft.toml +0 -5
  557. examples/task_apps/sokoban/task_app.py +0 -1058
  558. examples/task_apps/sokoban/tests/__init__.py +0 -4
  559. examples/task_apps/sokoban/tests/conftest.py +0 -113
  560. examples/task_apps/sokoban/tests/integration/__init__.py +0 -4
  561. examples/task_apps/sokoban/tests/integration/test_sokoban_eval.py +0 -57
  562. examples/task_apps/sokoban/tests/integration/test_sokoban_rollout.py +0 -198
  563. examples/task_apps/sokoban/tests/unit/__init__.py +0 -4
  564. examples/task_apps/sokoban/tests/unit/test_sokoban_environment.py +0 -114
  565. examples/task_apps/verilog/__init__.py +0 -1
  566. examples/task_apps/verilog/eval_groq_qwen32b.toml +0 -24
  567. examples/task_apps/verilog/filter_sft.toml +0 -5
  568. examples/task_apps/verilog/task_app/README.md +0 -12
  569. examples/task_apps/verilog/task_app/__init__.py +0 -1
  570. examples/task_apps/verilog/task_app/grpo_verilog.py +0 -1166
  571. examples/task_apps/verilog/task_app/grpo_verilog_task_app.py +0 -145
  572. examples/task_apps/verilog/tests/__init__.py +0 -4
  573. examples/task_apps/verilog/tests/conftest.py +0 -115
  574. examples/task_apps/verilog/tests/integration/__init__.py +0 -4
  575. examples/task_apps/verilog/tests/integration/test_verilog_eval.py +0 -181
  576. examples/task_apps/verilog/tests/integration/test_verilog_rollout.py +0 -55
  577. examples/task_apps/verilog/tests/unit/__init__.py +0 -4
  578. examples/task_apps/verilog/tests/unit/test_verilog_scoring.py +0 -118
  579. examples/vlm/PROPOSAL.md +0 -53
  580. examples/vlm/README.md +0 -68
  581. examples/vlm/configs/crafter_vlm_gpt4o.toml +0 -44
  582. examples/vlm/crafter_image_only_agent.py +0 -207
  583. examples/vlm/crafter_openai_vlm_agent.py +0 -277
  584. examples/vlm/filter_image_rows.py +0 -63
  585. examples/vlm/run_crafter_vlm_benchmark.py +0 -316
  586. examples/warming_up_to_rl/analyze_trace_db.py +0 -422
  587. examples/warming_up_to_rl/configs/crafter_fft.toml +0 -48
  588. examples/warming_up_to_rl/configs/crafter_fft_4b.toml +0 -54
  589. examples/warming_up_to_rl/configs/eval_fft_qwen4b.toml +0 -20
  590. examples/warming_up_to_rl/configs/eval_groq_qwen32b.toml +0 -13
  591. examples/warming_up_to_rl/configs/eval_modal_qwen4b.toml +0 -23
  592. examples/warming_up_to_rl/configs/eval_stepwise_complex.toml +0 -35
  593. examples/warming_up_to_rl/configs/eval_stepwise_consistent.toml +0 -26
  594. examples/warming_up_to_rl/configs/eval_stepwise_per_achievement.toml +0 -36
  595. examples/warming_up_to_rl/configs/eval_stepwise_simple.toml +0 -32
  596. examples/warming_up_to_rl/configs/rl_from_base_qwen4b.toml +0 -83
  597. examples/warming_up_to_rl/configs/rl_from_ft.toml +0 -56
  598. examples/warming_up_to_rl/export_trace_sft.py +0 -723
  599. examples/warming_up_to_rl/groq_test.py +0 -97
  600. examples/warming_up_to_rl/manage_secrets.py +0 -131
  601. examples/warming_up_to_rl/old/event_rewards.md +0 -234
  602. examples/warming_up_to_rl/old/notes.md +0 -73
  603. examples/warming_up_to_rl/readme.md +0 -179
  604. examples/warming_up_to_rl/run_eval.py +0 -736
  605. examples/warming_up_to_rl/run_fft_and_save.py +0 -380
  606. examples/warming_up_to_rl/run_local_rollout.py +0 -239
  607. examples/warming_up_to_rl/run_local_rollout_modal.py +0 -248
  608. examples/warming_up_to_rl/run_local_rollout_parallel.py +0 -405
  609. examples/warming_up_to_rl/run_local_rollout_traced.py +0 -477
  610. examples/warming_up_to_rl/run_rl_and_save.py +0 -124
  611. examples/warming_up_to_rl/run_rollout_remote.py +0 -156
  612. examples/workflows/__init__.py +0 -0
  613. examples/workflows/math_rl/__init__.py +0 -0
  614. examples/workflows/math_rl/configs/eval_base_qwen.toml +0 -15
  615. examples/workflows/math_rl/configs/eval_rl_qwen.toml +0 -11
  616. examples/workflows/math_rl/configs/rl_from_base_qwen.toml +0 -35
  617. examples/workflows/math_rl/configs/rl_from_base_qwen17.toml +0 -74
  618. examples/workflows/math_rl/configs/rl_from_ft_qwen.toml +0 -35
  619. examples/workflows/math_rl/download_dataset.py +0 -80
  620. examples/workflows/math_rl/run_eval.py +0 -436
  621. examples/workflows/math_rl/run_rl_and_save.py +0 -111
  622. synth_ai/api/models/supported.py +0 -377
  623. synth_ai/api/train/__init__.py +0 -5
  624. synth_ai/api/train/builders.py +0 -351
  625. synth_ai/api/train/cli.py +0 -635
  626. synth_ai/api/train/config_finder.py +0 -228
  627. synth_ai/api/train/configs/__init__.py +0 -44
  628. synth_ai/api/train/configs/rl.py +0 -134
  629. synth_ai/api/train/configs/sft.py +0 -95
  630. synth_ai/api/train/configs/shared.py +0 -24
  631. synth_ai/api/train/env_resolver.py +0 -349
  632. synth_ai/api/train/pollers.py +0 -75
  633. synth_ai/api/train/supported_algos.py +0 -147
  634. synth_ai/api/train/task_app.py +0 -195
  635. synth_ai/api/train/utils.py +0 -225
  636. synth_ai/cli/_modal_wrapper.py +0 -29
  637. synth_ai/cli/_storage.py +0 -20
  638. synth_ai/cli/_typer_patch.py +0 -49
  639. synth_ai/cli/_validate_task_app.py +0 -11
  640. synth_ai/cli/balance.py +0 -216
  641. synth_ai/cli/calc.py +0 -84
  642. synth_ai/cli/demo.py +0 -165
  643. synth_ai/cli/legacy_root_backup.py +0 -468
  644. synth_ai/cli/man.py +0 -106
  645. synth_ai/cli/recent.py +0 -132
  646. synth_ai/cli/rl_demo.py +0 -254
  647. synth_ai/cli/status.py +0 -134
  648. synth_ai/cli/task_apps.py +0 -4523
  649. synth_ai/cli/traces.py +0 -164
  650. synth_ai/cli/tui.py +0 -57
  651. synth_ai/cli/watch.py +0 -506
  652. synth_ai/compound/cais.py +0 -0
  653. synth_ai/config/base_url.py +0 -107
  654. synth_ai/core/experiment.py +0 -13
  655. synth_ai/core/system.py +0 -15
  656. synth_ai/demo_registry.py +0 -295
  657. synth_ai/demos/core/__init__.py +0 -1
  658. synth_ai/demos/core/cli.py +0 -1718
  659. synth_ai/demos/demo_task_apps/core.py +0 -440
  660. synth_ai/demos/demo_task_apps/crafter/grpo_crafter_task_app.py +0 -184
  661. synth_ai/demos/demo_task_apps/math/deploy_task_app.sh +0 -22
  662. synth_ai/demos/demo_task_apps/math/modal_task_app.py +0 -739
  663. synth_ai/demos/demo_task_apps/math/task_app_entry.py +0 -37
  664. synth_ai/environments/__init__.py +0 -31
  665. synth_ai/environments/environment/__init__.py +0 -1
  666. synth_ai/environments/environment/artifacts/__init__.py +0 -1
  667. synth_ai/environments/environment/artifacts/base.py +0 -52
  668. synth_ai/environments/environment/core.py +0 -67
  669. synth_ai/environments/environment/db/__init__.py +0 -1
  670. synth_ai/environments/environment/db/sqlite.py +0 -45
  671. synth_ai/environments/environment/registry.py +0 -233
  672. synth_ai/environments/environment/resources/sqlite.py +0 -45
  673. synth_ai/environments/environment/results.py +0 -1
  674. synth_ai/environments/environment/rewards/__init__.py +0 -1
  675. synth_ai/environments/environment/rewards/core.py +0 -29
  676. synth_ai/environments/environment/shared_engine.py +0 -26
  677. synth_ai/environments/environment/tools/__init__.py +0 -200
  678. synth_ai/environments/examples/__init__.py +0 -1
  679. synth_ai/environments/examples/bandit/__init__.py +0 -33
  680. synth_ai/environments/examples/bandit/engine.py +0 -302
  681. synth_ai/environments/examples/bandit/environment.py +0 -194
  682. synth_ai/environments/examples/bandit/taskset.py +0 -200
  683. synth_ai/environments/examples/crafter_classic/__init__.py +0 -8
  684. synth_ai/environments/examples/crafter_classic/agent_demos/analyze_semantic_words_markdown.py +0 -250
  685. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_comprehensive_evaluation.py +0 -59
  686. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_evaluation_browser.py +0 -152
  687. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_evaluation_config.toml +0 -24
  688. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_evaluation_framework.py +0 -1194
  689. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/crafter_synth_config.toml +0 -56
  690. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/filter_config_modal.toml +0 -32
  691. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/filter_traces_sft_turso.py +0 -738
  692. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/kick_off_ft_modal.py +0 -384
  693. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_action_results.py +0 -53
  694. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_agent_actions.py +0 -178
  695. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_latest_run.py +0 -222
  696. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_lm_traces.py +0 -183
  697. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_no_rewards.py +0 -210
  698. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_trace_issue.py +0 -206
  699. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/check_db_schema.py +0 -49
  700. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/check_latest_results.py +0 -64
  701. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/debug_agent_responses.py +0 -88
  702. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/quick_trace_check.py +0 -77
  703. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/compare_experiments.py +0 -324
  704. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/filter_traces_sft_turso.py +0 -580
  705. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/kick_off_ft_oai.py +0 -362
  706. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/multi_model_config.toml +0 -49
  707. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/analyze_enhanced_hooks.py +0 -332
  708. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/analyze_hook_events.py +0 -97
  709. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/analyze_hook_results.py +0 -217
  710. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/check_hook_storage.py +0 -87
  711. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/check_seeds.py +0 -88
  712. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/compare_seed_performance.py +0 -195
  713. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/custom_eval_pipelines.py +0 -400
  714. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/plot_hook_frequency.py +0 -195
  715. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/seed_analysis_summary.py +0 -56
  716. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/run_rollouts_for_models_and_compare_v3.py +0 -858
  717. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_quick_evaluation.py +0 -52
  718. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_react_agent.py +0 -874
  719. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_trace_evaluation.py +0 -1412
  720. synth_ai/environments/examples/crafter_classic/agent_demos/example_v3_usage.py +0 -216
  721. synth_ai/environments/examples/crafter_classic/agent_demos/old/compare_traces.py +0 -296
  722. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_comprehensive_evaluation.py +0 -58
  723. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_env_serialization.py +0 -464
  724. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_evaluation_browser.py +0 -152
  725. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_quick_evaluation.py +0 -51
  726. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_trace_evaluation.py +0 -1412
  727. synth_ai/environments/examples/crafter_classic/agent_demos/old/debug_player_loss.py +0 -112
  728. synth_ai/environments/examples/crafter_classic/agent_demos/old/diagnose_service.py +0 -203
  729. synth_ai/environments/examples/crafter_classic/agent_demos/old/diagnose_slowness.py +0 -305
  730. synth_ai/environments/examples/crafter_classic/agent_demos/old/eval_by_difficulty.py +0 -126
  731. synth_ai/environments/examples/crafter_classic/agent_demos/old/eval_example.py +0 -94
  732. synth_ai/environments/examples/crafter_classic/agent_demos/old/explore_saved_states.py +0 -142
  733. synth_ai/environments/examples/crafter_classic/agent_demos/old/filter_traces_sft.py +0 -26
  734. synth_ai/environments/examples/crafter_classic/agent_demos/old/filter_traces_sft_OLD.py +0 -984
  735. synth_ai/environments/examples/crafter_classic/agent_demos/old/generate_ft_data_gemini.py +0 -724
  736. synth_ai/environments/examples/crafter_classic/agent_demos/old/generate_ft_data_modal.py +0 -386
  737. synth_ai/environments/examples/crafter_classic/agent_demos/old/generate_ft_metadata.py +0 -205
  738. synth_ai/environments/examples/crafter_classic/agent_demos/old/kick_off_ft_gemini.py +0 -150
  739. synth_ai/environments/examples/crafter_classic/agent_demos/old/kick_off_ft_modal.py +0 -283
  740. synth_ai/environments/examples/crafter_classic/agent_demos/old/prepare_vertex_ft.py +0 -280
  741. synth_ai/environments/examples/crafter_classic/agent_demos/old/profile_env_slowness.py +0 -456
  742. synth_ai/environments/examples/crafter_classic/agent_demos/old/replicate_issue.py +0 -166
  743. synth_ai/environments/examples/crafter_classic/agent_demos/old/run_and_eval.py +0 -102
  744. synth_ai/environments/examples/crafter_classic/agent_demos/old/run_comparison.py +0 -128
  745. synth_ai/environments/examples/crafter_classic/agent_demos/old/run_qwen_rollouts.py +0 -655
  746. synth_ai/environments/examples/crafter_classic/agent_demos/old/trace_eval_OLD.py +0 -202
  747. synth_ai/environments/examples/crafter_classic/agent_demos/old/validate_openai_format.py +0 -166
  748. synth_ai/environments/examples/crafter_classic/config_logging.py +0 -111
  749. synth_ai/environments/examples/crafter_classic/debug_translation.py +0 -0
  750. synth_ai/environments/examples/crafter_classic/engine.py +0 -579
  751. synth_ai/environments/examples/crafter_classic/engine_deterministic_patch.py +0 -64
  752. synth_ai/environments/examples/crafter_classic/engine_helpers/action_map.py +0 -6
  753. synth_ai/environments/examples/crafter_classic/engine_helpers/serialization.py +0 -75
  754. synth_ai/environments/examples/crafter_classic/engine_serialization_patch_v3.py +0 -267
  755. synth_ai/environments/examples/crafter_classic/environment.py +0 -495
  756. synth_ai/environments/examples/crafter_classic/taskset.py +0 -233
  757. synth_ai/environments/examples/crafter_classic/trace_hooks_v3.py +0 -228
  758. synth_ai/environments/examples/crafter_classic/world_config_patch_simple.py +0 -299
  759. synth_ai/environments/examples/crafter_custom/__init__.py +0 -4
  760. synth_ai/environments/examples/crafter_custom/agent_demos/__init__.py +0 -1
  761. synth_ai/environments/examples/crafter_custom/agent_demos/trace_eval.py +0 -202
  762. synth_ai/environments/examples/crafter_custom/crafter/__init__.py +0 -7
  763. synth_ai/environments/examples/crafter_custom/crafter/config.py +0 -182
  764. synth_ai/environments/examples/crafter_custom/crafter/constants.py +0 -8
  765. synth_ai/environments/examples/crafter_custom/crafter/engine.py +0 -269
  766. synth_ai/environments/examples/crafter_custom/crafter/env.py +0 -262
  767. synth_ai/environments/examples/crafter_custom/crafter/objects.py +0 -417
  768. synth_ai/environments/examples/crafter_custom/crafter/recorder.py +0 -187
  769. synth_ai/environments/examples/crafter_custom/crafter/worldgen.py +0 -118
  770. synth_ai/environments/examples/crafter_custom/dataset_builder.py +0 -373
  771. synth_ai/environments/examples/crafter_custom/environment.py +0 -312
  772. synth_ai/environments/examples/crafter_custom/old/analyze_diamond_issue.py +0 -159
  773. synth_ai/environments/examples/crafter_custom/old/analyze_diamond_spawning.py +0 -158
  774. synth_ai/environments/examples/crafter_custom/old/compare_worlds.py +0 -71
  775. synth_ai/environments/examples/crafter_custom/old/dataset_stats.py +0 -105
  776. synth_ai/environments/examples/crafter_custom/old/diamond_spawning_summary.py +0 -119
  777. synth_ai/environments/examples/crafter_custom/old/example_dataset_usage.py +0 -52
  778. synth_ai/environments/examples/crafter_custom/run_dataset.py +0 -305
  779. synth_ai/environments/examples/enron/art_helpers/email_search_tools.py +0 -156
  780. synth_ai/environments/examples/enron/art_helpers/local_email_db.py +0 -281
  781. synth_ai/environments/examples/enron/art_helpers/types_enron.py +0 -25
  782. synth_ai/environments/examples/enron/engine.py +0 -300
  783. synth_ai/environments/examples/enron/environment.py +0 -234
  784. synth_ai/environments/examples/enron/taskset.py +0 -112
  785. synth_ai/environments/examples/enron/units/keyword_stats.py +0 -112
  786. synth_ai/environments/examples/minigrid/__init__.py +0 -48
  787. synth_ai/environments/examples/minigrid/agent_demos/minigrid_evaluation_framework.py +0 -1188
  788. synth_ai/environments/examples/minigrid/agent_demos/minigrid_quick_evaluation.py +0 -48
  789. synth_ai/environments/examples/minigrid/agent_demos/minigrid_react_agent.py +0 -562
  790. synth_ai/environments/examples/minigrid/agent_demos/minigrid_trace_evaluation.py +0 -221
  791. synth_ai/environments/examples/minigrid/engine.py +0 -589
  792. synth_ai/environments/examples/minigrid/environment.py +0 -274
  793. synth_ai/environments/examples/minigrid/environment_mapping.py +0 -242
  794. synth_ai/environments/examples/minigrid/puzzle_loader.py +0 -417
  795. synth_ai/environments/examples/minigrid/taskset.py +0 -583
  796. synth_ai/environments/examples/nethack/__init__.py +0 -7
  797. synth_ai/environments/examples/nethack/achievements.py +0 -337
  798. synth_ai/environments/examples/nethack/agent_demos/nethack_evaluation_framework.py +0 -981
  799. synth_ai/environments/examples/nethack/agent_demos/nethack_quick_evaluation.py +0 -74
  800. synth_ai/environments/examples/nethack/agent_demos/nethack_react_agent.py +0 -831
  801. synth_ai/environments/examples/nethack/engine.py +0 -739
  802. synth_ai/environments/examples/nethack/environment.py +0 -256
  803. synth_ai/environments/examples/nethack/helpers/__init__.py +0 -41
  804. synth_ai/environments/examples/nethack/helpers/action_mapping.py +0 -301
  805. synth_ai/environments/examples/nethack/helpers/nle_wrapper.py +0 -402
  806. synth_ai/environments/examples/nethack/helpers/observation_utils.py +0 -433
  807. synth_ai/environments/examples/nethack/helpers/recording_wrapper.py +0 -200
  808. synth_ai/environments/examples/nethack/helpers/trajectory_recorder.py +0 -269
  809. synth_ai/environments/examples/nethack/helpers/visualization/replay_viewer.py +0 -308
  810. synth_ai/environments/examples/nethack/helpers/visualization/visualizer.py +0 -431
  811. synth_ai/environments/examples/nethack/taskset.py +0 -323
  812. synth_ai/environments/examples/red/__init__.py +0 -7
  813. synth_ai/environments/examples/red/agent_demos/__init__.py +0 -1
  814. synth_ai/environments/examples/red/config_logging.py +0 -110
  815. synth_ai/environments/examples/red/engine.py +0 -721
  816. synth_ai/environments/examples/red/engine_helpers/__init__.py +0 -1
  817. synth_ai/environments/examples/red/engine_helpers/memory_map.py +0 -35
  818. synth_ai/environments/examples/red/engine_helpers/reward_components.py +0 -276
  819. synth_ai/environments/examples/red/engine_helpers/reward_library/__init__.py +0 -142
  820. synth_ai/environments/examples/red/engine_helpers/reward_library/adaptive_rewards.py +0 -57
  821. synth_ai/environments/examples/red/engine_helpers/reward_library/battle_rewards.py +0 -284
  822. synth_ai/environments/examples/red/engine_helpers/reward_library/composite_rewards.py +0 -150
  823. synth_ai/environments/examples/red/engine_helpers/reward_library/economy_rewards.py +0 -138
  824. synth_ai/environments/examples/red/engine_helpers/reward_library/efficiency_rewards.py +0 -57
  825. synth_ai/environments/examples/red/engine_helpers/reward_library/exploration_rewards.py +0 -331
  826. synth_ai/environments/examples/red/engine_helpers/reward_library/novelty_rewards.py +0 -121
  827. synth_ai/environments/examples/red/engine_helpers/reward_library/pallet_town_progression.py +0 -477
  828. synth_ai/environments/examples/red/engine_helpers/reward_library/pallet_town_rewards.py +0 -559
  829. synth_ai/environments/examples/red/engine_helpers/reward_library/pokemon_rewards.py +0 -313
  830. synth_ai/environments/examples/red/engine_helpers/reward_library/social_rewards.py +0 -148
  831. synth_ai/environments/examples/red/engine_helpers/reward_library/story_rewards.py +0 -247
  832. synth_ai/environments/examples/red/engine_helpers/screen_analysis.py +0 -368
  833. synth_ai/environments/examples/red/engine_helpers/state_extraction.py +0 -172
  834. synth_ai/environments/examples/red/environment.py +0 -298
  835. synth_ai/environments/examples/red/taskset.py +0 -79
  836. synth_ai/environments/examples/red/units/__init__.py +0 -1
  837. synth_ai/environments/examples/sokoban/__init__.py +0 -1
  838. synth_ai/environments/examples/sokoban/agent_demos/sokoban_full_eval.py +0 -899
  839. synth_ai/environments/examples/sokoban/engine.py +0 -678
  840. synth_ai/environments/examples/sokoban/engine_helpers/__init__.py +0 -1
  841. synth_ai/environments/examples/sokoban/engine_helpers/room_utils.py +0 -657
  842. synth_ai/environments/examples/sokoban/engine_helpers/vendored/__init__.py +0 -18
  843. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/__init__.py +0 -3
  844. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/boxoban_env.py +0 -131
  845. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/render_utils.py +0 -370
  846. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/room_utils.py +0 -332
  847. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env.py +0 -306
  848. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_fixed_targets.py +0 -67
  849. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_pull.py +0 -115
  850. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_two_player.py +0 -123
  851. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_variations.py +0 -394
  852. synth_ai/environments/examples/sokoban/environment.py +0 -229
  853. synth_ai/environments/examples/sokoban/generate_verified_puzzles.py +0 -440
  854. synth_ai/environments/examples/sokoban/puzzle_loader.py +0 -312
  855. synth_ai/environments/examples/sokoban/taskset.py +0 -544
  856. synth_ai/environments/examples/tictactoe/__init__.py +0 -1
  857. synth_ai/environments/examples/tictactoe/engine.py +0 -368
  858. synth_ai/environments/examples/tictactoe/environment.py +0 -240
  859. synth_ai/environments/examples/tictactoe/taskset.py +0 -215
  860. synth_ai/environments/examples/verilog/__init__.py +0 -10
  861. synth_ai/environments/examples/verilog/engine.py +0 -421
  862. synth_ai/environments/examples/verilog/environment.py +0 -350
  863. synth_ai/environments/examples/verilog/taskset.py +0 -420
  864. synth_ai/environments/examples/wordle/__init__.py +0 -29
  865. synth_ai/environments/examples/wordle/engine.py +0 -398
  866. synth_ai/environments/examples/wordle/environment.py +0 -159
  867. synth_ai/environments/examples/wordle/helpers/generate_instances_wordfreq.py +0 -75
  868. synth_ai/environments/examples/wordle/taskset.py +0 -230
  869. synth_ai/environments/reproducibility/core.py +0 -42
  870. synth_ai/environments/reproducibility/helpers.py +0 -0
  871. synth_ai/environments/reproducibility/tree.py +0 -363
  872. synth_ai/environments/service/app.py +0 -97
  873. synth_ai/environments/service/core_routes.py +0 -1021
  874. synth_ai/environments/service/external_registry.py +0 -56
  875. synth_ai/environments/service/registry.py +0 -9
  876. synth_ai/environments/stateful/__init__.py +0 -1
  877. synth_ai/environments/stateful/core.py +0 -163
  878. synth_ai/environments/stateful/engine.py +0 -21
  879. synth_ai/environments/stateful/state.py +0 -7
  880. synth_ai/environments/tasks/api.py +0 -19
  881. synth_ai/environments/tasks/core.py +0 -81
  882. synth_ai/environments/tasks/filters.py +0 -40
  883. synth_ai/environments/tasks/utils.py +0 -90
  884. synth_ai/environments/v0_observability/history.py +0 -3
  885. synth_ai/environments/v0_observability/log.py +0 -2
  886. synth_ai/evals/__init__.py +0 -15
  887. synth_ai/evals/base.py +0 -13
  888. synth_ai/evals/client.py +0 -82
  889. synth_ai/handshake.py +0 -109
  890. synth_ai/http.py +0 -26
  891. synth_ai/http_client.py +0 -136
  892. synth_ai/inference/__init__.py +0 -5
  893. synth_ai/inference/client.py +0 -34
  894. synth_ai/jobs/client.py +0 -295
  895. synth_ai/judge_schemas.py +0 -127
  896. synth_ai/learning/__init__.py +0 -59
  897. synth_ai/learning/client.py +0 -241
  898. synth_ai/learning/ft_client.py +0 -7
  899. synth_ai/learning/health.py +0 -49
  900. synth_ai/learning/jobs.py +0 -201
  901. synth_ai/learning/rl/client.py +0 -267
  902. synth_ai/learning/rl/contracts.py +0 -27
  903. synth_ai/learning/rl/env_keys.py +0 -166
  904. synth_ai/learning/rl/secrets.py +0 -13
  905. synth_ai/learning/sft/client.py +0 -68
  906. synth_ai/learning/sft/config.py +0 -270
  907. synth_ai/learning/sft/data.py +0 -295
  908. synth_ai/learning/validators.py +0 -49
  909. synth_ai/lm/__init__.py +0 -25
  910. synth_ai/task/__init__.py +0 -121
  911. synth_ai/task/apps/__init__.py +0 -129
  912. synth_ai/task/config.py +0 -257
  913. synth_ai/task/contracts.py +0 -236
  914. synth_ai/task/datasets.py +0 -108
  915. synth_ai/task/proxy.py +0 -251
  916. synth_ai/task/rubrics/__init__.py +0 -56
  917. synth_ai/task/rubrics/loaders.py +0 -152
  918. synth_ai/task/server.py +0 -432
  919. synth_ai/task/trace_correlation_helpers.py +0 -315
  920. synth_ai/task/tracing_utils.py +0 -84
  921. synth_ai/task/validators.py +0 -418
  922. synth_ai/tracing_v3/__init__.py +0 -97
  923. synth_ai/tracing_v3/abstractions.py +0 -302
  924. synth_ai/tracing_v3/config.py +0 -84
  925. synth_ai/tracing_v3/db_config.py +0 -194
  926. synth_ai/tracing_v3/decorators.py +0 -398
  927. synth_ai/tracing_v3/llm_call_record_helpers.py +0 -391
  928. synth_ai/tracing_v3/migration_helper.py +0 -120
  929. synth_ai/tracing_v3/session_tracer.py +0 -540
  930. synth_ai/tracing_v3/storage/base.py +0 -210
  931. synth_ai/tracing_v3/storage/config.py +0 -75
  932. synth_ai/tracing_v3/storage/factory.py +0 -39
  933. synth_ai/tracing_v3/trace_utils.py +0 -317
  934. synth_ai/tracing_v3/turso/daemon.py +0 -151
  935. synth_ai/tracing_v3/turso/models.py +0 -469
  936. synth_ai/tracing_v3/turso/native_manager.py +0 -1209
  937. synth_ai/tracing_v3/utils.py +0 -108
  938. synth_ai/tui/__init__.py +0 -5
  939. synth_ai/tui/__main__.py +0 -13
  940. synth_ai/tui/cli/__init__.py +0 -1
  941. synth_ai/tui/cli/query_experiments.py +0 -164
  942. synth_ai/tui/cli/query_experiments_v3.py +0 -164
  943. synth_ai/tui/dashboard.py +0 -906
  944. synth_ai/v0/api/__init__.py +0 -8
  945. synth_ai/v0/api/models/__init__.py +0 -8
  946. synth_ai/v0/api/models/supported.py +0 -8
  947. synth_ai/v0/config/__init__.py +0 -15
  948. synth_ai/v0/config/base_url.py +0 -12
  949. synth_ai/v0/lm/__init__.py +0 -51
  950. synth_ai/v0/lm/caching/__init__.py +0 -0
  951. synth_ai/v0/lm/caching/constants.py +0 -6
  952. synth_ai/v0/lm/caching/dbs.py +0 -0
  953. synth_ai/v0/lm/caching/ephemeral.py +0 -100
  954. synth_ai/v0/lm/caching/handler.py +0 -137
  955. synth_ai/v0/lm/caching/initialize.py +0 -11
  956. synth_ai/v0/lm/caching/persistent.py +0 -114
  957. synth_ai/v0/lm/config.py +0 -115
  958. synth_ai/v0/lm/constants.py +0 -32
  959. synth_ai/v0/lm/core/__init__.py +0 -8
  960. synth_ai/v0/lm/core/all.py +0 -73
  961. synth_ai/v0/lm/core/exceptions.py +0 -5
  962. synth_ai/v0/lm/core/main.py +0 -331
  963. synth_ai/v0/lm/core/main_v3.py +0 -594
  964. synth_ai/v0/lm/core/synth_models.py +0 -35
  965. synth_ai/v0/lm/core/vendor_clients.py +0 -190
  966. synth_ai/v0/lm/cost/__init__.py +0 -0
  967. synth_ai/v0/lm/cost/monitor.py +0 -1
  968. synth_ai/v0/lm/cost/statefulness.py +0 -1
  969. synth_ai/v0/lm/injection.py +0 -80
  970. synth_ai/v0/lm/overrides.py +0 -206
  971. synth_ai/v0/lm/provider_support/__init__.py +0 -8
  972. synth_ai/v0/lm/provider_support/anthropic.py +0 -972
  973. synth_ai/v0/lm/provider_support/openai.py +0 -1139
  974. synth_ai/v0/lm/provider_support/suppress_logging.py +0 -31
  975. synth_ai/v0/lm/structured_outputs/__init__.py +0 -0
  976. synth_ai/v0/lm/structured_outputs/handler.py +0 -440
  977. synth_ai/v0/lm/structured_outputs/inject.py +0 -297
  978. synth_ai/v0/lm/structured_outputs/rehabilitate.py +0 -185
  979. synth_ai/v0/lm/tools/__init__.py +0 -3
  980. synth_ai/v0/lm/tools/base.py +0 -172
  981. synth_ai/v0/lm/unified_interface.py +0 -202
  982. synth_ai/v0/lm/vendors/__init__.py +0 -0
  983. synth_ai/v0/lm/vendors/base.py +0 -81
  984. synth_ai/v0/lm/vendors/core/__init__.py +0 -0
  985. synth_ai/v0/lm/vendors/core/anthropic_api.py +0 -387
  986. synth_ai/v0/lm/vendors/core/gemini_api.py +0 -292
  987. synth_ai/v0/lm/vendors/core/mistral_api.py +0 -322
  988. synth_ai/v0/lm/vendors/core/openai_api.py +0 -227
  989. synth_ai/v0/lm/vendors/core/synth_dev_api.py +0 -0
  990. synth_ai/v0/lm/vendors/local/__init__.py +0 -0
  991. synth_ai/v0/lm/vendors/local/ollama.py +0 -0
  992. synth_ai/v0/lm/vendors/openai_standard.py +0 -782
  993. synth_ai/v0/lm/vendors/openai_standard_responses.py +0 -259
  994. synth_ai/v0/lm/vendors/retries.py +0 -22
  995. synth_ai/v0/lm/vendors/supported/__init__.py +0 -0
  996. synth_ai/v0/lm/vendors/supported/custom_endpoint.py +0 -415
  997. synth_ai/v0/lm/vendors/supported/deepseek.py +0 -69
  998. synth_ai/v0/lm/vendors/supported/grok.py +0 -75
  999. synth_ai/v0/lm/vendors/supported/groq.py +0 -16
  1000. synth_ai/v0/lm/vendors/supported/ollama.py +0 -15
  1001. synth_ai/v0/lm/vendors/supported/openrouter.py +0 -74
  1002. synth_ai/v0/lm/vendors/supported/together.py +0 -11
  1003. synth_ai/v0/lm/vendors/synth_client.py +0 -835
  1004. synth_ai/v0/lm/warmup.py +0 -186
  1005. synth_ai/v0/tracing/__init__.py +0 -0
  1006. synth_ai/v0/tracing/abstractions.py +0 -224
  1007. synth_ai/v0/tracing/base_client.py +0 -91
  1008. synth_ai/v0/tracing/client_manager.py +0 -131
  1009. synth_ai/v0/tracing/config.py +0 -142
  1010. synth_ai/v0/tracing/context.py +0 -146
  1011. synth_ai/v0/tracing/decorators.py +0 -682
  1012. synth_ai/v0/tracing/events/__init__.py +0 -0
  1013. synth_ai/v0/tracing/events/manage.py +0 -147
  1014. synth_ai/v0/tracing/events/scope.py +0 -86
  1015. synth_ai/v0/tracing/events/store.py +0 -228
  1016. synth_ai/v0/tracing/immediate_client.py +0 -151
  1017. synth_ai/v0/tracing/local.py +0 -18
  1018. synth_ai/v0/tracing/log_client_base.py +0 -73
  1019. synth_ai/v0/tracing/retry_queue.py +0 -186
  1020. synth_ai/v0/tracing/trackers.py +0 -515
  1021. synth_ai/v0/tracing/upload.py +0 -409
  1022. synth_ai/v0/tracing/utils.py +0 -9
  1023. synth_ai/v0/tracing_v1/__init__.py +0 -16
  1024. synth_ai/v0/tracing_v1/abstractions.py +0 -224
  1025. synth_ai/v0/tracing_v1/base_client.py +0 -91
  1026. synth_ai/v0/tracing_v1/client_manager.py +0 -131
  1027. synth_ai/v0/tracing_v1/config.py +0 -142
  1028. synth_ai/v0/tracing_v1/context.py +0 -146
  1029. synth_ai/v0/tracing_v1/decorators.py +0 -703
  1030. synth_ai/v0/tracing_v1/events/__init__.py +0 -0
  1031. synth_ai/v0/tracing_v1/events/manage.py +0 -147
  1032. synth_ai/v0/tracing_v1/events/scope.py +0 -86
  1033. synth_ai/v0/tracing_v1/events/store.py +0 -228
  1034. synth_ai/v0/tracing_v1/immediate_client.py +0 -151
  1035. synth_ai/v0/tracing_v1/local.py +0 -18
  1036. synth_ai/v0/tracing_v1/log_client_base.py +0 -73
  1037. synth_ai/v0/tracing_v1/retry_queue.py +0 -186
  1038. synth_ai/v0/tracing_v1/trackers.py +0 -515
  1039. synth_ai/v0/tracing_v1/upload.py +0 -527
  1040. synth_ai/v0/tracing_v1/utils.py +0 -9
  1041. synth_ai/v0/tracing_v3/__init__.py +0 -10
  1042. synth_ai/v0/tracing_v3/abstractions.py +0 -3
  1043. synth_ai/v0/tracing_v3/decorators.py +0 -3
  1044. synth_ai/v0/tracing_v3/llm_call_record_helpers.py +0 -3
  1045. synth_ai/v0/tracing_v3/session_tracer.py +0 -3
  1046. synth_ai-0.2.14.dist-info/METADATA +0 -139
  1047. synth_ai-0.2.14.dist-info/RECORD +0 -762
  1048. synth_ai-0.2.14.dist-info/top_level.txt +0 -2
  1049. /synth_ai/{demos/demo_task_apps → cli/demo_apps}/crafter/__init__.py +0 -0
  1050. /synth_ai/{demos → cli/demo_apps}/demo_task_apps/__init__.py +0 -0
  1051. /synth_ai/{demos → cli/demo_apps}/demo_task_apps/crafter/configs/crafter_fft_4b.toml +0 -0
  1052. /synth_ai/{demos → cli/demo_apps}/demo_task_apps/crafter/configs/rl_from_base_qwen4b.toml +0 -0
  1053. /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/__init__.py +0 -0
  1054. /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/_common.py +0 -0
  1055. /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/app.py +0 -0
  1056. /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/config.toml +0 -0
  1057. /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/deploy_modal.py +0 -0
  1058. {examples/task_apps → synth_ai/core/apps}/__init__.py +0 -0
  1059. /synth_ai/{tracing_v3 → core/tracing_v3}/examples/basic_usage.py +0 -0
  1060. /synth_ai/{tracing_v3 → core/tracing_v3}/hooks.py +0 -0
  1061. /synth_ai/{tracing_v3 → core/tracing_v3}/lm_call_record_abstractions.py +0 -0
  1062. /synth_ai/{tracing_v3 → core/tracing_v3}/replica_sync.py +0 -0
  1063. /synth_ai/{tracing_v3 → core/tracing_v3}/serialization.py +0 -0
  1064. /synth_ai/{tracing_v3 → core/tracing_v3}/storage/__init__.py +0 -0
  1065. /synth_ai/{tracing_v3 → core/tracing_v3}/storage/exceptions.py +0 -0
  1066. /synth_ai/{tracing_v3 → core/tracing_v3}/storage/types.py +0 -0
  1067. /synth_ai/{tracing_v3 → core/tracing_v3}/storage/utils.py +0 -0
  1068. /synth_ai/{tracing_v3 → core/tracing_v3}/turso/__init__.py +0 -0
  1069. /synth_ai/{evals → sdk/judging}/types.py +0 -0
  1070. /synth_ai/{learning → sdk/learning}/algorithms.py +0 -0
  1071. /synth_ai/{learning → sdk/learning}/config.py +0 -0
  1072. /synth_ai/{learning → sdk/learning}/constants.py +0 -0
  1073. /synth_ai/{learning → sdk/learning}/core.py +0 -0
  1074. /synth_ai/{learning → sdk/learning}/gateway.py +0 -0
  1075. /synth_ai/{learning → sdk/learning}/rl/__init__.py +0 -0
  1076. /synth_ai/{learning → sdk/learning}/rl/config.py +0 -0
  1077. /synth_ai/{learning → sdk/learning}/rl_client.py +0 -0
  1078. /synth_ai/{learning → sdk/learning}/sft/__init__.py +0 -0
  1079. /synth_ai/{learning → sdk/learning}/sse.py +0 -0
  1080. /synth_ai/{task → sdk/task}/auth.py +0 -0
  1081. /synth_ai/{task → sdk/task}/client.py +0 -0
  1082. /synth_ai/{task → sdk/task}/errors.py +0 -0
  1083. /synth_ai/{task → sdk/task}/health.py +0 -0
  1084. /synth_ai/{task → sdk/task}/json.py +0 -0
  1085. /synth_ai/{task → sdk/task}/rubrics/models.py +0 -0
  1086. /synth_ai/{task → sdk/task}/rubrics/scoring.py +0 -0
  1087. /synth_ai/{task → sdk/task}/rubrics/strict.py +0 -0
  1088. /synth_ai/{task → sdk/task}/vendors.py +0 -0
  1089. {synth_ai-0.2.14.dist-info → synth_ai-0.4.1.dist-info}/WHEEL +0 -0
  1090. {synth_ai-0.2.14.dist-info → synth_ai-0.4.1.dist-info}/entry_points.txt +0 -0
  1091. {synth_ai-0.2.14.dist-info → synth_ai-0.4.1.dist-info}/licenses/LICENSE +0 -0
@@ -0,0 +1,1113 @@
1
+ from __future__ import annotations
2
+
3
+ import asyncio
4
+ import contextlib
5
+ import importlib
6
+ import importlib.util
7
+ import json
8
+ import os
9
+ import sqlite3
10
+ import sys
11
+ import time
12
+ import uuid
13
+ from collections.abc import Sequence
14
+ from functools import lru_cache
15
+ from pathlib import Path
16
+ from typing import TYPE_CHECKING, Any, cast
17
+
18
+ import click
19
+
20
+ from synth_ai.cli.lib.task_app_discovery import discover_eval_config_paths
21
+ from synth_ai.core.tracing_v3.session_tracer import SessionTracer
22
+ from synth_ai.sdk.task.config import EvalConfig
23
+
24
+ from .errors import (
25
+ EvalCliError,
26
+ EvalConfigNotFoundError,
27
+ EvalConfigParseError,
28
+ InvalidEvalConfigError,
29
+ MetadataFilterFormatError,
30
+ MetadataSQLExecutionError,
31
+ MetadataSQLResultError,
32
+ MissingEvalTableError,
33
+ NoSeedsMatchedError,
34
+ SeedParseError,
35
+ TaskInfoUnavailableError,
36
+ TomlUnavailableError,
37
+ )
38
+ from .validation import validate_eval_options
39
+
40
+ try: # Python 3.11+
41
+ import tomllib as _toml
42
+ except Exception: # pragma: no cover - fallback
43
+ _toml = None # type: ignore[assignment]
44
+
45
+ __all__ = ["command", "get_command", "format_eval_error"]
46
+
47
+ if TYPE_CHECKING:
48
+ from synth_ai.cli.task_apps import AppChoice, TaskAppEntryType
49
+
50
+
51
+ @lru_cache(maxsize=1)
52
+ def _task_apps_module():
53
+ from synth_ai.cli import task_apps as module # local import to avoid circular deps
54
+
55
+ return module
56
+
57
+
58
+ @click.command(
59
+ "eval",
60
+ help="Run one-off rollouts against a task app and print judge/eval summaries.",
61
+ )
62
+ @click.argument("app_id", type=str, required=False)
63
+ @click.option(
64
+ "--config",
65
+ type=click.Path(),
66
+ default=None,
67
+ help="Path to eval TOML (short schema). Auto-discovers the first matching file when omitted.",
68
+ )
69
+ @click.option(
70
+ "--url",
71
+ "task_app_url",
72
+ type=str,
73
+ default=None,
74
+ help="Base URL of a running task app instead of spawning locally (requires --env-file for secrets).",
75
+ )
76
+ @click.option(
77
+ "--seeds",
78
+ default="0,1,2,3,4",
79
+ help="Comma-separated seeds/indices to evaluate. Use negative numbers to wrap around the dataset.",
80
+ )
81
+ @click.option("--split", default="train", show_default=True, help="Dataset split to use")
82
+ @click.option(
83
+ "--model",
84
+ default=None,
85
+ help="Model identifier. When omitted the CLI will prompt based on task metadata.",
86
+ )
87
+ @click.option(
88
+ "--env-file",
89
+ multiple=True,
90
+ type=click.Path(),
91
+ help="Env file(s) to load (API keys, etc.). Required when using --url or remote judges.",
92
+ )
93
+ @click.option(
94
+ "--trace-db",
95
+ default="traces/v3/synth_ai.db",
96
+ show_default=True,
97
+ help="SQLite/Turso URL for storing rollout traces set to 'none' to disable persistence.",
98
+ )
99
+ @click.option(
100
+ "--metadata",
101
+ multiple=True,
102
+ help="Filter tasks by key=value metadata (e.g., --metadata difficulty=easy)",
103
+ )
104
+ @click.option(
105
+ "--metadata-sql",
106
+ default=None,
107
+ help="SQLite query that returns seeds to evaluate (e.g., SELECT seed FROM tasks WHERE difficulty='easy' LIMIT 5)",
108
+ )
109
+ def eval_command(
110
+ app_id: str | None,
111
+ config: str | None,
112
+ task_app_url: str | None,
113
+ seeds: str,
114
+ split: str,
115
+ model: str | None,
116
+ env_file: Sequence[str],
117
+ trace_db: str,
118
+ metadata: Sequence[str],
119
+ metadata_sql: str | None,
120
+ ) -> None:
121
+ try:
122
+ return _eval_command_impl(
123
+ app_id=app_id,
124
+ config=config,
125
+ task_app_url=task_app_url,
126
+ seeds=seeds,
127
+ split=split,
128
+ model=model,
129
+ env_file=env_file,
130
+ trace_db=trace_db,
131
+ metadata=metadata,
132
+ metadata_sql=metadata_sql,
133
+ )
134
+ except EvalCliError as exc:
135
+ raise click.ClickException(format_eval_error(exc)) from exc
136
+
137
+
138
+ def _eval_command_impl(
139
+ app_id: str | None,
140
+ config: str | None,
141
+ task_app_url: str | None,
142
+ seeds: str,
143
+ split: str,
144
+ model: str | None,
145
+ env_file: Sequence[str],
146
+ trace_db: str,
147
+ metadata: Sequence[str],
148
+ metadata_sql: str | None,
149
+ ) -> None:
150
+ """Run rollouts against a task app and report judge statistics.
151
+
152
+ By default the command spins up the selected task app in-process, executes the
153
+ requested seeds, and prints aggregate scores (official and custom judges). When
154
+ pointing at a remote `--url`, supply matching `--env-file` values so the CLI can
155
+ forward authentication headers to the running service.
156
+ """
157
+ module = _task_apps_module()
158
+ task_app_config_type = module.TaskAppConfig
159
+ create_task_app = module.create_task_app
160
+ select_app_choice = module._select_app_choice
161
+ determine_env_files = module._determine_env_files
162
+ load_env_files_into_process = module._load_env_files_into_process
163
+ store_trace = getattr(module, "_store_trace", None)
164
+ pearson = module._pearson
165
+ judge_spec_cls = module.JudgeSpec
166
+ session_tracer_cls = getattr(module, "SessionTracer", None)
167
+
168
+ # Parse and validate TOML config
169
+
170
+ cfg: dict[str, Any] = {}
171
+ eval_cfg: EvalConfig | None = None
172
+ config_path: Path | None = None
173
+
174
+ if config:
175
+ config_path = Path(config)
176
+ else:
177
+ auto_configs = discover_eval_config_paths()
178
+ if auto_configs:
179
+ config_path = auto_configs[0]
180
+ click.echo(f"Using eval config: {config_path}")
181
+
182
+ if config_path:
183
+ if _toml is None:
184
+ raise TomlUnavailableError()
185
+ if not config_path.exists():
186
+ raise EvalConfigNotFoundError(str(config_path))
187
+ try:
188
+ data = config_path.read_bytes()
189
+ parsed = _toml.loads(data.decode("utf-8"))
190
+ if isinstance(parsed, dict):
191
+ section = parsed.get("eval")
192
+ if section is None:
193
+ cfg = dict(parsed)
194
+ elif isinstance(section, dict):
195
+ cfg = dict(section)
196
+ else:
197
+ raise MissingEvalTableError()
198
+ except Exception as exc:
199
+ raise EvalConfigParseError(path=str(config_path), detail=str(exc)) from exc
200
+
201
+ if cfg:
202
+ try:
203
+ normalized_cfg = validate_eval_options(cfg)
204
+ normalized_cfg_dict = dict(normalized_cfg)
205
+ eval_cfg = EvalConfig.from_dict(normalized_cfg_dict)
206
+ cfg = normalized_cfg_dict
207
+ click.echo(f"✓ Config validated: {len(eval_cfg.seeds)} seeds, model={eval_cfg.model}")
208
+ except (ValueError, TypeError) as validation_error:
209
+ raise InvalidEvalConfigError(detail=str(validation_error)) from validation_error
210
+ else:
211
+ cfg = {}
212
+
213
+ # CLI args override config
214
+ if eval_cfg:
215
+ app_id = app_id or eval_cfg.app_id
216
+ else:
217
+ app_id = app_id or (cfg.get("app_id") if isinstance(cfg.get("app_id"), str) else None) # type: ignore
218
+
219
+ metadata_filters: dict[str, str] = {}
220
+ if eval_cfg:
221
+ metadata_filters.update(eval_cfg.metadata)
222
+ else:
223
+ cfg_metadata = cfg.get("metadata")
224
+ if isinstance(cfg_metadata, dict):
225
+ for key, value in cfg_metadata.items():
226
+ metadata_filters[str(key)] = str(value)
227
+ elif isinstance(cfg_metadata, list):
228
+ for item in cfg_metadata:
229
+ if isinstance(item, str) and "=" in item:
230
+ key, value = item.split("=", 1)
231
+ metadata_filters[key.strip()] = value.strip()
232
+
233
+ for item in metadata or ():
234
+ if "=" not in item:
235
+ raise MetadataFilterFormatError(entry=item)
236
+ key, value = item.split("=", 1)
237
+ key = key.strip()
238
+ value = value.strip()
239
+ if not key or not value:
240
+ raise MetadataFilterFormatError(entry=item)
241
+ metadata_filters[key] = value
242
+
243
+ metadata_sql_query: str | None = None
244
+ if eval_cfg and eval_cfg.metadata_sql:
245
+ metadata_sql_query = eval_cfg.metadata_sql
246
+ else:
247
+ cfg_metadata_sql = cfg.get("metadata_sql")
248
+ if isinstance(cfg_metadata_sql, dict):
249
+ metadata_sql_query = cfg_metadata_sql.get("query") or cfg_metadata_sql.get("sql")
250
+ elif isinstance(cfg_metadata_sql, str):
251
+ metadata_sql_query = cfg_metadata_sql
252
+
253
+ if metadata_sql:
254
+ metadata_sql_query = metadata_sql
255
+ if metadata_sql_query is not None:
256
+ metadata_sql_query = str(metadata_sql_query)
257
+
258
+ trace_db_url: str | None = None
259
+ trace_db = (trace_db or "").strip()
260
+ if trace_db and trace_db.lower() not in {"none", "off", "disable"}:
261
+ if "://" in trace_db:
262
+ trace_db_url = trace_db
263
+ else:
264
+ trace_path = Path(trace_db).expanduser()
265
+ trace_path.parent.mkdir(parents=True, exist_ok=True)
266
+ trace_db_url = f"sqlite+aiosqlite:///{trace_path}"
267
+ trace_tracer: SessionTracer | None = None
268
+ if trace_db_url and session_tracer_cls is not None:
269
+ trace_tracer = cast(SessionTracer, session_tracer_cls(db_url=trace_db_url, auto_save=True))
270
+
271
+ # Determine selection params (CLI takes precedence; TOML only fills unset model/seeds/env)
272
+ if cfg.get("model") and not model:
273
+ model = str(cfg["model"]) # type: ignore[index]
274
+ if cfg.get("seeds") and seeds == "0,1,2,3,4":
275
+ val = cfg["seeds"]
276
+ if isinstance(val, list):
277
+ with contextlib.suppress(Exception):
278
+ seeds = ",".join(str(int(x)) for x in val)
279
+ elif isinstance(val, str):
280
+ seeds = val
281
+ elif isinstance(val, int):
282
+ seeds = str(val)
283
+ if cfg.get("env_file") and not env_file:
284
+ ef = cfg["env_file"]
285
+ if isinstance(ef, str):
286
+ env_file = (ef,) # type: ignore[assignment]
287
+ elif isinstance(ef, list):
288
+ env_file = tuple(str(x) for x in ef) # type: ignore[assignment]
289
+
290
+ choice_for_env: AppChoice | None = None
291
+ entry: TaskAppEntryType | None = None
292
+ if task_app_url is None:
293
+ choice_for_env = select_app_choice(app_id, purpose="eval")
294
+ entry = choice_for_env.ensure_entry()
295
+
296
+ env_paths: list[Path] = []
297
+ if entry is not None:
298
+ original_env_path = choice_for_env.path if choice_for_env is not None else None
299
+ env_paths = determine_env_files(entry, env_file, original_path=original_env_path)
300
+ else:
301
+ if not env_file:
302
+ raise click.ClickException("--env-file is required when using --url")
303
+ for candidate in env_file:
304
+ p = Path(candidate).expanduser()
305
+ if not p.exists():
306
+ raise click.ClickException(f"Env file not found: {p}")
307
+ env_paths.append(p)
308
+
309
+ click.echo("Using env file(s): " + ", ".join(str(p) for p in env_paths))
310
+ load_env_files_into_process([str(Path(p)) for p in env_paths])
311
+
312
+ if task_app_url is None:
313
+ config = entry.config_factory() # type: ignore[union-attr]
314
+ # Help the type checker; runtime check also enforced in server.run_task_app
315
+ if not isinstance(config, task_app_config_type):
316
+ raise click.ClickException(
317
+ "Invalid task app: config_factory did not return TaskAppConfig"
318
+ )
319
+ app = create_task_app(config)
320
+
321
+ # Determine supported models
322
+ inference_meta: dict[str, Any] = {}
323
+ supported: list[str] = []
324
+ seen_models: set[str] = set()
325
+
326
+ def _add_supported_model(candidate: Any) -> None:
327
+ if not candidate:
328
+ return
329
+ text = str(candidate).strip()
330
+ if not text or text in seen_models:
331
+ return
332
+ supported.append(text)
333
+ seen_models.add(text)
334
+
335
+ if task_app_url is None:
336
+ try:
337
+ if hasattr(config, "base_task_info") and config.base_task_info:
338
+ inf_obj = getattr(config.base_task_info, "inference", None)
339
+ if inf_obj is not None:
340
+ if hasattr(inf_obj, "model_dump"):
341
+ inference_meta = dict(inf_obj.model_dump(exclude_none=True)) # type: ignore[attr-defined]
342
+ elif isinstance(inf_obj, dict):
343
+ inference_meta = dict(inf_obj)
344
+ except Exception:
345
+ inference_meta = {}
346
+ else:
347
+ try:
348
+ import httpx as _hx
349
+
350
+ headers = {}
351
+ api_key = (os.environ.get("ENVIRONMENT_API_KEY") or "").strip()
352
+ if api_key:
353
+ headers["X-API-Key"] = api_key
354
+ with _hx.Client(base_url=task_app_url, headers=headers, timeout=15.0) as c:
355
+ info = c.get("/info").json()
356
+ inf = info.get("inference") if isinstance(info, dict) else None
357
+ if isinstance(inf, dict):
358
+ inference_meta = dict(inf)
359
+ except Exception:
360
+ inference_meta = {}
361
+
362
+ default_model = inference_meta.get("model")
363
+ if isinstance(default_model, str):
364
+ _add_supported_model(default_model)
365
+
366
+ models_field = inference_meta.get("models")
367
+ if isinstance(models_field, list):
368
+ for candidate in models_field:
369
+ _add_supported_model(candidate)
370
+
371
+ supported_models = inference_meta.get("supported_models")
372
+ if isinstance(supported_models, list):
373
+ for candidate in supported_models:
374
+ _add_supported_model(candidate)
375
+
376
+ providers = inference_meta.get("providers")
377
+ if isinstance(providers, list):
378
+ if "openai" in providers:
379
+ _add_supported_model("gpt-5")
380
+ if "groq" in providers:
381
+ _add_supported_model("groq:llama-3.1-70b-versatile")
382
+
383
+ _add_supported_model("synth:qwen-0.6b")
384
+
385
+ selected_model = model
386
+ if not selected_model:
387
+ if not supported:
388
+ raise click.ClickException(
389
+ "No supported models; supply --model or add base_task_info.inference.model"
390
+ )
391
+ click.echo("Select model to evaluate:")
392
+ for idx, m in enumerate(supported, start=1):
393
+ click.echo(f" {idx}) {m}")
394
+ choice_idx = click.prompt("Enter choice", type=click.IntRange(1, len(supported)))
395
+ selected_model = supported[choice_idx - 1]
396
+
397
+ try:
398
+ seed_values = [int(s.strip()) for s in seeds.split(",") if s.strip()]
399
+ except Exception as exc:
400
+ raise SeedParseError(value=seeds) from exc
401
+
402
+ import httpx
403
+
404
+ headers = {}
405
+ api_key = (os.environ.get("ENVIRONMENT_API_KEY") or "").strip()
406
+ if api_key:
407
+ headers["X-API-Key"] = api_key
408
+
409
+ # Precompute optional policy overrides from TOML
410
+ policy_overrides: dict[str, Any] = {}
411
+ try:
412
+ # Accept [eval.policy] table or top-level keys for convenience
413
+ if isinstance(cfg.get("policy"), dict):
414
+ policy_overrides.update(dict(cfg["policy"]))
415
+ # Back-compat: allow temperature/max_tokens at top level
416
+ for k in (
417
+ "temperature",
418
+ "max_tokens",
419
+ "reasoning_effort",
420
+ "system_hint",
421
+ "tool_choice",
422
+ "inference_url",
423
+ ):
424
+ if k in cfg and k not in policy_overrides:
425
+ policy_overrides[k] = cfg.get(k)
426
+ except Exception:
427
+ policy_overrides = {}
428
+
429
+ raw_concurrency = cfg.get("concurrency")
430
+ try:
431
+ concurrency_limit = int(raw_concurrency) if raw_concurrency is not None else 1
432
+ except Exception:
433
+ concurrency_limit = 1
434
+ if concurrency_limit <= 0:
435
+ concurrency_limit = 1
436
+ concurrency_limit = min(concurrency_limit, max(1, len(seed_values)))
437
+
438
+ judge_specs: list[Any] = []
439
+
440
+ def _register_judge(name_hint: str | None, judge_cfg: dict[str, Any]) -> None:
441
+ if not judge_cfg:
442
+ return
443
+ judge_module = judge_cfg.get("module")
444
+ judge_path = judge_cfg.get("path")
445
+ judge_callable_name = judge_cfg.get("callable") or judge_cfg.get("function")
446
+ if judge_module and judge_path:
447
+ raise click.ClickException("Judge config cannot set both 'module' and 'path'")
448
+ if not judge_module and not judge_path:
449
+ raise click.ClickException("Judge config requires 'module' or 'path'")
450
+ try:
451
+ if judge_module:
452
+ module = importlib.import_module(str(judge_module))
453
+ else:
454
+ path = Path(str(judge_path)).expanduser()
455
+ if not path.exists():
456
+ raise click.ClickException(f"Judge module path not found: {path}")
457
+ spec = importlib.util.spec_from_file_location(
458
+ f"_eval_judge_{path.stem}", path
459
+ )
460
+ if not spec or not spec.loader:
461
+ raise click.ClickException(f"Failed to load judge module from {path}")
462
+ module = importlib.util.module_from_spec(spec)
463
+ sys.modules[spec.name] = module
464
+ spec.loader.exec_module(module)
465
+ except click.ClickException:
466
+ raise
467
+ except Exception as exc:
468
+ raise click.ClickException(f"Unable to load judge module: {exc}") from exc
469
+
470
+ if judge_callable_name:
471
+ try:
472
+ judge_fn = getattr(module, str(judge_callable_name))
473
+ except AttributeError as exc:
474
+ raise click.ClickException(
475
+ f"Judge callable '{judge_callable_name}' not found in module"
476
+ ) from exc
477
+ else:
478
+ if hasattr(module, "judge"):
479
+ judge_fn = module.judge
480
+ else:
481
+ raise click.ClickException("Judge module must expose 'judge' callable")
482
+
483
+ if not callable(judge_fn):
484
+ raise click.ClickException("Judge callable is not callable")
485
+
486
+ judge_kwargs = {
487
+ k: v
488
+ for k, v in judge_cfg.items()
489
+ if k not in {"module", "path", "callable", "function", "name"}
490
+ }
491
+ display_name = str(
492
+ judge_cfg.get("name")
493
+ or name_hint
494
+ or f"judge{len(judge_specs) + 1}"
495
+ )
496
+ judge_specs.append(judge_spec_cls(display_name, judge_fn, judge_kwargs))
497
+
498
+ raw_judge_cfg = cfg.get("judge")
499
+ if isinstance(raw_judge_cfg, dict) and raw_judge_cfg:
500
+ direct_keys = {"module", "path", "callable", "function", "name"}
501
+ has_direct_keys = any(key in raw_judge_cfg for key in direct_keys)
502
+ nested_candidates = [
503
+ (key, value)
504
+ for key, value in raw_judge_cfg.items()
505
+ if isinstance(value, dict)
506
+ ]
507
+ if has_direct_keys and not nested_candidates:
508
+ _register_judge(None, raw_judge_cfg)
509
+ else:
510
+ for sub_name, sub_cfg in nested_candidates:
511
+ _register_judge(sub_name, sub_cfg)
512
+
513
+ raw_judges_list = cfg.get("judges")
514
+ if isinstance(raw_judges_list, list):
515
+ for _index, entry in enumerate(raw_judges_list, start=1):
516
+ if isinstance(entry, dict):
517
+ _register_judge(entry.get("name") or f"judge{len(judge_specs) + 1}", entry)
518
+
519
+ records: list[dict[str, Any]] = []
520
+
521
+ successes = 0
522
+ failures = 0
523
+ # Aggregate outcome stats across successful seeds
524
+ outcome_sum: float = 0.0
525
+ outcome_count: int = 0
526
+ outcome_correct: int = 0
527
+
528
+ def _build_task_rows(taskset: Any) -> dict[int, dict[str, Any]]:
529
+ rows: dict[int, dict[str, Any]] = {}
530
+ if not isinstance(taskset, dict):
531
+ return rows
532
+
533
+ scenario_ids = taskset.get("scenario_ids") or []
534
+ loop_ids = taskset.get("loop_ids") or []
535
+ thread_ids = taskset.get("thread_ids") or []
536
+ difficulty_map = taskset.get("difficulty_map") or {}
537
+
538
+ max_len = max(len(scenario_ids), len(loop_ids), len(thread_ids))
539
+ for seed in range(max_len):
540
+ scenario_id = scenario_ids[seed] if seed < len(scenario_ids) else None
541
+ loop_id = loop_ids[seed] if seed < len(loop_ids) else None
542
+ thread_id = thread_ids[seed] if seed < len(thread_ids) else None
543
+ difficulty = None
544
+ if isinstance(difficulty_map, dict):
545
+ if scenario_id and scenario_id in difficulty_map:
546
+ difficulty = difficulty_map.get(scenario_id)
547
+ elif str(seed) in difficulty_map:
548
+ difficulty = difficulty_map.get(str(seed))
549
+
550
+ rows[seed] = {
551
+ "seed": seed,
552
+ "scenario_id": scenario_id,
553
+ "loop_id": loop_id,
554
+ "thread_id": thread_id,
555
+ "difficulty": difficulty,
556
+ }
557
+ return rows
558
+
559
+ def _apply_metadata_filters(
560
+ rows: dict[int, dict[str, Any]], seeds_list: list[int], filters: dict[str, str]
561
+ ) -> list[int]:
562
+ if not filters:
563
+ return seeds_list
564
+ filtered: list[int] = []
565
+ for seed in seeds_list:
566
+ row = rows.get(seed)
567
+ if not row:
568
+ continue
569
+ include = True
570
+ for key, expected in filters.items():
571
+ actual = row.get(key)
572
+ if actual is None:
573
+ include = False
574
+ break
575
+ if str(actual).lower() != expected.lower():
576
+ include = False
577
+ break
578
+ if include:
579
+ filtered.append(seed)
580
+ return filtered
581
+
582
+ def _apply_metadata_sql(
583
+ rows: dict[int, dict[str, Any]], seeds_list: list[int], query: str
584
+ ) -> list[int]:
585
+ """Return seeds that satisfy an arbitrary SQL query.
586
+
587
+ The query is executed against an in-memory SQLite table named `tasks`
588
+ with columns (seed INTEGER, scenario_id TEXT, loop_id TEXT, thread_id TEXT, difficulty TEXT).
589
+ Any rows whose `seed` value (or first column if `seed` is absent) appear in the result set are retained.
590
+ """
591
+ if not query:
592
+ return seeds_list
593
+ conn = sqlite3.connect(":memory:")
594
+ try:
595
+ cur = conn.cursor()
596
+ cur.execute(
597
+ "CREATE TABLE tasks (seed INTEGER, scenario_id TEXT, loop_id TEXT, thread_id TEXT, difficulty TEXT)"
598
+ )
599
+ insert_stmt = (
600
+ "INSERT INTO tasks (seed, scenario_id, loop_id, thread_id, difficulty) VALUES (?,?,?,?,?)"
601
+ )
602
+ for seed in seeds_list:
603
+ row = rows.get(seed, {})
604
+ cur.execute(
605
+ insert_stmt,
606
+ [
607
+ seed,
608
+ row.get("scenario_id"),
609
+ row.get("loop_id"),
610
+ row.get("thread_id"),
611
+ row.get("difficulty"),
612
+ ],
613
+ )
614
+
615
+ result = cur.execute(query)
616
+ fetched = result.fetchall()
617
+ if not fetched:
618
+ return []
619
+ description = result.description or []
620
+ col_names = [col[0] for col in description]
621
+ seeds_out: list[int] = []
622
+ for entry in fetched:
623
+ value = entry[col_names.index("seed")] if "seed" in col_names else entry[0]
624
+ try:
625
+ seeds_out.append(int(value))
626
+ except Exception as exc:
627
+ raise MetadataSQLResultError(
628
+ query=query,
629
+ detail="non-integer value returned",
630
+ ) from exc
631
+ seeds_set = set(seeds_out)
632
+ return [seed for seed in seeds_list if seed in seeds_set]
633
+ except sqlite3.Error as exc:
634
+ raise MetadataSQLExecutionError(query=query, detail=str(exc)) from exc
635
+ finally:
636
+ conn.close()
637
+
638
+ async def _run_eval() -> None:
639
+ nonlocal successes, failures, outcome_sum, outcome_count, outcome_correct, records, seed_values
640
+
641
+ if trace_tracer is not None and trace_tracer.db is None:
642
+ await trace_tracer.initialize()
643
+
644
+ if task_app_url is None:
645
+ transport = httpx.ASGITransport(app=app) # type: ignore[name-defined]
646
+ async_client = httpx.AsyncClient(
647
+ transport=cast(Any, transport),
648
+ base_url="http://eval.local",
649
+ timeout=300.0,
650
+ follow_redirects=True,
651
+ headers=headers,
652
+ )
653
+ else:
654
+ async_client = httpx.AsyncClient(
655
+ base_url=task_app_url,
656
+ timeout=300.0,
657
+ follow_redirects=True,
658
+ headers=headers,
659
+ )
660
+
661
+ try:
662
+ taskset_payload: dict[str, Any] | None = None
663
+ try:
664
+ task_info_response = await async_client.get("/task_info")
665
+ except Exception:
666
+ task_info_response = None
667
+ if task_info_response is not None and task_info_response.status_code == 200:
668
+ with contextlib.suppress(Exception):
669
+ payload_json = task_info_response.json()
670
+ if isinstance(payload_json, dict) and "taskset" in payload_json:
671
+ taskset_payload = payload_json.get("taskset")
672
+ if not isinstance(taskset_payload, dict):
673
+ taskset_payload = None
674
+ elif isinstance(payload_json, dict):
675
+ taskset_payload = payload_json
676
+
677
+ available_seeds = list(seed_values)
678
+ if metadata_sql_query or metadata_filters:
679
+ if not taskset_payload:
680
+ raise TaskInfoUnavailableError()
681
+ rows = _build_task_rows(taskset_payload)
682
+ if metadata_sql_query:
683
+ available_seeds = _apply_metadata_sql(rows, available_seeds, metadata_sql_query)
684
+ if metadata_filters:
685
+ available_seeds = _apply_metadata_filters(rows, available_seeds, metadata_filters)
686
+ if not available_seeds:
687
+ raise NoSeedsMatchedError()
688
+ seed_values = available_seeds
689
+
690
+ semaphore = asyncio.Semaphore(concurrency_limit)
691
+
692
+ async def _run_seed(seed_val: int) -> None:
693
+ nonlocal successes, failures, outcome_sum, outcome_count, outcome_correct, records
694
+ # Read env_name and policy_name from config if available
695
+ env_name = cfg.get("env_name") or (cfg.get("env", {}).get("env_name") if isinstance(cfg.get("env"), dict) else None)
696
+ policy_name = cfg.get("policy_name") or (cfg.get("policy", {}).get("policy_name") if isinstance(cfg.get("policy"), dict) else None)
697
+ env_config_overrides = cfg.get("env_config", {}) if isinstance(cfg.get("env_config"), dict) else {}
698
+ policy_config_overrides = cfg.get("policy_config", {}) if isinstance(cfg.get("policy_config"), dict) else {}
699
+
700
+ # Debug: print config parsing
701
+ if seed_val == 0:
702
+ click.echo(f"[DEBUG] env_name from config: {env_name}")
703
+ click.echo(f"[DEBUG] policy_name from config: {policy_name}")
704
+
705
+ # Generate default ops sequence if not provided
706
+ max_llm_calls = policy_config_overrides.get("max_llm_calls", 10)
707
+ ops_list = cfg.get("ops", [])
708
+ if not ops_list:
709
+ # Generate default "agent, env" pairs for max_llm_calls
710
+ ops_list = ["agent", "env"] * int(max_llm_calls)
711
+
712
+ body = {
713
+ "run_id": str(uuid.uuid4()),
714
+ "env": {"config": {"split": split, "index": seed_val, **env_config_overrides}, "seed": seed_val},
715
+ "policy": {
716
+ "policy_name": policy_name or selected_model,
717
+ "config": {"model": selected_model, **policy_overrides, **policy_config_overrides},
718
+ },
719
+ "ops": ops_list,
720
+ "record": {
721
+ "return_trace": cfg.get("return_trace", True),
722
+ "trace_format": cfg.get("trace_format", "structured"),
723
+ },
724
+ "mode": "eval", # RolloutMode.EVAL: use inference URLs as-is, no transformations
725
+ }
726
+ if env_name:
727
+ env_section = body.get("env")
728
+ if isinstance(env_section, dict):
729
+ env_section["env_name"] = env_name
730
+ else:
731
+ body["env"] = {"env_name": env_name}
732
+
733
+ # Debug: print the body being sent
734
+ if seed_val == 0:
735
+ click.echo(f"[DEBUG] rollout body env: {body['env']}")
736
+ click.echo(f"[DEBUG] rollout body policy: {body['policy']}")
737
+ click.echo(f"[DEBUG] rollout body mode: {body.get('mode', 'NOT SET')}")
738
+ rollout_elapsed: float | None = None
739
+ rollout_start = time.perf_counter()
740
+ try:
741
+ import logging
742
+ _log = logging.getLogger(__name__)
743
+ _log.info(f"[EVAL_BODY_DEBUG] Sending body with mode={body.get('mode')}")
744
+ async with semaphore:
745
+ response = await async_client.post("/rollout", json=body)
746
+ rollout_elapsed = time.perf_counter() - rollout_start
747
+ except Exception as exc:
748
+ failures += 1
749
+ click.echo(f"seed={seed_val} error={exc}")
750
+ return
751
+
752
+ ok = 200 <= response.status_code < 300
753
+ if ok:
754
+ successes += 1
755
+ else:
756
+ failures += 1
757
+
758
+ summary = [f"seed={seed_val}", f"status={response.status_code}"]
759
+ data: Any
760
+ try:
761
+ data = response.json()
762
+ except Exception:
763
+ data = None
764
+
765
+ # Debug: print validation errors
766
+ if response.status_code == 422 and data:
767
+ click.echo(f"[DEBUG] 422 Validation Error: {data}")
768
+
769
+ metrics: dict[str, Any] | None = None
770
+ completion: str | None = None
771
+ prompt_index: int | None = None
772
+ prompt_text: str | None = None
773
+ task_id: str | None = None
774
+ task_split: str | None = None
775
+ task_rubric_id: str | None = None
776
+
777
+ trace_namespace: dict[str, Any] | None = None
778
+ session_trace_dict: dict[str, Any] | None = None
779
+
780
+ if isinstance(data, dict):
781
+ import logging
782
+ _logger = logging.getLogger(__name__)
783
+ _logger.info(f"[EVAL_DEBUG] Response data keys: {list(data.keys())}")
784
+ if "detail" in data:
785
+ _logger.error(f"[EVAL_DEBUG] Task app returned error: {data['detail']}")
786
+ trace_namespace = data.get("trace")
787
+ _logger.info(f"[EVAL_DEBUG] trace_namespace type: {type(trace_namespace)}, value: {trace_namespace if not isinstance(trace_namespace, dict) else 'dict with keys: ' + str(list(trace_namespace.keys()) if trace_namespace else 'None')}")
788
+ if not isinstance(trace_namespace, dict):
789
+ raise RuntimeError(
790
+ "The 'synth-ai eval' command requires trace payloads in rollout responses. "
791
+ "Ensure the rollout request includes 'trace_format': 'structured' and 'return_trace': true, "
792
+ "and that task app tracing is enabled (TASKAPP_TRACING_ENABLED=1). "
793
+ "Note: This is specific to the eval command - general rollout endpoints don't require traces."
794
+ )
795
+ # Handle both "compact" and "full" trace formats:
796
+ # - compact: trace_namespace contains {session_id, metadata, ...}
797
+ # - full: trace_namespace IS the full session_trace dict
798
+ session_trace_dict = trace_namespace.get("session_trace")
799
+ if not isinstance(session_trace_dict, dict):
800
+ # If no session_trace key, assume "full" format where trace itself is the session_trace
801
+ if "session_id" in trace_namespace:
802
+ session_trace_dict = trace_namespace
803
+ else:
804
+ raise RuntimeError(
805
+ "The 'synth-ai eval' command requires 'session_trace' in the trace payload or a valid full trace format. "
806
+ "Ensure the task app is using tracing_v3 and returning structured trace data."
807
+ )
808
+ metrics = data.get("metrics") if isinstance(data.get("metrics"), dict) else None
809
+ if metrics:
810
+ mean_return = metrics.get("mean_return") or metrics.get("total_reward")
811
+ outcome = metrics.get("outcome_score")
812
+ if mean_return is not None:
813
+ summary.append(f"mean_return={mean_return}")
814
+ if outcome is not None:
815
+ summary.append(f"outcome={outcome}")
816
+ try:
817
+ val = float(outcome)
818
+ outcome_sum += val
819
+ outcome_count += 1
820
+ if val >= 0.5:
821
+ outcome_correct += 1
822
+ except Exception:
823
+ pass
824
+ trajs = (
825
+ data.get("trajectories")
826
+ if isinstance(data.get("trajectories"), list)
827
+ else None
828
+ )
829
+ if trajs:
830
+ first = trajs[0] if trajs else None
831
+ steps = first.get("steps") if isinstance(first, dict) else None
832
+ if isinstance(steps, list) and steps:
833
+ step0 = steps[0]
834
+ tool_calls = step0.get("tool_calls") or step0.get("tools") or []
835
+ if isinstance(tool_calls, list):
836
+ summary.append(f"tool_calls={len(tool_calls)}")
837
+ obs = step0.get("obs") if isinstance(step0, dict) else None
838
+ if isinstance(obs, dict):
839
+ idx_val = obs.get("prompt_index")
840
+ if isinstance(idx_val, int):
841
+ prompt_index = idx_val
842
+ prompt_raw = obs.get("prompt")
843
+ if isinstance(prompt_raw, str):
844
+ prompt_text = prompt_raw
845
+ if task_id is None:
846
+ candidate_id = obs.get("task_id")
847
+ if isinstance(candidate_id, str) and candidate_id:
848
+ task_id = candidate_id
849
+ if task_split is None:
850
+ candidate_split = obs.get("task_split")
851
+ if isinstance(candidate_split, str) and candidate_split:
852
+ task_split = candidate_split
853
+ if task_rubric_id is None:
854
+ candidate_rid = obs.get("task_rubric_id")
855
+ if isinstance(candidate_rid, str) and candidate_rid:
856
+ task_rubric_id = candidate_rid
857
+ final = first.get("final") if isinstance(first, dict) else None
858
+ if isinstance(final, dict):
859
+ final_obs = final.get("observation")
860
+ if isinstance(final_obs, dict):
861
+ comp_val = final_obs.get("completion")
862
+ if isinstance(comp_val, str):
863
+ completion = comp_val
864
+ if task_id is None:
865
+ candidate_id = final_obs.get("task_id")
866
+ if isinstance(candidate_id, str) and candidate_id:
867
+ task_id = candidate_id
868
+ if task_split is None:
869
+ candidate_split = final_obs.get("task_split")
870
+ if isinstance(candidate_split, str) and candidate_split:
871
+ task_split = candidate_split
872
+ if task_rubric_id is None:
873
+ candidate_rid = final_obs.get("task_rubric_id")
874
+ if isinstance(candidate_rid, str) and candidate_rid:
875
+ task_rubric_id = candidate_rid
876
+ final_info = final.get("info")
877
+ if isinstance(final_info, dict):
878
+ if task_id is None:
879
+ candidate_id = final_info.get("task_id")
880
+ if isinstance(candidate_id, str) and candidate_id:
881
+ task_id = candidate_id
882
+ if task_split is None:
883
+ candidate_split = final_info.get("task_split")
884
+ if isinstance(candidate_split, str) and candidate_split:
885
+ task_split = candidate_split
886
+ if task_rubric_id is None:
887
+ candidate_rid = final_info.get("task_rubric_id")
888
+ if isinstance(candidate_rid, str) and candidate_rid:
889
+ task_rubric_id = candidate_rid
890
+ if task_id:
891
+ summary.append(f"task_id={task_id}")
892
+ click.echo(" ".join(summary))
893
+ with contextlib.suppress(Exception):
894
+ click.echo(json.dumps(data, indent=2))
895
+ else:
896
+ click.echo(" ".join(summary))
897
+
898
+ official_score = None
899
+ if isinstance(metrics, dict):
900
+ for key in ("mean_return", "total_reward", "outcome_score"):
901
+ val = metrics.get(key)
902
+ if isinstance(val, int | float):
903
+ official_score = float(val)
904
+ break
905
+ if official_score is None and isinstance(data, dict):
906
+ try:
907
+ reward_val = data["trajectories"][0]["steps"][0].get("reward")
908
+ if isinstance(reward_val, int | float):
909
+ official_score = float(reward_val)
910
+ except Exception:
911
+ pass
912
+
913
+ if official_score is not None:
914
+ if official_score < 0.0:
915
+ official_score = 0.0
916
+ elif official_score > 1.0:
917
+ official_score = min(1.0, official_score)
918
+
919
+ judge_scores: dict[str, float | None] = {}
920
+ judges_timings: dict[str, float | None] = {}
921
+ timings: dict[str, Any] = {
922
+ "rollout_s": rollout_elapsed,
923
+ "judges": judges_timings,
924
+ }
925
+ if judge_specs:
926
+ for spec in judge_specs:
927
+ score_value: float | None = None
928
+ judge_elapsed: float | None = None
929
+ # Run judges for all tasks (text-based and trajectory-based)
930
+ # Text-based tasks have completion, trajectory-based tasks use response
931
+ judge_payload = {
932
+ "seed": seed_val,
933
+ "prompt_index": prompt_index,
934
+ "prompt": prompt_text,
935
+ "completion": completion,
936
+ "metrics": metrics,
937
+ "response": data,
938
+ "trace": trace_namespace,
939
+ }
940
+ try:
941
+ judge_start = time.perf_counter()
942
+ result = spec.fn(judge_payload, **spec.kwargs)
943
+ judge_elapsed = time.perf_counter() - judge_start
944
+ if isinstance(result, int | float):
945
+ score_value = float(result)
946
+ except Exception as exc:
947
+ if judge_elapsed is None:
948
+ judge_elapsed = time.perf_counter() - judge_start
949
+ click.echo(f"seed={seed_val} judge[{spec.name}]_error={exc}")
950
+ judges_timings[spec.name] = judge_elapsed
951
+ judge_scores[spec.name] = score_value
952
+
953
+ if trace_tracer is not None and trace_namespace:
954
+ storage_metadata = {
955
+ "eval_seed": seed_val,
956
+ "prompt_index": prompt_index,
957
+ "task_id": task_id,
958
+ "task_split": task_split,
959
+ "task_rubric_id": task_rubric_id,
960
+ "official_score": official_score,
961
+ "judge_scores": judge_scores,
962
+ "model": selected_model,
963
+ "prompt": prompt_text,
964
+ "completion": completion,
965
+ }
966
+ if store_trace is not None:
967
+ await store_trace(trace_tracer, trace_namespace, storage_metadata)
968
+
969
+ records.append(
970
+ {
971
+ "seed": seed_val,
972
+ "prompt_index": prompt_index,
973
+ "task_id": task_id,
974
+ "task_split": task_split,
975
+ "task_rubric_id": task_rubric_id,
976
+ "official_score": official_score,
977
+ "judge_scores": judge_scores,
978
+ "timings": timings,
979
+ }
980
+ )
981
+
982
+ await asyncio.gather(*[_run_seed(seed_val) for seed_val in seed_values])
983
+ finally:
984
+ await async_client.aclose()
985
+
986
+ try:
987
+ asyncio.run(_run_eval())
988
+ finally:
989
+ if trace_tracer is not None and trace_tracer.db is not None:
990
+ asyncio.run(trace_tracer.db.close())
991
+
992
+ click.echo(
993
+ f"Eval complete: {successes} ok, {failures} failed; model={selected_model}, split={split}"
994
+ )
995
+
996
+ if outcome_count > 0:
997
+ mean_outcome = outcome_sum / float(outcome_count)
998
+ frac_right = outcome_correct / float(outcome_count)
999
+ click.echo(
1000
+ f"Outcome summary: correct={outcome_correct}/{outcome_count} ({frac_right:.2%}), mean_outcome={mean_outcome:.3f}"
1001
+ )
1002
+
1003
+ if records:
1004
+ judge_specs = judge_specs or [] # ensure iterable
1005
+ official_scores = [
1006
+ r["official_score"] for r in records if r["official_score"] is not None
1007
+ ]
1008
+ if official_scores:
1009
+ click.echo(f" Official mean: {sum(official_scores) / len(official_scores):.3f}")
1010
+ else:
1011
+ click.echo(" Official mean: n/a")
1012
+
1013
+ for spec in judge_specs:
1014
+ spec_scores = [
1015
+ record["judge_scores"].get(spec.name)
1016
+ for record in records
1017
+ if record["judge_scores"].get(spec.name) is not None
1018
+ ]
1019
+ if spec_scores:
1020
+ mean_spec = sum(spec_scores) / len(spec_scores)
1021
+ click.echo(f" [{spec.name}] mean: {mean_spec:.3f}")
1022
+ else:
1023
+ click.echo(f" [{spec.name}] mean: n/a")
1024
+
1025
+ paired = [
1026
+ (
1027
+ record["official_score"],
1028
+ record["judge_scores"].get(spec.name),
1029
+ )
1030
+ for record in records
1031
+ if record["official_score"] is not None
1032
+ and record["judge_scores"].get(spec.name) is not None
1033
+ ]
1034
+ if len(paired) >= 2:
1035
+ corr = pearson(
1036
+ [p[0] for p in paired if p[0] is not None],
1037
+ [p[1] for p in paired if p[1] is not None],
1038
+ )
1039
+ if corr is not None:
1040
+ click.echo(f" Pearson r: {corr:.3f}")
1041
+ else:
1042
+ click.echo(" Pearson r: undefined (zero variance)")
1043
+ else:
1044
+ click.echo(" Pearson r: n/a (need ≥2 paired scores)")
1045
+
1046
+ header = ["Seed", "Prompt", "Official"]
1047
+ header.extend(spec.name for spec in judge_specs)
1048
+ rows: list[list[str]] = []
1049
+ for record in sorted(records, key=lambda r: (r["seed"], r.get("prompt_index") or -1)):
1050
+ seed_val = str(record["seed"])
1051
+ prompt_idx = (
1052
+ str(record["prompt_index"])
1053
+ if record["prompt_index"] is not None
1054
+ else "-"
1055
+ )
1056
+ official_val = (
1057
+ f"{record['official_score']:.3f}"
1058
+ if record["official_score"] is not None
1059
+ else "-"
1060
+ )
1061
+ row = [seed_val, prompt_idx, official_val]
1062
+ for spec in judge_specs:
1063
+ score_val = record["judge_scores"].get(spec.name)
1064
+ row.append(f"{score_val:.3f}" if isinstance(score_val, int | float) else "-")
1065
+ rows.append(row)
1066
+
1067
+ widths = [len(col) for col in header]
1068
+ for row in rows:
1069
+ for idx, cell in enumerate(row):
1070
+ widths[idx] = max(widths[idx], len(cell))
1071
+
1072
+ click.echo("")
1073
+ click.echo(" ".join(h.ljust(widths[idx]) for idx, h in enumerate(header)))
1074
+ click.echo(" ".join("-" * widths[idx] for idx in range(len(header))))
1075
+ for row in rows:
1076
+ click.echo(" ".join(cell.ljust(widths[idx]) for idx, cell in enumerate(row)))
1077
+
1078
+
1079
+
1080
+ command = eval_command
1081
+
1082
+
1083
+ def get_command() -> click.Command:
1084
+ """Return the Click command implementing task-app evaluation."""
1085
+ return command
1086
+
1087
+
1088
+ def format_eval_error(err: EvalCliError) -> str:
1089
+ if isinstance(err, TomlUnavailableError):
1090
+ hint = err.hint or "Install tomli or use Python 3.11+."
1091
+ return f"TOML parser not available. {hint}"
1092
+ if isinstance(err, EvalConfigNotFoundError):
1093
+ return f"Eval config not found: {err.path}"
1094
+ if isinstance(err, EvalConfigParseError):
1095
+ return f"Failed to parse TOML '{err.path}': {err.detail}"
1096
+ if isinstance(err, MissingEvalTableError):
1097
+ return "Config must contain an [eval] table."
1098
+ if isinstance(err, InvalidEvalConfigError):
1099
+ return f"Invalid eval config: {err.detail}"
1100
+ if isinstance(err, SeedParseError):
1101
+ return f"Unable to parse seeds from '{err.value}'. Provide comma-separated integers."
1102
+ if isinstance(err, MetadataFilterFormatError):
1103
+ return f"Metadata filter '{err.entry}' must be key=value."
1104
+ if isinstance(err, TaskInfoUnavailableError):
1105
+ return "Task metadata filters require the task app to expose /task_info metadata."
1106
+ if isinstance(err, NoSeedsMatchedError):
1107
+ hint = err.hint or "Adjust the metadata filters or seed list."
1108
+ return f"No seeds match the provided metadata filters. {hint}"
1109
+ if isinstance(err, MetadataSQLExecutionError):
1110
+ return f"Failed to execute metadata SQL query '{err.query}': {err.detail}"
1111
+ if isinstance(err, MetadataSQLResultError):
1112
+ return f"metadata SQL query '{err.query}' must return integer seed values ({err.detail})"
1113
+ return str(err)