synth-ai 0.2.9.dev0__py3-none-any.whl → 0.2.23.dev3__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.
- examples/README.md +1 -0
- examples/__init__.py +16 -0
- examples/analyze_semantic_words.sh +17 -0
- examples/baseline/banking77_baseline.py +243 -0
- examples/baseline/banking77_pipeline_baseline.py +294 -0
- examples/baseline/crafter_baseline.py +407 -0
- examples/baseline/pokemon_red_baseline.py +326 -0
- examples/baseline/simple_baseline.py +56 -0
- examples/baseline/warming_up_to_rl_baseline.py +239 -0
- examples/blog_posts/gepa/README.md +355 -0
- examples/blog_posts/gepa/configs/banking77_gepa_local.toml +95 -0
- examples/blog_posts/gepa/configs/banking77_gepa_test.toml +80 -0
- examples/blog_posts/gepa/configs/banking77_mipro_local.toml +50 -0
- examples/blog_posts/gepa/configs/banking77_pipeline_gepa_local.toml +101 -0
- examples/blog_posts/gepa/configs/banking77_pipeline_gepa_test.toml +96 -0
- examples/blog_posts/gepa/configs/hotpotqa_gepa_local.toml +57 -0
- examples/blog_posts/gepa/configs/hotpotqa_gepa_qwen.toml +35 -0
- examples/blog_posts/gepa/configs/hotpotqa_mipro_local.toml +51 -0
- examples/blog_posts/gepa/configs/hover_gepa_local.toml +57 -0
- examples/blog_posts/gepa/configs/hover_gepa_qwen.toml +35 -0
- examples/blog_posts/gepa/configs/hover_mipro_local.toml +51 -0
- examples/blog_posts/gepa/configs/ifbench_gepa_local.toml +57 -0
- examples/blog_posts/gepa/configs/ifbench_gepa_qwen.toml +35 -0
- examples/blog_posts/gepa/configs/ifbench_mipro_local.toml +51 -0
- examples/blog_posts/gepa/configs/pupa_gepa_local.toml +58 -0
- examples/blog_posts/gepa/configs/pupa_mipro_local.toml +52 -0
- examples/blog_posts/gepa/deploy_banking77_task_app.sh +54 -0
- examples/blog_posts/gepa/gepa_baseline.py +204 -0
- examples/blog_posts/gepa/query_prompts_example.py +97 -0
- examples/blog_posts/gepa/run_gepa_banking77.sh +112 -0
- examples/blog_posts/gepa/run_gepa_banking77_pipeline.sh +163 -0
- examples/blog_posts/gepa/task_apps.py +105 -0
- examples/blog_posts/gepa/test_gepa_local.sh +67 -0
- examples/blog_posts/gepa/verify_banking77_setup.sh +123 -0
- examples/blog_posts/mipro/README.md +415 -0
- examples/blog_posts/mipro/configs/banking77_mipro_local.toml +91 -0
- examples/blog_posts/mipro/configs/banking77_mipro_test.toml +87 -0
- examples/blog_posts/mipro/configs/banking77_pipeline_mipro_gemini_flash_lite_local.toml +98 -0
- examples/blog_posts/mipro/configs/banking77_pipeline_mipro_gpt41mini_local.toml +96 -0
- examples/blog_posts/mipro/configs/banking77_pipeline_mipro_local.toml +94 -0
- examples/blog_posts/mipro/configs/banking77_pipeline_mipro_test.toml +170 -0
- examples/blog_posts/mipro/deploy_banking77_pipeline_task_app.sh +59 -0
- examples/blog_posts/mipro/deploy_banking77_task_app.sh +41 -0
- examples/blog_posts/mipro/multi_step.md +79 -0
- examples/blog_posts/mipro/run_mipro_banking77.sh +191 -0
- examples/blog_posts/mipro/run_mipro_banking77_pipeline.sh +171 -0
- examples/blog_posts/mipro/run_mipro_banking77_pipeline_gemini_flash_lite.sh +177 -0
- examples/blog_posts/mipro/run_mipro_banking77_pipeline_gpt41mini.sh +173 -0
- examples/blog_posts/mipro/verify_banking77_setup.sh +117 -0
- examples/blog_posts/pokemon_vl/README.md +98 -0
- examples/blog_posts/pokemon_vl/configs/eval_gpt5nano.toml +26 -0
- examples/blog_posts/pokemon_vl/configs/eval_qwen3_vl.toml +27 -0
- examples/blog_posts/pokemon_vl/configs/eval_rl_final.toml +24 -0
- examples/blog_posts/pokemon_vl/configs/filter_high_reward.toml +10 -0
- examples/blog_posts/pokemon_vl/configs/train_rl_from_sft.toml +43 -0
- examples/blog_posts/pokemon_vl/configs/train_sft_qwen4b_vl.toml +40 -0
- examples/blog_posts/pokemon_vl/extract_images.py +239 -0
- examples/blog_posts/pokemon_vl/pokemon_vl_baseline.py +326 -0
- examples/blog_posts/pokemon_vl/run_eval_extract_images.py +209 -0
- examples/blog_posts/pokemon_vl/run_qwen_eval_extract_images.py +212 -0
- examples/blog_posts/pokemon_vl/text_box_analysis.md +106 -0
- examples/blog_posts/warming_up_to_rl/ARCHITECTURE.md +195 -0
- examples/blog_posts/warming_up_to_rl/FINAL_TEST_RESULTS.md +127 -0
- examples/blog_posts/warming_up_to_rl/INFERENCE_SUCCESS.md +132 -0
- examples/blog_posts/warming_up_to_rl/README.md +158 -0
- examples/blog_posts/warming_up_to_rl/SMOKE_TESTING.md +164 -0
- examples/blog_posts/warming_up_to_rl/SMOKE_TEST_COMPLETE.md +253 -0
- examples/blog_posts/warming_up_to_rl/configs/eval_baseline_qwen32b_10x20.toml +25 -0
- examples/blog_posts/warming_up_to_rl/configs/eval_ft_qwen4b.toml +25 -0
- examples/blog_posts/warming_up_to_rl/configs/eval_ft_qwen4b_10x20.toml +26 -0
- examples/blog_posts/warming_up_to_rl/configs/eval_groq_qwen32b.toml +25 -0
- examples/blog_posts/warming_up_to_rl/configs/eval_openai_gpt_oss_120b.toml +29 -0
- examples/blog_posts/warming_up_to_rl/configs/filter_high_reward_dataset.toml +10 -0
- examples/blog_posts/warming_up_to_rl/configs/smoke_test.toml +75 -0
- examples/blog_posts/warming_up_to_rl/configs/train_rl_from_sft.toml +91 -0
- examples/blog_posts/warming_up_to_rl/configs/train_sft_qwen4b.toml +40 -0
- examples/blog_posts/warming_up_to_rl/warming_up_to_rl_baseline.py +187 -0
- examples/crafter_debug_render.py +186 -0
- examples/dev/qwen3_32b_qlora_4xh100.toml +45 -0
- examples/gepa/banking77_pipeline_gepa.toml +96 -0
- examples/gepa/multi_stage_gepa_example.toml +84 -0
- examples/gepa/run_gepa_banking77_pipeline.sh +157 -0
- examples/multi_step/SFT_README.md +147 -0
- examples/multi_step/configs/README_verilog_rl.md +77 -0
- examples/multi_step/configs/VERILOG_REWARDS.md +103 -0
- examples/multi_step/configs/VERILOG_RL_CHECKLIST.md +196 -0
- examples/multi_step/configs/crafter_eval_synth_qwen4b.toml +35 -0
- examples/multi_step/configs/crafter_eval_text_only_groq_qwen32b.toml +36 -0
- examples/multi_step/configs/crafter_rl_outcome.toml +75 -0
- examples/multi_step/configs/crafter_rl_stepwise_hosted_judge.toml +145 -0
- examples/multi_step/configs/crafter_rl_stepwise_shaped.toml +84 -0
- examples/multi_step/configs/crafter_rl_stepwise_simple.toml +79 -0
- examples/multi_step/configs/crafter_rl_stepwise_simple_NEW_FORMAT.toml +105 -0
- examples/multi_step/configs/crafter_sft_qwen30b_lora.toml +62 -0
- examples/multi_step/configs/crafter_synth_backend.md +40 -0
- examples/multi_step/configs/verilog_eval_groq_qwen32b.toml +31 -0
- examples/multi_step/configs/verilog_eval_synth_qwen8b.toml +33 -0
- examples/multi_step/configs/verilog_rl_lora.toml +147 -0
- examples/multi_step/convert_traces_to_sft.py +84 -0
- examples/multi_step/crafter_rl_lora.md +70 -0
- examples/multi_step/judges/crafter_backend_judge.py +220 -0
- examples/multi_step/judges/verilog_backend_judge.py +234 -0
- examples/multi_step/readme.md +48 -0
- examples/multi_step/run_sft_qwen30b.sh +45 -0
- examples/multi_step/sse_metrics_streaming_notes.md +357 -0
- examples/multi_step/task_app_config_notes.md +494 -0
- examples/multi_step/verilog_rl_lora.md +218 -0
- examples/qwen_coder/README.md +102 -0
- examples/qwen_coder/_shared.py +113 -0
- examples/qwen_coder/configs/coder_lora_30b.toml +60 -0
- examples/qwen_coder/configs/coder_lora_4b.toml +61 -0
- examples/qwen_coder/configs/coder_lora_small.toml +57 -0
- examples/qwen_coder/generate_dataset.py +98 -0
- examples/qwen_coder/infer_ft_smoke.py +65 -0
- examples/qwen_coder/infer_prod_proxy.py +73 -0
- examples/qwen_coder/infer_via_synth.py +87 -0
- examples/qwen_coder/scripts/infer_coder.sh +19 -0
- examples/qwen_coder/scripts/train_coder_30b.sh +22 -0
- examples/qwen_coder/sft_full_17b.py +103 -0
- examples/qwen_coder/sft_lora_30b.py +110 -0
- examples/qwen_coder/subset_jsonl.py +39 -0
- examples/qwen_coder/todos.md +38 -0
- examples/qwen_coder/validate_jsonl.py +60 -0
- examples/qwen_vl/BUGS_AND_FIXES.md +232 -0
- examples/qwen_vl/IMAGE_VALIDATION_COMPLETE.md +271 -0
- examples/qwen_vl/IMAGE_VALIDATION_SUMMARY.md +260 -0
- examples/qwen_vl/INFERENCE_SFT_TESTS.md +412 -0
- examples/qwen_vl/NEXT_STEPS_2B.md +325 -0
- examples/qwen_vl/QUICKSTART.md +327 -0
- examples/qwen_vl/QUICKSTART_RL_VISION.md +110 -0
- examples/qwen_vl/README.md +152 -0
- examples/qwen_vl/RL_VISION_COMPLETE.md +475 -0
- examples/qwen_vl/RL_VISION_TESTING.md +333 -0
- examples/qwen_vl/SDK_VISION_INTEGRATION.md +328 -0
- examples/qwen_vl/SETUP_COMPLETE.md +274 -0
- examples/qwen_vl/VISION_TESTS_COMPLETE.md +489 -0
- examples/qwen_vl/VLM_PIPELINE_COMPLETE.md +242 -0
- examples/qwen_vl/__init__.py +2 -0
- examples/qwen_vl/collect_data_via_cli.md +415 -0
- examples/qwen_vl/collect_vision_traces.py +368 -0
- examples/qwen_vl/configs/crafter_rl_vision_qwen3vl4b.toml +110 -0
- examples/qwen_vl/configs/crafter_vlm_sft_example.toml +59 -0
- examples/qwen_vl/configs/eval_gpt4o_mini_vision.toml +26 -0
- examples/qwen_vl/configs/eval_gpt4o_vision_proper.toml +29 -0
- examples/qwen_vl/configs/eval_gpt5nano_vision.toml +26 -0
- examples/qwen_vl/configs/eval_qwen3vl_vision.toml +26 -0
- examples/qwen_vl/configs/filter_qwen3vl_sft.toml +49 -0
- examples/qwen_vl/configs/filter_vision_sft.toml +52 -0
- examples/qwen_vl/configs/filter_vision_test.toml +8 -0
- examples/qwen_vl/configs/sft_qwen3_vl_2b_test.toml +54 -0
- examples/qwen_vl/crafter_gpt5nano_agent.py +308 -0
- examples/qwen_vl/crafter_qwen_vl_agent.py +300 -0
- examples/qwen_vl/run_vision_comparison.sh +61 -0
- examples/qwen_vl/run_vision_sft_pipeline.sh +175 -0
- examples/qwen_vl/test_image_validation.py +201 -0
- examples/qwen_vl/test_sft_vision_data.py +110 -0
- examples/rl/README.md +169 -0
- examples/rl/configs/eval_base_qwen.toml +17 -0
- examples/rl/configs/eval_rl_qwen.toml +13 -0
- examples/rl/configs/rl_from_base_qwen.toml +62 -0
- examples/rl/configs/rl_from_base_qwen17.toml +80 -0
- examples/rl/configs/rl_from_ft_qwen.toml +37 -0
- examples/rl/download_dataset.py +80 -0
- examples/rl/run_eval.py +436 -0
- examples/rl/run_rl_and_save.py +111 -0
- examples/rl/task_app/README.md +21 -0
- {synth_ai/task/apps → examples/rl/task_app}/math_single_step.py +188 -50
- examples/rl/task_app/math_task_app.py +111 -0
- examples/run_crafter_demo.sh +10 -0
- examples/sdk_prompt_learning_example.py +55 -0
- examples/sft/README.md +139 -0
- examples/sft/configs/crafter_fft_qwen0p6b.toml +49 -0
- examples/sft/configs/crafter_lora_qwen0p6b.toml +49 -0
- examples/sft/evaluate.py +117 -0
- examples/sft/export_dataset.py +120 -0
- examples/sft/generate_traces.py +164 -0
- examples/swe/__init__.py +12 -0
- examples/swe/task_app/README.md +135 -0
- examples/swe/task_app/__init__.py +2 -0
- examples/swe/task_app/grpo_swe_mini.py +604 -0
- examples/swe/task_app/grpo_swe_mini_task_app.py +124 -0
- examples/swe/task_app/hosted/README.md +173 -0
- examples/swe/task_app/hosted/__init__.py +5 -0
- examples/swe/task_app/hosted/branching.py +143 -0
- examples/swe/task_app/hosted/environment_routes.py +1289 -0
- examples/swe/task_app/hosted/envs/__init__.py +1 -0
- examples/swe/task_app/hosted/envs/crafter/__init__.py +6 -0
- examples/swe/task_app/hosted/envs/crafter/app.py +1 -0
- examples/swe/task_app/hosted/envs/crafter/environment.py +522 -0
- examples/swe/task_app/hosted/envs/crafter/policy.py +478 -0
- examples/swe/task_app/hosted/envs/crafter/react_agent.py +108 -0
- examples/swe/task_app/hosted/envs/crafter/shared.py +305 -0
- examples/swe/task_app/hosted/envs/crafter/tools.py +47 -0
- examples/swe/task_app/hosted/envs/mini_swe/__init__.py +8 -0
- examples/swe/task_app/hosted/envs/mini_swe/environment.py +1191 -0
- examples/swe/task_app/hosted/envs/mini_swe/policy.py +355 -0
- examples/swe/task_app/hosted/envs/mini_swe/shared.py +83 -0
- examples/swe/task_app/hosted/envs/mini_swe/tools.py +96 -0
- examples/swe/task_app/hosted/hosted_app.py +204 -0
- examples/swe/task_app/hosted/inference/__init__.py +5 -0
- examples/swe/task_app/hosted/inference/openai_client.py +584 -0
- examples/swe/task_app/hosted/main.py +100 -0
- examples/swe/task_app/hosted/policy_routes.py +1094 -0
- examples/swe/task_app/hosted/registry.py +195 -0
- examples/swe/task_app/hosted/rollout.py +1905 -0
- examples/swe/task_app/hosted/storage/__init__.py +5 -0
- examples/swe/task_app/hosted/storage/volume.py +211 -0
- examples/swe/task_app/hosted/test_agents.py +161 -0
- examples/swe/task_app/hosted/test_service.py +136 -0
- examples/swe/task_app/hosted/utils.py +62 -0
- examples/swe/task_app/morph_backend.py +178 -0
- examples/task_apps/IMAGE_ONLY_EVAL_QUICKSTART.md +258 -0
- examples/task_apps/TESTING.md +275 -0
- examples/task_apps/banking77/__init__.py +6 -0
- examples/task_apps/banking77/banking77_task_app.py +912 -0
- examples/task_apps/banking77/deploy_wrapper.py +46 -0
- examples/task_apps/banking77_pipeline/__init__.py +6 -0
- examples/task_apps/banking77_pipeline/banking77_pipeline_task_app.py +489 -0
- examples/task_apps/banking77_pipeline/deploy_wrapper.py +50 -0
- examples/task_apps/crafter/CREATE_SFT_DATASET.md +286 -0
- examples/task_apps/crafter/EVAL_IMAGE_ONLY_RESULTS.md +152 -0
- examples/task_apps/crafter/FILTER_COMMAND_STATUS.md +187 -0
- examples/task_apps/crafter/FILTER_COMMAND_SUCCESS.md +281 -0
- examples/task_apps/crafter/QUERY_EXAMPLES.md +203 -0
- examples/task_apps/crafter/README_IMAGE_ONLY_EVAL.md +316 -0
- examples/task_apps/crafter/eval_image_only_gpt4o.toml +28 -0
- examples/task_apps/crafter/eval_text_only_groq_llama.toml +36 -0
- examples/task_apps/crafter/filter_sft_dataset.toml +16 -0
- examples/task_apps/crafter/task_app/README.md +42 -0
- examples/task_apps/crafter/task_app/__init__.py +5 -0
- examples/task_apps/crafter/task_app/grpo_crafter.py +1055 -0
- examples/task_apps/crafter/task_app/grpo_crafter_task_app.py +146 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/README.md +173 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/__init__.py +5 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/branching.py +143 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/environment_routes.py +1226 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/__init__.py +1 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/__init__.py +6 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/app.py +1 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/environment.py +532 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/policy.py +583 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/react_agent.py +122 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/shared.py +305 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/envs/crafter/tools.py +47 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/hosted_app.py +253 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/inference/__init__.py +5 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/inference/openai_client.py +999 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/main.py +100 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/policy_routes.py +1252 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/registry.py +195 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/rollout.py +2233 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/storage/__init__.py +5 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/storage/volume.py +211 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/test_agents.py +161 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/test_service.py +136 -0
- examples/task_apps/crafter/task_app/synth_envs_hosted/utils.py +411 -0
- examples/task_apps/dev/pokemon_emerald/__init__.py +2 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/README.md +811 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/__init__.py +120 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/action.py +160 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/memory.py +155 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/perception.py +69 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/planning.py +96 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/simple.py +1502 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/agent/system_prompt.py +4 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/grab_map.py +68 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/manual.py +216 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/__init__.py +35 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/emerald_utils.py +631 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/emulator.py +1544 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/enums.py +1428 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/memory_reader.py +4848 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/types.py +41 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/utils.py +298 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pyproject.toml +95 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/run.py +204 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/app.py +2152 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/client.py +429 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server/frame_server.py +155 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/README.md +78 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/run_tests.py +122 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_agent_direct.py +76 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_agent_prompts.py +413 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_battle_state_formatting.py +204 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_dialogue_detection.py +133 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_dialogue_detection_comprehensive.py +229 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_direct_agent_emulator.py +300 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_fps_adjustment_pytest.py +205 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_house_to_outside_direct.py +200 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_house_to_outside_transition.py +284 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_map_ground_truth_comparison.py +468 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_memory_map.py +575 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_server_map_validation.py +311 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests/test_torchic_state.py +259 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/anticheat.py +372 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/checkpoint.py +296 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/error_handler.py +275 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/get_local_ip.py +22 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/helpers.py +44 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/llm_logger.py +514 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_formatter.py +415 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_stitcher.py +1763 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_stitcher_singleton.py +33 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_trimmer.py +106 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/map_visualizer.py +334 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/ocr_dialogue.py +1020 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/recording.py +188 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/state_formatter.py +1481 -0
- examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils/vlm.py +862 -0
- examples/task_apps/dev/pokemon_emerald/modal_app.py +114 -0
- examples/task_apps/dev/pokemon_emerald/task_app/README.md +81 -0
- examples/task_apps/dev/pokemon_emerald/task_app/__init__.py +6 -0
- examples/task_apps/dev/pokemon_emerald/task_app/pokemon_emerald.py +685 -0
- examples/task_apps/enron/__init__.py +2 -0
- examples/task_apps/enron/eval_groq_qwen32.toml +16 -0
- examples/task_apps/enron/filter_sft.toml +5 -0
- examples/task_apps/enron/task_app/README.md +14 -0
- examples/task_apps/enron/task_app/__init__.py +1 -0
- examples/task_apps/enron/task_app/grpo_enron.py +906 -0
- examples/task_apps/enron/task_app/grpo_enron_task_app.py +146 -0
- examples/task_apps/enron/tests/__init__.py +4 -0
- examples/task_apps/enron/tests/conftest.py +115 -0
- examples/task_apps/enron/tests/integration/__init__.py +4 -0
- examples/task_apps/enron/tests/integration/test_enron_eval.py +179 -0
- examples/task_apps/enron/tests/integration/test_enron_rollout.py +135 -0
- examples/task_apps/enron/tests/unit/__init__.py +4 -0
- examples/task_apps/enron/tests/unit/test_enron_environment.py +126 -0
- examples/task_apps/gepa_benchmarks/__init__.py +7 -0
- examples/task_apps/gepa_benchmarks/common.py +260 -0
- examples/task_apps/gepa_benchmarks/hotpotqa_task_app.py +507 -0
- examples/task_apps/gepa_benchmarks/hover_task_app.py +436 -0
- examples/task_apps/gepa_benchmarks/ifbench_task_app.py +563 -0
- examples/task_apps/gepa_benchmarks/pupa_task_app.py +460 -0
- examples/task_apps/math/README.md +21 -0
- examples/task_apps/math/math_single_step.py +1000 -0
- examples/task_apps/math/math_task_app.py +115 -0
- examples/task_apps/pokemon_battle/__init__.py +2 -0
- examples/task_apps/pokemon_battle/modal_app.py +104 -0
- examples/task_apps/pokemon_battle/task_app/README.md +68 -0
- examples/task_apps/pokemon_battle/task_app/__init__.py +6 -0
- examples/task_apps/pokemon_battle/task_app/pokemon_showdown.py +932 -0
- examples/task_apps/pokemon_red/EVAL_IMAGE_ONLY_COMPLETE.md +283 -0
- examples/task_apps/pokemon_red/EVAL_IMAGE_ONLY_STATUS.md +155 -0
- examples/task_apps/pokemon_red/README.md +356 -0
- examples/task_apps/pokemon_red/README_IMAGE_ONLY_EVAL.md +428 -0
- examples/task_apps/pokemon_red/__init__.py +3 -0
- examples/task_apps/pokemon_red/eval_image_only_gpt4o.toml +30 -0
- examples/task_apps/pokemon_red/eval_pokemon_red_policy.py +224 -0
- examples/task_apps/pokemon_red/pallet_town_rl_config.toml +75 -0
- examples/task_apps/pokemon_red/task_app.py +1048 -0
- examples/task_apps/pokemon_red/test_pallet_town_rewards.py +193 -0
- examples/task_apps/sokoban/README.md +306 -0
- examples/task_apps/sokoban/__init__.py +3 -0
- examples/task_apps/sokoban/eval_groq_qwen32.toml +16 -0
- examples/task_apps/sokoban/eval_openai_gpt5.toml +16 -0
- examples/task_apps/sokoban/filter_sft.toml +5 -0
- examples/task_apps/sokoban/task_app.py +1058 -0
- examples/task_apps/sokoban/tests/__init__.py +4 -0
- examples/task_apps/sokoban/tests/conftest.py +113 -0
- examples/task_apps/sokoban/tests/integration/__init__.py +4 -0
- examples/task_apps/sokoban/tests/integration/test_sokoban_eval.py +57 -0
- examples/task_apps/sokoban/tests/integration/test_sokoban_rollout.py +198 -0
- examples/task_apps/sokoban/tests/unit/__init__.py +4 -0
- examples/task_apps/sokoban/tests/unit/test_sokoban_environment.py +114 -0
- examples/task_apps/verilog/__init__.py +1 -0
- examples/task_apps/verilog/eval_groq_qwen32b.toml +22 -0
- examples/task_apps/verilog/filter_sft.toml +5 -0
- examples/task_apps/verilog/task_app/README.md +12 -0
- examples/task_apps/verilog/task_app/__init__.py +1 -0
- examples/task_apps/verilog/task_app/grpo_verilog.py +1166 -0
- examples/task_apps/verilog/task_app/grpo_verilog_task_app.py +145 -0
- examples/task_apps/verilog/tests/__init__.py +4 -0
- examples/task_apps/verilog/tests/conftest.py +115 -0
- examples/task_apps/verilog/tests/integration/__init__.py +4 -0
- examples/task_apps/verilog/tests/integration/test_verilog_eval.py +181 -0
- examples/task_apps/verilog/tests/integration/test_verilog_rollout.py +55 -0
- examples/task_apps/verilog/tests/unit/__init__.py +4 -0
- examples/task_apps/verilog/tests/unit/test_verilog_scoring.py +118 -0
- examples/tunnel_gepa_banking77/README.md +106 -0
- examples/tunnel_gepa_banking77/banking77_gepa_tunnel.toml +95 -0
- examples/tunnel_gepa_banking77/keep_tunnel_running.py +60 -0
- examples/tunnel_gepa_banking77/run_gepa_with_tunnel.sh +226 -0
- examples/vlm/PROPOSAL.md +53 -0
- examples/vlm/README.md +68 -0
- examples/vlm/configs/crafter_vlm_gpt4o.toml +49 -0
- examples/vlm/crafter_image_only_agent.py +207 -0
- examples/vlm/crafter_openai_vlm_agent.py +275 -0
- examples/vlm/filter_image_rows.py +63 -0
- examples/vlm/run_crafter_vlm_benchmark.py +316 -0
- examples/warming_up_to_rl/_utils.py +92 -0
- examples/warming_up_to_rl/analyze_trace_db.py +422 -0
- examples/warming_up_to_rl/configs/crafter_fft.toml +53 -0
- examples/warming_up_to_rl/configs/crafter_fft_4b.toml +54 -0
- examples/warming_up_to_rl/configs/eval_fft_qwen4b.toml +22 -0
- examples/warming_up_to_rl/configs/eval_groq_qwen32b.toml +15 -0
- examples/warming_up_to_rl/configs/eval_modal_qwen4b.toml +24 -0
- examples/warming_up_to_rl/configs/eval_stepwise_complex.toml +35 -0
- examples/warming_up_to_rl/configs/eval_stepwise_consistent.toml +26 -0
- examples/warming_up_to_rl/configs/eval_stepwise_per_achievement.toml +36 -0
- examples/warming_up_to_rl/configs/eval_stepwise_simple.toml +32 -0
- examples/warming_up_to_rl/configs/rl_from_base_qwen4b.toml +85 -0
- examples/warming_up_to_rl/configs/rl_from_ft.toml +58 -0
- examples/warming_up_to_rl/export_trace_sft.py +837 -0
- examples/warming_up_to_rl/groq_test.py +97 -0
- examples/warming_up_to_rl/manage_secrets.py +131 -0
- examples/warming_up_to_rl/old/event_rewards.md +234 -0
- examples/warming_up_to_rl/old/notes.md +73 -0
- examples/warming_up_to_rl/readme.md +110 -0
- examples/warming_up_to_rl/run_eval.py +736 -0
- examples/warming_up_to_rl/run_fft_and_save.py +380 -0
- examples/warming_up_to_rl/run_local_rollout.py +239 -0
- examples/warming_up_to_rl/run_local_rollout_modal.py +248 -0
- examples/warming_up_to_rl/run_local_rollout_parallel.py +405 -0
- examples/warming_up_to_rl/run_local_rollout_traced.py +477 -0
- examples/warming_up_to_rl/run_rl_and_save.py +124 -0
- examples/warming_up_to_rl/run_rollout_remote.py +156 -0
- examples/warming_up_to_rl/task_app/README.md +42 -0
- examples/warming_up_to_rl/task_app/grpo_crafter.py +876 -0
- examples/warming_up_to_rl/task_app/grpo_crafter_task_app.py +135 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/README.md +173 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/__init__.py +5 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/branching.py +143 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/environment_routes.py +1226 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/__init__.py +1 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/crafter/__init__.py +6 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/crafter/app.py +1 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/crafter/environment.py +522 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/crafter/policy.py +454 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/crafter/react_agent.py +108 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/crafter/shared.py +305 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/crafter/tools.py +47 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/hosted_app.py +253 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/inference/__init__.py +5 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/inference/openai_client.py +729 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/main.py +100 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/policy_routes.py +1114 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/registry.py +195 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/rollout.py +1891 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/storage/__init__.py +5 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/storage/volume.py +211 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/test_agents.py +161 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/test_service.py +137 -0
- examples/warming_up_to_rl/task_app/synth_envs_hosted/utils.py +129 -0
- examples/workflows/math_rl/configs/eval_base_qwen.toml +15 -0
- examples/workflows/math_rl/configs/eval_rl_qwen.toml +11 -0
- examples/workflows/math_rl/configs/rl_from_base_qwen.toml +62 -0
- examples/workflows/math_rl/configs/rl_from_base_qwen17.toml +80 -0
- examples/workflows/math_rl/configs/rl_from_ft_qwen.toml +35 -0
- examples/workflows/math_rl/download_dataset.py +80 -0
- examples/workflows/math_rl/run_eval.py +436 -0
- examples/workflows/math_rl/run_rl_and_save.py +111 -0
- synth_ai/__init__.py +47 -23
- synth_ai/_utils/__init__.py +47 -0
- synth_ai/_utils/base_url.py +10 -0
- synth_ai/_utils/http.py +10 -0
- synth_ai/_utils/prompts.py +10 -0
- synth_ai/_utils/task_app_state.py +12 -0
- synth_ai/_utils/user_config.py +10 -0
- synth_ai/api/models/supported.py +514 -0
- synth_ai/api/train/__init__.py +60 -2
- synth_ai/api/train/builders.py +347 -39
- synth_ai/api/train/cli.py +895 -160
- synth_ai/api/train/config_finder.py +103 -25
- synth_ai/api/train/configs/__init__.py +65 -0
- synth_ai/api/train/configs/prompt_learning.py +496 -0
- synth_ai/api/train/configs/rl.py +188 -0
- synth_ai/api/train/configs/sft.py +99 -0
- synth_ai/api/train/configs/shared.py +81 -0
- synth_ai/api/train/env_resolver.py +70 -20
- synth_ai/api/train/pollers.py +29 -4
- synth_ai/api/train/prompt_learning.py +425 -0
- synth_ai/api/train/sft.py +390 -0
- synth_ai/api/train/supported_algos.py +147 -0
- synth_ai/api/train/task_app.py +6 -4
- synth_ai/api/train/utils.py +64 -52
- synth_ai/api/train/validators.py +1117 -0
- synth_ai/api/tunnel.py +49 -0
- synth_ai/auth/credentials.py +94 -0
- synth_ai/baseline/__init__.py +25 -0
- synth_ai/baseline/config.py +209 -0
- synth_ai/baseline/discovery.py +214 -0
- synth_ai/baseline/execution.py +146 -0
- synth_ai/cfgs.py +227 -0
- synth_ai/cli/__init__.py +85 -63
- synth_ai/cli/_modal_wrapper.py +31 -0
- synth_ai/cli/_storage.py +20 -0
- synth_ai/cli/_typer_patch.py +47 -0
- synth_ai/cli/_validate_task_app.py +29 -0
- synth_ai/cli/balance.py +16 -4
- synth_ai/cli/calc.py +36 -21
- synth_ai/cli/claude.py +70 -0
- synth_ai/cli/codex.py +267 -0
- synth_ai/cli/commands/__init__.py +18 -0
- synth_ai/cli/commands/baseline/__init__.py +12 -0
- synth_ai/cli/commands/baseline/core.py +637 -0
- synth_ai/cli/commands/baseline/list.py +93 -0
- synth_ai/cli/commands/demo/__init__.py +6 -0
- synth_ai/cli/commands/demo/core.py +163 -0
- synth_ai/cli/commands/eval/__init__.py +19 -0
- synth_ai/cli/commands/eval/core.py +1112 -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/smoke/__init__.py +7 -0
- synth_ai/cli/commands/smoke/core.py +1437 -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 +22 -0
- synth_ai/cli/commands/status/subcommands/runs.py +81 -0
- synth_ai/cli/commands/status/subcommands/session.py +183 -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 +21 -0
- synth_ai/cli/commands/train/errors.py +117 -0
- synth_ai/cli/commands/train/judge_schemas.py +200 -0
- synth_ai/cli/commands/train/judge_validation.py +305 -0
- synth_ai/cli/commands/train/validation.py +386 -0
- synth_ai/cli/demo.py +32 -140
- synth_ai/cli/deploy.py +233 -0
- synth_ai/cli/eval/__init__.py +36 -0
- synth_ai/cli/eval/core.py +5 -0
- synth_ai/cli/eval/errors.py +31 -0
- synth_ai/cli/eval/validation.py +5 -0
- synth_ai/cli/filter/__init__.py +28 -0
- synth_ai/cli/filter/core.py +5 -0
- synth_ai/cli/filter/errors.py +23 -0
- synth_ai/cli/filter/validation.py +5 -0
- synth_ai/cli/legacy_root_backup.py +28 -22
- synth_ai/cli/lib/__init__.py +10 -0
- synth_ai/cli/lib/task_app_discovery.py +7 -0
- synth_ai/cli/lib/task_app_env.py +518 -0
- synth_ai/cli/mcp.py +34 -0
- synth_ai/cli/modal_serve/__init__.py +12 -0
- synth_ai/cli/modal_serve/core.py +14 -0
- synth_ai/cli/modal_serve/errors.py +8 -0
- synth_ai/cli/modal_serve/validation.py +11 -0
- synth_ai/cli/opencode.py +256 -0
- synth_ai/cli/recent.py +13 -7
- synth_ai/cli/rl_demo.py +156 -116
- synth_ai/cli/root.py +131 -132
- synth_ai/cli/serve/__init__.py +12 -0
- synth_ai/cli/serve/core.py +14 -0
- synth_ai/cli/serve/errors.py +8 -0
- synth_ai/cli/serve/validation.py +11 -0
- synth_ai/cli/setup.py +49 -0
- synth_ai/cli/status.py +7 -125
- synth_ai/cli/task_app_deploy.py +7 -0
- synth_ai/cli/task_app_list.py +25 -0
- synth_ai/cli/task_app_modal_serve.py +11 -0
- synth_ai/cli/task_app_serve.py +11 -0
- synth_ai/cli/task_apps.py +2284 -257
- synth_ai/cli/traces.py +9 -5
- synth_ai/cli/train/__init__.py +12 -0
- synth_ai/cli/train/core.py +21 -0
- synth_ai/cli/train/errors.py +8 -0
- synth_ai/cli/train/validation.py +24 -0
- synth_ai/cli/train.py +5 -0
- synth_ai/cli/turso.py +73 -0
- synth_ai/cli/watch.py +13 -18
- synth_ai/demos/__init__.py +10 -0
- synth_ai/demos/core/__init__.py +28 -1
- synth_ai/demos/core/cli.py +579 -291
- synth_ai/demos/crafter/__init__.py +1 -0
- synth_ai/demos/crafter/crafter_fft_4b.toml +55 -0
- synth_ai/demos/crafter/grpo_crafter_task_app.py +185 -0
- synth_ai/demos/crafter/rl_from_base_qwen4b.toml +74 -0
- synth_ai/demos/demo_registry.py +176 -0
- synth_ai/demos/demo_task_apps/__init__.py +3 -3
- synth_ai/demos/demo_task_apps/core.py +64 -28
- synth_ai/demos/demo_task_apps/crafter/__init__.py +1 -0
- synth_ai/demos/demo_task_apps/crafter/configs/crafter_fft_4b.toml +53 -0
- synth_ai/demos/demo_task_apps/crafter/configs/rl_from_base_qwen4b.toml +73 -0
- synth_ai/demos/demo_task_apps/crafter/grpo_crafter_task_app.py +184 -0
- synth_ai/demos/demo_task_apps/math/_common.py +1 -2
- synth_ai/demos/demo_task_apps/math/app.py +2 -1
- synth_ai/demos/demo_task_apps/math/deploy_modal.py +3 -6
- synth_ai/demos/demo_task_apps/math/modal_task_app.py +185 -83
- synth_ai/demos/demo_task_apps/math/task_app_entry.py +0 -2
- synth_ai/demos/math/__init__.py +1 -0
- synth_ai/demos/math/_common.py +16 -0
- synth_ai/demos/math/app.py +38 -0
- synth_ai/demos/math/config.toml +76 -0
- synth_ai/demos/math/deploy_modal.py +54 -0
- synth_ai/demos/math/modal_task_app.py +703 -0
- synth_ai/demos/math/task_app_entry.py +51 -0
- synth_ai/environments/environment/core.py +7 -1
- synth_ai/environments/examples/bandit/engine.py +12 -5
- synth_ai/environments/examples/bandit/environment.py +0 -1
- synth_ai/environments/examples/bandit/taskset.py +4 -4
- synth_ai/environments/examples/crafter_classic/engine_deterministic_patch.py +7 -4
- synth_ai/environments/examples/crafter_classic/engine_serialization_patch_v3.py +9 -5
- synth_ai/environments/examples/crafter_classic/environment.py +93 -2
- synth_ai/environments/examples/crafter_classic/world_config_patch_simple.py +4 -3
- synth_ai/environments/examples/enron/engine.py +7 -2
- synth_ai/environments/examples/enron/environment.py +68 -0
- synth_ai/environments/examples/red/engine.py +60 -12
- synth_ai/environments/examples/red/engine_helpers/memory_map.py +7 -0
- synth_ai/environments/examples/red/engine_helpers/reward_components.py +151 -179
- synth_ai/environments/examples/red/engine_helpers/reward_library/pallet_town_progression.py +477 -0
- synth_ai/environments/examples/red/engine_helpers/state_extraction.py +32 -0
- synth_ai/environments/examples/red/environment.py +86 -0
- synth_ai/environments/examples/red/trace_hooks_v3.py +168 -0
- synth_ai/environments/examples/sokoban/taskset.py +116 -0
- synth_ai/environments/examples/verilog/engine.py +104 -12
- synth_ai/environments/examples/wordle/environment.py +0 -1
- synth_ai/environments/reproducibility/tree.py +5 -6
- synth_ai/environments/service/app.py +11 -12
- synth_ai/environments/service/core_routes.py +10 -9
- synth_ai/environments/stateful/engine.py +1 -1
- synth_ai/environments/tasks/core.py +1 -0
- synth_ai/environments/tasks/filters.py +5 -6
- synth_ai/environments/tasks/utils.py +4 -5
- synth_ai/evals/__init__.py +15 -0
- synth_ai/evals/base.py +14 -5
- synth_ai/evals/client.py +82 -0
- synth_ai/evals/types.py +42 -0
- synth_ai/http.py +8 -22
- synth_ai/http_client.py +45 -12
- synth_ai/inference/__init__.py +0 -2
- synth_ai/inference/client.py +21 -7
- synth_ai/jobs/client.py +129 -80
- synth_ai/judge_schemas.py +127 -0
- synth_ai/learning/__init__.py +51 -6
- synth_ai/learning/algorithms.py +14 -0
- synth_ai/learning/client.py +122 -30
- synth_ai/learning/config.py +2 -40
- synth_ai/learning/constants.py +0 -2
- synth_ai/learning/ft_client.py +4 -56
- synth_ai/learning/health.py +14 -8
- synth_ai/learning/jobs.py +43 -47
- synth_ai/learning/prompt_learning_client.py +276 -0
- synth_ai/learning/prompt_learning_types.py +185 -0
- synth_ai/{rl → learning/rl}/__init__.py +14 -5
- synth_ai/learning/rl/client.py +269 -0
- synth_ai/learning/rl/config.py +31 -0
- synth_ai/{rl → learning/rl}/contracts.py +5 -10
- synth_ai/{rl → learning/rl}/env_keys.py +45 -16
- synth_ai/learning/rl/secrets.py +13 -0
- synth_ai/learning/rl_client.py +2 -253
- synth_ai/learning/sft/__init__.py +29 -0
- synth_ai/learning/sft/client.py +68 -0
- synth_ai/learning/sft/config.py +270 -0
- synth_ai/learning/sft/data.py +698 -0
- synth_ai/learning/sse.py +25 -26
- synth_ai/learning/validators.py +29 -25
- synth_ai/mcp/__init__.py +5 -0
- synth_ai/mcp/__main__.py +8 -0
- synth_ai/mcp/main.py +254 -0
- synth_ai/mcp/setup.py +100 -0
- synth_ai/modal.py +257 -0
- synth_ai/pricing/__init__.py +3 -0
- synth_ai/pricing/model_pricing.py +64 -0
- synth_ai/session/__init__.py +75 -0
- synth_ai/session/client.py +383 -0
- synth_ai/session/constants.py +63 -0
- synth_ai/session/exceptions.py +105 -0
- synth_ai/session/manager.py +139 -0
- synth_ai/session/models.py +89 -0
- synth_ai/session/query.py +110 -0
- synth_ai/spec/__init__.py +46 -0
- synth_ai/spec/dataclasses.py +149 -0
- synth_ai/spec/loader.py +144 -0
- synth_ai/spec/serializer.py +199 -0
- synth_ai/spec/validation.py +250 -0
- synth_ai/streaming/__init__.py +29 -0
- synth_ai/streaming/config.py +94 -0
- synth_ai/streaming/handlers.py +589 -0
- synth_ai/streaming/streamer.py +320 -0
- synth_ai/streaming/types.py +95 -0
- synth_ai/task/__init__.py +50 -30
- synth_ai/task/apps/__init__.py +63 -19
- synth_ai/task/auth.py +35 -23
- synth_ai/task/client.py +15 -13
- synth_ai/task/config.py +261 -0
- synth_ai/task/contracts.py +165 -64
- synth_ai/task/datasets.py +9 -6
- synth_ai/task/errors.py +11 -10
- synth_ai/task/health.py +17 -11
- synth_ai/task/inference_api.py +101 -0
- synth_ai/task/json.py +58 -24
- synth_ai/task/proxy.py +59 -66
- synth_ai/task/rubrics/__init__.py +55 -0
- synth_ai/task/rubrics/loaders.py +156 -0
- synth_ai/task/rubrics/models.py +57 -0
- synth_ai/task/rubrics/scoring.py +116 -0
- synth_ai/task/rubrics/strict.py +149 -0
- synth_ai/task/rubrics.py +22 -15
- synth_ai/task/server.py +65 -31
- synth_ai/task/trace_correlation_helpers.py +328 -0
- synth_ai/task/tracing_utils.py +44 -28
- synth_ai/task/validators.py +449 -6
- synth_ai/task/vendors.py +5 -7
- synth_ai/tracing_v3/__init__.py +4 -0
- synth_ai/tracing_v3/abstractions.py +21 -4
- synth_ai/tracing_v3/config.py +167 -22
- synth_ai/tracing_v3/constants.py +21 -0
- synth_ai/tracing_v3/db_config.py +42 -29
- synth_ai/tracing_v3/decorators.py +80 -45
- synth_ai/tracing_v3/examples/basic_usage.py +15 -9
- synth_ai/tracing_v3/hooks.py +6 -4
- synth_ai/tracing_v3/llm_call_record_helpers.py +161 -61
- synth_ai/tracing_v3/migration_helper.py +1 -2
- synth_ai/tracing_v3/replica_sync.py +12 -7
- synth_ai/tracing_v3/serialization.py +130 -0
- synth_ai/tracing_v3/session_tracer.py +73 -16
- synth_ai/tracing_v3/storage/base.py +89 -1
- synth_ai/tracing_v3/storage/config.py +63 -16
- synth_ai/tracing_v3/storage/factory.py +11 -9
- synth_ai/tracing_v3/storage/utils.py +15 -11
- synth_ai/tracing_v3/trace_utils.py +317 -0
- synth_ai/tracing_v3/turso/__init__.py +8 -21
- synth_ai/tracing_v3/turso/daemon.py +123 -15
- synth_ai/tracing_v3/turso/models.py +5 -2
- synth_ai/tracing_v3/turso/native_manager.py +1293 -0
- synth_ai/tracing_v3/utils.py +5 -4
- synth_ai/tunnel.py +143 -0
- synth_ai/tunnel_deploy.py +278 -0
- synth_ai/types.py +8 -0
- synth_ai/urls.py +11 -0
- synth_ai/utils/__init__.py +166 -0
- synth_ai/utils/agents.py +74 -0
- synth_ai/utils/apps.py +152 -0
- synth_ai/utils/base_url.py +94 -0
- synth_ai/utils/bin.py +39 -0
- synth_ai/utils/claude.py +36 -0
- synth_ai/utils/cli.py +284 -0
- synth_ai/utils/config.py +81 -0
- synth_ai/utils/env.py +346 -0
- synth_ai/utils/errors.py +85 -0
- synth_ai/utils/http.py +172 -0
- synth_ai/utils/json.py +72 -0
- synth_ai/utils/log_filter.py +99 -0
- synth_ai/utils/logging.py +198 -0
- synth_ai/utils/modal.py +299 -0
- synth_ai/utils/paths.py +95 -0
- synth_ai/utils/process.py +233 -0
- synth_ai/utils/prompts.py +39 -0
- synth_ai/utils/sqld.py +122 -0
- synth_ai/utils/ssl.py +25 -0
- synth_ai/utils/task_app_discovery.py +882 -0
- synth_ai/utils/task_app_env.py +186 -0
- synth_ai/utils/task_app_state.py +318 -0
- synth_ai/utils/tunnel/__init__.py +12 -0
- synth_ai/utils/tunnel/config.py +55 -0
- synth_ai/utils/user_config.py +137 -0
- synth_ai/uvicorn.py +77 -0
- synth_ai-0.2.23.dev3.dist-info/METADATA +357 -0
- synth_ai-0.2.23.dev3.dist-info/RECORD +983 -0
- {synth_ai-0.2.9.dev0.dist-info → synth_ai-0.2.23.dev3.dist-info}/entry_points.txt +0 -1
- {synth_ai-0.2.9.dev0.dist-info → synth_ai-0.2.23.dev3.dist-info}/top_level.txt +1 -0
- synth_ai/cli/man.py +0 -106
- synth_ai/core/experiment.py +0 -15
- synth_ai/core/system.py +0 -15
- synth_ai/demo_registry.py +0 -258
- synth_ai/environments/examples/sokoban/units/astar_common.py +0 -95
- synth_ai/experimental/synth_oss.py +0 -446
- synth_ai/handshake.py +0 -107
- synth_ai/install_sqld.sh +0 -40
- synth_ai/learning/offline/dpo.py +0 -0
- synth_ai/learning/offline/providers.py +0 -7
- synth_ai/learning/offline/sft.py +0 -0
- synth_ai/learning/offline/shared.py +0 -0
- synth_ai/learning/online/grpo.py +0 -0
- synth_ai/learning/online/irft.py +0 -0
- synth_ai/learning/prompts/banking77_injection_eval.py +0 -168
- synth_ai/learning/prompts/gepa.py +0 -0
- synth_ai/learning/prompts/hello_world_in_context_injection_ex.py +0 -213
- synth_ai/learning/prompts/mipro.py +0 -289
- synth_ai/learning/prompts/random_search.py +0 -246
- synth_ai/learning/prompts/run_mipro_banking77.py +0 -172
- synth_ai/learning/prompts/run_random_search_banking77.py +0 -324
- synth_ai/lm/__init__.py +0 -51
- synth_ai/lm/caching/constants.py +0 -6
- synth_ai/lm/caching/dbs.py +0 -0
- synth_ai/lm/caching/ephemeral.py +0 -102
- synth_ai/lm/caching/handler.py +0 -137
- synth_ai/lm/caching/initialize.py +0 -11
- synth_ai/lm/caching/persistent.py +0 -114
- synth_ai/lm/config.py +0 -110
- synth_ai/lm/constants.py +0 -32
- synth_ai/lm/core/__init__.py +0 -8
- synth_ai/lm/core/all.py +0 -73
- synth_ai/lm/core/exceptions.py +0 -7
- synth_ai/lm/core/main.py +0 -319
- synth_ai/lm/core/main_v3.py +0 -594
- synth_ai/lm/core/synth_models.py +0 -48
- synth_ai/lm/core/vendor_clients.py +0 -188
- synth_ai/lm/cost/monitor.py +0 -1
- synth_ai/lm/cost/statefulness.py +0 -1
- synth_ai/lm/injection.py +0 -80
- synth_ai/lm/overrides.py +0 -206
- synth_ai/lm/provider_support/__init__.py +0 -8
- synth_ai/lm/provider_support/anthropic.py +0 -972
- synth_ai/lm/provider_support/openai.py +0 -1139
- synth_ai/lm/provider_support/suppress_logging.py +0 -31
- synth_ai/lm/structured_outputs/handler.py +0 -440
- synth_ai/lm/structured_outputs/inject.py +0 -297
- synth_ai/lm/structured_outputs/rehabilitate.py +0 -185
- synth_ai/lm/tools/__init__.py +0 -3
- synth_ai/lm/tools/base.py +0 -172
- synth_ai/lm/unified_interface.py +0 -202
- synth_ai/lm/vendors/base.py +0 -81
- synth_ai/lm/vendors/core/anthropic_api.py +0 -387
- synth_ai/lm/vendors/core/gemini_api.py +0 -292
- synth_ai/lm/vendors/core/mistral_api.py +0 -322
- synth_ai/lm/vendors/core/openai_api.py +0 -225
- synth_ai/lm/vendors/core/synth_dev_api.py +0 -0
- synth_ai/lm/vendors/local/ollama.py +0 -0
- synth_ai/lm/vendors/openai_standard.py +0 -780
- synth_ai/lm/vendors/openai_standard_responses.py +0 -256
- synth_ai/lm/vendors/retries.py +0 -22
- synth_ai/lm/vendors/supported/custom_endpoint.py +0 -417
- synth_ai/lm/vendors/supported/deepseek.py +0 -69
- synth_ai/lm/vendors/supported/grok.py +0 -75
- synth_ai/lm/vendors/supported/groq.py +0 -16
- synth_ai/lm/vendors/supported/ollama.py +0 -15
- synth_ai/lm/vendors/supported/openrouter.py +0 -74
- synth_ai/lm/vendors/supported/together.py +0 -11
- synth_ai/lm/vendors/synth_client.py +0 -808
- synth_ai/lm/warmup.py +0 -186
- synth_ai/rl/secrets.py +0 -19
- synth_ai/scripts/verify_rewards.py +0 -100
- synth_ai/task/apps/grpo_crafter.py +0 -438
- synth_ai/tracing/__init__.py +0 -30
- synth_ai/tracing_v1/__init__.py +0 -33
- synth_ai/tracing_v3/turso/manager.py +0 -774
- 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 -512
- 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/zyk/__init__.py +0 -30
- synth_ai-0.2.9.dev0.dist-info/METADATA +0 -131
- synth_ai-0.2.9.dev0.dist-info/RECORD +0 -444
- {synth_ai/lm/caching → examples/task_apps}/__init__.py +0 -0
- {synth_ai/lm/cost → examples/task_apps/crafter}/__init__.py +0 -0
- {synth_ai/lm/structured_outputs → examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/server}/__init__.py +0 -0
- {synth_ai/lm/vendors → examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/tests}/__init__.py +0 -0
- {synth_ai/lm/vendors/core → examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/utils}/__init__.py +0 -0
- {synth_ai/lm/vendors/local → examples/task_apps/math}/__init__.py +0 -0
- {synth_ai/lm/vendors/supported → examples/workflows}/__init__.py +0 -0
- {synth_ai/v0/tracing → examples/workflows/math_rl}/__init__.py +0 -0
- /synth_ai/{compound/cais.py → cli/__main__.py} +0 -0
- /synth_ai/{learning/filtering.py → py.typed} +0 -0
- {synth_ai-0.2.9.dev0.dist-info → synth_ai-0.2.23.dev3.dist-info}/WHEEL +0 -0
- {synth_ai-0.2.9.dev0.dist-info → synth_ai-0.2.23.dev3.dist-info}/licenses/LICENSE +0 -0
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
# MIPROv2 Prompt Learning for Banking77
|
|
2
|
+
# Local backend configuration targeting the Banking77 intent classification task app.
|
|
3
|
+
|
|
4
|
+
[prompt_learning]
|
|
5
|
+
algorithm = "mipro"
|
|
6
|
+
task_app_url = "https://synth-laboratories-dev--synth-banking77-web-web.modal.run"
|
|
7
|
+
task_app_id = "banking77"
|
|
8
|
+
|
|
9
|
+
# Initial prompt pattern (pattern-based mode)
|
|
10
|
+
[prompt_learning.initial_prompt]
|
|
11
|
+
id = "banking77_pattern"
|
|
12
|
+
name = "Banking77 Classification Pattern"
|
|
13
|
+
|
|
14
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
15
|
+
role = "system"
|
|
16
|
+
pattern = "You are an expert banking assistant that classifies customer queries into banking intents. Given a customer message, respond with exactly one intent label from the provided list using the `banking77_classify` tool."
|
|
17
|
+
order = 0
|
|
18
|
+
|
|
19
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
20
|
+
role = "user"
|
|
21
|
+
pattern = "Customer Query: {query}\n\nClassify this query into one of the banking intents using the tool call."
|
|
22
|
+
order = 1
|
|
23
|
+
|
|
24
|
+
[prompt_learning.initial_prompt.wildcards]
|
|
25
|
+
query = "REQUIRED" # Will be provided by task app at runtime
|
|
26
|
+
|
|
27
|
+
# Policy configuration (model, provider, etc.)
|
|
28
|
+
[prompt_learning.policy]
|
|
29
|
+
inference_mode = "synth_hosted"
|
|
30
|
+
model = "openai/gpt-oss-20b"
|
|
31
|
+
provider = "groq"
|
|
32
|
+
temperature = 0.0
|
|
33
|
+
max_completion_tokens = 128
|
|
34
|
+
policy_name = "banking77-mipro" # Required for Banking77 task app
|
|
35
|
+
|
|
36
|
+
# Training split config
|
|
37
|
+
[prompt_learning.env_config]
|
|
38
|
+
pool = "train"
|
|
39
|
+
|
|
40
|
+
# MIPROv2-specific configuration
|
|
41
|
+
[prompt_learning.mipro]
|
|
42
|
+
env_name = "banking77"
|
|
43
|
+
num_iterations = 5 # 5 iterations × 2 per iteration = 10 total prompt combos
|
|
44
|
+
num_evaluations_per_iteration = 2 # 2 proposals per iteration
|
|
45
|
+
batch_size = 6
|
|
46
|
+
max_concurrent = 16
|
|
47
|
+
meta_model = "gpt-4o-mini"
|
|
48
|
+
meta_model_provider = "openai"
|
|
49
|
+
few_shot_score_threshold = 0.85
|
|
50
|
+
max_instructions = 3
|
|
51
|
+
|
|
52
|
+
# Retry guard for duplicate proposals/candidates
|
|
53
|
+
duplicate_retry_limit = 10
|
|
54
|
+
|
|
55
|
+
# Seed pools for different phases - tons of data for reliable evaluation
|
|
56
|
+
bootstrap_train_seeds = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14] # 15 seeds for bootstrap
|
|
57
|
+
online_pool = [15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39] # 25 seeds for online evaluation
|
|
58
|
+
test_pool = [40, 41, 42, 43, 44, 45, 46, 47, 48, 49] # 10 held-out seeds for final evaluation
|
|
59
|
+
|
|
60
|
+
# TPE hyperparameters
|
|
61
|
+
[prompt_learning.mipro.tpe]
|
|
62
|
+
gamma = 0.25 # Quantile for splitting good/bad observations
|
|
63
|
+
n_candidates = 32 # Number of candidates to evaluate Expected Improvement
|
|
64
|
+
n_startup_trials = 10 # Uniform random trials before TPE kicks in
|
|
65
|
+
epsilon = 0.1 # Exploration probability (uniform sample)
|
|
66
|
+
categorical_alpha = 1.0 # Laplace smoothing for categorical variables
|
|
67
|
+
prior_weight = 0.05 # Prior mixing weight to avoid zero-densities
|
|
68
|
+
|
|
69
|
+
# Demo set selection hyperparameters
|
|
70
|
+
[prompt_learning.mipro.demo]
|
|
71
|
+
max_few_shot_examples = 5 # Maximum number of few-shot examples to include
|
|
72
|
+
sets_per_size = 6 # Number of demo sets to generate per size
|
|
73
|
+
include_empty = true # Whether to include empty demo set (zero-shot)
|
|
74
|
+
|
|
75
|
+
# Grounding configuration for instruction proposals
|
|
76
|
+
[prompt_learning.mipro.grounding]
|
|
77
|
+
n = 10 # Number of instruction proposals to generate
|
|
78
|
+
temperature = 0.7 # Temperature for LLM proposal generation
|
|
79
|
+
max_tokens = 600 # Max tokens for proposal generation
|
|
80
|
+
|
|
81
|
+
# Meta-update configuration
|
|
82
|
+
[prompt_learning.mipro.meta_update]
|
|
83
|
+
enabled = true # Enable periodic meta-updates
|
|
84
|
+
every_iterations = 3 # Regenerate instructions every N iterations
|
|
85
|
+
topk_success = 5 # Number of top success examples to use for grounding
|
|
86
|
+
topk_failure = 5 # Number of top failure examples to use for grounding
|
|
87
|
+
validate_on_batch = 32 # Batch size for proposal validation
|
|
88
|
+
keep_k = 12 # Maximum number of instruction variants to keep
|
|
89
|
+
dedup_token_overlap = 0.8 # Token overlap threshold for deduplication
|
|
90
|
+
regen_temperature_decay = 0.95 # Temperature decay factor per meta-update
|
|
91
|
+
|
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
# MIPROv2 Prompt Learning for Banking77
|
|
2
|
+
# Test configuration with reduced iterations for faster testing
|
|
3
|
+
|
|
4
|
+
[prompt_learning]
|
|
5
|
+
algorithm = "mipro"
|
|
6
|
+
task_app_url = "https://synth-laboratories-dev--synth-banking77-web-web.modal.run"
|
|
7
|
+
task_app_id = "banking77"
|
|
8
|
+
|
|
9
|
+
# Initial prompt pattern (pattern-based mode)
|
|
10
|
+
[prompt_learning.initial_prompt]
|
|
11
|
+
id = "banking77_pattern"
|
|
12
|
+
name = "Banking77 Classification Pattern"
|
|
13
|
+
|
|
14
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
15
|
+
role = "system"
|
|
16
|
+
pattern = "You are an expert banking assistant that classifies customer queries into banking intents. Given a customer message, respond with exactly one intent label from the provided list using the `banking77_classify` tool."
|
|
17
|
+
order = 0
|
|
18
|
+
|
|
19
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
20
|
+
role = "user"
|
|
21
|
+
pattern = "Customer Query: {query}\n\nClassify this query into one of the banking intents using the tool call."
|
|
22
|
+
order = 1
|
|
23
|
+
|
|
24
|
+
[prompt_learning.initial_prompt.wildcards]
|
|
25
|
+
query = "REQUIRED"
|
|
26
|
+
|
|
27
|
+
# Policy configuration
|
|
28
|
+
[prompt_learning.policy]
|
|
29
|
+
inference_mode = "synth_hosted"
|
|
30
|
+
model = "openai/gpt-oss-20b"
|
|
31
|
+
provider = "groq"
|
|
32
|
+
temperature = 0.0
|
|
33
|
+
max_completion_tokens = 128
|
|
34
|
+
policy_name = "banking77-mipro"
|
|
35
|
+
|
|
36
|
+
# Training split config
|
|
37
|
+
[prompt_learning.env_config]
|
|
38
|
+
pool = "train"
|
|
39
|
+
|
|
40
|
+
# MIPROv2-specific configuration (reduced for testing)
|
|
41
|
+
[prompt_learning.mipro]
|
|
42
|
+
env_name = "banking77"
|
|
43
|
+
num_iterations = 10 # Reduced for testing (default: 16)
|
|
44
|
+
num_evaluations_per_iteration = 4 # Reduced for testing (default: 6)
|
|
45
|
+
batch_size = 4
|
|
46
|
+
max_concurrent = 10
|
|
47
|
+
meta_model = "gpt-4o-mini"
|
|
48
|
+
meta_model_provider = "openai"
|
|
49
|
+
few_shot_score_threshold = 0.80 # Lower threshold for testing (default: 0.85)
|
|
50
|
+
|
|
51
|
+
# Seed pools for different phases
|
|
52
|
+
bootstrap_train_seeds = [0, 1, 2, 3, 4] # 5 seeds for bootstrap phase
|
|
53
|
+
online_pool = [5, 6, 7, 8, 9] # 5 seeds for online evaluation
|
|
54
|
+
test_pool = [10, 11, 12, 13, 14] # 5 seeds for final test
|
|
55
|
+
|
|
56
|
+
# TPE hyperparameters
|
|
57
|
+
[prompt_learning.mipro.tpe]
|
|
58
|
+
gamma = 0.25 # Quantile for splitting good/bad observations
|
|
59
|
+
n_candidates = 16 # Number of candidates to evaluate Expected Improvement (reduced for testing)
|
|
60
|
+
n_startup_trials = 5 # Uniform random trials before TPE kicks in
|
|
61
|
+
epsilon = 0.1 # Exploration probability (uniform sample)
|
|
62
|
+
categorical_alpha = 1.0 # Laplace smoothing for categorical variables
|
|
63
|
+
prior_weight = 0.05 # Prior mixing weight to avoid zero-densities
|
|
64
|
+
|
|
65
|
+
# Demo set selection hyperparameters
|
|
66
|
+
[prompt_learning.mipro.demo]
|
|
67
|
+
max_few_shot_examples = 5 # Maximum number of few-shot examples to include
|
|
68
|
+
sets_per_size = 3 # Number of demo sets to generate per size (reduced for testing)
|
|
69
|
+
include_empty = true # Whether to include empty demo set (zero-shot)
|
|
70
|
+
|
|
71
|
+
# Grounding configuration for instruction proposals
|
|
72
|
+
[prompt_learning.mipro.grounding]
|
|
73
|
+
n = 8 # Number of instruction proposals to generate (reduced for testing)
|
|
74
|
+
temperature = 0.7 # Temperature for LLM proposal generation
|
|
75
|
+
max_tokens = 600 # Max tokens for proposal generation
|
|
76
|
+
|
|
77
|
+
# Meta-update configuration
|
|
78
|
+
[prompt_learning.mipro.meta_update]
|
|
79
|
+
enabled = true # Enable periodic meta-updates
|
|
80
|
+
every_iterations = 3 # Regenerate instructions every N iterations
|
|
81
|
+
topk_success = 3 # Number of top success examples to use for grounding (reduced for testing)
|
|
82
|
+
topk_failure = 3 # Number of top failure examples to use for grounding (reduced for testing)
|
|
83
|
+
validate_on_batch = 16 # Batch size for proposal validation (reduced for testing)
|
|
84
|
+
keep_k = 8 # Maximum number of instruction variants to keep (reduced for testing)
|
|
85
|
+
dedup_token_overlap = 0.8 # Token overlap threshold for deduplication
|
|
86
|
+
regen_temperature_decay = 0.95 # Temperature decay factor per meta-update
|
|
87
|
+
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
# MIPROv2 Prompt Learning for the Banking77 Two-Step Pipeline
|
|
2
|
+
# Using gemini-2.5-flash-lite as the policy model
|
|
3
|
+
|
|
4
|
+
[prompt_learning]
|
|
5
|
+
algorithm = "mipro"
|
|
6
|
+
task_app_url = "https://synth-laboratories-dev--synth-banking77-pipeline-web-web.modal.run"
|
|
7
|
+
task_app_id = "banking77-pipeline"
|
|
8
|
+
|
|
9
|
+
[prompt_learning.initial_prompt]
|
|
10
|
+
id = "banking77_pipeline_pattern"
|
|
11
|
+
name = "Banking77 Pipeline Pattern"
|
|
12
|
+
|
|
13
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
14
|
+
role = "system"
|
|
15
|
+
pattern = "Pipeline placeholder message. Actual classifier/calibrator instructions are provided via metadata.pipeline_modules."
|
|
16
|
+
order = 0
|
|
17
|
+
|
|
18
|
+
[prompt_learning.initial_prompt.metadata]
|
|
19
|
+
pipeline_modules = [
|
|
20
|
+
{ name = "classifier", instruction_text = "You are an expert banking assistant. Given a customer query, choose the correct Banking77 intent and call the banking77_classify tool with that label.", few_shots = [] },
|
|
21
|
+
{ name = "calibrator", instruction_text = "You review the classifier's suggestion. Confirm it if appropriate; otherwise pick the best Banking77 intent and return it via the banking77_classify tool.", few_shots = [] }
|
|
22
|
+
]
|
|
23
|
+
|
|
24
|
+
[prompt_learning.policy]
|
|
25
|
+
inference_mode = "synth_hosted"
|
|
26
|
+
model = "gemini-2.5-flash-lite"
|
|
27
|
+
provider = "google"
|
|
28
|
+
temperature = 0.0
|
|
29
|
+
max_completion_tokens = 128
|
|
30
|
+
policy_name = "banking77-pipeline"
|
|
31
|
+
|
|
32
|
+
[prompt_learning.env_config]
|
|
33
|
+
pool = "train"
|
|
34
|
+
|
|
35
|
+
[prompt_learning.mipro]
|
|
36
|
+
env_name = "banking77_pipeline"
|
|
37
|
+
num_iterations = 5
|
|
38
|
+
num_evaluations_per_iteration = 2
|
|
39
|
+
batch_size = 6
|
|
40
|
+
max_concurrent = 16
|
|
41
|
+
meta_model = "gpt-4o-mini"
|
|
42
|
+
meta_model_provider = "openai"
|
|
43
|
+
few_shot_score_threshold = 0.85
|
|
44
|
+
max_instructions = 3
|
|
45
|
+
duplicate_retry_limit = 10
|
|
46
|
+
bootstrap_train_seeds = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
|
|
47
|
+
online_pool = [15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39]
|
|
48
|
+
test_pool = [40, 41, 42, 43, 44, 45, 46, 47, 48, 49]
|
|
49
|
+
# Reference pool: Used to generate rich dataset context (~50k tokens) in the meta-prompt
|
|
50
|
+
# Must not overlap with bootstrap/online/test pools to avoid data leakage
|
|
51
|
+
reference_pool = [50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149]
|
|
52
|
+
|
|
53
|
+
# System specification (pipeline-level, covers both analyzer and classifier stages)
|
|
54
|
+
spec_path = "examples/task_apps/banking77_pipeline/banking77_pipeline_spec.json"
|
|
55
|
+
spec_max_tokens = 5000
|
|
56
|
+
spec_include_examples = true
|
|
57
|
+
spec_priority_threshold = 8 # Only include high-priority rules (8+)
|
|
58
|
+
|
|
59
|
+
[prompt_learning.mipro.tpe]
|
|
60
|
+
gamma = 0.25
|
|
61
|
+
n_candidates = 32
|
|
62
|
+
n_startup_trials = 10
|
|
63
|
+
epsilon = 0.1
|
|
64
|
+
categorical_alpha = 1.0
|
|
65
|
+
prior_weight = 0.05
|
|
66
|
+
|
|
67
|
+
[prompt_learning.mipro.demo]
|
|
68
|
+
max_few_shot_examples = 5
|
|
69
|
+
sets_per_size = 6
|
|
70
|
+
include_empty = true
|
|
71
|
+
|
|
72
|
+
[prompt_learning.mipro.grounding]
|
|
73
|
+
n = 10
|
|
74
|
+
temperature = 0.7
|
|
75
|
+
max_tokens = 600
|
|
76
|
+
|
|
77
|
+
[prompt_learning.mipro.meta_update]
|
|
78
|
+
enabled = true
|
|
79
|
+
every_iterations = 3
|
|
80
|
+
topk_success = 5
|
|
81
|
+
topk_failure = 5
|
|
82
|
+
validate_on_batch = 32
|
|
83
|
+
keep_k = 12
|
|
84
|
+
dedup_token_overlap = 0.8
|
|
85
|
+
regen_temperature_decay = 0.95
|
|
86
|
+
|
|
87
|
+
[[prompt_learning.mipro.modules]]
|
|
88
|
+
module_id = "classifier"
|
|
89
|
+
max_instruction_slots = 3
|
|
90
|
+
max_demo_slots = 5
|
|
91
|
+
|
|
92
|
+
[[prompt_learning.mipro.modules]]
|
|
93
|
+
module_id = "calibrator"
|
|
94
|
+
max_instruction_slots = 3
|
|
95
|
+
max_demo_slots = 5
|
|
96
|
+
|
|
97
|
+
|
|
98
|
+
|
|
@@ -0,0 +1,96 @@
|
|
|
1
|
+
# MIPROv2 Prompt Learning for the Banking77 Two-Step Pipeline
|
|
2
|
+
# Using gpt-4.1-mini as the policy model
|
|
3
|
+
|
|
4
|
+
[prompt_learning]
|
|
5
|
+
algorithm = "mipro"
|
|
6
|
+
task_app_url = "https://synth-laboratories-dev--synth-banking77-pipeline-web-web.modal.run"
|
|
7
|
+
task_app_id = "banking77-pipeline"
|
|
8
|
+
|
|
9
|
+
[prompt_learning.initial_prompt]
|
|
10
|
+
id = "banking77_pipeline_pattern"
|
|
11
|
+
name = "Banking77 Pipeline Pattern"
|
|
12
|
+
|
|
13
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
14
|
+
role = "system"
|
|
15
|
+
pattern = "Pipeline placeholder message. Actual classifier/calibrator instructions are provided via metadata.pipeline_modules."
|
|
16
|
+
order = 0
|
|
17
|
+
|
|
18
|
+
[prompt_learning.initial_prompt.metadata]
|
|
19
|
+
pipeline_modules = [
|
|
20
|
+
{ name = "classifier", instruction_text = "You are an expert banking assistant. Given a customer query, choose the correct Banking77 intent and call the banking77_classify tool with that label.", few_shots = [] },
|
|
21
|
+
{ name = "calibrator", instruction_text = "You review the classifier's suggestion. Confirm it if appropriate; otherwise pick the best Banking77 intent and return it via the banking77_classify tool.", few_shots = [] }
|
|
22
|
+
]
|
|
23
|
+
|
|
24
|
+
[prompt_learning.policy]
|
|
25
|
+
inference_mode = "synth_hosted"
|
|
26
|
+
model = "gpt-4.1-mini"
|
|
27
|
+
provider = "openai"
|
|
28
|
+
temperature = 0.0
|
|
29
|
+
max_completion_tokens = 128
|
|
30
|
+
policy_name = "banking77-pipeline"
|
|
31
|
+
|
|
32
|
+
[prompt_learning.env_config]
|
|
33
|
+
pool = "train"
|
|
34
|
+
|
|
35
|
+
[prompt_learning.mipro]
|
|
36
|
+
env_name = "banking77_pipeline"
|
|
37
|
+
num_iterations = 5
|
|
38
|
+
num_evaluations_per_iteration = 2
|
|
39
|
+
batch_size = 6
|
|
40
|
+
max_concurrent = 16
|
|
41
|
+
meta_model = "gpt-4o-mini"
|
|
42
|
+
meta_model_provider = "openai"
|
|
43
|
+
few_shot_score_threshold = 0.85
|
|
44
|
+
max_instructions = 3
|
|
45
|
+
duplicate_retry_limit = 10
|
|
46
|
+
bootstrap_train_seeds = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
|
|
47
|
+
online_pool = [15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39]
|
|
48
|
+
test_pool = [40, 41, 42, 43, 44, 45, 46, 47, 48, 49]
|
|
49
|
+
# Reference pool: Used to generate rich dataset context (~50k tokens) in the meta-prompt
|
|
50
|
+
# Must not overlap with bootstrap/online/test pools to avoid data leakage
|
|
51
|
+
reference_pool = [50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149]
|
|
52
|
+
|
|
53
|
+
# System specification (pipeline-level, covers both analyzer and classifier stages)
|
|
54
|
+
spec_path = "examples/task_apps/banking77_pipeline/banking77_pipeline_spec.json"
|
|
55
|
+
spec_max_tokens = 5000
|
|
56
|
+
spec_include_examples = true
|
|
57
|
+
spec_priority_threshold = 8 # Only include high-priority rules (8+)
|
|
58
|
+
|
|
59
|
+
[prompt_learning.mipro.tpe]
|
|
60
|
+
gamma = 0.25
|
|
61
|
+
n_candidates = 32
|
|
62
|
+
n_startup_trials = 10
|
|
63
|
+
epsilon = 0.1
|
|
64
|
+
categorical_alpha = 1.0
|
|
65
|
+
prior_weight = 0.05
|
|
66
|
+
|
|
67
|
+
[prompt_learning.mipro.demo]
|
|
68
|
+
max_few_shot_examples = 5
|
|
69
|
+
sets_per_size = 6
|
|
70
|
+
include_empty = true
|
|
71
|
+
|
|
72
|
+
[prompt_learning.mipro.grounding]
|
|
73
|
+
n = 10
|
|
74
|
+
temperature = 0.7
|
|
75
|
+
max_tokens = 600
|
|
76
|
+
|
|
77
|
+
[prompt_learning.mipro.meta_update]
|
|
78
|
+
enabled = true
|
|
79
|
+
every_iterations = 3
|
|
80
|
+
topk_success = 5
|
|
81
|
+
topk_failure = 5
|
|
82
|
+
validate_on_batch = 32
|
|
83
|
+
keep_k = 12
|
|
84
|
+
dedup_token_overlap = 0.8
|
|
85
|
+
regen_temperature_decay = 0.95
|
|
86
|
+
|
|
87
|
+
[[prompt_learning.mipro.modules]]
|
|
88
|
+
module_id = "classifier"
|
|
89
|
+
max_instruction_slots = 3
|
|
90
|
+
max_demo_slots = 5
|
|
91
|
+
|
|
92
|
+
[[prompt_learning.mipro.modules]]
|
|
93
|
+
module_id = "calibrator"
|
|
94
|
+
max_instruction_slots = 3
|
|
95
|
+
max_demo_slots = 5
|
|
96
|
+
|
|
@@ -0,0 +1,94 @@
|
|
|
1
|
+
# MIPROv2 Prompt Learning for the Banking77 Two-Step Pipeline
|
|
2
|
+
|
|
3
|
+
[prompt_learning]
|
|
4
|
+
algorithm = "mipro"
|
|
5
|
+
task_app_url = "https://synth-laboratories-dev--synth-banking77-pipeline-web-web.modal.run"
|
|
6
|
+
task_app_id = "banking77-pipeline"
|
|
7
|
+
|
|
8
|
+
[prompt_learning.initial_prompt]
|
|
9
|
+
id = "banking77_pipeline_pattern"
|
|
10
|
+
name = "Banking77 Pipeline Pattern"
|
|
11
|
+
|
|
12
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
13
|
+
role = "system"
|
|
14
|
+
pattern = "Pipeline placeholder message. Actual classifier/calibrator instructions are provided via metadata.pipeline_modules."
|
|
15
|
+
order = 0
|
|
16
|
+
|
|
17
|
+
[prompt_learning.initial_prompt.metadata]
|
|
18
|
+
pipeline_modules = [
|
|
19
|
+
{ name = "classifier", instruction_text = "You are an expert banking assistant. Given a customer query, choose the correct Banking77 intent and call the banking77_classify tool with that label.", few_shots = [] },
|
|
20
|
+
{ name = "calibrator", instruction_text = "You review the classifier's suggestion. Confirm it if appropriate; otherwise pick the best Banking77 intent and return it via the banking77_classify tool.", few_shots = [] }
|
|
21
|
+
]
|
|
22
|
+
|
|
23
|
+
[prompt_learning.policy]
|
|
24
|
+
inference_mode = "synth_hosted"
|
|
25
|
+
model = "openai/gpt-oss-120b"
|
|
26
|
+
provider = "groq"
|
|
27
|
+
temperature = 0.0
|
|
28
|
+
max_completion_tokens = 128
|
|
29
|
+
policy_name = "banking77-pipeline"
|
|
30
|
+
|
|
31
|
+
[prompt_learning.env_config]
|
|
32
|
+
pool = "train"
|
|
33
|
+
|
|
34
|
+
[prompt_learning.mipro]
|
|
35
|
+
env_name = "banking77_pipeline"
|
|
36
|
+
num_iterations = 5
|
|
37
|
+
num_evaluations_per_iteration = 2
|
|
38
|
+
batch_size = 6
|
|
39
|
+
max_concurrent = 16
|
|
40
|
+
meta_model = "gpt-4o-mini"
|
|
41
|
+
meta_model_provider = "openai"
|
|
42
|
+
few_shot_score_threshold = 0.85
|
|
43
|
+
max_instructions = 3
|
|
44
|
+
duplicate_retry_limit = 10
|
|
45
|
+
bootstrap_train_seeds = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
|
|
46
|
+
online_pool = [15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39]
|
|
47
|
+
test_pool = [40, 41, 42, 43, 44, 45, 46, 47, 48, 49]
|
|
48
|
+
# Reference pool: Used to generate rich dataset context (~50k tokens) in the meta-prompt
|
|
49
|
+
# Must not overlap with bootstrap/online/test pools to avoid data leakage
|
|
50
|
+
reference_pool = [50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149]
|
|
51
|
+
|
|
52
|
+
# System specification (pipeline-level, covers both analyzer and classifier stages)
|
|
53
|
+
spec_path = "examples/task_apps/banking77_pipeline/banking77_pipeline_spec.json"
|
|
54
|
+
spec_max_tokens = 5000
|
|
55
|
+
spec_include_examples = true
|
|
56
|
+
spec_priority_threshold = 8 # Only include high-priority rules (8+)
|
|
57
|
+
|
|
58
|
+
[prompt_learning.mipro.tpe]
|
|
59
|
+
gamma = 0.25
|
|
60
|
+
n_candidates = 32
|
|
61
|
+
n_startup_trials = 10
|
|
62
|
+
epsilon = 0.1
|
|
63
|
+
categorical_alpha = 1.0
|
|
64
|
+
prior_weight = 0.05
|
|
65
|
+
|
|
66
|
+
[prompt_learning.mipro.demo]
|
|
67
|
+
max_few_shot_examples = 5
|
|
68
|
+
sets_per_size = 6
|
|
69
|
+
include_empty = true
|
|
70
|
+
|
|
71
|
+
[prompt_learning.mipro.grounding]
|
|
72
|
+
n = 10
|
|
73
|
+
temperature = 0.7
|
|
74
|
+
max_tokens = 600
|
|
75
|
+
|
|
76
|
+
[prompt_learning.mipro.meta_update]
|
|
77
|
+
enabled = true
|
|
78
|
+
every_iterations = 3
|
|
79
|
+
topk_success = 5
|
|
80
|
+
topk_failure = 5
|
|
81
|
+
validate_on_batch = 32
|
|
82
|
+
keep_k = 12
|
|
83
|
+
dedup_token_overlap = 0.8
|
|
84
|
+
regen_temperature_decay = 0.95
|
|
85
|
+
|
|
86
|
+
[[prompt_learning.mipro.modules]]
|
|
87
|
+
module_id = "classifier"
|
|
88
|
+
max_instruction_slots = 3
|
|
89
|
+
max_demo_slots = 5
|
|
90
|
+
|
|
91
|
+
[[prompt_learning.mipro.modules]]
|
|
92
|
+
module_id = "calibrator"
|
|
93
|
+
max_instruction_slots = 3
|
|
94
|
+
max_demo_slots = 5
|
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
# Lightweight smoke-test config for the Banking77 two-step pipeline
|
|
2
|
+
|
|
3
|
+
[prompt_learning]
|
|
4
|
+
algorithm = "mipro"
|
|
5
|
+
task_app_url = "https://synth-laboratories-dev--synth-banking77-pipeline-web-web.modal.run"
|
|
6
|
+
task_app_id = "banking77-pipeline"
|
|
7
|
+
|
|
8
|
+
[prompt_learning.initial_prompt]
|
|
9
|
+
id = "banking77_pipeline_pattern_test"
|
|
10
|
+
name = "Banking77 Pipeline Pattern (Test)"
|
|
11
|
+
|
|
12
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
13
|
+
role = "system"
|
|
14
|
+
pattern = "Pipeline placeholder message; see metadata.pipeline_modules for concrete instructions."
|
|
15
|
+
order = 0
|
|
16
|
+
|
|
17
|
+
[prompt_learning.initial_prompt.metadata]
|
|
18
|
+
pipeline_modules = [
|
|
19
|
+
{
|
|
20
|
+
name = "classifier",
|
|
21
|
+
instruction_text = "Classify the customer query into a Banking77 intent and invoke the tool.",
|
|
22
|
+
few_shots = [],
|
|
23
|
+
},
|
|
24
|
+
{
|
|
25
|
+
name = "calibrator",
|
|
26
|
+
instruction_text = "Validate or correct the classifier suggestion, then emit the final intent via the tool.",
|
|
27
|
+
few_shots = [],
|
|
28
|
+
},
|
|
29
|
+
]
|
|
30
|
+
|
|
31
|
+
[prompt_learning.policy]
|
|
32
|
+
inference_mode = "synth_hosted"
|
|
33
|
+
model = "openai/gpt-oss-20b"
|
|
34
|
+
provider = "groq"
|
|
35
|
+
temperature = 0.0
|
|
36
|
+
max_completion_tokens = 128
|
|
37
|
+
policy_name = "banking77-pipeline-test"
|
|
38
|
+
|
|
39
|
+
[prompt_learning.env_config]
|
|
40
|
+
pool = "train"
|
|
41
|
+
|
|
42
|
+
[prompt_learning.mipro]
|
|
43
|
+
env_name = "banking77_pipeline"
|
|
44
|
+
num_iterations = 2
|
|
45
|
+
num_evaluations_per_iteration = 2
|
|
46
|
+
batch_size = 4
|
|
47
|
+
max_concurrent = 8
|
|
48
|
+
meta_model = "gpt-4o-mini"
|
|
49
|
+
meta_model_provider = "openai"
|
|
50
|
+
few_shot_score_threshold = 0.75
|
|
51
|
+
max_instructions = 2
|
|
52
|
+
duplicate_retry_limit = 6
|
|
53
|
+
bootstrap_train_seeds = [0, 1, 2, 3]
|
|
54
|
+
online_pool = [4, 5, 6, 7]
|
|
55
|
+
test_pool = [8, 9]
|
|
56
|
+
|
|
57
|
+
[prompt_learning.mipro.tpe]
|
|
58
|
+
gamma = 0.25
|
|
59
|
+
n_candidates = 24
|
|
60
|
+
n_startup_trials = 4
|
|
61
|
+
epsilon = 0.2
|
|
62
|
+
categorical_alpha = 1.0
|
|
63
|
+
prior_weight = 0.05
|
|
64
|
+
|
|
65
|
+
[prompt_learning.mipro.demo]
|
|
66
|
+
max_few_shot_examples = 3
|
|
67
|
+
sets_per_size = 4
|
|
68
|
+
include_empty = true
|
|
69
|
+
|
|
70
|
+
[prompt_learning.mipro.grounding]
|
|
71
|
+
n = 6
|
|
72
|
+
temperature = 0.7
|
|
73
|
+
max_tokens = 400
|
|
74
|
+
|
|
75
|
+
[prompt_learning.mipro.meta_update]
|
|
76
|
+
enabled = false
|
|
77
|
+
# Banking77 Multi-Step Pipeline (MIPROv2) - Test Configuration
|
|
78
|
+
# Reduced-iteration settings for CI and quick verification
|
|
79
|
+
|
|
80
|
+
[prompt_learning]
|
|
81
|
+
algorithm = "mipro"
|
|
82
|
+
task_app_url = "http://127.0.0.1:8112"
|
|
83
|
+
task_app_id = "banking77-pipeline"
|
|
84
|
+
|
|
85
|
+
[prompt_learning.initial_prompt]
|
|
86
|
+
id = "banking77_pipeline_pattern"
|
|
87
|
+
name = "Banking77 Pipeline Classification Pattern"
|
|
88
|
+
|
|
89
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
90
|
+
role = "system"
|
|
91
|
+
pattern = "You orchestrate a two-stage Banking77 pipeline. The classifier proposes an intent and the calibrator verifies it. Always return the final intent label using the `banking77_classify` tool."
|
|
92
|
+
order = 0
|
|
93
|
+
|
|
94
|
+
[[prompt_learning.initial_prompt.messages]]
|
|
95
|
+
role = "user"
|
|
96
|
+
pattern = "Customer Query: {query}\n\nEnsure both classifier and calibrator stay within the Banking77 intent set."
|
|
97
|
+
order = 1
|
|
98
|
+
|
|
99
|
+
[prompt_learning.initial_prompt.wildcards]
|
|
100
|
+
query = "REQUIRED"
|
|
101
|
+
|
|
102
|
+
[prompt_learning.initial_prompt.metadata]
|
|
103
|
+
pipeline_modules = [
|
|
104
|
+
{ name = "classifier", instruction_text = "Map the customer query to the closest Banking77 intent. Stay precise, lean on the official label list, and respond via `banking77_classify`.", few_shots = [] },
|
|
105
|
+
{ name = "calibrator", instruction_text = "Audit the classifier suggestion. Confirm when it matches the query; otherwise pick the best Banking77 intent and reply with `banking77_classify`.", few_shots = [] }
|
|
106
|
+
]
|
|
107
|
+
|
|
108
|
+
[prompt_learning.policy]
|
|
109
|
+
inference_mode = "synth_hosted"
|
|
110
|
+
model = "openai/gpt-oss-20b"
|
|
111
|
+
provider = "groq"
|
|
112
|
+
temperature = 0.0
|
|
113
|
+
max_completion_tokens = 128
|
|
114
|
+
policy_name = "banking77-pipeline-mipro"
|
|
115
|
+
|
|
116
|
+
[prompt_learning.env_config]
|
|
117
|
+
pool = "train"
|
|
118
|
+
|
|
119
|
+
[prompt_learning.mipro]
|
|
120
|
+
env_name = "banking77-pipeline"
|
|
121
|
+
num_iterations = 4
|
|
122
|
+
num_evaluations_per_iteration = 3
|
|
123
|
+
batch_size = 3
|
|
124
|
+
max_concurrent = 8
|
|
125
|
+
meta_model = "gpt-4o-mini"
|
|
126
|
+
meta_model_provider = "openai"
|
|
127
|
+
few_shot_score_threshold = 0.80
|
|
128
|
+
max_instructions = 3
|
|
129
|
+
|
|
130
|
+
bootstrap_train_seeds = [0, 1, 2, 3]
|
|
131
|
+
online_pool = [4, 5, 6, 7]
|
|
132
|
+
test_pool = [8, 9, 10, 11]
|
|
133
|
+
|
|
134
|
+
[prompt_learning.mipro.tpe]
|
|
135
|
+
gamma = 0.25
|
|
136
|
+
n_candidates = 16
|
|
137
|
+
n_startup_trials = 5
|
|
138
|
+
epsilon = 0.1
|
|
139
|
+
categorical_alpha = 1.0
|
|
140
|
+
prior_weight = 0.05
|
|
141
|
+
|
|
142
|
+
[prompt_learning.mipro.demo]
|
|
143
|
+
max_few_shot_examples = 4
|
|
144
|
+
sets_per_size = 3
|
|
145
|
+
include_empty = true
|
|
146
|
+
|
|
147
|
+
[prompt_learning.mipro.grounding]
|
|
148
|
+
n = 6
|
|
149
|
+
temperature = 0.7
|
|
150
|
+
max_tokens = 500
|
|
151
|
+
|
|
152
|
+
[prompt_learning.mipro.meta_update]
|
|
153
|
+
enabled = true
|
|
154
|
+
every_iterations = 2
|
|
155
|
+
topk_success = 3
|
|
156
|
+
topk_failure = 3
|
|
157
|
+
validate_on_batch = 12
|
|
158
|
+
keep_k = 8
|
|
159
|
+
dedup_token_overlap = 0.8
|
|
160
|
+
regen_temperature_decay = 0.95
|
|
161
|
+
|
|
162
|
+
[[prompt_learning.mipro.modules]]
|
|
163
|
+
module_id = "classifier"
|
|
164
|
+
max_instruction_slots = 2
|
|
165
|
+
max_demo_slots = 4
|
|
166
|
+
|
|
167
|
+
[[prompt_learning.mipro.modules]]
|
|
168
|
+
module_id = "calibrator"
|
|
169
|
+
max_instruction_slots = 2
|
|
170
|
+
max_demo_slots = 4
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
#!/bin/bash
|
|
2
|
+
# Deploy the Banking77 pipeline task app locally via uvicorn
|
|
3
|
+
|
|
4
|
+
set -euo pipefail
|
|
5
|
+
|
|
6
|
+
REPO_ROOT="$(cd "$(dirname "$0")/../../.." && pwd)"
|
|
7
|
+
cd "$REPO_ROOT"
|
|
8
|
+
|
|
9
|
+
PORT="8112"
|
|
10
|
+
|
|
11
|
+
echo "🚀 Launching banking77-pipeline task app on port ${PORT}"
|
|
12
|
+
uvx synth-ai deploy banking77-pipeline \
|
|
13
|
+
--runtime uvicorn \
|
|
14
|
+
--port "${PORT}" \
|
|
15
|
+
--env-file .env \
|
|
16
|
+
--follow
|
|
17
|
+
#!/bin/bash
|
|
18
|
+
# Deploy the Banking77 multi-step pipeline task app locally for MIPROv2 optimisation
|
|
19
|
+
|
|
20
|
+
set -e
|
|
21
|
+
|
|
22
|
+
echo "🚀 Deploying Banking77 Pipeline Task App for MIPROv2..."
|
|
23
|
+
echo "======================================================"
|
|
24
|
+
|
|
25
|
+
# Ensure ENVIRONMENT_API_KEY exists (shared with backend + CLI)
|
|
26
|
+
export ENVIRONMENT_API_KEY="${ENVIRONMENT_API_KEY:-$(python -c 'import secrets; print(secrets.token_urlsafe(32))')}"
|
|
27
|
+
|
|
28
|
+
# Optional providers (helpful during local experimentation)
|
|
29
|
+
if [ -z "$GROQ_API_KEY" ] && [ -z "$OPENAI_API_KEY" ]; then
|
|
30
|
+
echo "⚠️ Neither GROQ_API_KEY nor OPENAI_API_KEY is set."
|
|
31
|
+
echo " The task app can still start, but hosted inference may fail."
|
|
32
|
+
else
|
|
33
|
+
if [ -n "$GROQ_API_KEY" ]; then
|
|
34
|
+
echo "✅ GROQ_API_KEY: ${GROQ_API_KEY:0:20}..."
|
|
35
|
+
fi
|
|
36
|
+
if [ -n "$OPENAI_API_KEY" ]; then
|
|
37
|
+
echo "✅ OPENAI_API_KEY: ${OPENAI_API_KEY:0:20}..."
|
|
38
|
+
fi
|
|
39
|
+
fi
|
|
40
|
+
|
|
41
|
+
echo "✅ ENVIRONMENT_API_KEY: ${ENVIRONMENT_API_KEY:0:20}..."
|
|
42
|
+
|
|
43
|
+
# Navigate to repo root
|
|
44
|
+
cd "$(dirname "$0")/../../.."
|
|
45
|
+
|
|
46
|
+
echo ""
|
|
47
|
+
echo "📦 Ensuring dependencies are installed..."
|
|
48
|
+
uv pip install -e . --quiet || true
|
|
49
|
+
|
|
50
|
+
echo ""
|
|
51
|
+
echo "🏃 Starting Banking77 pipeline task app on http://127.0.0.1:8112"
|
|
52
|
+
echo "Press Ctrl+C to stop."
|
|
53
|
+
echo ""
|
|
54
|
+
|
|
55
|
+
python -m examples.task_apps.banking77_pipeline.banking77_pipeline_task_app \
|
|
56
|
+
--host 0.0.0.0 \
|
|
57
|
+
--port 8112 \
|
|
58
|
+
--reload
|
|
59
|
+
|