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.
- synth_ai/__init__.py +19 -40
- synth_ai/__main__.py +30 -3
- synth_ai/cli/__init__.py +105 -70
- synth_ai/cli/__main__.py +42 -0
- synth_ai/cli/_internal/__init__.py +5 -0
- synth_ai/cli/_internal/modal_wrapper.py +31 -0
- synth_ai/cli/_internal/storage.py +20 -0
- synth_ai/cli/_internal/typer_patch.py +47 -0
- synth_ai/cli/_internal/validate_task_app.py +29 -0
- synth_ai/cli/agents/__init__.py +17 -0
- synth_ai/cli/agents/claude.py +77 -0
- synth_ai/cli/agents/codex.py +265 -0
- synth_ai/cli/agents/opencode.py +253 -0
- synth_ai/cli/commands/__init__.py +18 -0
- synth_ai/cli/commands/artifacts/__init__.py +13 -0
- synth_ai/cli/commands/artifacts/client.py +119 -0
- synth_ai/cli/commands/artifacts/config.py +57 -0
- synth_ai/cli/commands/artifacts/core.py +24 -0
- synth_ai/cli/commands/artifacts/download.py +188 -0
- synth_ai/cli/commands/artifacts/export.py +186 -0
- synth_ai/cli/commands/artifacts/list.py +156 -0
- synth_ai/cli/commands/artifacts/parsing.py +250 -0
- synth_ai/cli/commands/artifacts/show.py +336 -0
- synth_ai/cli/commands/baseline/__init__.py +12 -0
- synth_ai/cli/commands/baseline/core.py +636 -0
- synth_ai/cli/commands/baseline/list.py +94 -0
- synth_ai/cli/commands/demo/__init__.py +3 -0
- synth_ai/cli/commands/demo/core.py +153 -0
- synth_ai/cli/commands/eval/__init__.py +19 -0
- synth_ai/cli/commands/eval/core.py +1113 -0
- synth_ai/cli/commands/eval/errors.py +81 -0
- synth_ai/cli/commands/eval/validation.py +133 -0
- synth_ai/cli/commands/filter/__init__.py +12 -0
- synth_ai/cli/commands/filter/core.py +424 -0
- synth_ai/cli/commands/filter/errors.py +55 -0
- synth_ai/cli/commands/filter/validation.py +77 -0
- synth_ai/cli/commands/help/__init__.py +185 -0
- synth_ai/cli/commands/help/core.py +72 -0
- synth_ai/cli/commands/scan/__init__.py +19 -0
- synth_ai/cli/commands/scan/cloudflare_scanner.py +403 -0
- synth_ai/cli/commands/scan/core.py +344 -0
- synth_ai/cli/commands/scan/health_checker.py +242 -0
- synth_ai/cli/commands/scan/local_scanner.py +278 -0
- synth_ai/cli/commands/scan/models.py +83 -0
- synth_ai/cli/commands/smoke/__init__.py +7 -0
- synth_ai/cli/commands/smoke/core.py +1438 -0
- synth_ai/cli/commands/status/__init__.py +66 -0
- synth_ai/cli/commands/status/client.py +192 -0
- synth_ai/cli/commands/status/config.py +92 -0
- synth_ai/cli/commands/status/errors.py +20 -0
- synth_ai/cli/commands/status/formatters.py +164 -0
- synth_ai/cli/commands/status/subcommands/__init__.py +9 -0
- synth_ai/cli/commands/status/subcommands/files.py +79 -0
- synth_ai/cli/commands/status/subcommands/jobs.py +334 -0
- synth_ai/cli/commands/status/subcommands/models.py +79 -0
- synth_ai/cli/commands/status/subcommands/pricing.py +23 -0
- synth_ai/cli/commands/status/subcommands/runs.py +81 -0
- synth_ai/cli/commands/status/subcommands/session.py +182 -0
- synth_ai/cli/commands/status/subcommands/summary.py +47 -0
- synth_ai/cli/commands/status/subcommands/usage.py +203 -0
- synth_ai/cli/commands/status/utils.py +114 -0
- synth_ai/cli/commands/train/__init__.py +53 -0
- synth_ai/cli/commands/train/core.py +22 -0
- synth_ai/cli/commands/train/errors.py +117 -0
- synth_ai/cli/commands/train/judge_schemas.py +201 -0
- synth_ai/cli/commands/train/judge_validation.py +305 -0
- synth_ai/cli/commands/train/prompt_learning_validation.py +633 -0
- synth_ai/cli/commands/train/validation.py +392 -0
- synth_ai/cli/demo_apps/__init__.py +10 -0
- synth_ai/cli/demo_apps/core/__init__.py +28 -0
- synth_ai/cli/demo_apps/core/cli.py +1735 -0
- synth_ai/cli/demo_apps/crafter/crafter_fft_4b.toml +55 -0
- synth_ai/cli/demo_apps/crafter/grpo_crafter_task_app.py +186 -0
- synth_ai/cli/demo_apps/crafter/rl_from_base_qwen4b.toml +74 -0
- synth_ai/cli/demo_apps/demo_registry.py +176 -0
- synth_ai/cli/demo_apps/demo_task_apps/core.py +440 -0
- synth_ai/cli/demo_apps/demo_task_apps/crafter/__init__.py +1 -0
- synth_ai/cli/demo_apps/demo_task_apps/crafter/grpo_crafter_task_app.py +185 -0
- synth_ai/cli/demo_apps/demo_task_apps/math/modal_task_app.py +742 -0
- synth_ai/cli/demo_apps/demo_task_apps/math/task_app_entry.py +39 -0
- synth_ai/cli/demo_apps/math/__init__.py +1 -0
- synth_ai/cli/demo_apps/math/_common.py +16 -0
- synth_ai/cli/demo_apps/math/app.py +38 -0
- synth_ai/cli/demo_apps/math/config.toml +76 -0
- synth_ai/cli/demo_apps/math/deploy_modal.py +54 -0
- synth_ai/cli/demo_apps/math/modal_task_app.py +702 -0
- synth_ai/cli/demo_apps/math/task_app_entry.py +53 -0
- synth_ai/cli/demo_apps/mipro/main.py +271 -0
- synth_ai/cli/demo_apps/mipro/task_app.py +933 -0
- synth_ai/cli/demo_apps/mipro/train_cfg.toml +92 -0
- synth_ai/cli/demos/__init__.py +12 -0
- synth_ai/cli/demos/demo.py +32 -0
- synth_ai/cli/demos/rl_demo.py +254 -0
- synth_ai/cli/deploy.py +216 -0
- synth_ai/cli/infra/__init__.py +14 -0
- synth_ai/cli/infra/balance.py +216 -0
- synth_ai/cli/infra/mcp.py +35 -0
- synth_ai/cli/infra/modal_app.py +36 -0
- synth_ai/cli/infra/setup.py +69 -0
- synth_ai/cli/infra/status.py +16 -0
- synth_ai/cli/infra/turso.py +77 -0
- synth_ai/cli/lib/__init__.py +10 -0
- synth_ai/cli/lib/agents.py +76 -0
- synth_ai/cli/lib/apps/modal_app.py +101 -0
- synth_ai/cli/lib/apps/task_app.py +643 -0
- synth_ai/cli/lib/bin.py +39 -0
- synth_ai/cli/lib/env.py +375 -0
- synth_ai/cli/lib/errors.py +85 -0
- synth_ai/cli/lib/modal.py +315 -0
- synth_ai/cli/lib/plotting.py +126 -0
- synth_ai/cli/lib/prompt_args.py +39 -0
- synth_ai/cli/lib/prompts.py +284 -0
- synth_ai/cli/lib/sqld.py +122 -0
- synth_ai/cli/lib/task_app_discovery.py +884 -0
- synth_ai/cli/lib/task_app_env.py +295 -0
- synth_ai/cli/lib/train_cfgs.py +300 -0
- synth_ai/cli/lib/tunnel_records.py +207 -0
- synth_ai/cli/local/__init__.py +14 -0
- synth_ai/cli/local/experiment_queue/__init__.py +72 -0
- synth_ai/cli/local/experiment_queue/api_schemas.py +221 -0
- synth_ai/cli/local/experiment_queue/celery_app.py +208 -0
- synth_ai/cli/local/experiment_queue/config.py +128 -0
- synth_ai/cli/local/experiment_queue/config_utils.py +272 -0
- synth_ai/cli/local/experiment_queue/database.py +175 -0
- synth_ai/cli/local/experiment_queue/dispatcher.py +119 -0
- synth_ai/cli/local/experiment_queue/models.py +231 -0
- synth_ai/cli/local/experiment_queue/progress_info.py +160 -0
- synth_ai/cli/local/experiment_queue/results.py +373 -0
- synth_ai/cli/local/experiment_queue/schemas.py +131 -0
- synth_ai/cli/local/experiment_queue/service.py +344 -0
- synth_ai/cli/local/experiment_queue/status.py +372 -0
- synth_ai/cli/local/experiment_queue/status_tracker.py +360 -0
- synth_ai/cli/local/experiment_queue/tasks.py +1984 -0
- synth_ai/cli/local/experiment_queue/trace_storage.py +65 -0
- synth_ai/cli/local/experiment_queue/validation.py +157 -0
- synth_ai/cli/local/session/__init__.py +92 -0
- synth_ai/cli/local/session/client.py +383 -0
- synth_ai/cli/local/session/constants.py +63 -0
- synth_ai/cli/local/session/exceptions.py +105 -0
- synth_ai/cli/local/session/manager.py +139 -0
- synth_ai/cli/local/session/models.py +89 -0
- synth_ai/cli/local/session/query.py +110 -0
- synth_ai/cli/root.py +30 -6
- synth_ai/cli/task_apps/__init__.py +26 -0
- synth_ai/cli/task_apps/commands.py +3153 -0
- synth_ai/cli/task_apps/deploy.py +7 -0
- synth_ai/cli/task_apps/list.py +26 -0
- synth_ai/cli/task_apps/main.py +36 -0
- synth_ai/cli/task_apps/modal_serve.py +11 -0
- synth_ai/cli/task_apps/serve.py +11 -0
- synth_ai/cli/training/__init__.py +8 -0
- synth_ai/cli/training/train.py +5 -0
- synth_ai/cli/training/train_cfg.py +34 -0
- synth_ai/cli/training/watch.py +506 -0
- synth_ai/cli/turso.py +34 -55
- synth_ai/cli/usage.py +159 -0
- synth_ai/cli/utils/__init__.py +8 -0
- synth_ai/cli/utils/experiments.py +235 -0
- synth_ai/cli/utils/queue.py +504 -0
- synth_ai/cli/utils/recent.py +133 -0
- synth_ai/cli/utils/traces.py +164 -0
- synth_ai/contracts/__init__.py +67 -0
- synth_ai/core/__init__.py +100 -0
- synth_ai/core/_utils/__init__.py +54 -0
- synth_ai/core/_utils/base_url.py +10 -0
- synth_ai/core/_utils/http.py +10 -0
- synth_ai/core/_utils/prompts.py +14 -0
- synth_ai/core/_utils/task_app_state.py +12 -0
- synth_ai/core/_utils/user_config.py +10 -0
- synth_ai/core/apps/common.py +116 -0
- synth_ai/core/auth.py +95 -0
- synth_ai/core/cfgs.py +240 -0
- synth_ai/core/config/__init__.py +16 -0
- synth_ai/core/config/base.py +168 -0
- synth_ai/core/config/resolver.py +89 -0
- synth_ai/core/env.py +220 -0
- synth_ai/core/errors.py +126 -0
- synth_ai/core/http.py +230 -0
- synth_ai/core/integrations/__init__.py +11 -0
- synth_ai/core/integrations/cloudflare.py +1710 -0
- synth_ai/core/integrations/mcp/__init__.py +6 -0
- synth_ai/core/integrations/mcp/__main__.py +8 -0
- synth_ai/core/integrations/mcp/claude.py +36 -0
- synth_ai/core/integrations/mcp/main.py +254 -0
- synth_ai/core/integrations/mcp/setup.py +100 -0
- synth_ai/core/integrations/modal.py +277 -0
- synth_ai/core/json.py +72 -0
- synth_ai/core/log_filter.py +99 -0
- synth_ai/core/logging.py +82 -0
- synth_ai/core/paths.py +107 -0
- synth_ai/core/pricing.py +109 -0
- synth_ai/core/process.py +233 -0
- synth_ai/core/ssl.py +25 -0
- synth_ai/core/storage/__init__.py +71 -0
- synth_ai/core/task_app_state.py +318 -0
- synth_ai/core/telemetry.py +282 -0
- synth_ai/core/tracing_v3/__init__.py +99 -0
- synth_ai/core/tracing_v3/abstractions.py +302 -0
- synth_ai/core/tracing_v3/config.py +229 -0
- synth_ai/core/tracing_v3/constants.py +21 -0
- synth_ai/core/tracing_v3/db_config.py +182 -0
- synth_ai/core/tracing_v3/decorators.py +401 -0
- synth_ai/core/tracing_v3/llm_call_record_helpers.py +437 -0
- synth_ai/core/tracing_v3/migration_helper.py +119 -0
- synth_ai/core/tracing_v3/session_tracer.py +542 -0
- synth_ai/core/tracing_v3/storage/base.py +211 -0
- synth_ai/core/tracing_v3/storage/config.py +109 -0
- synth_ai/core/tracing_v3/storage/factory.py +39 -0
- synth_ai/core/tracing_v3/trace_utils.py +326 -0
- synth_ai/core/tracing_v3/turso/daemon.py +278 -0
- synth_ai/core/tracing_v3/turso/models.py +470 -0
- synth_ai/core/tracing_v3/turso/native_manager.py +1385 -0
- synth_ai/core/tracing_v3/utils.py +108 -0
- synth_ai/core/urls.py +18 -0
- synth_ai/core/user_config.py +137 -0
- synth_ai/core/uvicorn.py +222 -0
- synth_ai/data/__init__.py +110 -0
- synth_ai/data/enums.py +141 -0
- synth_ai/data/rewards.py +152 -0
- synth_ai/data/specs.py +36 -0
- synth_ai/data/traces.py +35 -0
- synth_ai/products/__init__.py +6 -0
- synth_ai/products/graph_evolve/__init__.py +46 -0
- synth_ai/products/graph_evolve/client.py +226 -0
- synth_ai/products/graph_evolve/config.py +591 -0
- synth_ai/products/graph_evolve/converters/__init__.py +42 -0
- synth_ai/products/graph_evolve/converters/openai_sft.py +484 -0
- synth_ai/products/graph_evolve/examples/hotpotqa/config.toml +109 -0
- synth_ai/products/graph_evolve/run.py +222 -0
- synth_ai/sdk/__init__.py +119 -0
- synth_ai/sdk/api/__init__.py +1 -0
- synth_ai/sdk/api/models/supported.py +514 -0
- synth_ai/sdk/api/research_agent/__init__.py +86 -0
- synth_ai/sdk/api/research_agent/cli.py +428 -0
- synth_ai/sdk/api/research_agent/config.py +357 -0
- synth_ai/sdk/api/research_agent/job.py +717 -0
- synth_ai/sdk/api/train/__init__.py +85 -0
- synth_ai/sdk/api/train/builders.py +895 -0
- synth_ai/sdk/api/train/cli.py +2188 -0
- synth_ai/sdk/api/train/config_finder.py +267 -0
- synth_ai/sdk/api/train/configs/__init__.py +65 -0
- synth_ai/sdk/api/train/configs/prompt_learning.py +1706 -0
- synth_ai/sdk/api/train/configs/rl.py +188 -0
- synth_ai/sdk/api/train/configs/sft.py +99 -0
- synth_ai/sdk/api/train/configs/shared.py +81 -0
- synth_ai/sdk/api/train/context_learning.py +312 -0
- synth_ai/sdk/api/train/env_resolver.py +418 -0
- synth_ai/sdk/api/train/graph_validators.py +216 -0
- synth_ai/sdk/api/train/graphgen.py +984 -0
- synth_ai/sdk/api/train/graphgen_models.py +823 -0
- synth_ai/sdk/api/train/graphgen_validators.py +109 -0
- synth_ai/sdk/api/train/pollers.py +124 -0
- synth_ai/sdk/api/train/progress/__init__.py +97 -0
- synth_ai/sdk/api/train/progress/dataclasses.py +569 -0
- synth_ai/sdk/api/train/progress/events.py +326 -0
- synth_ai/sdk/api/train/progress/results.py +428 -0
- synth_ai/sdk/api/train/progress/tracker.py +641 -0
- synth_ai/sdk/api/train/prompt_learning.py +470 -0
- synth_ai/sdk/api/train/rl.py +442 -0
- synth_ai/sdk/api/train/sft.py +396 -0
- synth_ai/sdk/api/train/summary.py +522 -0
- synth_ai/sdk/api/train/supported_algos.py +147 -0
- synth_ai/sdk/api/train/task_app.py +331 -0
- synth_ai/sdk/api/train/utils.py +279 -0
- synth_ai/sdk/api/train/validators.py +2424 -0
- synth_ai/sdk/baseline/__init__.py +25 -0
- synth_ai/sdk/baseline/config.py +209 -0
- synth_ai/sdk/baseline/discovery.py +216 -0
- synth_ai/sdk/baseline/execution.py +154 -0
- synth_ai/sdk/graphs/__init__.py +15 -0
- synth_ai/sdk/graphs/completions.py +570 -0
- synth_ai/sdk/inference/__init__.py +6 -0
- synth_ai/sdk/inference/client.py +128 -0
- synth_ai/sdk/jobs/__init__.py +16 -0
- synth_ai/sdk/jobs/client.py +371 -0
- synth_ai/sdk/judging/__init__.py +15 -0
- synth_ai/sdk/judging/base.py +24 -0
- synth_ai/sdk/judging/client.py +191 -0
- synth_ai/sdk/judging/schemas.py +222 -0
- synth_ai/sdk/learning/__init__.py +69 -0
- synth_ai/sdk/learning/client.py +240 -0
- synth_ai/sdk/learning/ft_client.py +7 -0
- synth_ai/sdk/learning/health.py +49 -0
- synth_ai/sdk/learning/jobs.py +202 -0
- synth_ai/sdk/learning/prompt_extraction.py +334 -0
- synth_ai/sdk/learning/prompt_learning_client.py +455 -0
- synth_ai/sdk/learning/prompt_learning_types.py +185 -0
- synth_ai/sdk/learning/rl/client.py +268 -0
- synth_ai/sdk/learning/rl/contracts.py +27 -0
- synth_ai/sdk/learning/rl/env_keys.py +166 -0
- synth_ai/sdk/learning/rl/secrets.py +13 -0
- synth_ai/sdk/learning/sft/client.py +95 -0
- synth_ai/sdk/learning/sft/config.py +270 -0
- synth_ai/sdk/learning/sft/data.py +698 -0
- synth_ai/sdk/learning/validators.py +52 -0
- synth_ai/sdk/research_agent/__init__.py +34 -0
- synth_ai/sdk/research_agent/container_builder.py +328 -0
- synth_ai/sdk/research_agent/container_spec.py +198 -0
- synth_ai/sdk/research_agent/defaults.py +34 -0
- synth_ai/sdk/research_agent/results_collector.py +69 -0
- synth_ai/sdk/specs/__init__.py +46 -0
- synth_ai/sdk/specs/dataclasses.py +149 -0
- synth_ai/sdk/specs/loader.py +144 -0
- synth_ai/sdk/specs/serializer.py +199 -0
- synth_ai/sdk/specs/validation.py +250 -0
- synth_ai/sdk/streaming/__init__.py +35 -0
- synth_ai/sdk/streaming/config.py +94 -0
- synth_ai/sdk/streaming/handlers.py +1997 -0
- synth_ai/sdk/streaming/streamer.py +704 -0
- synth_ai/sdk/streaming/types.py +112 -0
- synth_ai/sdk/task/__init__.py +151 -0
- synth_ai/sdk/task/apps/__init__.py +133 -0
- synth_ai/sdk/task/config.py +261 -0
- synth_ai/sdk/task/contracts.py +298 -0
- synth_ai/sdk/task/datasets.py +108 -0
- synth_ai/sdk/task/in_process.py +1190 -0
- synth_ai/sdk/task/in_process_runner.py +309 -0
- synth_ai/sdk/task/inference_api.py +299 -0
- synth_ai/sdk/task/proxy.py +287 -0
- synth_ai/sdk/task/rubrics/__init__.py +55 -0
- synth_ai/sdk/task/rubrics/loaders.py +156 -0
- synth_ai/sdk/task/rubrics.py +219 -0
- synth_ai/sdk/task/server.py +580 -0
- synth_ai/sdk/task/trace_correlation_helpers.py +506 -0
- synth_ai/sdk/task/tracing_utils.py +95 -0
- synth_ai/sdk/task/validators.py +456 -0
- synth_ai/sdk/tracing/__init__.py +39 -0
- synth_ai/sdk/training/__init__.py +102 -0
- synth_ai/sdk/usage/__init__.py +37 -0
- synth_ai/sdk/usage/client.py +171 -0
- synth_ai/sdk/usage/models.py +261 -0
- synth_ai/utils/__init__.py +213 -0
- synth_ai-0.4.1.dist-info/METADATA +195 -0
- synth_ai-0.4.1.dist-info/RECORD +379 -0
- synth_ai-0.4.1.dist-info/top_level.txt +1 -0
- examples/__init__.py +0 -16
- examples/analyze_semantic_words.sh +0 -17
- examples/crafter_debug_render.py +0 -186
- examples/dev/qwen3_32b_qlora_4xh100.toml +0 -40
- examples/multi_step/configs/README_verilog_rl.md +0 -77
- examples/multi_step/configs/VERILOG_REWARDS.md +0 -90
- examples/multi_step/configs/VERILOG_RL_CHECKLIST.md +0 -183
- examples/multi_step/configs/crafter_eval_synth_qwen4b.toml +0 -35
- examples/multi_step/configs/crafter_eval_text_only_groq_qwen32b.toml +0 -36
- examples/multi_step/configs/crafter_rl_outcome.toml +0 -74
- examples/multi_step/configs/crafter_rl_stepwise_hosted_judge.toml +0 -187
- examples/multi_step/configs/crafter_rl_stepwise_shaped.toml +0 -83
- examples/multi_step/configs/crafter_rl_stepwise_simple.toml +0 -78
- examples/multi_step/configs/crafter_synth_backend.md +0 -40
- examples/multi_step/configs/verilog_eval_groq_qwen32b.toml +0 -31
- examples/multi_step/configs/verilog_eval_synth_qwen8b.toml +0 -33
- examples/multi_step/configs/verilog_rl_lora.toml +0 -190
- examples/multi_step/crafter_rl_lora.md +0 -70
- examples/multi_step/judges/crafter_backend_judge.py +0 -220
- examples/multi_step/judges/verilog_backend_judge.py +0 -234
- examples/multi_step/readme.md +0 -48
- examples/multi_step/sse_metrics_streaming_notes.md +0 -357
- examples/multi_step/task_app_config_notes.md +0 -494
- examples/multi_step/verilog_rl_lora.md +0 -218
- examples/qwen_coder/README.md +0 -102
- examples/qwen_coder/_shared.py +0 -113
- examples/qwen_coder/configs/coder_lora_30b.toml +0 -61
- examples/qwen_coder/configs/coder_lora_4b.toml +0 -57
- examples/qwen_coder/configs/coder_lora_small.toml +0 -58
- examples/qwen_coder/generate_dataset.py +0 -98
- examples/qwen_coder/infer_ft_smoke.py +0 -65
- examples/qwen_coder/infer_prod_proxy.py +0 -73
- examples/qwen_coder/infer_via_synth.py +0 -87
- examples/qwen_coder/scripts/infer_coder.sh +0 -19
- examples/qwen_coder/scripts/train_coder_30b.sh +0 -22
- examples/qwen_coder/sft_full_17b.py +0 -103
- examples/qwen_coder/sft_lora_30b.py +0 -110
- examples/qwen_coder/subset_jsonl.py +0 -39
- examples/qwen_coder/todos.md +0 -38
- examples/qwen_coder/validate_jsonl.py +0 -60
- examples/rl/README.md +0 -169
- examples/rl/download_dataset.py +0 -80
- examples/run_crafter_demo.sh +0 -10
- examples/sft/README.md +0 -139
- examples/sft/configs/crafter_fft_qwen0p6b.toml +0 -44
- examples/sft/configs/crafter_lora_qwen0p6b.toml +0 -45
- examples/sft/evaluate.py +0 -119
- examples/sft/export_dataset.py +0 -117
- examples/sft/generate_traces.py +0 -164
- examples/swe/__init__.py +0 -12
- examples/swe/task_app/README.md +0 -105
- examples/swe/task_app/__init__.py +0 -2
- examples/swe/task_app/grpo_swe_mini.py +0 -601
- examples/swe/task_app/grpo_swe_mini_task_app.py +0 -136
- examples/swe/task_app/hosted/README.md +0 -173
- examples/swe/task_app/hosted/__init__.py +0 -5
- examples/swe/task_app/hosted/branching.py +0 -143
- examples/swe/task_app/hosted/environment_routes.py +0 -1289
- examples/swe/task_app/hosted/envs/__init__.py +0 -1
- examples/swe/task_app/hosted/envs/crafter/__init__.py +0 -6
- examples/swe/task_app/hosted/envs/crafter/app.py +0 -1
- examples/swe/task_app/hosted/envs/crafter/environment.py +0 -522
- examples/swe/task_app/hosted/envs/crafter/policy.py +0 -478
- examples/swe/task_app/hosted/envs/crafter/react_agent.py +0 -108
- examples/swe/task_app/hosted/envs/crafter/shared.py +0 -305
- examples/swe/task_app/hosted/envs/crafter/tools.py +0 -47
- examples/swe/task_app/hosted/envs/mini_swe/__init__.py +0 -8
- examples/swe/task_app/hosted/envs/mini_swe/environment.py +0 -1164
- examples/swe/task_app/hosted/envs/mini_swe/policy.py +0 -355
- examples/swe/task_app/hosted/envs/mini_swe/shared.py +0 -83
- examples/swe/task_app/hosted/envs/mini_swe/tools.py +0 -96
- examples/swe/task_app/hosted/hosted_app.py +0 -204
- examples/swe/task_app/hosted/inference/__init__.py +0 -5
- examples/swe/task_app/hosted/inference/openai_client.py +0 -618
- examples/swe/task_app/hosted/main.py +0 -100
- examples/swe/task_app/hosted/policy_routes.py +0 -1079
- examples/swe/task_app/hosted/registry.py +0 -195
- examples/swe/task_app/hosted/rollout.py +0 -1911
- examples/swe/task_app/hosted/storage/__init__.py +0 -5
- examples/swe/task_app/hosted/storage/volume.py +0 -211
- examples/swe/task_app/hosted/test_agents.py +0 -161
- examples/swe/task_app/hosted/test_service.py +0 -136
- examples/swe/task_app/hosted/utils.py +0 -62
- examples/task_apps/IMAGE_ONLY_EVAL_QUICKSTART.md +0 -258
- examples/task_apps/TESTING.md +0 -275
- examples/task_apps/crafter/CREATE_SFT_DATASET.md +0 -273
- examples/task_apps/crafter/EVAL_IMAGE_ONLY_RESULTS.md +0 -152
- examples/task_apps/crafter/FILTER_COMMAND_STATUS.md +0 -174
- examples/task_apps/crafter/FILTER_COMMAND_SUCCESS.md +0 -268
- examples/task_apps/crafter/QUERY_EXAMPLES.md +0 -203
- examples/task_apps/crafter/README_IMAGE_ONLY_EVAL.md +0 -316
- examples/task_apps/crafter/__init__.py +0 -0
- examples/task_apps/crafter/eval_image_only_gpt4o.toml +0 -28
- examples/task_apps/crafter/eval_text_only_groq_llama.toml +0 -36
- examples/task_apps/crafter/filter_sft_dataset.toml +0 -16
- examples/task_apps/crafter/task_app/README.md +0 -42
- examples/task_apps/crafter/task_app/__init__.py +0 -5
- examples/task_apps/crafter/task_app/grpo_crafter.py +0 -973
- examples/task_apps/crafter/task_app/grpo_crafter_task_app.py +0 -146
- examples/task_apps/crafter/task_app/synth_envs_hosted/README.md +0 -173
- examples/task_apps/crafter/task_app/synth_envs_hosted/__init__.py +0 -5
- examples/task_apps/crafter/task_app/synth_envs_hosted/branching.py +0 -143
- examples/task_apps/crafter/task_app/synth_envs_hosted/environment_routes.py +0 -1226
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/__init__.py +0 -1
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/__init__.py +0 -6
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/app.py +0 -1
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/environment.py +0 -532
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/policy.py +0 -547
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/react_agent.py +0 -123
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/shared.py +0 -305
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/tools.py +0 -47
- examples/task_apps/crafter/task_app/synth_envs_hosted/hosted_app.py +0 -204
- examples/task_apps/crafter/task_app/synth_envs_hosted/inference/__init__.py +0 -5
- examples/task_apps/crafter/task_app/synth_envs_hosted/inference/openai_client.py +0 -704
- examples/task_apps/crafter/task_app/synth_envs_hosted/main.py +0 -100
- examples/task_apps/crafter/task_app/synth_envs_hosted/policy_routes.py +0 -1152
- examples/task_apps/crafter/task_app/synth_envs_hosted/registry.py +0 -195
- examples/task_apps/crafter/task_app/synth_envs_hosted/rollout.py +0 -2160
- examples/task_apps/crafter/task_app/synth_envs_hosted/storage/__init__.py +0 -5
- examples/task_apps/crafter/task_app/synth_envs_hosted/storage/volume.py +0 -211
- examples/task_apps/crafter/task_app/synth_envs_hosted/test_agents.py +0 -161
- examples/task_apps/crafter/task_app/synth_envs_hosted/test_service.py +0 -136
- examples/task_apps/crafter/task_app/synth_envs_hosted/utils.py +0 -218
- examples/task_apps/dev/pokemon_emerald/__init__.py +0 -2
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/README.md +0 -811
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/__init__.py +0 -120
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/action.py +0 -160
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/memory.py +0 -155
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/perception.py +0 -69
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/planning.py +0 -96
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/simple.py +0 -1502
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/system_prompt.py +0 -4
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/grab_map.py +0 -68
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/manual.py +0 -216
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/__init__.py +0 -35
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/emerald_utils.py +0 -631
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/emulator.py +0 -1544
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/enums.py +0 -1428
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/memory_reader.py +0 -4848
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/types.py +0 -41
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/utils.py +0 -298
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pyproject.toml +0 -95
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/run.py +0 -204
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/__init__.py +0 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/app.py +0 -2152
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/client.py +0 -429
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/frame_server.py +0 -155
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/README.md +0 -78
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/__init__.py +0 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/run_tests.py +0 -122
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_agent_direct.py +0 -76
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_agent_prompts.py +0 -413
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_battle_state_formatting.py +0 -204
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_dialogue_detection.py +0 -133
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_dialogue_detection_comprehensive.py +0 -229
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_direct_agent_emulator.py +0 -300
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_fps_adjustment_pytest.py +0 -205
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_house_to_outside_direct.py +0 -200
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_house_to_outside_transition.py +0 -284
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_map_ground_truth_comparison.py +0 -468
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_memory_map.py +0 -575
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_server_map_validation.py +0 -311
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_torchic_state.py +0 -259
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/__init__.py +0 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/anticheat.py +0 -372
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/checkpoint.py +0 -296
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/error_handler.py +0 -275
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/get_local_ip.py +0 -22
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/helpers.py +0 -44
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/llm_logger.py +0 -514
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_formatter.py +0 -415
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_stitcher.py +0 -1763
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_stitcher_singleton.py +0 -33
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_trimmer.py +0 -106
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_visualizer.py +0 -334
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/ocr_dialogue.py +0 -1020
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/recording.py +0 -188
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/state_formatter.py +0 -1481
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/vlm.py +0 -862
- examples/task_apps/dev/pokemon_emerald/modal_app.py +0 -114
- examples/task_apps/dev/pokemon_emerald/task_app/README.md +0 -81
- examples/task_apps/dev/pokemon_emerald/task_app/__init__.py +0 -6
- examples/task_apps/dev/pokemon_emerald/task_app/pokemon_emerald.py +0 -685
- examples/task_apps/enron/__init__.py +0 -1
- examples/task_apps/enron/eval_groq_qwen32.toml +0 -16
- examples/task_apps/enron/filter_sft.toml +0 -5
- examples/task_apps/enron/task_app/README.md +0 -14
- examples/task_apps/enron/task_app/__init__.py +0 -1
- examples/task_apps/enron/task_app/grpo_enron.py +0 -906
- examples/task_apps/enron/task_app/grpo_enron_task_app.py +0 -146
- examples/task_apps/enron/tests/__init__.py +0 -4
- examples/task_apps/enron/tests/conftest.py +0 -115
- examples/task_apps/enron/tests/integration/__init__.py +0 -4
- examples/task_apps/enron/tests/integration/test_enron_eval.py +0 -179
- examples/task_apps/enron/tests/integration/test_enron_rollout.py +0 -135
- examples/task_apps/enron/tests/unit/__init__.py +0 -4
- examples/task_apps/enron/tests/unit/test_enron_environment.py +0 -126
- examples/task_apps/math/README.md +0 -22
- examples/task_apps/math/__init__.py +0 -0
- examples/task_apps/math/math_single_step.py +0 -1000
- examples/task_apps/math/math_task_app.py +0 -115
- examples/task_apps/pokemon_battle/__init__.py +0 -2
- examples/task_apps/pokemon_battle/modal_app.py +0 -104
- examples/task_apps/pokemon_battle/task_app/README.md +0 -68
- examples/task_apps/pokemon_battle/task_app/__init__.py +0 -6
- examples/task_apps/pokemon_battle/task_app/pokemon_showdown.py +0 -932
- examples/task_apps/pokemon_red/EVAL_IMAGE_ONLY_COMPLETE.md +0 -283
- examples/task_apps/pokemon_red/EVAL_IMAGE_ONLY_STATUS.md +0 -155
- examples/task_apps/pokemon_red/README.md +0 -357
- examples/task_apps/pokemon_red/README_IMAGE_ONLY_EVAL.md +0 -415
- examples/task_apps/pokemon_red/__init__.py +0 -3
- examples/task_apps/pokemon_red/eval_image_only_gpt4o.toml +0 -29
- examples/task_apps/pokemon_red/eval_pokemon_red_policy.py +0 -225
- examples/task_apps/pokemon_red/pallet_town_rl_config.toml +0 -75
- examples/task_apps/pokemon_red/task_app.py +0 -799
- examples/task_apps/pokemon_red/test_pallet_town_rewards.py +0 -193
- examples/task_apps/sokoban/README.md +0 -307
- examples/task_apps/sokoban/__init__.py +0 -3
- examples/task_apps/sokoban/eval_groq_qwen32.toml +0 -16
- examples/task_apps/sokoban/eval_openai_gpt5.toml +0 -16
- examples/task_apps/sokoban/filter_sft.toml +0 -5
- examples/task_apps/sokoban/task_app.py +0 -1058
- examples/task_apps/sokoban/tests/__init__.py +0 -4
- examples/task_apps/sokoban/tests/conftest.py +0 -113
- examples/task_apps/sokoban/tests/integration/__init__.py +0 -4
- examples/task_apps/sokoban/tests/integration/test_sokoban_eval.py +0 -57
- examples/task_apps/sokoban/tests/integration/test_sokoban_rollout.py +0 -198
- examples/task_apps/sokoban/tests/unit/__init__.py +0 -4
- examples/task_apps/sokoban/tests/unit/test_sokoban_environment.py +0 -114
- examples/task_apps/verilog/__init__.py +0 -1
- examples/task_apps/verilog/eval_groq_qwen32b.toml +0 -24
- examples/task_apps/verilog/filter_sft.toml +0 -5
- examples/task_apps/verilog/task_app/README.md +0 -12
- examples/task_apps/verilog/task_app/__init__.py +0 -1
- examples/task_apps/verilog/task_app/grpo_verilog.py +0 -1166
- examples/task_apps/verilog/task_app/grpo_verilog_task_app.py +0 -145
- examples/task_apps/verilog/tests/__init__.py +0 -4
- examples/task_apps/verilog/tests/conftest.py +0 -115
- examples/task_apps/verilog/tests/integration/__init__.py +0 -4
- examples/task_apps/verilog/tests/integration/test_verilog_eval.py +0 -181
- examples/task_apps/verilog/tests/integration/test_verilog_rollout.py +0 -55
- examples/task_apps/verilog/tests/unit/__init__.py +0 -4
- examples/task_apps/verilog/tests/unit/test_verilog_scoring.py +0 -118
- examples/vlm/PROPOSAL.md +0 -53
- examples/vlm/README.md +0 -68
- examples/vlm/configs/crafter_vlm_gpt4o.toml +0 -44
- examples/vlm/crafter_image_only_agent.py +0 -207
- examples/vlm/crafter_openai_vlm_agent.py +0 -277
- examples/vlm/filter_image_rows.py +0 -63
- examples/vlm/run_crafter_vlm_benchmark.py +0 -316
- examples/warming_up_to_rl/analyze_trace_db.py +0 -422
- examples/warming_up_to_rl/configs/crafter_fft.toml +0 -48
- examples/warming_up_to_rl/configs/crafter_fft_4b.toml +0 -54
- examples/warming_up_to_rl/configs/eval_fft_qwen4b.toml +0 -20
- examples/warming_up_to_rl/configs/eval_groq_qwen32b.toml +0 -13
- examples/warming_up_to_rl/configs/eval_modal_qwen4b.toml +0 -23
- examples/warming_up_to_rl/configs/eval_stepwise_complex.toml +0 -35
- examples/warming_up_to_rl/configs/eval_stepwise_consistent.toml +0 -26
- examples/warming_up_to_rl/configs/eval_stepwise_per_achievement.toml +0 -36
- examples/warming_up_to_rl/configs/eval_stepwise_simple.toml +0 -32
- examples/warming_up_to_rl/configs/rl_from_base_qwen4b.toml +0 -83
- examples/warming_up_to_rl/configs/rl_from_ft.toml +0 -56
- examples/warming_up_to_rl/export_trace_sft.py +0 -723
- examples/warming_up_to_rl/groq_test.py +0 -97
- examples/warming_up_to_rl/manage_secrets.py +0 -131
- examples/warming_up_to_rl/old/event_rewards.md +0 -234
- examples/warming_up_to_rl/old/notes.md +0 -73
- examples/warming_up_to_rl/readme.md +0 -179
- examples/warming_up_to_rl/run_eval.py +0 -736
- examples/warming_up_to_rl/run_fft_and_save.py +0 -380
- examples/warming_up_to_rl/run_local_rollout.py +0 -239
- examples/warming_up_to_rl/run_local_rollout_modal.py +0 -248
- examples/warming_up_to_rl/run_local_rollout_parallel.py +0 -405
- examples/warming_up_to_rl/run_local_rollout_traced.py +0 -477
- examples/warming_up_to_rl/run_rl_and_save.py +0 -124
- examples/warming_up_to_rl/run_rollout_remote.py +0 -156
- examples/workflows/__init__.py +0 -0
- examples/workflows/math_rl/__init__.py +0 -0
- examples/workflows/math_rl/configs/eval_base_qwen.toml +0 -15
- examples/workflows/math_rl/configs/eval_rl_qwen.toml +0 -11
- examples/workflows/math_rl/configs/rl_from_base_qwen.toml +0 -35
- examples/workflows/math_rl/configs/rl_from_base_qwen17.toml +0 -74
- examples/workflows/math_rl/configs/rl_from_ft_qwen.toml +0 -35
- examples/workflows/math_rl/download_dataset.py +0 -80
- examples/workflows/math_rl/run_eval.py +0 -436
- examples/workflows/math_rl/run_rl_and_save.py +0 -111
- synth_ai/api/models/supported.py +0 -377
- synth_ai/api/train/__init__.py +0 -5
- synth_ai/api/train/builders.py +0 -351
- synth_ai/api/train/cli.py +0 -635
- synth_ai/api/train/config_finder.py +0 -228
- synth_ai/api/train/configs/__init__.py +0 -44
- synth_ai/api/train/configs/rl.py +0 -134
- synth_ai/api/train/configs/sft.py +0 -95
- synth_ai/api/train/configs/shared.py +0 -24
- synth_ai/api/train/env_resolver.py +0 -349
- synth_ai/api/train/pollers.py +0 -75
- synth_ai/api/train/supported_algos.py +0 -147
- synth_ai/api/train/task_app.py +0 -195
- synth_ai/api/train/utils.py +0 -225
- synth_ai/cli/_modal_wrapper.py +0 -29
- synth_ai/cli/_storage.py +0 -20
- synth_ai/cli/_typer_patch.py +0 -49
- synth_ai/cli/_validate_task_app.py +0 -11
- synth_ai/cli/balance.py +0 -216
- synth_ai/cli/calc.py +0 -84
- synth_ai/cli/demo.py +0 -165
- synth_ai/cli/legacy_root_backup.py +0 -468
- synth_ai/cli/man.py +0 -106
- synth_ai/cli/recent.py +0 -132
- synth_ai/cli/rl_demo.py +0 -254
- synth_ai/cli/status.py +0 -134
- synth_ai/cli/task_apps.py +0 -4523
- synth_ai/cli/traces.py +0 -164
- synth_ai/cli/tui.py +0 -57
- synth_ai/cli/watch.py +0 -506
- synth_ai/compound/cais.py +0 -0
- synth_ai/config/base_url.py +0 -107
- synth_ai/core/experiment.py +0 -13
- synth_ai/core/system.py +0 -15
- synth_ai/demo_registry.py +0 -295
- synth_ai/demos/core/__init__.py +0 -1
- synth_ai/demos/core/cli.py +0 -1718
- synth_ai/demos/demo_task_apps/core.py +0 -440
- synth_ai/demos/demo_task_apps/crafter/grpo_crafter_task_app.py +0 -184
- synth_ai/demos/demo_task_apps/math/deploy_task_app.sh +0 -22
- synth_ai/demos/demo_task_apps/math/modal_task_app.py +0 -739
- synth_ai/demos/demo_task_apps/math/task_app_entry.py +0 -37
- synth_ai/environments/__init__.py +0 -31
- synth_ai/environments/environment/__init__.py +0 -1
- synth_ai/environments/environment/artifacts/__init__.py +0 -1
- synth_ai/environments/environment/artifacts/base.py +0 -52
- synth_ai/environments/environment/core.py +0 -67
- synth_ai/environments/environment/db/__init__.py +0 -1
- synth_ai/environments/environment/db/sqlite.py +0 -45
- synth_ai/environments/environment/registry.py +0 -233
- synth_ai/environments/environment/resources/sqlite.py +0 -45
- synth_ai/environments/environment/results.py +0 -1
- synth_ai/environments/environment/rewards/__init__.py +0 -1
- synth_ai/environments/environment/rewards/core.py +0 -29
- synth_ai/environments/environment/shared_engine.py +0 -26
- synth_ai/environments/environment/tools/__init__.py +0 -200
- synth_ai/environments/examples/__init__.py +0 -1
- synth_ai/environments/examples/bandit/__init__.py +0 -33
- synth_ai/environments/examples/bandit/engine.py +0 -302
- synth_ai/environments/examples/bandit/environment.py +0 -194
- synth_ai/environments/examples/bandit/taskset.py +0 -200
- synth_ai/environments/examples/crafter_classic/__init__.py +0 -8
- synth_ai/environments/examples/crafter_classic/agent_demos/analyze_semantic_words_markdown.py +0 -250
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_comprehensive_evaluation.py +0 -59
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_evaluation_browser.py +0 -152
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_evaluation_config.toml +0 -24
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_evaluation_framework.py +0 -1194
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/crafter_synth_config.toml +0 -56
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/filter_config_modal.toml +0 -32
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/filter_traces_sft_turso.py +0 -738
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/kick_off_ft_modal.py +0 -384
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_action_results.py +0 -53
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_agent_actions.py +0 -178
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_latest_run.py +0 -222
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_lm_traces.py +0 -183
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_no_rewards.py +0 -210
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/analyze_trace_issue.py +0 -206
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/check_db_schema.py +0 -49
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/check_latest_results.py +0 -64
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/debug_agent_responses.py +0 -88
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_modal_ft/old/quick_trace_check.py +0 -77
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/compare_experiments.py +0 -324
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/filter_traces_sft_turso.py +0 -580
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/kick_off_ft_oai.py +0 -362
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/multi_model_config.toml +0 -49
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/analyze_enhanced_hooks.py +0 -332
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/analyze_hook_events.py +0 -97
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/analyze_hook_results.py +0 -217
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/check_hook_storage.py +0 -87
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/check_seeds.py +0 -88
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/compare_seed_performance.py +0 -195
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/custom_eval_pipelines.py +0 -400
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/plot_hook_frequency.py +0 -195
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/old/seed_analysis_summary.py +0 -56
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_openai_ft/run_rollouts_for_models_and_compare_v3.py +0 -858
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_quick_evaluation.py +0 -52
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_react_agent.py +0 -874
- synth_ai/environments/examples/crafter_classic/agent_demos/crafter_trace_evaluation.py +0 -1412
- synth_ai/environments/examples/crafter_classic/agent_demos/example_v3_usage.py +0 -216
- synth_ai/environments/examples/crafter_classic/agent_demos/old/compare_traces.py +0 -296
- synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_comprehensive_evaluation.py +0 -58
- synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_env_serialization.py +0 -464
- synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_evaluation_browser.py +0 -152
- synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_quick_evaluation.py +0 -51
- synth_ai/environments/examples/crafter_classic/agent_demos/old/crafter_trace_evaluation.py +0 -1412
- synth_ai/environments/examples/crafter_classic/agent_demos/old/debug_player_loss.py +0 -112
- synth_ai/environments/examples/crafter_classic/agent_demos/old/diagnose_service.py +0 -203
- synth_ai/environments/examples/crafter_classic/agent_demos/old/diagnose_slowness.py +0 -305
- synth_ai/environments/examples/crafter_classic/agent_demos/old/eval_by_difficulty.py +0 -126
- synth_ai/environments/examples/crafter_classic/agent_demos/old/eval_example.py +0 -94
- synth_ai/environments/examples/crafter_classic/agent_demos/old/explore_saved_states.py +0 -142
- synth_ai/environments/examples/crafter_classic/agent_demos/old/filter_traces_sft.py +0 -26
- synth_ai/environments/examples/crafter_classic/agent_demos/old/filter_traces_sft_OLD.py +0 -984
- synth_ai/environments/examples/crafter_classic/agent_demos/old/generate_ft_data_gemini.py +0 -724
- synth_ai/environments/examples/crafter_classic/agent_demos/old/generate_ft_data_modal.py +0 -386
- synth_ai/environments/examples/crafter_classic/agent_demos/old/generate_ft_metadata.py +0 -205
- synth_ai/environments/examples/crafter_classic/agent_demos/old/kick_off_ft_gemini.py +0 -150
- synth_ai/environments/examples/crafter_classic/agent_demos/old/kick_off_ft_modal.py +0 -283
- synth_ai/environments/examples/crafter_classic/agent_demos/old/prepare_vertex_ft.py +0 -280
- synth_ai/environments/examples/crafter_classic/agent_demos/old/profile_env_slowness.py +0 -456
- synth_ai/environments/examples/crafter_classic/agent_demos/old/replicate_issue.py +0 -166
- synth_ai/environments/examples/crafter_classic/agent_demos/old/run_and_eval.py +0 -102
- synth_ai/environments/examples/crafter_classic/agent_demos/old/run_comparison.py +0 -128
- synth_ai/environments/examples/crafter_classic/agent_demos/old/run_qwen_rollouts.py +0 -655
- synth_ai/environments/examples/crafter_classic/agent_demos/old/trace_eval_OLD.py +0 -202
- synth_ai/environments/examples/crafter_classic/agent_demos/old/validate_openai_format.py +0 -166
- synth_ai/environments/examples/crafter_classic/config_logging.py +0 -111
- synth_ai/environments/examples/crafter_classic/debug_translation.py +0 -0
- synth_ai/environments/examples/crafter_classic/engine.py +0 -579
- synth_ai/environments/examples/crafter_classic/engine_deterministic_patch.py +0 -64
- synth_ai/environments/examples/crafter_classic/engine_helpers/action_map.py +0 -6
- synth_ai/environments/examples/crafter_classic/engine_helpers/serialization.py +0 -75
- synth_ai/environments/examples/crafter_classic/engine_serialization_patch_v3.py +0 -267
- synth_ai/environments/examples/crafter_classic/environment.py +0 -495
- synth_ai/environments/examples/crafter_classic/taskset.py +0 -233
- synth_ai/environments/examples/crafter_classic/trace_hooks_v3.py +0 -228
- synth_ai/environments/examples/crafter_classic/world_config_patch_simple.py +0 -299
- synth_ai/environments/examples/crafter_custom/__init__.py +0 -4
- synth_ai/environments/examples/crafter_custom/agent_demos/__init__.py +0 -1
- synth_ai/environments/examples/crafter_custom/agent_demos/trace_eval.py +0 -202
- synth_ai/environments/examples/crafter_custom/crafter/__init__.py +0 -7
- synth_ai/environments/examples/crafter_custom/crafter/config.py +0 -182
- synth_ai/environments/examples/crafter_custom/crafter/constants.py +0 -8
- synth_ai/environments/examples/crafter_custom/crafter/engine.py +0 -269
- synth_ai/environments/examples/crafter_custom/crafter/env.py +0 -262
- synth_ai/environments/examples/crafter_custom/crafter/objects.py +0 -417
- synth_ai/environments/examples/crafter_custom/crafter/recorder.py +0 -187
- synth_ai/environments/examples/crafter_custom/crafter/worldgen.py +0 -118
- synth_ai/environments/examples/crafter_custom/dataset_builder.py +0 -373
- synth_ai/environments/examples/crafter_custom/environment.py +0 -312
- synth_ai/environments/examples/crafter_custom/old/analyze_diamond_issue.py +0 -159
- synth_ai/environments/examples/crafter_custom/old/analyze_diamond_spawning.py +0 -158
- synth_ai/environments/examples/crafter_custom/old/compare_worlds.py +0 -71
- synth_ai/environments/examples/crafter_custom/old/dataset_stats.py +0 -105
- synth_ai/environments/examples/crafter_custom/old/diamond_spawning_summary.py +0 -119
- synth_ai/environments/examples/crafter_custom/old/example_dataset_usage.py +0 -52
- synth_ai/environments/examples/crafter_custom/run_dataset.py +0 -305
- synth_ai/environments/examples/enron/art_helpers/email_search_tools.py +0 -156
- synth_ai/environments/examples/enron/art_helpers/local_email_db.py +0 -281
- synth_ai/environments/examples/enron/art_helpers/types_enron.py +0 -25
- synth_ai/environments/examples/enron/engine.py +0 -300
- synth_ai/environments/examples/enron/environment.py +0 -234
- synth_ai/environments/examples/enron/taskset.py +0 -112
- synth_ai/environments/examples/enron/units/keyword_stats.py +0 -112
- synth_ai/environments/examples/minigrid/__init__.py +0 -48
- synth_ai/environments/examples/minigrid/agent_demos/minigrid_evaluation_framework.py +0 -1188
- synth_ai/environments/examples/minigrid/agent_demos/minigrid_quick_evaluation.py +0 -48
- synth_ai/environments/examples/minigrid/agent_demos/minigrid_react_agent.py +0 -562
- synth_ai/environments/examples/minigrid/agent_demos/minigrid_trace_evaluation.py +0 -221
- synth_ai/environments/examples/minigrid/engine.py +0 -589
- synth_ai/environments/examples/minigrid/environment.py +0 -274
- synth_ai/environments/examples/minigrid/environment_mapping.py +0 -242
- synth_ai/environments/examples/minigrid/puzzle_loader.py +0 -417
- synth_ai/environments/examples/minigrid/taskset.py +0 -583
- synth_ai/environments/examples/nethack/__init__.py +0 -7
- synth_ai/environments/examples/nethack/achievements.py +0 -337
- synth_ai/environments/examples/nethack/agent_demos/nethack_evaluation_framework.py +0 -981
- synth_ai/environments/examples/nethack/agent_demos/nethack_quick_evaluation.py +0 -74
- synth_ai/environments/examples/nethack/agent_demos/nethack_react_agent.py +0 -831
- synth_ai/environments/examples/nethack/engine.py +0 -739
- synth_ai/environments/examples/nethack/environment.py +0 -256
- synth_ai/environments/examples/nethack/helpers/__init__.py +0 -41
- synth_ai/environments/examples/nethack/helpers/action_mapping.py +0 -301
- synth_ai/environments/examples/nethack/helpers/nle_wrapper.py +0 -402
- synth_ai/environments/examples/nethack/helpers/observation_utils.py +0 -433
- synth_ai/environments/examples/nethack/helpers/recording_wrapper.py +0 -200
- synth_ai/environments/examples/nethack/helpers/trajectory_recorder.py +0 -269
- synth_ai/environments/examples/nethack/helpers/visualization/replay_viewer.py +0 -308
- synth_ai/environments/examples/nethack/helpers/visualization/visualizer.py +0 -431
- synth_ai/environments/examples/nethack/taskset.py +0 -323
- synth_ai/environments/examples/red/__init__.py +0 -7
- synth_ai/environments/examples/red/agent_demos/__init__.py +0 -1
- synth_ai/environments/examples/red/config_logging.py +0 -110
- synth_ai/environments/examples/red/engine.py +0 -721
- synth_ai/environments/examples/red/engine_helpers/__init__.py +0 -1
- synth_ai/environments/examples/red/engine_helpers/memory_map.py +0 -35
- synth_ai/environments/examples/red/engine_helpers/reward_components.py +0 -276
- synth_ai/environments/examples/red/engine_helpers/reward_library/__init__.py +0 -142
- synth_ai/environments/examples/red/engine_helpers/reward_library/adaptive_rewards.py +0 -57
- synth_ai/environments/examples/red/engine_helpers/reward_library/battle_rewards.py +0 -284
- synth_ai/environments/examples/red/engine_helpers/reward_library/composite_rewards.py +0 -150
- synth_ai/environments/examples/red/engine_helpers/reward_library/economy_rewards.py +0 -138
- synth_ai/environments/examples/red/engine_helpers/reward_library/efficiency_rewards.py +0 -57
- synth_ai/environments/examples/red/engine_helpers/reward_library/exploration_rewards.py +0 -331
- synth_ai/environments/examples/red/engine_helpers/reward_library/novelty_rewards.py +0 -121
- synth_ai/environments/examples/red/engine_helpers/reward_library/pallet_town_progression.py +0 -477
- synth_ai/environments/examples/red/engine_helpers/reward_library/pallet_town_rewards.py +0 -559
- synth_ai/environments/examples/red/engine_helpers/reward_library/pokemon_rewards.py +0 -313
- synth_ai/environments/examples/red/engine_helpers/reward_library/social_rewards.py +0 -148
- synth_ai/environments/examples/red/engine_helpers/reward_library/story_rewards.py +0 -247
- synth_ai/environments/examples/red/engine_helpers/screen_analysis.py +0 -368
- synth_ai/environments/examples/red/engine_helpers/state_extraction.py +0 -172
- synth_ai/environments/examples/red/environment.py +0 -298
- synth_ai/environments/examples/red/taskset.py +0 -79
- synth_ai/environments/examples/red/units/__init__.py +0 -1
- synth_ai/environments/examples/sokoban/__init__.py +0 -1
- synth_ai/environments/examples/sokoban/agent_demos/sokoban_full_eval.py +0 -899
- synth_ai/environments/examples/sokoban/engine.py +0 -678
- synth_ai/environments/examples/sokoban/engine_helpers/__init__.py +0 -1
- synth_ai/environments/examples/sokoban/engine_helpers/room_utils.py +0 -657
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/__init__.py +0 -18
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/__init__.py +0 -3
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/boxoban_env.py +0 -131
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/render_utils.py +0 -370
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/room_utils.py +0 -332
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env.py +0 -306
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_fixed_targets.py +0 -67
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_pull.py +0 -115
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_two_player.py +0 -123
- synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_variations.py +0 -394
- synth_ai/environments/examples/sokoban/environment.py +0 -229
- synth_ai/environments/examples/sokoban/generate_verified_puzzles.py +0 -440
- synth_ai/environments/examples/sokoban/puzzle_loader.py +0 -312
- synth_ai/environments/examples/sokoban/taskset.py +0 -544
- synth_ai/environments/examples/tictactoe/__init__.py +0 -1
- synth_ai/environments/examples/tictactoe/engine.py +0 -368
- synth_ai/environments/examples/tictactoe/environment.py +0 -240
- synth_ai/environments/examples/tictactoe/taskset.py +0 -215
- synth_ai/environments/examples/verilog/__init__.py +0 -10
- synth_ai/environments/examples/verilog/engine.py +0 -421
- synth_ai/environments/examples/verilog/environment.py +0 -350
- synth_ai/environments/examples/verilog/taskset.py +0 -420
- synth_ai/environments/examples/wordle/__init__.py +0 -29
- synth_ai/environments/examples/wordle/engine.py +0 -398
- synth_ai/environments/examples/wordle/environment.py +0 -159
- synth_ai/environments/examples/wordle/helpers/generate_instances_wordfreq.py +0 -75
- synth_ai/environments/examples/wordle/taskset.py +0 -230
- synth_ai/environments/reproducibility/core.py +0 -42
- synth_ai/environments/reproducibility/helpers.py +0 -0
- synth_ai/environments/reproducibility/tree.py +0 -363
- synth_ai/environments/service/app.py +0 -97
- synth_ai/environments/service/core_routes.py +0 -1021
- synth_ai/environments/service/external_registry.py +0 -56
- synth_ai/environments/service/registry.py +0 -9
- synth_ai/environments/stateful/__init__.py +0 -1
- synth_ai/environments/stateful/core.py +0 -163
- synth_ai/environments/stateful/engine.py +0 -21
- synth_ai/environments/stateful/state.py +0 -7
- synth_ai/environments/tasks/api.py +0 -19
- synth_ai/environments/tasks/core.py +0 -81
- synth_ai/environments/tasks/filters.py +0 -40
- synth_ai/environments/tasks/utils.py +0 -90
- synth_ai/environments/v0_observability/history.py +0 -3
- synth_ai/environments/v0_observability/log.py +0 -2
- synth_ai/evals/__init__.py +0 -15
- synth_ai/evals/base.py +0 -13
- synth_ai/evals/client.py +0 -82
- synth_ai/handshake.py +0 -109
- synth_ai/http.py +0 -26
- synth_ai/http_client.py +0 -136
- synth_ai/inference/__init__.py +0 -5
- synth_ai/inference/client.py +0 -34
- synth_ai/jobs/client.py +0 -295
- synth_ai/judge_schemas.py +0 -127
- synth_ai/learning/__init__.py +0 -59
- synth_ai/learning/client.py +0 -241
- synth_ai/learning/ft_client.py +0 -7
- synth_ai/learning/health.py +0 -49
- synth_ai/learning/jobs.py +0 -201
- synth_ai/learning/rl/client.py +0 -267
- synth_ai/learning/rl/contracts.py +0 -27
- synth_ai/learning/rl/env_keys.py +0 -166
- synth_ai/learning/rl/secrets.py +0 -13
- synth_ai/learning/sft/client.py +0 -68
- synth_ai/learning/sft/config.py +0 -270
- synth_ai/learning/sft/data.py +0 -295
- synth_ai/learning/validators.py +0 -49
- synth_ai/lm/__init__.py +0 -25
- synth_ai/task/__init__.py +0 -121
- synth_ai/task/apps/__init__.py +0 -129
- synth_ai/task/config.py +0 -257
- synth_ai/task/contracts.py +0 -236
- synth_ai/task/datasets.py +0 -108
- synth_ai/task/proxy.py +0 -251
- synth_ai/task/rubrics/__init__.py +0 -56
- synth_ai/task/rubrics/loaders.py +0 -152
- synth_ai/task/server.py +0 -432
- synth_ai/task/trace_correlation_helpers.py +0 -315
- synth_ai/task/tracing_utils.py +0 -84
- synth_ai/task/validators.py +0 -418
- synth_ai/tracing_v3/__init__.py +0 -97
- synth_ai/tracing_v3/abstractions.py +0 -302
- synth_ai/tracing_v3/config.py +0 -84
- synth_ai/tracing_v3/db_config.py +0 -194
- synth_ai/tracing_v3/decorators.py +0 -398
- synth_ai/tracing_v3/llm_call_record_helpers.py +0 -391
- synth_ai/tracing_v3/migration_helper.py +0 -120
- synth_ai/tracing_v3/session_tracer.py +0 -540
- synth_ai/tracing_v3/storage/base.py +0 -210
- synth_ai/tracing_v3/storage/config.py +0 -75
- synth_ai/tracing_v3/storage/factory.py +0 -39
- synth_ai/tracing_v3/trace_utils.py +0 -317
- synth_ai/tracing_v3/turso/daemon.py +0 -151
- synth_ai/tracing_v3/turso/models.py +0 -469
- synth_ai/tracing_v3/turso/native_manager.py +0 -1209
- synth_ai/tracing_v3/utils.py +0 -108
- synth_ai/tui/__init__.py +0 -5
- synth_ai/tui/__main__.py +0 -13
- synth_ai/tui/cli/__init__.py +0 -1
- synth_ai/tui/cli/query_experiments.py +0 -164
- synth_ai/tui/cli/query_experiments_v3.py +0 -164
- synth_ai/tui/dashboard.py +0 -906
- synth_ai/v0/api/__init__.py +0 -8
- synth_ai/v0/api/models/__init__.py +0 -8
- synth_ai/v0/api/models/supported.py +0 -8
- synth_ai/v0/config/__init__.py +0 -15
- synth_ai/v0/config/base_url.py +0 -12
- synth_ai/v0/lm/__init__.py +0 -51
- synth_ai/v0/lm/caching/__init__.py +0 -0
- synth_ai/v0/lm/caching/constants.py +0 -6
- synth_ai/v0/lm/caching/dbs.py +0 -0
- synth_ai/v0/lm/caching/ephemeral.py +0 -100
- synth_ai/v0/lm/caching/handler.py +0 -137
- synth_ai/v0/lm/caching/initialize.py +0 -11
- synth_ai/v0/lm/caching/persistent.py +0 -114
- synth_ai/v0/lm/config.py +0 -115
- synth_ai/v0/lm/constants.py +0 -32
- synth_ai/v0/lm/core/__init__.py +0 -8
- synth_ai/v0/lm/core/all.py +0 -73
- synth_ai/v0/lm/core/exceptions.py +0 -5
- synth_ai/v0/lm/core/main.py +0 -331
- synth_ai/v0/lm/core/main_v3.py +0 -594
- synth_ai/v0/lm/core/synth_models.py +0 -35
- synth_ai/v0/lm/core/vendor_clients.py +0 -190
- synth_ai/v0/lm/cost/__init__.py +0 -0
- synth_ai/v0/lm/cost/monitor.py +0 -1
- synth_ai/v0/lm/cost/statefulness.py +0 -1
- synth_ai/v0/lm/injection.py +0 -80
- synth_ai/v0/lm/overrides.py +0 -206
- synth_ai/v0/lm/provider_support/__init__.py +0 -8
- synth_ai/v0/lm/provider_support/anthropic.py +0 -972
- synth_ai/v0/lm/provider_support/openai.py +0 -1139
- synth_ai/v0/lm/provider_support/suppress_logging.py +0 -31
- synth_ai/v0/lm/structured_outputs/__init__.py +0 -0
- synth_ai/v0/lm/structured_outputs/handler.py +0 -440
- synth_ai/v0/lm/structured_outputs/inject.py +0 -297
- synth_ai/v0/lm/structured_outputs/rehabilitate.py +0 -185
- synth_ai/v0/lm/tools/__init__.py +0 -3
- synth_ai/v0/lm/tools/base.py +0 -172
- synth_ai/v0/lm/unified_interface.py +0 -202
- synth_ai/v0/lm/vendors/__init__.py +0 -0
- synth_ai/v0/lm/vendors/base.py +0 -81
- synth_ai/v0/lm/vendors/core/__init__.py +0 -0
- synth_ai/v0/lm/vendors/core/anthropic_api.py +0 -387
- synth_ai/v0/lm/vendors/core/gemini_api.py +0 -292
- synth_ai/v0/lm/vendors/core/mistral_api.py +0 -322
- synth_ai/v0/lm/vendors/core/openai_api.py +0 -227
- synth_ai/v0/lm/vendors/core/synth_dev_api.py +0 -0
- synth_ai/v0/lm/vendors/local/__init__.py +0 -0
- synth_ai/v0/lm/vendors/local/ollama.py +0 -0
- synth_ai/v0/lm/vendors/openai_standard.py +0 -782
- synth_ai/v0/lm/vendors/openai_standard_responses.py +0 -259
- synth_ai/v0/lm/vendors/retries.py +0 -22
- synth_ai/v0/lm/vendors/supported/__init__.py +0 -0
- synth_ai/v0/lm/vendors/supported/custom_endpoint.py +0 -415
- synth_ai/v0/lm/vendors/supported/deepseek.py +0 -69
- synth_ai/v0/lm/vendors/supported/grok.py +0 -75
- synth_ai/v0/lm/vendors/supported/groq.py +0 -16
- synth_ai/v0/lm/vendors/supported/ollama.py +0 -15
- synth_ai/v0/lm/vendors/supported/openrouter.py +0 -74
- synth_ai/v0/lm/vendors/supported/together.py +0 -11
- synth_ai/v0/lm/vendors/synth_client.py +0 -835
- synth_ai/v0/lm/warmup.py +0 -186
- synth_ai/v0/tracing/__init__.py +0 -0
- synth_ai/v0/tracing/abstractions.py +0 -224
- synth_ai/v0/tracing/base_client.py +0 -91
- synth_ai/v0/tracing/client_manager.py +0 -131
- synth_ai/v0/tracing/config.py +0 -142
- synth_ai/v0/tracing/context.py +0 -146
- synth_ai/v0/tracing/decorators.py +0 -682
- synth_ai/v0/tracing/events/__init__.py +0 -0
- synth_ai/v0/tracing/events/manage.py +0 -147
- synth_ai/v0/tracing/events/scope.py +0 -86
- synth_ai/v0/tracing/events/store.py +0 -228
- synth_ai/v0/tracing/immediate_client.py +0 -151
- synth_ai/v0/tracing/local.py +0 -18
- synth_ai/v0/tracing/log_client_base.py +0 -73
- synth_ai/v0/tracing/retry_queue.py +0 -186
- synth_ai/v0/tracing/trackers.py +0 -515
- synth_ai/v0/tracing/upload.py +0 -409
- synth_ai/v0/tracing/utils.py +0 -9
- synth_ai/v0/tracing_v1/__init__.py +0 -16
- synth_ai/v0/tracing_v1/abstractions.py +0 -224
- synth_ai/v0/tracing_v1/base_client.py +0 -91
- synth_ai/v0/tracing_v1/client_manager.py +0 -131
- synth_ai/v0/tracing_v1/config.py +0 -142
- synth_ai/v0/tracing_v1/context.py +0 -146
- synth_ai/v0/tracing_v1/decorators.py +0 -703
- synth_ai/v0/tracing_v1/events/__init__.py +0 -0
- synth_ai/v0/tracing_v1/events/manage.py +0 -147
- synth_ai/v0/tracing_v1/events/scope.py +0 -86
- synth_ai/v0/tracing_v1/events/store.py +0 -228
- synth_ai/v0/tracing_v1/immediate_client.py +0 -151
- synth_ai/v0/tracing_v1/local.py +0 -18
- synth_ai/v0/tracing_v1/log_client_base.py +0 -73
- synth_ai/v0/tracing_v1/retry_queue.py +0 -186
- synth_ai/v0/tracing_v1/trackers.py +0 -515
- synth_ai/v0/tracing_v1/upload.py +0 -527
- synth_ai/v0/tracing_v1/utils.py +0 -9
- synth_ai/v0/tracing_v3/__init__.py +0 -10
- synth_ai/v0/tracing_v3/abstractions.py +0 -3
- synth_ai/v0/tracing_v3/decorators.py +0 -3
- synth_ai/v0/tracing_v3/llm_call_record_helpers.py +0 -3
- synth_ai/v0/tracing_v3/session_tracer.py +0 -3
- synth_ai-0.2.14.dist-info/METADATA +0 -139
- synth_ai-0.2.14.dist-info/RECORD +0 -762
- synth_ai-0.2.14.dist-info/top_level.txt +0 -2
- /synth_ai/{demos/demo_task_apps → cli/demo_apps}/crafter/__init__.py +0 -0
- /synth_ai/{demos → cli/demo_apps}/demo_task_apps/__init__.py +0 -0
- /synth_ai/{demos → cli/demo_apps}/demo_task_apps/crafter/configs/crafter_fft_4b.toml +0 -0
- /synth_ai/{demos → cli/demo_apps}/demo_task_apps/crafter/configs/rl_from_base_qwen4b.toml +0 -0
- /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/__init__.py +0 -0
- /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/_common.py +0 -0
- /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/app.py +0 -0
- /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/config.toml +0 -0
- /synth_ai/{demos → cli/demo_apps}/demo_task_apps/math/deploy_modal.py +0 -0
- {examples/task_apps → synth_ai/core/apps}/__init__.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/examples/basic_usage.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/hooks.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/lm_call_record_abstractions.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/replica_sync.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/serialization.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/storage/__init__.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/storage/exceptions.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/storage/types.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/storage/utils.py +0 -0
- /synth_ai/{tracing_v3 → core/tracing_v3}/turso/__init__.py +0 -0
- /synth_ai/{evals → sdk/judging}/types.py +0 -0
- /synth_ai/{learning → sdk/learning}/algorithms.py +0 -0
- /synth_ai/{learning → sdk/learning}/config.py +0 -0
- /synth_ai/{learning → sdk/learning}/constants.py +0 -0
- /synth_ai/{learning → sdk/learning}/core.py +0 -0
- /synth_ai/{learning → sdk/learning}/gateway.py +0 -0
- /synth_ai/{learning → sdk/learning}/rl/__init__.py +0 -0
- /synth_ai/{learning → sdk/learning}/rl/config.py +0 -0
- /synth_ai/{learning → sdk/learning}/rl_client.py +0 -0
- /synth_ai/{learning → sdk/learning}/sft/__init__.py +0 -0
- /synth_ai/{learning → sdk/learning}/sse.py +0 -0
- /synth_ai/{task → sdk/task}/auth.py +0 -0
- /synth_ai/{task → sdk/task}/client.py +0 -0
- /synth_ai/{task → sdk/task}/errors.py +0 -0
- /synth_ai/{task → sdk/task}/health.py +0 -0
- /synth_ai/{task → sdk/task}/json.py +0 -0
- /synth_ai/{task → sdk/task}/rubrics/models.py +0 -0
- /synth_ai/{task → sdk/task}/rubrics/scoring.py +0 -0
- /synth_ai/{task → sdk/task}/rubrics/strict.py +0 -0
- /synth_ai/{task → sdk/task}/vendors.py +0 -0
- {synth_ai-0.2.14.dist-info → synth_ai-0.4.1.dist-info}/WHEEL +0 -0
- {synth_ai-0.2.14.dist-info → synth_ai-0.4.1.dist-info}/entry_points.txt +0 -0
- {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)
|