synth-ai 0.2.8.dev4__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
- examples/rl/task_app/math_single_step.py +990 -0
- 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 +63 -0
- synth_ai/api/train/builders.py +473 -0
- synth_ai/api/train/cli.py +1185 -0
- synth_ai/api/train/config_finder.py +246 -0
- 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 +352 -0
- synth_ai/api/train/pollers.py +91 -0
- 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 +195 -0
- synth_ai/api/train/utils.py +244 -0
- 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 +90 -45
- 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 +166 -114
- synth_ai/cli/root.py +143 -112
- 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 +3134 -0
- 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 +745 -416
- 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 +7 -1
- synth_ai/demos/demo_task_apps/core.py +75 -37
- 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/config.toml +55 -110
- synth_ai/demos/demo_task_apps/math/deploy_modal.py +3 -6
- synth_ai/demos/demo_task_apps/math/modal_task_app.py +491 -166
- synth_ai/demos/demo_task_apps/math/task_app_entry.py +37 -0
- 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 +116 -3
- synth_ai/task/apps/__init__.py +132 -0
- synth_ai/task/auth.py +165 -0
- synth_ai/task/client.py +167 -0
- synth_ai/task/config.py +261 -0
- synth_ai/task/contracts.py +173 -57
- synth_ai/task/datasets.py +108 -0
- synth_ai/task/errors.py +50 -0
- synth_ai/task/health.py +17 -11
- synth_ai/task/inference_api.py +101 -0
- synth_ai/task/json.py +111 -0
- synth_ai/task/proxy.py +251 -0
- 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 +219 -0
- synth_ai/task/server.py +432 -0
- synth_ai/task/trace_correlation_helpers.py +328 -0
- synth_ai/task/tracing_utils.py +95 -0
- synth_ai/task/validators.py +449 -6
- synth_ai/task/vendors.py +59 -0
- 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 +86 -21
- synth_ai/tracing_v3/storage/base.py +98 -12
- 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.8.dev4.dist-info → synth_ai-0.2.23.dev3.dist-info}/entry_points.txt +0 -1
- {synth_ai-0.2.8.dev4.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/environments/examples/sokoban/units/astar_common.py +0 -95
- synth_ai/experimental/synth_oss.py +0 -446
- synth_ai/handshake.py +0 -63
- 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/tracing/__init__.py +0 -30
- synth_ai/tracing_v1/__init__.py +0 -33
- synth_ai/tracing_v3/turso/manager.py +0 -760
- 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.8.dev4.dist-info/METADATA +0 -129
- synth_ai-0.2.8.dev4.dist-info/RECORD +0 -420
- {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.8.dev4.dist-info → synth_ai-0.2.23.dev3.dist-info}/WHEEL +0 -0
- {synth_ai-0.2.8.dev4.dist-info → synth_ai-0.2.23.dev3.dist-info}/licenses/LICENSE +0 -0
examples/task_apps/dev/pokemon_emerald/external/pokeagent-speedrun/pokemon_env/emerald_utils.py
ADDED
|
@@ -0,0 +1,631 @@
|
|
|
1
|
+
# This file is taken from https://github.com/dvruette/pygba/blob/main/pygba/utils.py
|
|
2
|
+
# and modified to work with the Pokemon Emerald game.
|
|
3
|
+
|
|
4
|
+
import functools
|
|
5
|
+
import struct
|
|
6
|
+
from collections import namedtuple
|
|
7
|
+
|
|
8
|
+
from pokemon_env.enums import Move, PokemonType, StatusCondition
|
|
9
|
+
from pokemon_env.types import PokemonData
|
|
10
|
+
|
|
11
|
+
|
|
12
|
+
class BaseCharmap:
|
|
13
|
+
charmap: list[str]
|
|
14
|
+
terminator: int
|
|
15
|
+
|
|
16
|
+
def decode(self, chars: bytes) -> str:
|
|
17
|
+
string = ""
|
|
18
|
+
for i in range(len(chars)):
|
|
19
|
+
if chars[i] == self.terminator:
|
|
20
|
+
break
|
|
21
|
+
string += self.charmap[chars[i]]
|
|
22
|
+
return string
|
|
23
|
+
|
|
24
|
+
class AsciiCharmap(BaseCharmap):
|
|
25
|
+
charmap = [
|
|
26
|
+
"", "", "", "", "", "", "", "", "\r", "\t", " ", " ", " ", "\n", "", "",
|
|
27
|
+
"", "", "", "", "", "", "", "", "", "", "SUB", "ESC", "", "", "", "",
|
|
28
|
+
" ", "!", "\"", "#", "$", "%", "&", "'", "(", ")", "*", "+", ",", "-", ".", "/", "0", "1",
|
|
29
|
+
"2", "3", "4", "5", "6", "7", "8", "9", ":", ";", "<", "=", ">", "?", "@", "A",
|
|
30
|
+
"B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q",
|
|
31
|
+
"R", "S", "T", "U", "V", "W", "X", "Y", "Z", "[", "\\", "]", "^", "_", "`", "a",
|
|
32
|
+
"b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q",
|
|
33
|
+
"r", "s", "t", "u", "v", "w", "x", "y", "z", "{", "|", "}", "~", "DEL", "Ç", "ü",
|
|
34
|
+
"é", "â", "ä", "à", "å", "ç", "ê", "ë", "è", "ï", "î", "ì", "Ä", "Å", "É", "æ",
|
|
35
|
+
"Æ", "ô", "ö", "ò", "û", "ù", "ÿ", "Ö", "Ü", "ø", "£", "Ø", "×", "ƒ", "á", "í",
|
|
36
|
+
"ó", "ú", "ñ", "Ñ", "ª", "º", "¿", "®", "¬", "½", "¼", "¡", "«", "»", "░", "▒",
|
|
37
|
+
"▓", "│", "┤", "Á", "Â", "À", "©", "╣", "║", "╗", "╝", "¢", "¥", "┐", "└", "┴",
|
|
38
|
+
"┬", "├", "─", "┼", "ã", "Ã", "╚", "╔", "╩", "╦", "╠", "═", "╬", "¤", "ð", "Ð",
|
|
39
|
+
"Ê", "Ë", "È", "ı", "Í", "Î", "Ï", "┘", "┌", "█", "▄", "¦", "Ì", "▀", "Ó", "ß",
|
|
40
|
+
"Ô", "Ò", "õ", "Õ", "µ", "þ", "Þ", "Ú", "Û", "Ù", "ý", "Ý", "¯", "´", "¬", "±",
|
|
41
|
+
"‗", "¾", "¶", "§", "÷", "¸", "°", "¨", "•", "¹", "³", "²", "■", "\u00a0",
|
|
42
|
+
]
|
|
43
|
+
terminator = 0x00
|
|
44
|
+
|
|
45
|
+
|
|
46
|
+
|
|
47
|
+
# Pokemon Emerald Sym Addresses
|
|
48
|
+
# https://raw.githubusercontent.com/pret/pokeemerald/symbols/pokeemerald.sym
|
|
49
|
+
|
|
50
|
+
ADDRESSES = {
|
|
51
|
+
"gPlayerPartyCount": 0x020244e9,
|
|
52
|
+
"gPlayerParty": 0x020244ec,
|
|
53
|
+
"gSaveBlock1Ptr": 0x03005d8c,
|
|
54
|
+
"gSaveBlock2Ptr": 0x03005d90,
|
|
55
|
+
"gPokemonStoragePtr": 0x03005d94,
|
|
56
|
+
"gSpeciesNames": 0x083185c8,
|
|
57
|
+
"sSpeciesToHoennPokedexNum": 0x0831d94c,
|
|
58
|
+
"sSpeciesToNationalPokedexNum": 0x0831dc82,
|
|
59
|
+
"sHoennToNationalOrder": 0x0831dfb8,
|
|
60
|
+
"gExperienceTables": 0x0831f72c,
|
|
61
|
+
"gSpeciesInfo": 0x083203cc,
|
|
62
|
+
"gItems": 0x085839a0,
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
|
|
66
|
+
# Struct layouts and constants taken from pret/pokeemerald:
|
|
67
|
+
# https://github.com/pret/pokeemerald/blob/master/include/pokemon.h
|
|
68
|
+
# https://github.com/pret/pokeemerald/blob/master/include/global.h
|
|
69
|
+
|
|
70
|
+
|
|
71
|
+
## Constants
|
|
72
|
+
|
|
73
|
+
POKEMON_NAME_LENGTH = 10
|
|
74
|
+
PLAYER_NAME_LENGTH = 7
|
|
75
|
+
PC_ITEMS_COUNT = 50
|
|
76
|
+
BAG_ITEMS_COUNT = 30
|
|
77
|
+
BAG_KEYITEMS_COUNT = 30
|
|
78
|
+
BAG_POKEBALLS_COUNT = 16
|
|
79
|
+
BAG_TMHM_COUNT = 64
|
|
80
|
+
BAG_BERRIES_COUNT = 46
|
|
81
|
+
|
|
82
|
+
NUM_SPECIES = 412
|
|
83
|
+
NUM_DEX_FLAG_BYTES = (NUM_SPECIES + 7) // 8
|
|
84
|
+
|
|
85
|
+
TOTAL_BOXES_COUNT = 14
|
|
86
|
+
IN_BOX_COUNT = 30
|
|
87
|
+
BOX_NAME_LENGTH = 8
|
|
88
|
+
|
|
89
|
+
|
|
90
|
+
## Flag IDs
|
|
91
|
+
|
|
92
|
+
SCRIPT_FLAGS_START = 0x50
|
|
93
|
+
TRAINER_FLAGS_START = 0x500
|
|
94
|
+
SYSTEM_FLAGS_START = 0x860
|
|
95
|
+
DAILY_FLAGS_START = 0x920
|
|
96
|
+
|
|
97
|
+
FLAG_DEFEATED_RUSTBORO_GYM = 0x4F0
|
|
98
|
+
FLAG_DEFEATED_DEWFORD_GYM = 0x4F1
|
|
99
|
+
FLAG_DEFEATED_MAUVILLE_GYM = 0x4F2
|
|
100
|
+
FLAG_DEFEATED_LAVARIDGE_GYM = 0x4F3
|
|
101
|
+
FLAG_DEFEATED_PETALBURG_GYM = 0x4F4
|
|
102
|
+
FLAG_DEFEATED_FORTREE_GYM = 0x4F5
|
|
103
|
+
FLAG_DEFEATED_MOSSDEEP_GYM = 0x4F6
|
|
104
|
+
FLAG_DEFEATED_SOOTOPOLIS_GYM = 0x4F7
|
|
105
|
+
FLAG_DEFEATED_METEOR_FALLS_STEVEN = 0x4F8
|
|
106
|
+
|
|
107
|
+
FLAG_DEFEATED_ELITE_4_SIDNEY = 0x4FB
|
|
108
|
+
FLAG_DEFEATED_ELITE_4_PHOEBE = 0x4FC
|
|
109
|
+
FLAG_DEFEATED_ELITE_4_GLACIA = 0x4FD
|
|
110
|
+
FLAG_DEFEATED_ELITE_4_DRAKE = 0x4FE
|
|
111
|
+
|
|
112
|
+
|
|
113
|
+
FLAG_SYS_POKEMON_GET = SYSTEM_FLAGS_START + 0x0
|
|
114
|
+
FLAG_SYS_POKEDEX_GET = SYSTEM_FLAGS_START + 0x1
|
|
115
|
+
FLAG_SYS_POKENAV_GET = SYSTEM_FLAGS_START + 0x2
|
|
116
|
+
FLAG_RECEIVED_POKEDEX_FROM_BIRCH = SYSTEM_FLAGS_START + 0x84
|
|
117
|
+
|
|
118
|
+
FLAG_BADGE01_GET = SYSTEM_FLAGS_START + 0x7
|
|
119
|
+
FLAG_BADGE02_GET = SYSTEM_FLAGS_START + 0x8
|
|
120
|
+
FLAG_BADGE03_GET = SYSTEM_FLAGS_START + 0x9
|
|
121
|
+
FLAG_BADGE04_GET = SYSTEM_FLAGS_START + 0xa
|
|
122
|
+
FLAG_BADGE05_GET = SYSTEM_FLAGS_START + 0xb
|
|
123
|
+
FLAG_BADGE06_GET = SYSTEM_FLAGS_START + 0xc
|
|
124
|
+
FLAG_BADGE07_GET = SYSTEM_FLAGS_START + 0xd
|
|
125
|
+
FLAG_BADGE08_GET = SYSTEM_FLAGS_START + 0xe
|
|
126
|
+
|
|
127
|
+
FLAG_VISITED_LITTLEROOT_TOWN = SYSTEM_FLAGS_START + 0xF
|
|
128
|
+
FLAG_VISITED_OLDALE_TOWN = SYSTEM_FLAGS_START + 0x10
|
|
129
|
+
FLAG_VISITED_DEWFORD_TOWN = SYSTEM_FLAGS_START + 0x11
|
|
130
|
+
FLAG_VISITED_LAVARIDGE_TOWN = SYSTEM_FLAGS_START + 0x12
|
|
131
|
+
FLAG_VISITED_FALLARBOR_TOWN = SYSTEM_FLAGS_START + 0x13
|
|
132
|
+
FLAG_VISITED_VERDANTURF_TOWN = SYSTEM_FLAGS_START + 0x14
|
|
133
|
+
FLAG_VISITED_PACIFIDLOG_TOWN = SYSTEM_FLAGS_START + 0x15
|
|
134
|
+
FLAG_VISITED_PETALBURG_CITY = SYSTEM_FLAGS_START + 0x16
|
|
135
|
+
FLAG_VISITED_SLATEPORT_CITY = SYSTEM_FLAGS_START + 0x17
|
|
136
|
+
FLAG_VISITED_MAUVILLE_CITY = SYSTEM_FLAGS_START + 0x18
|
|
137
|
+
FLAG_VISITED_RUSTBORO_CITY = SYSTEM_FLAGS_START + 0x19
|
|
138
|
+
FLAG_VISITED_FORTREE_CITY = SYSTEM_FLAGS_START + 0x1A
|
|
139
|
+
FLAG_VISITED_LILYCOVE_CITY = SYSTEM_FLAGS_START + 0x1B
|
|
140
|
+
FLAG_VISITED_MOSSDEEP_CITY = SYSTEM_FLAGS_START + 0x1C
|
|
141
|
+
FLAG_VISITED_SOOTOPOLIS_CITY = SYSTEM_FLAGS_START + 0x1D
|
|
142
|
+
FLAG_VISITED_EVER_GRANDE_CITY = SYSTEM_FLAGS_START + 0x1E
|
|
143
|
+
|
|
144
|
+
FLAG_IS_CHAMPION = SYSTEM_FLAGS_START + 0x1F
|
|
145
|
+
|
|
146
|
+
|
|
147
|
+
|
|
148
|
+
|
|
149
|
+
class EmeraldCharmap(BaseCharmap):
|
|
150
|
+
charmap = [
|
|
151
|
+
" ", "À", "Á", "Â", "Ç", "È", "É", "Ê", "Ë", "Ì", "こ", "Î", "Ï", "Ò", "Ó", "Ô",
|
|
152
|
+
"Œ", "Ù", "Ú", "Û", "Ñ", "ß", "à", "á", "ね", "ç", "è", "é", "ê", "ë", "ì", "ま",
|
|
153
|
+
"î", "ï", "ò", "ó", "ô", "œ", "ù", "ú", "û", "ñ", "º", "ª", "�", "&", "+", "あ",
|
|
154
|
+
"ぃ", "ぅ", "ぇ", "ぉ", "Lv", "=", ";", "が", "ぎ", "ぐ", "げ", "ご", "ざ", "じ", "ず", "ぜ",
|
|
155
|
+
"ぞ", "だ", "ぢ", "づ", "で", "ど", "ば", "び", "ぶ", "べ", "ぼ", "ぱ", "ぴ", "ぷ", "ぺ", "ぽ",
|
|
156
|
+
"っ", "¿", "¡", "P\u200dk", "M\u200dn", "P\u200do", "K\u200dé", "B\u200dL", "O\u200dC", "\u200dK", "Í", "%", "(", ")", "セ", "ソ",
|
|
157
|
+
"タ", "チ", "ツ", "テ", "ト", "ナ", "ニ", "ヌ", "â", "ノ", "ハ", "ヒ", "フ", "ヘ", "ホ", "í",
|
|
158
|
+
"ミ", "ム", "メ", "モ", "ヤ", "ユ", "ヨ", "ラ", "リ", "⬆", "⬇", "⬅", "➡", "ヲ", "ン", "ァ",
|
|
159
|
+
"ィ", "ゥ", "ェ", "ォ", "ャ", "ュ", "ョ", "ガ", "ギ", "グ", "ゲ", "ゴ", "ザ", "ジ", "ズ", "ゼ",
|
|
160
|
+
"ゾ", "ダ", "ヂ", "ヅ", "デ", "ド", "バ", "ビ", "ブ", "ベ", "ボ", "パ", "ピ", "プ", "ペ", "ポ",
|
|
161
|
+
"ッ", "0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "!", "?", ".", "-", "・",
|
|
162
|
+
"…", "“", "”", "‘", "’", "♂", "♀", "$", ",", "×", "/", "A", "B", "C", "D", "E",
|
|
163
|
+
"F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U",
|
|
164
|
+
"V", "W", "X", "Y", "Z", "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k",
|
|
165
|
+
"l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z", "▶",
|
|
166
|
+
":", "Ä", "Ö", "Ü", "ä", "ö", "ü", "⬆", "⬇", "⬅", "�", "�", "�", "�", "�", "",
|
|
167
|
+
]
|
|
168
|
+
terminator = 0xFF
|
|
169
|
+
|
|
170
|
+
|
|
171
|
+
PokemonSubstruct0_spec = (
|
|
172
|
+
("species", "H"),
|
|
173
|
+
("heldItem", "H"),
|
|
174
|
+
("experience", "I"),
|
|
175
|
+
("ppBonuses", "B"),
|
|
176
|
+
("friendship", "B"),
|
|
177
|
+
("unknown", "H"),
|
|
178
|
+
)
|
|
179
|
+
PokemonSubstruct0 = namedtuple("PokemonSubstruct0", [x[0] for x in PokemonSubstruct0_spec])
|
|
180
|
+
PokemonSubstruct0_format = "".join([x[1] for x in PokemonSubstruct0_spec])
|
|
181
|
+
|
|
182
|
+
PokemonSubstruct1_spec = None
|
|
183
|
+
PokemonSubstruct1 = namedtuple("PokemonSubstruct1", ("moves", "pp"))
|
|
184
|
+
PokemonSubstruct1_format = "4H4B"
|
|
185
|
+
|
|
186
|
+
PokemonSubstruct2_spec = (
|
|
187
|
+
("hpEV", "B"),
|
|
188
|
+
("attackEV", "B"),
|
|
189
|
+
("defenseEV", "B"),
|
|
190
|
+
("speedEV", "B"),
|
|
191
|
+
("spAttackEV", "B"),
|
|
192
|
+
("spDefenseEV", "B"),
|
|
193
|
+
("cool", "B"),
|
|
194
|
+
("beauty", "B"),
|
|
195
|
+
("cute", "B"),
|
|
196
|
+
("smart", "B"),
|
|
197
|
+
("tough", "B"),
|
|
198
|
+
("sheen", "B"),
|
|
199
|
+
)
|
|
200
|
+
PokemonSubstruct2 = namedtuple("PokemonSubstruct2", [x[0] for x in PokemonSubstruct2_spec])
|
|
201
|
+
PokemonSubstruct2_format = "".join([x[1] for x in PokemonSubstruct2_spec])
|
|
202
|
+
|
|
203
|
+
PokemonSubstruct3_spec = None
|
|
204
|
+
PokemonSubstruct3_format = "III"
|
|
205
|
+
PokemonSubstruct3 = namedtuple("PokemonSubstruct3", (
|
|
206
|
+
"pokerus",
|
|
207
|
+
"metLocation",
|
|
208
|
+
"metLevel",
|
|
209
|
+
"metGame",
|
|
210
|
+
"pokeball",
|
|
211
|
+
"otGender",
|
|
212
|
+
"hpIV",
|
|
213
|
+
"attackIV",
|
|
214
|
+
"defenseIV",
|
|
215
|
+
"speedIV",
|
|
216
|
+
"spAttackIV",
|
|
217
|
+
"spDefenseIV",
|
|
218
|
+
"isEgg",
|
|
219
|
+
"abilityNum",
|
|
220
|
+
"ribbons",
|
|
221
|
+
))
|
|
222
|
+
|
|
223
|
+
|
|
224
|
+
BoxPokemon_spec = (
|
|
225
|
+
("personality", "I"),
|
|
226
|
+
("otId", "I"),
|
|
227
|
+
("nickname", f"{POKEMON_NAME_LENGTH}s"),
|
|
228
|
+
("language", "B"),
|
|
229
|
+
("flags", "B"),
|
|
230
|
+
("otName", f"{PLAYER_NAME_LENGTH}s"),
|
|
231
|
+
("markings", "B"),
|
|
232
|
+
("checksum", "H"),
|
|
233
|
+
("unknown", "H"),
|
|
234
|
+
("substructs", f"{48}s"),
|
|
235
|
+
)
|
|
236
|
+
BoxPokemon = namedtuple("BoxPokemon", [x[0] for x in BoxPokemon_spec])
|
|
237
|
+
BoxPokemon_format = "".join([x[1] for x in BoxPokemon_spec])
|
|
238
|
+
|
|
239
|
+
|
|
240
|
+
Pokemon_spec = (
|
|
241
|
+
("box", f"{struct.calcsize(BoxPokemon_format)}s"),
|
|
242
|
+
("status", "I"),
|
|
243
|
+
("level", "B"),
|
|
244
|
+
("mail", "B"),
|
|
245
|
+
("hp", "H"),
|
|
246
|
+
("maxHp", "H"),
|
|
247
|
+
("attack", "H"),
|
|
248
|
+
("defense", "H"),
|
|
249
|
+
("speed", "H"),
|
|
250
|
+
("spAttack", "H"),
|
|
251
|
+
("spDefense", "H"),
|
|
252
|
+
)
|
|
253
|
+
Pokemon = namedtuple("Pokemon", [x[0] for x in Pokemon_spec])
|
|
254
|
+
Pokemon_format = "".join([x[1] for x in Pokemon_spec])
|
|
255
|
+
|
|
256
|
+
|
|
257
|
+
Pokedex_spec = (
|
|
258
|
+
("order", "B"),
|
|
259
|
+
("mode", "B"),
|
|
260
|
+
("nationalMagic", "B"),
|
|
261
|
+
("padding1", "B"),
|
|
262
|
+
("unownPersonality", "I"),
|
|
263
|
+
("spindaPersonality", "I"),
|
|
264
|
+
("padding2", "4s"),
|
|
265
|
+
("owned", f"{NUM_DEX_FLAG_BYTES}s"),
|
|
266
|
+
("seen", f"{NUM_DEX_FLAG_BYTES}s"),
|
|
267
|
+
)
|
|
268
|
+
Pokedex = namedtuple("Pokedex", [x[0] for x in Pokedex_spec])
|
|
269
|
+
Pokedex_format = "".join([x[1] for x in Pokedex_spec])
|
|
270
|
+
|
|
271
|
+
|
|
272
|
+
SpeciesInfo_spec = (
|
|
273
|
+
("baseHP", "B"),
|
|
274
|
+
("baseAttack", "B"),
|
|
275
|
+
("baseDefense", "B"),
|
|
276
|
+
("baseSpeed", "B"),
|
|
277
|
+
("baseSpAttack", "B"),
|
|
278
|
+
("baseSpDefense", "B"),
|
|
279
|
+
("type1", "B"),
|
|
280
|
+
("type2", "B"),
|
|
281
|
+
("catchRate", "B"),
|
|
282
|
+
("expYield", "B"),
|
|
283
|
+
("evYield", "H"),
|
|
284
|
+
("itemCommon", "H"),
|
|
285
|
+
("itemRare", "H"),
|
|
286
|
+
("genderRatio", "B"),
|
|
287
|
+
("eggCycles", "B"),
|
|
288
|
+
("friendship", "B"),
|
|
289
|
+
("growthRate", "B"),
|
|
290
|
+
("eggGroup1", "B"),
|
|
291
|
+
("eggGroup2", "B"),
|
|
292
|
+
("ability1", "B"),
|
|
293
|
+
("ability2", "B"),
|
|
294
|
+
("safariZoneFleeRate", "B"),
|
|
295
|
+
("bodyColor", "Bxx"),
|
|
296
|
+
)
|
|
297
|
+
SpeciesInfo = namedtuple("SpeciesInfo", [x[0] for x in SpeciesInfo_spec])
|
|
298
|
+
SpeciesInfo_format = "".join([x[1] for x in SpeciesInfo_spec])
|
|
299
|
+
|
|
300
|
+
Coords16_spec = (
|
|
301
|
+
("x", "H"),
|
|
302
|
+
("y", "H"),
|
|
303
|
+
)
|
|
304
|
+
Coords16 = namedtuple("Coords16", [x[0] for x in Coords16_spec])
|
|
305
|
+
Coords16_format = "".join([x[1] for x in Coords16_spec])
|
|
306
|
+
|
|
307
|
+
WarpData_spec = (
|
|
308
|
+
("mapGroup", "b"),
|
|
309
|
+
("mapNum", "b"),
|
|
310
|
+
("warpId", "bx"),
|
|
311
|
+
("x", "H"),
|
|
312
|
+
("y", "H"),
|
|
313
|
+
)
|
|
314
|
+
WarpData = namedtuple("WarpData", [x[0] for x in WarpData_spec])
|
|
315
|
+
WarpData_format = "".join([x[1] for x in WarpData_spec])
|
|
316
|
+
|
|
317
|
+
ItemSlot_spec = (
|
|
318
|
+
("itemId", "H"),
|
|
319
|
+
("quantity", "H"),
|
|
320
|
+
)
|
|
321
|
+
ItemSlot = namedtuple("ItemSlot", [x[0] for x in ItemSlot_spec])
|
|
322
|
+
ItemSlot_format = "".join([x[1] for x in ItemSlot_spec])
|
|
323
|
+
|
|
324
|
+
SaveBlock2_spec = (
|
|
325
|
+
("playerName", f"{PLAYER_NAME_LENGTH + 1}s"),
|
|
326
|
+
("playerGender", "B"),
|
|
327
|
+
("specialSaveWarpFlags", "B"),
|
|
328
|
+
("playerTrainerId", "4s"),
|
|
329
|
+
("playTimeHours", "H"),
|
|
330
|
+
("playTimeMinutes", "B"),
|
|
331
|
+
("playTimeSeconds", "B"),
|
|
332
|
+
("playTimeVBlanks", "B"),
|
|
333
|
+
("optionsButtonMode", "B"),
|
|
334
|
+
("options", "H"),
|
|
335
|
+
("padding1", "2s"),
|
|
336
|
+
("pokedex", f"{struct.calcsize(Pokedex_format)}s"),
|
|
337
|
+
("filler_90", "8s"),
|
|
338
|
+
("localTimeOffset", "8s"),
|
|
339
|
+
("lastBerryTreeUpdate", "8s"),
|
|
340
|
+
("gcnLinkFlags", "I"),
|
|
341
|
+
("encryptionKey", "I"),
|
|
342
|
+
("rest", f"{0xe7c}s"),
|
|
343
|
+
)
|
|
344
|
+
SaveBlock2 = namedtuple("SaveBlock2", [x[0] for x in SaveBlock2_spec])
|
|
345
|
+
SaveBlock2_format = "".join([x[1] for x in SaveBlock2_spec])
|
|
346
|
+
|
|
347
|
+
SaveBlock1_spec = (
|
|
348
|
+
("pos", f"{struct.calcsize(Coords16_format)}s"),
|
|
349
|
+
("location", f"{struct.calcsize(WarpData_format)}s"),
|
|
350
|
+
("continueGameWarp", f"{struct.calcsize(WarpData_format)}s"),
|
|
351
|
+
("dynamicWarp", f"{struct.calcsize(WarpData_format)}s"),
|
|
352
|
+
("lastHealLocation", f"{struct.calcsize(WarpData_format)}s"),
|
|
353
|
+
("escapeWarp", f"{struct.calcsize(WarpData_format)}s"),
|
|
354
|
+
("savedMusic", "H"),
|
|
355
|
+
("weather", "B"),
|
|
356
|
+
("weatherCycleStage", "B"),
|
|
357
|
+
("flashLevel", "B"),
|
|
358
|
+
("padding1", "B"),
|
|
359
|
+
("mapLayoutId", "H"),
|
|
360
|
+
("mapView", f"{0x200}s"),
|
|
361
|
+
("playerPartyCount", "B"),
|
|
362
|
+
("padding2", "3s"),
|
|
363
|
+
("playerParty", f"{600}s"),
|
|
364
|
+
("money", "I"),
|
|
365
|
+
("coins", "H"),
|
|
366
|
+
("registeredItem", "H"),
|
|
367
|
+
("pcItems", f"{struct.calcsize(ItemSlot_format) * PC_ITEMS_COUNT}s"),
|
|
368
|
+
("bagPocket_Items", f"{struct.calcsize(ItemSlot_format) * BAG_ITEMS_COUNT}s"),
|
|
369
|
+
("bagPocket_KeyItems", f"{struct.calcsize(ItemSlot_format) * BAG_KEYITEMS_COUNT}s"),
|
|
370
|
+
("bagPocket_PokeBalls", f"{struct.calcsize(ItemSlot_format) * BAG_POKEBALLS_COUNT}s"),
|
|
371
|
+
("bagPocket_TMHM", f"{struct.calcsize(ItemSlot_format) * BAG_TMHM_COUNT}s"),
|
|
372
|
+
("bagPocket_Berries", f"{struct.calcsize(ItemSlot_format) * BAG_BERRIES_COUNT}s"),
|
|
373
|
+
("pokeblocks", f"{320}s"),
|
|
374
|
+
("seen1", f"{NUM_DEX_FLAG_BYTES}s"),
|
|
375
|
+
("berryBlenderRecords", "6s"),
|
|
376
|
+
("unused", "6s"),
|
|
377
|
+
("trainerRematchStepCounter", "H"),
|
|
378
|
+
("trainedRematches", "100s"),
|
|
379
|
+
("padding3", "2s"),
|
|
380
|
+
("objectEvents", f"{576}s"),
|
|
381
|
+
("objectEventTemplates", f"{1536}s"),
|
|
382
|
+
("flags", f"{300}s"),
|
|
383
|
+
("rest", f"{0x29ec}s"),
|
|
384
|
+
)
|
|
385
|
+
SaveBlock1 = namedtuple("SaveBlock1", [x[0] for x in SaveBlock1_spec])
|
|
386
|
+
SaveBlock1_format = "".join([x[1] for x in SaveBlock1_spec])
|
|
387
|
+
|
|
388
|
+
|
|
389
|
+
PokemonStorage_spec = (
|
|
390
|
+
("currentBox", "B"),
|
|
391
|
+
("padding", "3s"), # 3 bytes padding
|
|
392
|
+
("boxes", f"{struct.calcsize(BoxPokemon_format) * TOTAL_BOXES_COUNT * IN_BOX_COUNT}s"),
|
|
393
|
+
("boxNames", f"{TOTAL_BOXES_COUNT * (BOX_NAME_LENGTH + 1)}s"),
|
|
394
|
+
("boxWallpapers", f"{TOTAL_BOXES_COUNT}s"),
|
|
395
|
+
)
|
|
396
|
+
PokemonStorage = namedtuple("PokemonStorage", [x[0] for x in PokemonStorage_spec])
|
|
397
|
+
PokemonStorage_format = "".join([x[1] for x in PokemonStorage_spec])
|
|
398
|
+
|
|
399
|
+
|
|
400
|
+
def parse_box_pokemon(data):
|
|
401
|
+
if int.from_bytes(data[:4], "little") == 0:
|
|
402
|
+
return None
|
|
403
|
+
|
|
404
|
+
box = BoxPokemon._make(struct.unpack("<" + BoxPokemon_format, data))
|
|
405
|
+
|
|
406
|
+
key = box.otId ^ box.personality
|
|
407
|
+
substructs_raw = struct.unpack("<" + "I" * 12, box.substructs)
|
|
408
|
+
substructs = [x ^ key for x in substructs_raw]
|
|
409
|
+
|
|
410
|
+
substructSelector = [
|
|
411
|
+
[0, 1, 2, 3], [0, 1, 3, 2], [0, 2, 1, 3], [0, 3, 1, 2],
|
|
412
|
+
[0, 2, 3, 1], [0, 3, 2, 1], [1, 0, 2, 3], [1, 0, 3, 2],
|
|
413
|
+
[2, 0, 1, 3], [3, 0, 1, 2], [2, 0, 3, 1], [3, 0, 2, 1],
|
|
414
|
+
[1, 2, 0, 3], [1, 3, 0, 2], [2, 1, 0, 3], [3, 1, 0, 2],
|
|
415
|
+
[2, 3, 0, 1], [3, 2, 0, 1], [1, 2, 3, 0], [1, 3, 2, 0],
|
|
416
|
+
[2, 1, 3, 0], [3, 1, 2, 0], [2, 3, 1, 0], [3, 2, 1, 0],
|
|
417
|
+
]
|
|
418
|
+
# get substruct permutation by personality mod 24
|
|
419
|
+
perm = substructSelector[box.personality % 24]
|
|
420
|
+
substruct0 = substructs[3 * perm[0] : 3 * (perm[0] + 1)]
|
|
421
|
+
substruct1 = substructs[3 * perm[1] : 3 * (perm[1] + 1)]
|
|
422
|
+
substruct2 = substructs[3 * perm[2] : 3 * (perm[2] + 1)]
|
|
423
|
+
substruct3 = substructs[3 * perm[3] : 3 * (perm[3] + 1)]
|
|
424
|
+
|
|
425
|
+
substruct0 = PokemonSubstruct0._make(struct.unpack("<" + PokemonSubstruct0_format, struct.pack("<" + "I" * 3, *substruct0)))
|
|
426
|
+
substruct2 = PokemonSubstruct2._make(struct.unpack("<" + PokemonSubstruct2_format, struct.pack("<" + "I" * 3, *substruct2)))
|
|
427
|
+
|
|
428
|
+
x1, x2, x3 = substruct1
|
|
429
|
+
substruct1 = PokemonSubstruct1(
|
|
430
|
+
[
|
|
431
|
+
(x1 >> 0) & 0xFFFF,
|
|
432
|
+
(x1 >> 16) & 0xFFFF,
|
|
433
|
+
(x2 >> 0) & 0xFFFF,
|
|
434
|
+
(x2 >> 16) & 0xFFFF,
|
|
435
|
+
],
|
|
436
|
+
[
|
|
437
|
+
(x3 >> 0) & 0xFF,
|
|
438
|
+
(x3 >> 8) & 0xFF,
|
|
439
|
+
(x3 >> 16) & 0xFF,
|
|
440
|
+
(x3 >> 24) & 0xFF,
|
|
441
|
+
]
|
|
442
|
+
)
|
|
443
|
+
|
|
444
|
+
x1, x2, x3 = substruct3
|
|
445
|
+
substruct3 = PokemonSubstruct3(
|
|
446
|
+
(x1 >> 0) & 0xFF,
|
|
447
|
+
(x1 >> 8) & 0xFFFF,
|
|
448
|
+
(x1 >> 16) & 0b01111111,
|
|
449
|
+
(x1 >> 23) & 0xF,
|
|
450
|
+
(x1 >> 27) & 0xF,
|
|
451
|
+
(x1 >> 31) & 0b1,
|
|
452
|
+
(x2 >> 0) & 0b00011111,
|
|
453
|
+
(x2 >> 5) & 0b00011111,
|
|
454
|
+
(x2 >> 10) & 0b00011111,
|
|
455
|
+
(x2 >> 15) & 0b00011111,
|
|
456
|
+
(x2 >> 20) & 0b00011111,
|
|
457
|
+
(x2 >> 25) & 0b00011111,
|
|
458
|
+
(x2 >> 30) & 0b1,
|
|
459
|
+
(x2 >> 31) & 0b1,
|
|
460
|
+
x3,
|
|
461
|
+
)
|
|
462
|
+
|
|
463
|
+
box = box._replace(
|
|
464
|
+
nickname=EmeraldCharmap().decode(box.nickname),
|
|
465
|
+
otName=EmeraldCharmap().decode(box.otName),
|
|
466
|
+
substructs=(
|
|
467
|
+
substruct0._asdict(),
|
|
468
|
+
substruct1._asdict(),
|
|
469
|
+
substruct2._asdict(),
|
|
470
|
+
substruct3._asdict(),
|
|
471
|
+
),
|
|
472
|
+
)
|
|
473
|
+
|
|
474
|
+
box = box._asdict()
|
|
475
|
+
del box["unknown"]
|
|
476
|
+
del box["substructs"][0]["unknown"]
|
|
477
|
+
return box
|
|
478
|
+
|
|
479
|
+
def parse_pokemon(data):
|
|
480
|
+
pokemon = Pokemon._make(struct.unpack("<" + Pokemon_format, data))
|
|
481
|
+
box = parse_box_pokemon(pokemon.box)
|
|
482
|
+
pokemon = pokemon._replace(box=box)
|
|
483
|
+
# Construct a PokemonData object
|
|
484
|
+
return PokemonData(
|
|
485
|
+
species_id=box['substructs'][0]['species'],
|
|
486
|
+
species_name=box['nickname'],
|
|
487
|
+
current_hp=pokemon.hp,
|
|
488
|
+
max_hp=pokemon.maxHp,
|
|
489
|
+
level=pokemon.level,
|
|
490
|
+
status=StatusCondition(pokemon.status),
|
|
491
|
+
type1=PokemonType(box['type1']) if 'type1' in box.keys() else None,
|
|
492
|
+
type2=PokemonType(box['type2']) if 'type2' in box.keys() else None,
|
|
493
|
+
moves=[Move(move).name for move in box['substructs'][1]['moves']],
|
|
494
|
+
move_pp=box['substructs'][1]['pp'],
|
|
495
|
+
trainer_id=box['otId'],
|
|
496
|
+
nickname=box['nickname'],
|
|
497
|
+
experience=box['substructs'][0]['experience']
|
|
498
|
+
)
|
|
499
|
+
|
|
500
|
+
|
|
501
|
+
def read_save_block_2(gba):
|
|
502
|
+
save_block_2_ptr = gba.read_u32(ADDRESSES["gSaveBlock2Ptr"])
|
|
503
|
+
if save_block_2_ptr == 0:
|
|
504
|
+
return None
|
|
505
|
+
|
|
506
|
+
save_block_2_data = gba.read_memory(save_block_2_ptr, struct.calcsize(SaveBlock2_format))
|
|
507
|
+
save_block_2 = SaveBlock2._make(struct.unpack("<" + SaveBlock2_format, save_block_2_data))
|
|
508
|
+
save_block_2 = save_block_2._replace(pokedex=Pokedex._make(struct.unpack("<" + Pokedex_format, save_block_2.pokedex))._asdict())
|
|
509
|
+
return save_block_2._asdict()
|
|
510
|
+
|
|
511
|
+
def read_save_block_1(gba, parse_items: bool = False):
|
|
512
|
+
save_block_1_ptr = gba.read_u32(ADDRESSES["gSaveBlock1Ptr"])
|
|
513
|
+
if save_block_1_ptr == 0:
|
|
514
|
+
return None
|
|
515
|
+
|
|
516
|
+
save_block_1_data = gba.read_memory(save_block_1_ptr, struct.calcsize(SaveBlock1_format))
|
|
517
|
+
save_block_1 = SaveBlock1._make(struct.unpack("<" + SaveBlock1_format, save_block_1_data))
|
|
518
|
+
|
|
519
|
+
player_party_count = gba.read_u8(ADDRESSES["gPlayerPartyCount"])
|
|
520
|
+
player_party_data = gba.read_memory(ADDRESSES["gPlayerParty"], player_party_count * struct.calcsize(Pokemon_format))
|
|
521
|
+
|
|
522
|
+
# parse nested structs
|
|
523
|
+
save_block_1 = save_block_1._replace(
|
|
524
|
+
pos=Coords16._make(struct.unpack("<" + Coords16_format, save_block_1.pos))._asdict(),
|
|
525
|
+
location=WarpData._make(struct.unpack("<" + WarpData_format, save_block_1.location))._asdict(),
|
|
526
|
+
continueGameWarp=WarpData._make(struct.unpack("<" + WarpData_format, save_block_1.continueGameWarp))._asdict(),
|
|
527
|
+
dynamicWarp=WarpData._make(struct.unpack("<" + WarpData_format, save_block_1.dynamicWarp))._asdict(),
|
|
528
|
+
lastHealLocation=WarpData._make(struct.unpack("<" + WarpData_format, save_block_1.lastHealLocation))._asdict(),
|
|
529
|
+
escapeWarp=WarpData._make(struct.unpack("<" + WarpData_format, save_block_1.escapeWarp))._asdict(),
|
|
530
|
+
playerParty=[
|
|
531
|
+
parse_pokemon(player_party_data[i:i+struct.calcsize(Pokemon_format)])
|
|
532
|
+
for i in range(0, player_party_count * struct.calcsize(Pokemon_format), struct.calcsize(Pokemon_format))
|
|
533
|
+
],
|
|
534
|
+
)
|
|
535
|
+
if parse_items:
|
|
536
|
+
save_block_1 = save_block_1._replace(
|
|
537
|
+
pcItems=[
|
|
538
|
+
ItemSlot._make(struct.unpack("<" + ItemSlot_format, save_block_1.pcItems[i:i+struct.calcsize(ItemSlot_format)]))._asdict()
|
|
539
|
+
for i in range(0, len(save_block_1.pcItems), struct.calcsize(ItemSlot_format))
|
|
540
|
+
],
|
|
541
|
+
bagPocket_Items=[
|
|
542
|
+
ItemSlot._make(struct.unpack("<" + ItemSlot_format, save_block_1.bagPocket_Items[i:i+struct.calcsize(ItemSlot_format)]))._asdict()
|
|
543
|
+
for i in range(0, len(save_block_1.bagPocket_Items), struct.calcsize(ItemSlot_format))
|
|
544
|
+
],
|
|
545
|
+
bagPocket_KeyItems=[
|
|
546
|
+
ItemSlot._make(struct.unpack("<" + ItemSlot_format, save_block_1.bagPocket_KeyItems[i:i+struct.calcsize(ItemSlot_format)]))._asdict()
|
|
547
|
+
for i in range(0, len(save_block_1.bagPocket_KeyItems), struct.calcsize(ItemSlot_format))
|
|
548
|
+
],
|
|
549
|
+
bagPocket_PokeBalls=[
|
|
550
|
+
ItemSlot._make(struct.unpack("<" + ItemSlot_format, save_block_1.bagPocket_PokeBalls[i:i+struct.calcsize(ItemSlot_format)]))._asdict()
|
|
551
|
+
for i in range(0, len(save_block_1.bagPocket_PokeBalls), struct.calcsize(ItemSlot_format))
|
|
552
|
+
],
|
|
553
|
+
bagPocket_TMHM=[
|
|
554
|
+
ItemSlot._make(struct.unpack("<" + ItemSlot_format, save_block_1.bagPocket_TMHM[i:i+struct.calcsize(ItemSlot_format)]))._asdict()
|
|
555
|
+
for i in range(0, len(save_block_1.bagPocket_TMHM), struct.calcsize(ItemSlot_format))
|
|
556
|
+
],
|
|
557
|
+
bagPocket_Berries=[
|
|
558
|
+
ItemSlot._make(struct.unpack("<" + ItemSlot_format, save_block_1.bagPocket_Berries[i:i+struct.calcsize(ItemSlot_format)]))._asdict()
|
|
559
|
+
for i in range(0, len(save_block_1.bagPocket_Berries), struct.calcsize(ItemSlot_format))
|
|
560
|
+
]
|
|
561
|
+
)
|
|
562
|
+
|
|
563
|
+
return save_block_1._asdict()
|
|
564
|
+
|
|
565
|
+
|
|
566
|
+
def read_pokemon_storage(gba):
|
|
567
|
+
pokemon_storage_ptr = gba.read_u32(ADDRESSES["gPokemonStoragePtr"])
|
|
568
|
+
if pokemon_storage_ptr == 0:
|
|
569
|
+
return None
|
|
570
|
+
|
|
571
|
+
pokemon_storage_data = gba.read_memory(pokemon_storage_ptr, struct.calcsize(PokemonStorage_format))
|
|
572
|
+
pokemon_storage = PokemonStorage._make(struct.unpack("<" + PokemonStorage_format, pokemon_storage_data))
|
|
573
|
+
|
|
574
|
+
box_mon_size = struct.calcsize(BoxPokemon_format)
|
|
575
|
+
box_size = box_mon_size * IN_BOX_COUNT
|
|
576
|
+
parsed_boxes = []
|
|
577
|
+
for j in range(TOTAL_BOXES_COUNT):
|
|
578
|
+
parsed_boxes.append([
|
|
579
|
+
parse_box_pokemon(pokemon_storage.boxes[i:i+box_mon_size])
|
|
580
|
+
for i in range(j * box_size, (j + 1) * box_size, box_mon_size)
|
|
581
|
+
])
|
|
582
|
+
pokemon_storage = pokemon_storage._replace(
|
|
583
|
+
boxes=parsed_boxes,
|
|
584
|
+
boxNames=[
|
|
585
|
+
pokemon_storage.boxNames[i:i+BOX_NAME_LENGTH]
|
|
586
|
+
for i in range(0, len(pokemon_storage.boxNames), BOX_NAME_LENGTH + 1)
|
|
587
|
+
]
|
|
588
|
+
)
|
|
589
|
+
return pokemon_storage._asdict()
|
|
590
|
+
|
|
591
|
+
@functools.lru_cache(maxsize=1)
|
|
592
|
+
def read_species_names(gba):
|
|
593
|
+
species_names_ptr = ADDRESSES["gSpeciesNames"]
|
|
594
|
+
if species_names_ptr == 0:
|
|
595
|
+
return None
|
|
596
|
+
|
|
597
|
+
species_names_data = gba.read_memory(species_names_ptr, NUM_SPECIES * (POKEMON_NAME_LENGTH +1))
|
|
598
|
+
species_names = [
|
|
599
|
+
EmeraldCharmap().decode(species_names_data[i:i+POKEMON_NAME_LENGTH+1])
|
|
600
|
+
for i in range(0, len(species_names_data), POKEMON_NAME_LENGTH+1)
|
|
601
|
+
]
|
|
602
|
+
return species_names
|
|
603
|
+
|
|
604
|
+
@functools.lru_cache(maxsize=1)
|
|
605
|
+
def read_species_info(gba):
|
|
606
|
+
species_info_ptr = ADDRESSES["gSpeciesInfo"]
|
|
607
|
+
if species_info_ptr == 0:
|
|
608
|
+
return None
|
|
609
|
+
|
|
610
|
+
species_info_data = gba.read_memory(species_info_ptr, NUM_SPECIES * struct.calcsize(SpeciesInfo_format))
|
|
611
|
+
species_info = [
|
|
612
|
+
SpeciesInfo._make(struct.unpack("<" + SpeciesInfo_format, species_info_data[i:i+struct.calcsize(SpeciesInfo_format)]))
|
|
613
|
+
for i in range(0, len(species_info_data), struct.calcsize(SpeciesInfo_format))
|
|
614
|
+
]
|
|
615
|
+
return species_info
|
|
616
|
+
|
|
617
|
+
@functools.lru_cache(maxsize=1)
|
|
618
|
+
def read_experience_tables(gba):
|
|
619
|
+
exp_table_ptr = ADDRESSES["gExperienceTables"]
|
|
620
|
+
if exp_table_ptr == 0:
|
|
621
|
+
return None
|
|
622
|
+
|
|
623
|
+
# there's 6 different growth rates and 101 different levels, each being a 4-byte int
|
|
624
|
+
num_ints = 6 * 101
|
|
625
|
+
exp_table_data = gba.read_memory(exp_table_ptr, num_ints * 4)
|
|
626
|
+
exp_table_format = "<" + "I" * num_ints
|
|
627
|
+
exp_table_flat = struct.unpack(exp_table_format, exp_table_data)
|
|
628
|
+
exp_tables = []
|
|
629
|
+
for i in range(0, len(exp_table_flat), 101):
|
|
630
|
+
exp_tables.append(exp_table_flat[i:i+101])
|
|
631
|
+
return exp_tables
|