synth-ai 0.2.6.dev1__py3-none-any.whl → 0.4.3__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.
Files changed (738) hide show
  1. synth_ai/__init__.py +44 -24
  2. synth_ai/__main__.py +30 -3
  3. synth_ai/cli/__init__.py +103 -48
  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/demo/__init__.py +3 -0
  25. synth_ai/cli/commands/demo/core.py +153 -0
  26. synth_ai/cli/commands/eval/__init__.py +10 -0
  27. synth_ai/cli/commands/eval/config.py +338 -0
  28. synth_ai/cli/commands/eval/core.py +256 -0
  29. synth_ai/cli/commands/eval/runner.py +704 -0
  30. synth_ai/cli/commands/eval/validation.py +60 -0
  31. synth_ai/cli/commands/filter/__init__.py +12 -0
  32. synth_ai/cli/commands/filter/core.py +424 -0
  33. synth_ai/cli/commands/filter/errors.py +55 -0
  34. synth_ai/cli/commands/filter/validation.py +77 -0
  35. synth_ai/cli/commands/help/__init__.py +185 -0
  36. synth_ai/cli/commands/help/core.py +72 -0
  37. synth_ai/cli/commands/scan/__init__.py +19 -0
  38. synth_ai/cli/commands/scan/cloudflare_scanner.py +403 -0
  39. synth_ai/cli/commands/scan/core.py +344 -0
  40. synth_ai/cli/commands/scan/health_checker.py +242 -0
  41. synth_ai/cli/commands/scan/local_scanner.py +278 -0
  42. synth_ai/cli/commands/scan/models.py +83 -0
  43. synth_ai/cli/commands/smoke/__init__.py +7 -0
  44. synth_ai/cli/commands/smoke/core.py +1428 -0
  45. synth_ai/cli/commands/status/__init__.py +3 -0
  46. synth_ai/cli/commands/status/client.py +91 -0
  47. synth_ai/cli/commands/status/config.py +12 -0
  48. synth_ai/cli/commands/status/errors.py +11 -0
  49. synth_ai/cli/commands/status/subcommands/__init__.py +3 -0
  50. synth_ai/cli/commands/status/subcommands/config.py +13 -0
  51. synth_ai/cli/commands/status/subcommands/files.py +34 -0
  52. synth_ai/cli/commands/status/subcommands/jobs.py +51 -0
  53. synth_ai/cli/commands/status/subcommands/models.py +35 -0
  54. synth_ai/cli/commands/status/subcommands/runs.py +34 -0
  55. synth_ai/cli/commands/status/subcommands/session.py +77 -0
  56. synth_ai/cli/commands/status/subcommands/summary.py +39 -0
  57. synth_ai/cli/commands/status/subcommands/utils.py +41 -0
  58. synth_ai/cli/commands/status/utils.py +23 -0
  59. synth_ai/cli/commands/train/__init__.py +53 -0
  60. synth_ai/cli/commands/train/core.py +22 -0
  61. synth_ai/cli/commands/train/errors.py +117 -0
  62. synth_ai/cli/commands/train/judge_schemas.py +201 -0
  63. synth_ai/cli/commands/train/judge_validation.py +305 -0
  64. synth_ai/cli/commands/train/prompt_learning_validation.py +633 -0
  65. synth_ai/cli/commands/train/validation.py +392 -0
  66. synth_ai/cli/demo_apps/__init__.py +10 -0
  67. synth_ai/cli/demo_apps/core/__init__.py +28 -0
  68. synth_ai/cli/demo_apps/core/cli.py +1735 -0
  69. synth_ai/cli/demo_apps/crafter/__init__.py +1 -0
  70. synth_ai/cli/demo_apps/crafter/crafter_fft_4b.toml +55 -0
  71. synth_ai/cli/demo_apps/crafter/grpo_crafter_task_app.py +186 -0
  72. synth_ai/cli/demo_apps/crafter/rl_from_base_qwen4b.toml +74 -0
  73. synth_ai/cli/demo_apps/demo_registry.py +176 -0
  74. synth_ai/cli/demo_apps/demo_task_apps/__init__.py +7 -0
  75. synth_ai/{demos → cli/demo_apps}/demo_task_apps/core.py +117 -51
  76. synth_ai/cli/demo_apps/demo_task_apps/crafter/__init__.py +1 -0
  77. synth_ai/cli/demo_apps/demo_task_apps/crafter/configs/crafter_fft_4b.toml +53 -0
  78. synth_ai/cli/demo_apps/demo_task_apps/crafter/configs/rl_from_base_qwen4b.toml +73 -0
  79. synth_ai/cli/demo_apps/demo_task_apps/crafter/grpo_crafter_task_app.py +185 -0
  80. synth_ai/cli/demo_apps/demo_task_apps/math/_common.py +16 -0
  81. synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/app.py +2 -1
  82. synth_ai/cli/demo_apps/demo_task_apps/math/config.toml +73 -0
  83. synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/deploy_modal.py +3 -6
  84. synth_ai/cli/demo_apps/demo_task_apps/math/modal_task_app.py +738 -0
  85. synth_ai/cli/demo_apps/demo_task_apps/math/task_app_entry.py +39 -0
  86. synth_ai/cli/demo_apps/math/__init__.py +1 -0
  87. synth_ai/cli/demo_apps/math/_common.py +16 -0
  88. synth_ai/cli/demo_apps/math/app.py +38 -0
  89. synth_ai/cli/demo_apps/math/config.toml +75 -0
  90. synth_ai/cli/demo_apps/math/deploy_modal.py +54 -0
  91. synth_ai/cli/demo_apps/math/modal_task_app.py +698 -0
  92. synth_ai/cli/demo_apps/math/task_app_entry.py +53 -0
  93. synth_ai/cli/demo_apps/mipro/main.py +271 -0
  94. synth_ai/cli/demo_apps/mipro/task_app.py +922 -0
  95. synth_ai/cli/demo_apps/mipro/train_cfg.toml +92 -0
  96. synth_ai/cli/demos/__init__.py +12 -0
  97. synth_ai/cli/demos/demo.py +32 -0
  98. synth_ai/cli/demos/rl_demo.py +254 -0
  99. synth_ai/cli/deploy.py +216 -0
  100. synth_ai/cli/infra/__init__.py +14 -0
  101. synth_ai/cli/{balance.py → infra/balance.py} +21 -3
  102. synth_ai/cli/infra/mcp.py +35 -0
  103. synth_ai/cli/infra/modal_app.py +36 -0
  104. synth_ai/cli/infra/setup.py +69 -0
  105. synth_ai/cli/infra/status.py +16 -0
  106. synth_ai/cli/infra/turso.py +77 -0
  107. synth_ai/cli/lib/__init__.py +10 -0
  108. synth_ai/cli/lib/agents.py +76 -0
  109. synth_ai/cli/lib/apps/modal_app.py +101 -0
  110. synth_ai/cli/lib/apps/task_app.py +642 -0
  111. synth_ai/cli/lib/bin.py +39 -0
  112. synth_ai/cli/lib/env.py +375 -0
  113. synth_ai/cli/lib/errors.py +85 -0
  114. synth_ai/cli/lib/modal.py +315 -0
  115. synth_ai/cli/lib/plotting.py +126 -0
  116. synth_ai/cli/lib/prompt_args.py +39 -0
  117. synth_ai/cli/lib/prompts.py +284 -0
  118. synth_ai/cli/lib/sqld.py +122 -0
  119. synth_ai/cli/lib/task_app_discovery.py +884 -0
  120. synth_ai/cli/lib/task_app_env.py +295 -0
  121. synth_ai/cli/lib/train_cfgs.py +300 -0
  122. synth_ai/cli/lib/tunnel_records.py +207 -0
  123. synth_ai/cli/local/__init__.py +14 -0
  124. synth_ai/cli/local/experiment_queue/__init__.py +72 -0
  125. synth_ai/cli/local/experiment_queue/api_schemas.py +221 -0
  126. synth_ai/cli/local/experiment_queue/celery_app.py +208 -0
  127. synth_ai/cli/local/experiment_queue/config.py +128 -0
  128. synth_ai/cli/local/experiment_queue/config_utils.py +272 -0
  129. synth_ai/cli/local/experiment_queue/database.py +175 -0
  130. synth_ai/cli/local/experiment_queue/dispatcher.py +119 -0
  131. synth_ai/cli/local/experiment_queue/models.py +231 -0
  132. synth_ai/cli/local/experiment_queue/progress_info.py +160 -0
  133. synth_ai/cli/local/experiment_queue/results.py +373 -0
  134. synth_ai/cli/local/experiment_queue/schemas.py +131 -0
  135. synth_ai/cli/local/experiment_queue/service.py +344 -0
  136. synth_ai/cli/local/experiment_queue/status.py +372 -0
  137. synth_ai/cli/local/experiment_queue/status_tracker.py +360 -0
  138. synth_ai/cli/local/experiment_queue/tasks.py +1984 -0
  139. synth_ai/cli/local/experiment_queue/trace_storage.py +65 -0
  140. synth_ai/cli/local/experiment_queue/validation.py +157 -0
  141. synth_ai/cli/local/session/__init__.py +92 -0
  142. synth_ai/cli/local/session/client.py +383 -0
  143. synth_ai/cli/local/session/constants.py +63 -0
  144. synth_ai/cli/local/session/exceptions.py +105 -0
  145. synth_ai/cli/local/session/manager.py +139 -0
  146. synth_ai/cli/local/session/models.py +89 -0
  147. synth_ai/cli/local/session/query.py +110 -0
  148. synth_ai/cli/root.py +150 -102
  149. synth_ai/cli/task_apps/__init__.py +37 -0
  150. synth_ai/cli/task_apps/commands.py +3145 -0
  151. synth_ai/cli/task_apps/deploy.py +7 -0
  152. synth_ai/cli/task_apps/list.py +26 -0
  153. synth_ai/cli/task_apps/main.py +36 -0
  154. synth_ai/cli/task_apps/modal_serve.py +11 -0
  155. synth_ai/cli/task_apps/serve.py +11 -0
  156. synth_ai/cli/training/__init__.py +8 -0
  157. synth_ai/cli/training/train.py +5 -0
  158. synth_ai/cli/training/train_cfg.py +34 -0
  159. synth_ai/cli/{watch.py → training/watch.py} +13 -18
  160. synth_ai/cli/turso.py +52 -0
  161. synth_ai/cli/utils/__init__.py +8 -0
  162. synth_ai/cli/utils/experiments.py +235 -0
  163. synth_ai/cli/utils/queue.py +504 -0
  164. synth_ai/cli/{recent.py → utils/recent.py} +13 -7
  165. synth_ai/cli/{traces.py → utils/traces.py} +9 -5
  166. synth_ai/contracts/__init__.py +67 -0
  167. synth_ai/core/__init__.py +100 -0
  168. synth_ai/core/_utils/__init__.py +54 -0
  169. synth_ai/core/_utils/base_url.py +10 -0
  170. synth_ai/core/_utils/http.py +10 -0
  171. synth_ai/core/_utils/prompts.py +14 -0
  172. synth_ai/core/_utils/task_app_state.py +12 -0
  173. synth_ai/core/_utils/user_config.py +10 -0
  174. synth_ai/core/apps/common.py +116 -0
  175. synth_ai/core/auth.py +95 -0
  176. synth_ai/core/cfgs.py +240 -0
  177. synth_ai/core/config/__init__.py +16 -0
  178. synth_ai/core/config/base.py +168 -0
  179. synth_ai/core/config/resolver.py +89 -0
  180. synth_ai/core/env.py +231 -0
  181. synth_ai/core/errors.py +126 -0
  182. synth_ai/core/http.py +230 -0
  183. synth_ai/core/integrations/__init__.py +11 -0
  184. synth_ai/core/integrations/cloudflare.py +1710 -0
  185. synth_ai/core/integrations/mcp/__init__.py +6 -0
  186. synth_ai/core/integrations/mcp/__main__.py +8 -0
  187. synth_ai/core/integrations/mcp/claude.py +36 -0
  188. synth_ai/core/integrations/mcp/main.py +254 -0
  189. synth_ai/core/integrations/mcp/setup.py +100 -0
  190. synth_ai/core/integrations/modal.py +277 -0
  191. synth_ai/core/json.py +72 -0
  192. synth_ai/core/log_filter.py +99 -0
  193. synth_ai/core/logging.py +82 -0
  194. synth_ai/core/paths.py +107 -0
  195. synth_ai/core/pricing.py +109 -0
  196. synth_ai/core/process.py +233 -0
  197. synth_ai/core/ssl.py +25 -0
  198. synth_ai/core/storage/__init__.py +71 -0
  199. synth_ai/core/task_app_state.py +318 -0
  200. synth_ai/core/telemetry.py +282 -0
  201. synth_ai/{tracing_v3 → core/tracing_v3}/__init__.py +5 -1
  202. synth_ai/{tracing_v3 → core/tracing_v3}/abstractions.py +21 -4
  203. synth_ai/core/tracing_v3/config.py +229 -0
  204. synth_ai/core/tracing_v3/constants.py +21 -0
  205. synth_ai/{tracing_v3 → core/tracing_v3}/db_config.py +42 -29
  206. synth_ai/{tracing_v3 → core/tracing_v3}/decorators.py +80 -45
  207. synth_ai/{tracing_v3 → core/tracing_v3}/examples/basic_usage.py +15 -9
  208. synth_ai/{tracing_v3 → core/tracing_v3}/hooks.py +6 -4
  209. synth_ai/{tracing_v3 → core/tracing_v3}/llm_call_record_helpers.py +161 -61
  210. synth_ai/{tracing_v3 → core/tracing_v3}/migration_helper.py +1 -2
  211. synth_ai/{tracing_v3 → core/tracing_v3}/replica_sync.py +12 -7
  212. synth_ai/core/tracing_v3/serialization.py +130 -0
  213. synth_ai/{tracing_v3 → core/tracing_v3}/session_tracer.py +88 -21
  214. synth_ai/{tracing_v3 → core/tracing_v3}/storage/base.py +99 -12
  215. synth_ai/core/tracing_v3/storage/config.py +109 -0
  216. synth_ai/{tracing_v3 → core/tracing_v3}/storage/factory.py +11 -9
  217. synth_ai/{tracing_v3 → core/tracing_v3}/storage/utils.py +15 -11
  218. synth_ai/core/tracing_v3/trace_utils.py +326 -0
  219. synth_ai/core/tracing_v3/turso/__init__.py +12 -0
  220. synth_ai/core/tracing_v3/turso/daemon.py +278 -0
  221. synth_ai/{tracing_v3 → core/tracing_v3}/turso/models.py +7 -3
  222. synth_ai/core/tracing_v3/turso/native_manager.py +1385 -0
  223. synth_ai/{tracing_v3 → core/tracing_v3}/utils.py +5 -4
  224. synth_ai/core/urls.py +18 -0
  225. synth_ai/core/user_config.py +137 -0
  226. synth_ai/core/uvicorn.py +222 -0
  227. synth_ai/data/__init__.py +83 -0
  228. synth_ai/data/enums.py +123 -0
  229. synth_ai/data/rewards.py +152 -0
  230. synth_ai/data/traces.py +35 -0
  231. synth_ai/products/__init__.py +6 -0
  232. synth_ai/products/graph_evolve/__init__.py +46 -0
  233. synth_ai/products/graph_evolve/client.py +226 -0
  234. synth_ai/products/graph_evolve/config.py +591 -0
  235. synth_ai/products/graph_evolve/converters/__init__.py +42 -0
  236. synth_ai/products/graph_evolve/converters/openai_sft.py +484 -0
  237. synth_ai/products/graph_evolve/examples/hotpotqa/config.toml +109 -0
  238. synth_ai/products/graph_evolve/run.py +222 -0
  239. synth_ai/products/graph_gepa/__init__.py +23 -0
  240. synth_ai/products/graph_gepa/converters/__init__.py +19 -0
  241. synth_ai/products/graph_gepa/converters/openai_sft.py +29 -0
  242. synth_ai/sdk/__init__.py +123 -0
  243. synth_ai/sdk/api/__init__.py +1 -0
  244. synth_ai/sdk/api/models/supported.py +514 -0
  245. synth_ai/sdk/api/research_agent/__init__.py +296 -0
  246. synth_ai/sdk/api/train/__init__.py +85 -0
  247. synth_ai/sdk/api/train/builders.py +895 -0
  248. synth_ai/sdk/api/train/cli.py +2199 -0
  249. synth_ai/sdk/api/train/config_finder.py +267 -0
  250. synth_ai/sdk/api/train/configs/__init__.py +65 -0
  251. synth_ai/sdk/api/train/configs/prompt_learning.py +1706 -0
  252. synth_ai/sdk/api/train/configs/rl.py +187 -0
  253. synth_ai/sdk/api/train/configs/sft.py +99 -0
  254. synth_ai/sdk/api/train/configs/shared.py +81 -0
  255. synth_ai/sdk/api/train/context_learning.py +312 -0
  256. synth_ai/sdk/api/train/env_resolver.py +418 -0
  257. synth_ai/sdk/api/train/graph_validators.py +216 -0
  258. synth_ai/sdk/api/train/graphgen.py +984 -0
  259. synth_ai/sdk/api/train/graphgen_models.py +823 -0
  260. synth_ai/sdk/api/train/graphgen_validators.py +109 -0
  261. synth_ai/sdk/api/train/local_api.py +10 -0
  262. synth_ai/sdk/api/train/pollers.py +124 -0
  263. synth_ai/sdk/api/train/progress/__init__.py +97 -0
  264. synth_ai/sdk/api/train/progress/dataclasses.py +569 -0
  265. synth_ai/sdk/api/train/progress/events.py +326 -0
  266. synth_ai/sdk/api/train/progress/results.py +428 -0
  267. synth_ai/sdk/api/train/progress/tracker.py +641 -0
  268. synth_ai/sdk/api/train/prompt_learning.py +469 -0
  269. synth_ai/sdk/api/train/rl.py +441 -0
  270. synth_ai/sdk/api/train/sft.py +396 -0
  271. synth_ai/sdk/api/train/summary.py +522 -0
  272. synth_ai/sdk/api/train/supported_algos.py +147 -0
  273. synth_ai/sdk/api/train/task_app.py +351 -0
  274. synth_ai/sdk/api/train/utils.py +279 -0
  275. synth_ai/sdk/api/train/validators.py +2424 -0
  276. synth_ai/sdk/graphs/__init__.py +15 -0
  277. synth_ai/sdk/graphs/completions.py +570 -0
  278. synth_ai/{inference → sdk/inference}/__init__.py +0 -1
  279. synth_ai/sdk/inference/client.py +128 -0
  280. synth_ai/sdk/jobs/__init__.py +16 -0
  281. synth_ai/sdk/jobs/client.py +371 -0
  282. synth_ai/sdk/judging/__init__.py +14 -0
  283. synth_ai/sdk/judging/base.py +24 -0
  284. synth_ai/sdk/judging/client.py +40 -0
  285. synth_ai/sdk/judging/schemas.py +222 -0
  286. synth_ai/sdk/judging/types.py +42 -0
  287. synth_ai/sdk/learning/__init__.py +99 -0
  288. synth_ai/sdk/learning/algorithms.py +14 -0
  289. synth_ai/{learning → sdk/learning}/client.py +121 -30
  290. synth_ai/sdk/learning/config.py +5 -0
  291. synth_ai/{learning → sdk/learning}/constants.py +0 -2
  292. synth_ai/sdk/learning/context_learning_client.py +531 -0
  293. synth_ai/sdk/learning/context_learning_types.py +292 -0
  294. synth_ai/sdk/learning/ft_client.py +7 -0
  295. synth_ai/{learning → sdk/learning}/health.py +15 -9
  296. synth_ai/{learning → sdk/learning}/jobs.py +44 -47
  297. synth_ai/sdk/learning/prompt_extraction.py +334 -0
  298. synth_ai/sdk/learning/prompt_learning_client.py +455 -0
  299. synth_ai/sdk/learning/prompt_learning_types.py +186 -0
  300. synth_ai/{rl → sdk/learning/rl}/__init__.py +13 -8
  301. synth_ai/{learning/rl_client.py → sdk/learning/rl/client.py} +89 -77
  302. synth_ai/sdk/learning/rl/config.py +31 -0
  303. synth_ai/{rl → sdk/learning/rl}/contracts.py +5 -14
  304. synth_ai/{rl → sdk/learning/rl}/env_keys.py +45 -16
  305. synth_ai/sdk/learning/rl/secrets.py +13 -0
  306. synth_ai/sdk/learning/rl_client.py +5 -0
  307. synth_ai/sdk/learning/sft/__init__.py +29 -0
  308. synth_ai/sdk/learning/sft/client.py +95 -0
  309. synth_ai/sdk/learning/sft/config.py +270 -0
  310. synth_ai/sdk/learning/sft/data.py +698 -0
  311. synth_ai/sdk/learning/sse.py +57 -0
  312. synth_ai/sdk/learning/validators.py +52 -0
  313. synth_ai/sdk/localapi/__init__.py +40 -0
  314. synth_ai/sdk/localapi/apps/__init__.py +28 -0
  315. synth_ai/sdk/localapi/client.py +10 -0
  316. synth_ai/sdk/localapi/contracts.py +10 -0
  317. synth_ai/sdk/localapi/helpers.py +519 -0
  318. synth_ai/sdk/localapi/rollouts.py +87 -0
  319. synth_ai/sdk/localapi/server.py +29 -0
  320. synth_ai/sdk/localapi/template.py +70 -0
  321. synth_ai/sdk/streaming/__init__.py +35 -0
  322. synth_ai/sdk/streaming/config.py +94 -0
  323. synth_ai/sdk/streaming/handlers.py +1997 -0
  324. synth_ai/sdk/streaming/streamer.py +713 -0
  325. synth_ai/sdk/streaming/types.py +112 -0
  326. synth_ai/sdk/task/__init__.py +164 -0
  327. synth_ai/sdk/task/apps/__init__.py +169 -0
  328. synth_ai/sdk/task/auth.py +165 -0
  329. synth_ai/sdk/task/client.py +175 -0
  330. synth_ai/sdk/task/config.py +257 -0
  331. synth_ai/sdk/task/contracts.py +219 -0
  332. synth_ai/sdk/task/datasets.py +108 -0
  333. synth_ai/sdk/task/errors.py +50 -0
  334. synth_ai/sdk/task/health.py +34 -0
  335. synth_ai/sdk/task/in_process.py +1190 -0
  336. synth_ai/sdk/task/in_process_runner.py +314 -0
  337. synth_ai/sdk/task/inference_api.py +299 -0
  338. synth_ai/sdk/task/json.py +111 -0
  339. synth_ai/sdk/task/proxy.py +287 -0
  340. synth_ai/sdk/task/rubrics/__init__.py +55 -0
  341. synth_ai/sdk/task/rubrics/loaders.py +156 -0
  342. synth_ai/sdk/task/rubrics/models.py +57 -0
  343. synth_ai/sdk/task/rubrics/scoring.py +116 -0
  344. synth_ai/sdk/task/rubrics/strict.py +149 -0
  345. synth_ai/sdk/task/rubrics.py +219 -0
  346. synth_ai/sdk/task/server.py +631 -0
  347. synth_ai/sdk/task/trace_correlation_helpers.py +539 -0
  348. synth_ai/sdk/task/tracing_utils.py +95 -0
  349. synth_ai/sdk/task/validators.py +441 -0
  350. synth_ai/sdk/task/vendors.py +59 -0
  351. synth_ai/sdk/training/__init__.py +102 -0
  352. synth_ai/sdk/tunnels/__init__.py +83 -0
  353. synth_ai/sdk/tunnels/cleanup.py +83 -0
  354. synth_ai/sdk/tunnels/ports.py +120 -0
  355. synth_ai/utils/__init__.py +213 -0
  356. synth_ai-0.4.3.dist-info/METADATA +262 -0
  357. synth_ai-0.4.3.dist-info/RECORD +370 -0
  358. {synth_ai-0.2.6.dev1.dist-info → synth_ai-0.4.3.dist-info}/entry_points.txt +0 -1
  359. synth_ai/cli/calc.py +0 -69
  360. synth_ai/cli/demo.py +0 -131
  361. synth_ai/cli/legacy_root_backup.py +0 -470
  362. synth_ai/cli/man.py +0 -106
  363. synth_ai/cli/rl_demo.py +0 -137
  364. synth_ai/cli/status.py +0 -133
  365. synth_ai/config/base_url.py +0 -98
  366. synth_ai/core/experiment.py +0 -15
  367. synth_ai/core/system.py +0 -15
  368. synth_ai/demos/core/__init__.py +0 -1
  369. synth_ai/demos/core/cli.py +0 -685
  370. synth_ai/demos/demo_task_apps/__init__.py +0 -1
  371. synth_ai/demos/demo_task_apps/math/config.toml +0 -44
  372. synth_ai/demos/demo_task_apps/math/deploy_task_app.sh +0 -22
  373. synth_ai/environments/__init__.py +0 -31
  374. synth_ai/environments/environment/__init__.py +0 -1
  375. synth_ai/environments/environment/artifacts/__init__.py +0 -1
  376. synth_ai/environments/environment/artifacts/base.py +0 -52
  377. synth_ai/environments/environment/core.py +0 -67
  378. synth_ai/environments/environment/db/__init__.py +0 -1
  379. synth_ai/environments/environment/db/sqlite.py +0 -45
  380. synth_ai/environments/environment/registry.py +0 -233
  381. synth_ai/environments/environment/resources/sqlite.py +0 -45
  382. synth_ai/environments/environment/results.py +0 -1
  383. synth_ai/environments/environment/rewards/__init__.py +0 -1
  384. synth_ai/environments/environment/rewards/core.py +0 -29
  385. synth_ai/environments/environment/shared_engine.py +0 -26
  386. synth_ai/environments/environment/tools/__init__.py +0 -200
  387. synth_ai/environments/examples/__init__.py +0 -1
  388. synth_ai/environments/examples/bandit/__init__.py +0 -33
  389. synth_ai/environments/examples/bandit/engine.py +0 -294
  390. synth_ai/environments/examples/bandit/environment.py +0 -194
  391. synth_ai/environments/examples/bandit/taskset.py +0 -200
  392. synth_ai/environments/examples/crafter_classic/__init__.py +0 -8
  393. synth_ai/environments/examples/crafter_classic/agent_demos/analyze_semantic_words_markdown.py +0 -250
  394. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_comprehensive_evaluation.py +0 -59
  395. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_evaluation_browser.py +0 -152
  396. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_evaluation_config.toml +0 -24
  397. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_evaluation_framework.py +0 -1194
  398. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/crafter_synth_config.toml +0 -56
  399. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/filter_config_modal.toml +0 -32
  400. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/filter_traces_sft_turso.py +0 -724
  401. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/kick_off_ft_modal.py +0 -384
  402. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_action_results.py +0 -53
  403. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_agent_actions.py +0 -178
  404. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_latest_run.py +0 -222
  405. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_lm_traces.py +0 -183
  406. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_no_rewards.py +0 -210
  407. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_trace_issue.py +0 -206
  408. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/check_db_schema.py +0 -49
  409. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/check_latest_results.py +0 -64
  410. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/debug_agent_responses.py +0 -88
  411. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/quick_trace_check.py +0 -77
  412. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/compare_experiments.py +0 -324
  413. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/filter_traces_sft_turso.py +0 -580
  414. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/kick_off_ft_oai.py +0 -362
  415. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/multi_model_config.toml +0 -49
  416. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/analyze_enhanced_hooks.py +0 -332
  417. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/analyze_hook_events.py +0 -97
  418. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/analyze_hook_results.py +0 -217
  419. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/check_hook_storage.py +0 -87
  420. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/check_seeds.py +0 -88
  421. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/compare_seed_performance.py +0 -195
  422. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/custom_eval_pipelines.py +0 -400
  423. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/plot_hook_frequency.py +0 -195
  424. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/seed_analysis_summary.py +0 -56
  425. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/run_rollouts_for_models_and_compare_v3.py +0 -858
  426. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_quick_evaluation.py +0 -52
  427. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_react_agent.py +0 -874
  428. synth_ai/environments/examples/crafter_classic/agent_demos/crafter_trace_evaluation.py +0 -1412
  429. synth_ai/environments/examples/crafter_classic/agent_demos/example_v3_usage.py +0 -216
  430. synth_ai/environments/examples/crafter_classic/agent_demos/old/compare_traces.py +0 -296
  431. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_comprehensive_evaluation.py +0 -58
  432. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_env_serialization.py +0 -464
  433. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_evaluation_browser.py +0 -152
  434. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_quick_evaluation.py +0 -51
  435. synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_trace_evaluation.py +0 -1412
  436. synth_ai/environments/examples/crafter_classic/agent_demos/old/debug_player_loss.py +0 -112
  437. synth_ai/environments/examples/crafter_classic/agent_demos/old/diagnose_service.py +0 -203
  438. synth_ai/environments/examples/crafter_classic/agent_demos/old/diagnose_slowness.py +0 -305
  439. synth_ai/environments/examples/crafter_classic/agent_demos/old/eval_by_difficulty.py +0 -126
  440. synth_ai/environments/examples/crafter_classic/agent_demos/old/eval_example.py +0 -94
  441. synth_ai/environments/examples/crafter_classic/agent_demos/old/explore_saved_states.py +0 -142
  442. synth_ai/environments/examples/crafter_classic/agent_demos/old/filter_traces_sft.py +0 -26
  443. synth_ai/environments/examples/crafter_classic/agent_demos/old/filter_traces_sft_OLD.py +0 -984
  444. synth_ai/environments/examples/crafter_classic/agent_demos/old/generate_ft_data_gemini.py +0 -724
  445. synth_ai/environments/examples/crafter_classic/agent_demos/old/generate_ft_data_modal.py +0 -386
  446. synth_ai/environments/examples/crafter_classic/agent_demos/old/generate_ft_metadata.py +0 -205
  447. synth_ai/environments/examples/crafter_classic/agent_demos/old/kick_off_ft_gemini.py +0 -150
  448. synth_ai/environments/examples/crafter_classic/agent_demos/old/kick_off_ft_modal.py +0 -283
  449. synth_ai/environments/examples/crafter_classic/agent_demos/old/prepare_vertex_ft.py +0 -280
  450. synth_ai/environments/examples/crafter_classic/agent_demos/old/profile_env_slowness.py +0 -456
  451. synth_ai/environments/examples/crafter_classic/agent_demos/old/replicate_issue.py +0 -166
  452. synth_ai/environments/examples/crafter_classic/agent_demos/old/run_and_eval.py +0 -102
  453. synth_ai/environments/examples/crafter_classic/agent_demos/old/run_comparison.py +0 -128
  454. synth_ai/environments/examples/crafter_classic/agent_demos/old/run_qwen_rollouts.py +0 -655
  455. synth_ai/environments/examples/crafter_classic/agent_demos/old/trace_eval_OLD.py +0 -202
  456. synth_ai/environments/examples/crafter_classic/agent_demos/old/validate_openai_format.py +0 -166
  457. synth_ai/environments/examples/crafter_classic/config_logging.py +0 -111
  458. synth_ai/environments/examples/crafter_classic/debug_translation.py +0 -0
  459. synth_ai/environments/examples/crafter_classic/engine.py +0 -579
  460. synth_ai/environments/examples/crafter_classic/engine_deterministic_patch.py +0 -64
  461. synth_ai/environments/examples/crafter_classic/engine_helpers/action_map.py +0 -6
  462. synth_ai/environments/examples/crafter_classic/engine_helpers/serialization.py +0 -75
  463. synth_ai/environments/examples/crafter_classic/engine_serialization_patch_v3.py +0 -267
  464. synth_ai/environments/examples/crafter_classic/environment.py +0 -404
  465. synth_ai/environments/examples/crafter_classic/taskset.py +0 -233
  466. synth_ai/environments/examples/crafter_classic/trace_hooks_v3.py +0 -228
  467. synth_ai/environments/examples/crafter_classic/world_config_patch_simple.py +0 -299
  468. synth_ai/environments/examples/crafter_custom/__init__.py +0 -4
  469. synth_ai/environments/examples/crafter_custom/agent_demos/__init__.py +0 -1
  470. synth_ai/environments/examples/crafter_custom/agent_demos/trace_eval.py +0 -202
  471. synth_ai/environments/examples/crafter_custom/crafter/__init__.py +0 -7
  472. synth_ai/environments/examples/crafter_custom/crafter/config.py +0 -182
  473. synth_ai/environments/examples/crafter_custom/crafter/constants.py +0 -8
  474. synth_ai/environments/examples/crafter_custom/crafter/engine.py +0 -269
  475. synth_ai/environments/examples/crafter_custom/crafter/env.py +0 -262
  476. synth_ai/environments/examples/crafter_custom/crafter/objects.py +0 -417
  477. synth_ai/environments/examples/crafter_custom/crafter/recorder.py +0 -187
  478. synth_ai/environments/examples/crafter_custom/crafter/worldgen.py +0 -118
  479. synth_ai/environments/examples/crafter_custom/dataset_builder.py +0 -373
  480. synth_ai/environments/examples/crafter_custom/environment.py +0 -312
  481. synth_ai/environments/examples/crafter_custom/old/analyze_diamond_issue.py +0 -159
  482. synth_ai/environments/examples/crafter_custom/old/analyze_diamond_spawning.py +0 -158
  483. synth_ai/environments/examples/crafter_custom/old/compare_worlds.py +0 -71
  484. synth_ai/environments/examples/crafter_custom/old/dataset_stats.py +0 -105
  485. synth_ai/environments/examples/crafter_custom/old/diamond_spawning_summary.py +0 -119
  486. synth_ai/environments/examples/crafter_custom/old/example_dataset_usage.py +0 -52
  487. synth_ai/environments/examples/crafter_custom/run_dataset.py +0 -305
  488. synth_ai/environments/examples/enron/art_helpers/email_search_tools.py +0 -156
  489. synth_ai/environments/examples/enron/art_helpers/local_email_db.py +0 -281
  490. synth_ai/environments/examples/enron/art_helpers/types_enron.py +0 -25
  491. synth_ai/environments/examples/enron/engine.py +0 -295
  492. synth_ai/environments/examples/enron/environment.py +0 -166
  493. synth_ai/environments/examples/enron/taskset.py +0 -112
  494. synth_ai/environments/examples/enron/units/keyword_stats.py +0 -112
  495. synth_ai/environments/examples/minigrid/__init__.py +0 -48
  496. synth_ai/environments/examples/minigrid/agent_demos/minigrid_evaluation_framework.py +0 -1188
  497. synth_ai/environments/examples/minigrid/agent_demos/minigrid_quick_evaluation.py +0 -48
  498. synth_ai/environments/examples/minigrid/agent_demos/minigrid_react_agent.py +0 -562
  499. synth_ai/environments/examples/minigrid/agent_demos/minigrid_trace_evaluation.py +0 -221
  500. synth_ai/environments/examples/minigrid/engine.py +0 -589
  501. synth_ai/environments/examples/minigrid/environment.py +0 -274
  502. synth_ai/environments/examples/minigrid/environment_mapping.py +0 -242
  503. synth_ai/environments/examples/minigrid/puzzle_loader.py +0 -417
  504. synth_ai/environments/examples/minigrid/taskset.py +0 -583
  505. synth_ai/environments/examples/nethack/__init__.py +0 -7
  506. synth_ai/environments/examples/nethack/achievements.py +0 -337
  507. synth_ai/environments/examples/nethack/agent_demos/nethack_evaluation_framework.py +0 -981
  508. synth_ai/environments/examples/nethack/agent_demos/nethack_quick_evaluation.py +0 -74
  509. synth_ai/environments/examples/nethack/agent_demos/nethack_react_agent.py +0 -831
  510. synth_ai/environments/examples/nethack/engine.py +0 -739
  511. synth_ai/environments/examples/nethack/environment.py +0 -256
  512. synth_ai/environments/examples/nethack/helpers/__init__.py +0 -41
  513. synth_ai/environments/examples/nethack/helpers/action_mapping.py +0 -301
  514. synth_ai/environments/examples/nethack/helpers/nle_wrapper.py +0 -402
  515. synth_ai/environments/examples/nethack/helpers/observation_utils.py +0 -433
  516. synth_ai/environments/examples/nethack/helpers/recording_wrapper.py +0 -200
  517. synth_ai/environments/examples/nethack/helpers/trajectory_recorder.py +0 -269
  518. synth_ai/environments/examples/nethack/helpers/visualization/replay_viewer.py +0 -308
  519. synth_ai/environments/examples/nethack/helpers/visualization/visualizer.py +0 -431
  520. synth_ai/environments/examples/nethack/taskset.py +0 -323
  521. synth_ai/environments/examples/red/__init__.py +0 -7
  522. synth_ai/environments/examples/red/agent_demos/__init__.py +0 -1
  523. synth_ai/environments/examples/red/config_logging.py +0 -110
  524. synth_ai/environments/examples/red/engine.py +0 -694
  525. synth_ai/environments/examples/red/engine_helpers/__init__.py +0 -1
  526. synth_ai/environments/examples/red/engine_helpers/memory_map.py +0 -28
  527. synth_ai/environments/examples/red/engine_helpers/reward_components.py +0 -276
  528. synth_ai/environments/examples/red/engine_helpers/reward_library/__init__.py +0 -142
  529. synth_ai/environments/examples/red/engine_helpers/reward_library/adaptive_rewards.py +0 -57
  530. synth_ai/environments/examples/red/engine_helpers/reward_library/battle_rewards.py +0 -284
  531. synth_ai/environments/examples/red/engine_helpers/reward_library/composite_rewards.py +0 -150
  532. synth_ai/environments/examples/red/engine_helpers/reward_library/economy_rewards.py +0 -138
  533. synth_ai/environments/examples/red/engine_helpers/reward_library/efficiency_rewards.py +0 -57
  534. synth_ai/environments/examples/red/engine_helpers/reward_library/exploration_rewards.py +0 -331
  535. synth_ai/environments/examples/red/engine_helpers/reward_library/novelty_rewards.py +0 -121
  536. synth_ai/environments/examples/red/engine_helpers/reward_library/pallet_town_rewards.py +0 -559
  537. synth_ai/environments/examples/red/engine_helpers/reward_library/pokemon_rewards.py +0 -313
  538. synth_ai/environments/examples/red/engine_helpers/reward_library/social_rewards.py +0 -148
  539. synth_ai/environments/examples/red/engine_helpers/reward_library/story_rewards.py +0 -247
  540. synth_ai/environments/examples/red/engine_helpers/screen_analysis.py +0 -368
  541. synth_ai/environments/examples/red/engine_helpers/state_extraction.py +0 -140
  542. synth_ai/environments/examples/red/environment.py +0 -238
  543. synth_ai/environments/examples/red/taskset.py +0 -79
  544. synth_ai/environments/examples/red/units/__init__.py +0 -1
  545. synth_ai/environments/examples/sokoban/__init__.py +0 -1
  546. synth_ai/environments/examples/sokoban/agent_demos/sokoban_full_eval.py +0 -899
  547. synth_ai/environments/examples/sokoban/engine.py +0 -678
  548. synth_ai/environments/examples/sokoban/engine_helpers/__init__.py +0 -1
  549. synth_ai/environments/examples/sokoban/engine_helpers/room_utils.py +0 -657
  550. synth_ai/environments/examples/sokoban/engine_helpers/vendored/__init__.py +0 -18
  551. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/__init__.py +0 -3
  552. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/boxoban_env.py +0 -131
  553. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/render_utils.py +0 -370
  554. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/room_utils.py +0 -332
  555. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env.py +0 -306
  556. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_fixed_targets.py +0 -67
  557. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_pull.py +0 -115
  558. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_two_player.py +0 -123
  559. synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_variations.py +0 -394
  560. synth_ai/environments/examples/sokoban/environment.py +0 -229
  561. synth_ai/environments/examples/sokoban/generate_verified_puzzles.py +0 -440
  562. synth_ai/environments/examples/sokoban/puzzle_loader.py +0 -312
  563. synth_ai/environments/examples/sokoban/taskset.py +0 -428
  564. synth_ai/environments/examples/sokoban/units/astar_common.py +0 -95
  565. synth_ai/environments/examples/tictactoe/__init__.py +0 -1
  566. synth_ai/environments/examples/tictactoe/engine.py +0 -368
  567. synth_ai/environments/examples/tictactoe/environment.py +0 -240
  568. synth_ai/environments/examples/tictactoe/taskset.py +0 -215
  569. synth_ai/environments/examples/verilog/__init__.py +0 -10
  570. synth_ai/environments/examples/verilog/engine.py +0 -329
  571. synth_ai/environments/examples/verilog/environment.py +0 -350
  572. synth_ai/environments/examples/verilog/taskset.py +0 -420
  573. synth_ai/environments/examples/wordle/__init__.py +0 -29
  574. synth_ai/environments/examples/wordle/engine.py +0 -398
  575. synth_ai/environments/examples/wordle/environment.py +0 -159
  576. synth_ai/environments/examples/wordle/helpers/generate_instances_wordfreq.py +0 -75
  577. synth_ai/environments/examples/wordle/taskset.py +0 -230
  578. synth_ai/environments/reproducibility/core.py +0 -42
  579. synth_ai/environments/reproducibility/helpers.py +0 -0
  580. synth_ai/environments/reproducibility/tree.py +0 -364
  581. synth_ai/environments/service/app.py +0 -91
  582. synth_ai/environments/service/core_routes.py +0 -1020
  583. synth_ai/environments/service/external_registry.py +0 -56
  584. synth_ai/environments/service/registry.py +0 -9
  585. synth_ai/environments/stateful/__init__.py +0 -1
  586. synth_ai/environments/stateful/core.py +0 -163
  587. synth_ai/environments/stateful/engine.py +0 -21
  588. synth_ai/environments/stateful/state.py +0 -7
  589. synth_ai/environments/tasks/api.py +0 -19
  590. synth_ai/environments/tasks/core.py +0 -80
  591. synth_ai/environments/tasks/filters.py +0 -41
  592. synth_ai/environments/tasks/utils.py +0 -91
  593. synth_ai/environments/v0_observability/history.py +0 -3
  594. synth_ai/environments/v0_observability/log.py +0 -2
  595. synth_ai/evals/base.py +0 -15
  596. synth_ai/experimental/synth_oss.py +0 -446
  597. synth_ai/http.py +0 -102
  598. synth_ai/inference/client.py +0 -20
  599. synth_ai/install_sqld.sh +0 -40
  600. synth_ai/jobs/client.py +0 -246
  601. synth_ai/learning/__init__.py +0 -24
  602. synth_ai/learning/config.py +0 -43
  603. synth_ai/learning/filtering.py +0 -0
  604. synth_ai/learning/ft_client.py +0 -59
  605. synth_ai/learning/offline/dpo.py +0 -0
  606. synth_ai/learning/offline/providers.py +0 -7
  607. synth_ai/learning/offline/sft.py +0 -0
  608. synth_ai/learning/offline/shared.py +0 -0
  609. synth_ai/learning/online/grpo.py +0 -0
  610. synth_ai/learning/online/irft.py +0 -0
  611. synth_ai/learning/prompts/banking77_injection_eval.py +0 -168
  612. synth_ai/learning/prompts/gepa.py +0 -0
  613. synth_ai/learning/prompts/hello_world_in_context_injection_ex.py +0 -213
  614. synth_ai/learning/prompts/mipro.py +0 -289
  615. synth_ai/learning/prompts/random_search.py +0 -246
  616. synth_ai/learning/prompts/run_mipro_banking77.py +0 -172
  617. synth_ai/learning/prompts/run_random_search_banking77.py +0 -324
  618. synth_ai/learning/sse.py +0 -58
  619. synth_ai/learning/validators.py +0 -48
  620. synth_ai/lm/__init__.py +0 -51
  621. synth_ai/lm/caching/constants.py +0 -6
  622. synth_ai/lm/caching/dbs.py +0 -0
  623. synth_ai/lm/caching/ephemeral.py +0 -102
  624. synth_ai/lm/caching/handler.py +0 -137
  625. synth_ai/lm/caching/initialize.py +0 -11
  626. synth_ai/lm/caching/persistent.py +0 -114
  627. synth_ai/lm/config.py +0 -110
  628. synth_ai/lm/constants.py +0 -32
  629. synth_ai/lm/core/__init__.py +0 -8
  630. synth_ai/lm/core/all.py +0 -73
  631. synth_ai/lm/core/exceptions.py +0 -7
  632. synth_ai/lm/core/main.py +0 -319
  633. synth_ai/lm/core/main_v3.py +0 -594
  634. synth_ai/lm/core/synth_models.py +0 -48
  635. synth_ai/lm/core/vendor_clients.py +0 -188
  636. synth_ai/lm/cost/__init__.py +0 -0
  637. synth_ai/lm/cost/monitor.py +0 -1
  638. synth_ai/lm/cost/statefulness.py +0 -1
  639. synth_ai/lm/injection.py +0 -80
  640. synth_ai/lm/overrides.py +0 -206
  641. synth_ai/lm/provider_support/__init__.py +0 -8
  642. synth_ai/lm/provider_support/anthropic.py +0 -972
  643. synth_ai/lm/provider_support/openai.py +0 -1139
  644. synth_ai/lm/provider_support/suppress_logging.py +0 -31
  645. synth_ai/lm/structured_outputs/__init__.py +0 -0
  646. synth_ai/lm/structured_outputs/handler.py +0 -440
  647. synth_ai/lm/structured_outputs/inject.py +0 -297
  648. synth_ai/lm/structured_outputs/rehabilitate.py +0 -185
  649. synth_ai/lm/tools/__init__.py +0 -3
  650. synth_ai/lm/tools/base.py +0 -172
  651. synth_ai/lm/unified_interface.py +0 -202
  652. synth_ai/lm/vendors/__init__.py +0 -0
  653. synth_ai/lm/vendors/base.py +0 -81
  654. synth_ai/lm/vendors/core/__init__.py +0 -0
  655. synth_ai/lm/vendors/core/anthropic_api.py +0 -387
  656. synth_ai/lm/vendors/core/gemini_api.py +0 -292
  657. synth_ai/lm/vendors/core/mistral_api.py +0 -322
  658. synth_ai/lm/vendors/core/openai_api.py +0 -220
  659. synth_ai/lm/vendors/core/synth_dev_api.py +0 -0
  660. synth_ai/lm/vendors/local/__init__.py +0 -0
  661. synth_ai/lm/vendors/local/ollama.py +0 -0
  662. synth_ai/lm/vendors/openai_standard.py +0 -780
  663. synth_ai/lm/vendors/openai_standard_responses.py +0 -256
  664. synth_ai/lm/vendors/retries.py +0 -22
  665. synth_ai/lm/vendors/supported/__init__.py +0 -0
  666. synth_ai/lm/vendors/supported/custom_endpoint.py +0 -417
  667. synth_ai/lm/vendors/supported/deepseek.py +0 -69
  668. synth_ai/lm/vendors/supported/grok.py +0 -75
  669. synth_ai/lm/vendors/supported/groq.py +0 -16
  670. synth_ai/lm/vendors/supported/ollama.py +0 -15
  671. synth_ai/lm/vendors/supported/openrouter.py +0 -74
  672. synth_ai/lm/vendors/supported/together.py +0 -11
  673. synth_ai/lm/vendors/synth_client.py +0 -808
  674. synth_ai/lm/warmup.py +0 -186
  675. synth_ai/rl/secrets.py +0 -19
  676. synth_ai/scripts/verify_rewards.py +0 -100
  677. synth_ai/task/__init__.py +0 -10
  678. synth_ai/task/contracts.py +0 -120
  679. synth_ai/task/health.py +0 -28
  680. synth_ai/task/validators.py +0 -12
  681. synth_ai/tracing/__init__.py +0 -30
  682. synth_ai/tracing_v1/__init__.py +0 -33
  683. synth_ai/tracing_v3/config.py +0 -84
  684. synth_ai/tracing_v3/storage/config.py +0 -62
  685. synth_ai/tracing_v3/turso/__init__.py +0 -25
  686. synth_ai/tracing_v3/turso/daemon.py +0 -144
  687. synth_ai/tracing_v3/turso/manager.py +0 -760
  688. synth_ai/v0/tracing/__init__.py +0 -0
  689. synth_ai/v0/tracing/abstractions.py +0 -224
  690. synth_ai/v0/tracing/base_client.py +0 -91
  691. synth_ai/v0/tracing/client_manager.py +0 -131
  692. synth_ai/v0/tracing/config.py +0 -140
  693. synth_ai/v0/tracing/context.py +0 -146
  694. synth_ai/v0/tracing/decorators.py +0 -680
  695. synth_ai/v0/tracing/events/__init__.py +0 -0
  696. synth_ai/v0/tracing/events/manage.py +0 -147
  697. synth_ai/v0/tracing/events/scope.py +0 -86
  698. synth_ai/v0/tracing/events/store.py +0 -228
  699. synth_ai/v0/tracing/immediate_client.py +0 -151
  700. synth_ai/v0/tracing/local.py +0 -18
  701. synth_ai/v0/tracing/log_client_base.py +0 -73
  702. synth_ai/v0/tracing/retry_queue.py +0 -186
  703. synth_ai/v0/tracing/trackers.py +0 -515
  704. synth_ai/v0/tracing/upload.py +0 -510
  705. synth_ai/v0/tracing/utils.py +0 -9
  706. synth_ai/v0/tracing_v1/__init__.py +0 -16
  707. synth_ai/v0/tracing_v1/abstractions.py +0 -224
  708. synth_ai/v0/tracing_v1/base_client.py +0 -91
  709. synth_ai/v0/tracing_v1/client_manager.py +0 -131
  710. synth_ai/v0/tracing_v1/config.py +0 -140
  711. synth_ai/v0/tracing_v1/context.py +0 -146
  712. synth_ai/v0/tracing_v1/decorators.py +0 -701
  713. synth_ai/v0/tracing_v1/events/__init__.py +0 -0
  714. synth_ai/v0/tracing_v1/events/manage.py +0 -147
  715. synth_ai/v0/tracing_v1/events/scope.py +0 -86
  716. synth_ai/v0/tracing_v1/events/store.py +0 -228
  717. synth_ai/v0/tracing_v1/immediate_client.py +0 -151
  718. synth_ai/v0/tracing_v1/local.py +0 -18
  719. synth_ai/v0/tracing_v1/log_client_base.py +0 -73
  720. synth_ai/v0/tracing_v1/retry_queue.py +0 -186
  721. synth_ai/v0/tracing_v1/trackers.py +0 -515
  722. synth_ai/v0/tracing_v1/upload.py +0 -525
  723. synth_ai/v0/tracing_v1/utils.py +0 -9
  724. synth_ai/zyk/__init__.py +0 -30
  725. synth_ai-0.2.6.dev1.dist-info/METADATA +0 -106
  726. synth_ai-0.2.6.dev1.dist-info/RECORD +0 -416
  727. /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/__init__.py +0 -0
  728. /synth_ai/{lm/caching → core/apps}/__init__.py +0 -0
  729. /synth_ai/{tracing_v3 → core/tracing_v3}/lm_call_record_abstractions.py +0 -0
  730. /synth_ai/{tracing_v3 → core/tracing_v3}/storage/__init__.py +0 -0
  731. /synth_ai/{tracing_v3 → core/tracing_v3}/storage/exceptions.py +0 -0
  732. /synth_ai/{tracing_v3 → core/tracing_v3}/storage/types.py +0 -0
  733. /synth_ai/{compound/cais.py → py.typed} +0 -0
  734. /synth_ai/{learning → sdk/learning}/core.py +0 -0
  735. /synth_ai/{learning → sdk/learning}/gateway.py +0 -0
  736. {synth_ai-0.2.6.dev1.dist-info → synth_ai-0.4.3.dist-info}/WHEEL +0 -0
  737. {synth_ai-0.2.6.dev1.dist-info → synth_ai-0.4.3.dist-info}/licenses/LICENSE +0 -0
  738. {synth_ai-0.2.6.dev1.dist-info → synth_ai-0.4.3.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,441 @@
1
+ """Task app validation utilities."""
2
+
3
+ from __future__ import annotations
4
+
5
+ import re
6
+ from typing import Any
7
+ from urllib.parse import urlparse, urlunparse
8
+
9
+ import click
10
+ import httpx
11
+
12
+ from synth_ai.sdk.task.contracts import TaskAppEndpoints # type: ignore[attr-defined]
13
+
14
+
15
+ def validate_rollout_response_for_rl(response_data: dict[str, Any], *, warn_only: bool = False) -> list[str]:
16
+ """Validate that a task app rollout response has required fields for RL training.
17
+
18
+ The backend RL trainer requires:
19
+ 1. A v3 trace with event_history (preferred), OR
20
+ 2. pipeline_metadata["inference_url"] with ?cid= for trace hydration fallback
21
+
22
+ Args:
23
+ response_data: The rollout response dict from task app
24
+ warn_only: If True, return warnings instead of raising exceptions
25
+
26
+ Returns:
27
+ List of validation warnings/errors
28
+
29
+ Raises:
30
+ ValueError: If critical fields are missing (unless warn_only=True)
31
+ """
32
+ issues = []
33
+
34
+ trace_block = response_data.get("trace")
35
+ event_history = None
36
+ if isinstance(trace_block, dict):
37
+ event_history = trace_block.get("event_history")
38
+ if not event_history and isinstance(trace_block.get("session_trace"), dict):
39
+ event_history = trace_block["session_trace"].get("event_history")
40
+
41
+ has_event_history = isinstance(event_history, list) and len(event_history) > 0
42
+
43
+ trace_correlation_id = response_data.get("trace_correlation_id")
44
+ if not trace_correlation_id and isinstance(trace_block, dict):
45
+ trace_meta = trace_block.get("metadata")
46
+ if isinstance(trace_meta, dict):
47
+ trace_correlation_id = trace_meta.get("trace_correlation_id")
48
+
49
+ if not trace_correlation_id:
50
+ issues.append(
51
+ "Missing trace_correlation_id (top-level or trace.metadata). "
52
+ "RL trainer requires this to link traces."
53
+ )
54
+
55
+ if not has_event_history:
56
+ issues.append(
57
+ "trace.event_history is missing or empty. "
58
+ "Return a v3 trace or provide inference_url for hydration."
59
+ )
60
+
61
+ # Check pipeline_metadata inference_url only when trace is missing/empty
62
+ pipeline_metadata = response_data.get("pipeline_metadata")
63
+ inference_url = None
64
+ if isinstance(pipeline_metadata, dict):
65
+ inference_url = pipeline_metadata.get("inference_url")
66
+ if not has_event_history:
67
+ if not inference_url:
68
+ issues.append(
69
+ "pipeline_metadata['inference_url'] is missing. "
70
+ "RL trainer needs this to hydrate traces when event_history is absent."
71
+ )
72
+ elif not isinstance(inference_url, str):
73
+ issues.append(
74
+ f"pipeline_metadata['inference_url'] must be a string, got: {type(inference_url).__name__}"
75
+ )
76
+ elif "?cid=" not in inference_url:
77
+ issues.append(
78
+ f"pipeline_metadata['inference_url'] should contain '?cid=' for trace correlation. "
79
+ f"Got: {inference_url[:80]}..."
80
+ )
81
+
82
+ if issues and not warn_only:
83
+ error_msg = "Task app response validation failed for RL training:\n" + "\n".join(
84
+ f" - {issue}" for issue in issues
85
+ )
86
+ raise ValueError(error_msg)
87
+
88
+ return issues
89
+
90
+
91
+ def normalize_inference_url(url: str | None, *, default: str = "https://api.openai.com/v1/chat/completions") -> str:
92
+ """Normalize an inference URL to include the /v1/chat/completions path.
93
+
94
+ This utility ensures inference URLs have the correct path structure for OpenAI-compatible
95
+ chat completions endpoints, while preserving query parameters (e.g., ?cid=trace_123)
96
+ that may be added for tracing.
97
+
98
+ Args:
99
+ url: The inference URL to normalize (may be None or incomplete)
100
+ default: Default URL to use if url is None/empty
101
+
102
+ Returns:
103
+ Normalized URL with proper path and preserved query parameters
104
+
105
+ Examples:
106
+ >>> normalize_inference_url("https://api.groq.com")
107
+ 'https://api.groq.com/v1/chat/completions'
108
+
109
+ >>> normalize_inference_url("https://modal.host?cid=trace_123")
110
+ 'https://modal.host/v1/chat/completions?cid=trace_123'
111
+
112
+ >>> normalize_inference_url("https://api.openai.com/v1")
113
+ 'https://api.openai.com/v1/chat/completions'
114
+
115
+ >>> normalize_inference_url("https://api.groq.com/openai/v1/chat/completions")
116
+ 'https://api.groq.com/openai/v1/chat/completions'
117
+ """
118
+ candidate = (url or default).strip()
119
+ if not candidate:
120
+ candidate = default
121
+
122
+ parsed = urlparse(candidate)
123
+ path = (parsed.path or "").rstrip("/")
124
+ query = parsed.query or ""
125
+
126
+ # Repair malformed URLs where the completions path ended up in the query string.
127
+ # Example: https://host?cid=trace/v1/chat/completions -> https://host/v1/chat/completions?cid=trace
128
+ if query and "/" in query:
129
+ base_query, remainder = query.split("/", 1)
130
+ remainder_path = remainder
131
+ extra_query = ""
132
+ for separator in ("&", "?"):
133
+ idx = remainder_path.find(separator)
134
+ if idx != -1:
135
+ extra_query = remainder_path[idx + 1 :]
136
+ remainder_path = remainder_path[:idx]
137
+ break
138
+
139
+ query_path = "/" + remainder_path.lstrip("/")
140
+ merged_query_parts: list[str] = []
141
+ if base_query:
142
+ merged_query_parts.append(base_query)
143
+ if extra_query:
144
+ merged_query_parts.append(extra_query)
145
+ merged_query = "&".join(part for part in merged_query_parts if part)
146
+
147
+ if query_path and query_path != "/":
148
+ combined_path = f"{path.rstrip('/')}{query_path}" if path else query_path
149
+ else:
150
+ combined_path = path
151
+
152
+ parsed = parsed._replace(path=combined_path or "", query=merged_query)
153
+ path = (parsed.path or "").rstrip("/")
154
+ query = parsed.query or ""
155
+
156
+ # Check if path already ends with a completions endpoint
157
+ if path.endswith("/v1/chat/completions") or path.endswith("/chat/completions"):
158
+ final_query = parsed.query or ""
159
+ if final_query and "/" in final_query:
160
+ parsed = parsed._replace(query=final_query.split("/", 1)[0])
161
+ result = urlunparse(parsed)
162
+ return str(result)
163
+
164
+ # Determine what to append based on existing path
165
+ if path.endswith("/v1"):
166
+ new_path = f"{path}/chat/completions"
167
+ elif path.endswith("/chat"):
168
+ new_path = f"{path}/completions"
169
+ else:
170
+ new_path = f"{path}/v1/chat/completions" if path else "/v1/chat/completions"
171
+
172
+ parsed = parsed._replace(path=new_path)
173
+ final_query = parsed.query or ""
174
+ if final_query and "/" in final_query:
175
+ parsed = parsed._replace(query=final_query.split("/", 1)[0])
176
+
177
+ result = urlunparse(parsed)
178
+ return str(result)
179
+
180
+
181
+ def validate_task_app_url(url: str | None) -> str:
182
+ """Validate and normalize a task app URL.
183
+
184
+ Args:
185
+ url: URL to validate
186
+
187
+ Returns:
188
+ Normalized URL
189
+
190
+ Raises:
191
+ ValueError: If URL is invalid
192
+ """
193
+ if not url:
194
+ raise ValueError("Task app URL is required")
195
+
196
+ url = url.strip().rstrip("/")
197
+
198
+ # Basic URL validation
199
+ url_pattern = re.compile(
200
+ r"^https?://" # http:// or https://
201
+ r"(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+[A-Z]{2,6}\.?|" # domain...
202
+ r"localhost|" # localhost...
203
+ r"\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})" # ...or ip
204
+ r"(?::\d+)?" # optional port
205
+ r"(?:/?|[/?]\S+)$",
206
+ re.IGNORECASE,
207
+ )
208
+
209
+ if not url_pattern.match(url):
210
+ raise ValueError(f"Invalid task app URL: {url}")
211
+
212
+ return url
213
+
214
+
215
+ def _print_success(msg: str) -> None:
216
+ """Print success message in green."""
217
+ click.echo(click.style(f"✓ {msg}", fg="green"))
218
+
219
+
220
+ def _print_error(msg: str) -> None:
221
+ """Print error message in red."""
222
+ click.echo(click.style(f"✗ {msg}", fg="red"), err=True)
223
+
224
+
225
+ def _print_warning(msg: str) -> None:
226
+ """Print warning message in yellow."""
227
+ click.echo(click.style(f"⚠ {msg}", fg="yellow"))
228
+
229
+
230
+ def _print_info(msg: str) -> None:
231
+ """Print info message."""
232
+ click.echo(f" {msg}")
233
+
234
+
235
+ async def validate_task_app_endpoint(
236
+ url: str,
237
+ api_key: str | None = None,
238
+ min_instances: int = 10,
239
+ verbose: bool = False,
240
+ ) -> tuple[bool, dict[str, Any]]:
241
+ """Validate a task app deployment.
242
+
243
+ Returns:
244
+ (success: bool, results: dict)
245
+ """
246
+ results: dict[str, Any] = {
247
+ "url": url,
248
+ "endpoints": {},
249
+ "auth": {},
250
+ "task_instances": {},
251
+ "overall": False,
252
+ }
253
+
254
+ all_passed = True
255
+ endpoints = TaskAppEndpoints()
256
+
257
+ # Set up headers
258
+ headers = {}
259
+ if api_key:
260
+ headers["X-API-Key"] = api_key
261
+
262
+ click.echo(f"\n{'='*60}")
263
+ click.echo(f"Validating Task App: {url}")
264
+ click.echo(f"{'='*60}\n")
265
+
266
+ async with httpx.AsyncClient(timeout=30.0, follow_redirects=True) as client:
267
+ # 1. Check root endpoint
268
+ click.echo("1. Checking root endpoint...")
269
+ try:
270
+ resp = await client.get(f"{url}{endpoints.root}")
271
+ if resp.status_code == 200:
272
+ data = resp.json()
273
+ _print_success(f"Root endpoint responds (status: {data.get('status')})")
274
+ results["endpoints"]["root"] = {"passed": True, "data": data}
275
+ if verbose:
276
+ _print_info(f"Service: {data.get('service', 'N/A')}")
277
+ else:
278
+ _print_error(f"Root endpoint returned {resp.status_code}")
279
+ results["endpoints"]["root"] = {"passed": False, "status": resp.status_code}
280
+ all_passed = False
281
+ except Exception as e:
282
+ _print_error(f"Root endpoint failed: {e}")
283
+ results["endpoints"]["root"] = {"passed": False, "error": str(e)}
284
+ all_passed = False
285
+
286
+ # 2. Check health endpoint
287
+ click.echo("\n2. Checking health endpoint...")
288
+ try:
289
+ resp = await client.get(f"{url}{endpoints.health}", headers=headers)
290
+ if resp.status_code == 200:
291
+ data = resp.json()
292
+ _print_success(f"Health endpoint responds (healthy: {data.get('healthy')})")
293
+ results["endpoints"]["health"] = {"passed": True, "data": data}
294
+
295
+ # Check auth configuration
296
+ auth_info = data.get("auth", {})
297
+ if auth_info.get("required"):
298
+ _print_info(f"Auth required: {auth_info.get('required')}")
299
+ _print_info(f"Expected key prefix: {auth_info.get('expected_prefix', 'N/A')}")
300
+
301
+ if api_key:
302
+ _print_success("API key provided and accepted")
303
+ results["auth"]["provided"] = True
304
+ results["auth"]["accepted"] = True
305
+ else:
306
+ _print_warning("No API key provided but may be required")
307
+ results["auth"]["provided"] = False
308
+ results["auth"]["required"] = True
309
+ else:
310
+ _print_error(f"Health endpoint returned {resp.status_code}")
311
+ results["endpoints"]["health"] = {"passed": False, "status": resp.status_code}
312
+ all_passed = False
313
+
314
+ if resp.status_code == 403:
315
+ _print_error("Authentication failed - provide API key with --api-key")
316
+ results["auth"]["error"] = "Authentication failed"
317
+
318
+ except Exception as e:
319
+ _print_error(f"Health endpoint failed: {e}")
320
+ results["endpoints"]["health"] = {"passed": False, "error": str(e)}
321
+ all_passed = False
322
+
323
+ # 3. Check info endpoint
324
+ click.echo("\n3. Checking info endpoint...")
325
+ try:
326
+ resp = await client.get(f"{url}{endpoints.info}", headers=headers)
327
+ if resp.status_code == 200:
328
+ data = resp.json()
329
+ _print_success("Info endpoint responds")
330
+ results["endpoints"]["info"] = {"passed": True, "data": data}
331
+
332
+ if verbose:
333
+ service = data.get("service", {})
334
+ task_info = service.get("task", {})
335
+ if isinstance(task_info, dict):
336
+ _print_info(f"Task: {task_info.get('name', 'N/A')}")
337
+ _print_info(f"Version: {service.get('version', 'N/A')}")
338
+
339
+ dataset = data.get("dataset", {})
340
+ if isinstance(dataset, dict):
341
+ _print_info(f"Dataset: {dataset.get('id', 'N/A')}")
342
+ else:
343
+ _print_error(f"Info endpoint returned {resp.status_code}")
344
+ results["endpoints"]["info"] = {"passed": False, "status": resp.status_code}
345
+ all_passed = False
346
+ except Exception as e:
347
+ _print_error(f"Info endpoint failed: {e}")
348
+ results["endpoints"]["info"] = {"passed": False, "error": str(e)}
349
+ all_passed = False
350
+
351
+ # 4. Check task_info endpoint and instance count
352
+ click.echo("\n4. Checking task_info endpoint and instance availability...")
353
+ try:
354
+ # Get taskset descriptor first
355
+ resp = await client.get(f"{url}{endpoints.task_info}", headers=headers)
356
+ if resp.status_code == 200:
357
+ data = resp.json()
358
+ _print_success("Task info endpoint responds")
359
+ results["endpoints"]["task_info"] = {"passed": True}
360
+
361
+ taskset = data.get("taskset", {})
362
+ if verbose and taskset:
363
+ if isinstance(taskset, dict):
364
+ _print_info(f"Taskset: {taskset.get('id', 'N/A')}")
365
+ else:
366
+ _print_info(f"Taskset: {taskset}")
367
+
368
+ # Try to get specific task instances (seeds 0-19)
369
+ # Fetch instances one by one to verify we can get at least min_instances
370
+ instances = []
371
+ for seed in range(min_instances + 5): # Try a few extra
372
+ try:
373
+ resp_seed = await client.get(
374
+ f"{url}{endpoints.task_info}",
375
+ params={"seed": seed},
376
+ headers=headers,
377
+ )
378
+ if resp_seed.status_code == 200:
379
+ instance = resp_seed.json()
380
+ instances.append(instance)
381
+ else:
382
+ break # Stop if we hit an invalid seed
383
+ except Exception:
384
+ break
385
+
386
+ instance_count = len(instances)
387
+ results["task_instances"]["count"] = instance_count
388
+ results["task_instances"]["requested"] = min_instances
389
+
390
+ if instance_count >= min_instances:
391
+ _print_success(f"Found {instance_count} task instances (≥ {min_instances} required)")
392
+ results["task_instances"]["passed"] = True
393
+
394
+ if verbose and instances:
395
+ sample = instances[0]
396
+ task_info_sample = sample.get('task', {})
397
+ if isinstance(task_info_sample, dict):
398
+ _print_info(f"Sample task: {task_info_sample.get('name', 'N/A')}")
399
+ _print_info(f"Environment: {sample.get('environment', 'N/A')}")
400
+ else:
401
+ _print_error(f"Only {instance_count} task instances available (need ≥ {min_instances})")
402
+ results["task_instances"]["passed"] = False
403
+ all_passed = False
404
+ else:
405
+ _print_error(f"Task info endpoint returned {resp.status_code}")
406
+ results["endpoints"]["task_info"] = {"passed": False, "status": resp.status_code}
407
+ all_passed = False
408
+ except Exception as e:
409
+ _print_error(f"Task info endpoint failed: {e}")
410
+ results["endpoints"]["task_info"] = {"passed": False, "error": str(e)}
411
+ results["task_instances"]["passed"] = False
412
+ all_passed = False
413
+
414
+ # 5. Check rollout endpoint structure (don't actually run a rollout)
415
+ click.echo("\n5. Checking rollout endpoint availability...")
416
+ try:
417
+ # Just check if it's registered (OPTIONS or a lightweight probe)
418
+ resp = await client.options(f"{url}{endpoints.rollout}", headers=headers)
419
+ # Many servers return 200 for OPTIONS, some return 405
420
+ if resp.status_code in (200, 204, 405):
421
+ _print_success("Rollout endpoint is registered")
422
+ results["endpoints"]["rollout"] = {"passed": True}
423
+ else:
424
+ _print_warning(f"Rollout endpoint returned unexpected status: {resp.status_code}")
425
+ results["endpoints"]["rollout"] = {"passed": True, "note": "endpoint exists"}
426
+ except Exception as e:
427
+ # OPTIONS might not be supported, that's okay
428
+ _print_info(f"Rollout endpoint check skipped (OPTIONS not supported): {e}")
429
+ results["endpoints"]["rollout"] = {"passed": True, "note": "assumed present"}
430
+
431
+ # Summary
432
+ click.echo(f"\n{'='*60}")
433
+ if all_passed:
434
+ _print_success("All validations passed!")
435
+ click.echo(f"{'='*60}\n")
436
+ else:
437
+ _print_error("Some validations failed. See errors above.")
438
+ click.echo(f"{'='*60}\n")
439
+
440
+ results["overall"] = all_passed
441
+ return all_passed, results
@@ -0,0 +1,59 @@
1
+ """Vendor API key helpers shared by Task Apps."""
2
+
3
+ from __future__ import annotations
4
+
5
+ import os
6
+
7
+ from .errors import http_exception
8
+
9
+ _VENDOR_KEYS = {
10
+ "OPENAI_API_KEY": ("dev_openai_api_key", "DEV_OPENAI_API_KEY"),
11
+ "GROQ_API_KEY": ("dev_groq_api_key", "DEV_GROQ_API_KEY"),
12
+ }
13
+
14
+
15
+ def _mask(value: str, *, prefix: int = 4) -> str:
16
+ if not value:
17
+ return "<empty>"
18
+ visible = value[:prefix]
19
+ return f"{visible}{'…' if len(value) > prefix else ''}"
20
+
21
+
22
+ def _normalize_single(key: str) -> str | None:
23
+ direct = os.getenv(key)
24
+ if direct:
25
+ return direct
26
+ fallbacks = _VENDOR_KEYS.get(key, ())
27
+ for env in fallbacks:
28
+ candidate = os.getenv(env)
29
+ if candidate:
30
+ os.environ[key] = candidate
31
+ print(
32
+ f"[task:vendor] {key} set from {env} (prefix={_mask(candidate)})",
33
+ flush=True,
34
+ )
35
+ return candidate
36
+ return None
37
+
38
+
39
+ def normalize_vendor_keys() -> dict[str, str | None]:
40
+ """Normalise known vendor keys from dev fallbacks and return the mapping."""
41
+
42
+ resolved: dict[str, str | None] = {}
43
+ for key in _VENDOR_KEYS:
44
+ resolved[key] = _normalize_single(key)
45
+ return resolved
46
+
47
+
48
+ def get_openai_key_or_503() -> str:
49
+ key = _normalize_single("OPENAI_API_KEY")
50
+ if not key:
51
+ raise http_exception(503, "missing_openai_api_key", "OPENAI_API_KEY is not configured")
52
+ return key
53
+
54
+
55
+ def get_groq_key_or_503() -> str:
56
+ key = _normalize_single("GROQ_API_KEY")
57
+ if not key:
58
+ raise http_exception(503, "missing_groq_api_key", "GROQ_API_KEY is not configured")
59
+ return key
@@ -0,0 +1,102 @@
1
+ """Training SDK - prompt learning, SFT, RL, and GraphGen jobs.
2
+
3
+ This module provides high-level APIs for running training jobs:
4
+ - PromptLearningJob: GEPA and MIPRO prompt optimization
5
+ - SFTJob: Supervised fine-tuning
6
+ - RLJob: Reinforcement learning (GSPO, GRPO, PPO, etc.)
7
+ - GraphGenJob: Automated Design of Agentic Systems (simplified workflows API)
8
+
9
+ Example:
10
+ from synth_ai.sdk.training import PromptLearningJob, RLJob, GraphGenJob
11
+ from synth_ai.sdk.task.in_process import InProcessTaskApp
12
+
13
+ # Prompt optimization
14
+ job = PromptLearningJob.from_config("config.toml")
15
+ job.submit()
16
+ result = job.poll_until_complete()
17
+
18
+ # RL training with in-process task app
19
+ async with InProcessTaskApp(task_app_path="my_task_app.py", port=8114) as task_app:
20
+ rl_job = RLJob.from_config("rl_config.toml", task_app_url=task_app.url)
21
+ rl_job.submit()
22
+ rl_result = rl_job.poll_until_complete()
23
+
24
+ # GraphGen workflow optimization (no task app needed)
25
+ graphgen_job = GraphGenJob.from_dataset("my_tasks.json", rollout_budget=100)
26
+ graphgen_job.submit()
27
+ result = graphgen_job.stream_until_complete()
28
+ """
29
+
30
+ from __future__ import annotations
31
+
32
+ # Pollers and utilities
33
+ from synth_ai.sdk.api.train.pollers import JobPoller, PollOutcome, RLJobPoller
34
+
35
+ # Re-export from existing locations
36
+ from synth_ai.sdk.api.train.prompt_learning import (
37
+ PromptLearningJob,
38
+ PromptLearningJobConfig,
39
+ PromptLearningJobPoller,
40
+ )
41
+ from synth_ai.sdk.api.train.rl import RLJob, RLJobConfig
42
+ from synth_ai.sdk.api.train.sft import SFTJob
43
+
44
+ # GraphGen (formerly GraphGen)
45
+ from synth_ai.sdk.api.train.graphgen import GraphGenJob, GraphGenJobResult, GraphGenSubmitResult
46
+ from synth_ai.sdk.api.train.graphgen_models import (
47
+ GraphGenJobConfig,
48
+ GraphGenTaskSet,
49
+ GraphGenTask,
50
+ GraphGenGoldOutput,
51
+ GraphGenRubric,
52
+ GraphGenJudgeConfig,
53
+ load_graphgen_taskset,
54
+ parse_graphgen_taskset,
55
+ # GraphGen aliases
56
+ GraphGenJobConfig,
57
+ GraphGenTaskSet,
58
+ GraphGenTask,
59
+ GraphGenGoldOutput,
60
+ GraphGenRubric,
61
+ GraphGenJudgeConfig,
62
+ load_graphgen_taskset,
63
+ parse_graphgen_taskset,
64
+ )
65
+
66
+ __all__ = [
67
+ # Prompt Learning
68
+ "PromptLearningJob",
69
+ "PromptLearningJobConfig",
70
+ "PromptLearningJobPoller",
71
+ # RL
72
+ "RLJob",
73
+ "RLJobConfig",
74
+ "RLJobPoller",
75
+ # SFT
76
+ "SFTJob",
77
+ # GraphGen (preferred names)
78
+ "GraphGenJobConfig",
79
+ "GraphGenTaskSet",
80
+ "GraphGenTask",
81
+ "GraphGenGoldOutput",
82
+ "GraphGenRubric",
83
+ "GraphGenJudgeConfig",
84
+ "load_graphgen_taskset",
85
+ "parse_graphgen_taskset",
86
+ # GraphGen (legacy aliases)
87
+ "GraphGenJob",
88
+ "GraphGenJobConfig",
89
+ "GraphGenJobResult",
90
+ "GraphGenSubmitResult",
91
+ "GraphGenTaskSet",
92
+ "GraphGenTask",
93
+ "GraphGenGoldOutput",
94
+ "GraphGenRubric",
95
+ "GraphGenJudgeConfig",
96
+ "load_graphgen_taskset",
97
+ "parse_graphgen_taskset",
98
+ # Utils
99
+ "JobPoller",
100
+ "PollOutcome",
101
+ ]
102
+
@@ -0,0 +1,83 @@
1
+ """Cloudflare tunnel helpers - direct re-exports.
2
+
3
+ This module provides convenient access to tunnel management functions
4
+ that already exist in synth_ai.core.integrations.cloudflare.
5
+
6
+ Example:
7
+ from synth_ai.sdk.tunnels import (
8
+ rotate_tunnel,
9
+ open_managed_tunnel,
10
+ track_process,
11
+ verify_tunnel_dns_resolution,
12
+ )
13
+
14
+ # Get a managed tunnel from backend
15
+ tunnel = await rotate_tunnel(API_KEY, port=8001, reason="demo")
16
+
17
+ # Start cloudflared with the token
18
+ proc = track_process(open_managed_tunnel(tunnel['tunnel_token']))
19
+
20
+ # Verify the tunnel is ready
21
+ await verify_tunnel_dns_resolution(f"https://{tunnel['hostname']}")
22
+
23
+ Note:
24
+ Processes registered with track_process() are automatically cleaned up
25
+ when Python exits (via atexit). You can also call cleanup_all() manually.
26
+ """
27
+
28
+ from __future__ import annotations
29
+
30
+ # Re-export from cloudflare.py (no wrappers - these are the actual functions)
31
+ from synth_ai.core.integrations.cloudflare import (
32
+ # Tunnel lifecycle
33
+ create_tunnel,
34
+ open_managed_tunnel,
35
+ open_quick_tunnel,
36
+ open_quick_tunnel_with_dns_verification,
37
+ rotate_tunnel,
38
+ stop_tunnel,
39
+ # Verification
40
+ verify_tunnel_dns_resolution,
41
+ wait_for_health_check,
42
+ # Installation
43
+ ensure_cloudflared_installed,
44
+ get_cloudflared_path,
45
+ require_cloudflared,
46
+ # Discovery
47
+ fetch_managed_tunnels,
48
+ ManagedTunnelRecord,
49
+ )
50
+
51
+ # New: process tracking with atexit cleanup
52
+ from synth_ai.sdk.tunnels.cleanup import cleanup_all, track_process, tracked_processes
53
+
54
+ # New: port management utilities (was private in in_process.py)
55
+ from synth_ai.sdk.tunnels.ports import find_available_port, is_port_available, kill_port
56
+
57
+ __all__ = [
58
+ # Tunnel lifecycle
59
+ "rotate_tunnel",
60
+ "create_tunnel",
61
+ "open_managed_tunnel",
62
+ "open_quick_tunnel",
63
+ "open_quick_tunnel_with_dns_verification",
64
+ "stop_tunnel",
65
+ # Verification
66
+ "verify_tunnel_dns_resolution",
67
+ "wait_for_health_check",
68
+ # Installation
69
+ "require_cloudflared",
70
+ "ensure_cloudflared_installed",
71
+ "get_cloudflared_path",
72
+ # Discovery
73
+ "fetch_managed_tunnels",
74
+ "ManagedTunnelRecord",
75
+ # Process tracking (NEW)
76
+ "track_process",
77
+ "cleanup_all",
78
+ "tracked_processes",
79
+ # Port management (NEW - was private)
80
+ "kill_port",
81
+ "is_port_available",
82
+ "find_available_port",
83
+ ]