fix: preserve Claude Code default budget with real configs

openevolve/llm/claude_code.py

@@ -46,6 +46,8 @@ def __init__(self, model_cfg=None):
         self.retries = getattr(model_cfg, "retries", 3)
         self.retry_delay = getattr(model_cfg, "retry_delay", 5)
         self.max_budget_usd = getattr(model_cfg, "max_budget_usd", 1.0)
+        if self.max_budget_usd is None:
+            self.max_budget_usd = 1.0
         self.cwd = getattr(model_cfg, "cwd", None)
         logger.info(f"Initialized ClaudeCodeLLM with model: {self.model}")
 

tests/test_claude_code_llm.py

@@ -168,5 +168,44 @@ def test_ensemble_creates_claude_code(self):
         self.assertIsInstance(model, ClaudeCodeLLM)
 
 
+class TestRealConfigBudget(unittest.TestCase):
+    def _assert_cli_budget(self, model_options, expected, **call_options):
+        from openevolve.config import Config
+        from openevolve.llm.ensemble import _create_model
+
+        config = Config.from_dict(
+            {
+                "llm": {
+                    "provider": "claude_code",
+                    "retries": 0,
+                    "models": [{"name": "sonnet", **model_options}],
+                }
+            }
+        )
+        model = _create_model(config.llm.models[0])
+        with patch("openevolve.llm.claude_code.subprocess.run") as mock_run:
+            mock_run.return_value = MagicMock(returncode=0, stdout="response", stderr="")
+            self.assertEqual(asyncio.run(model.generate("prompt", **call_options)), "response")
+            mock_run.assert_called_once()
+            cmd = mock_run.call_args.args[0]
+            self.assertEqual(cmd[cmd.index("--max-budget-usd") + 1], str(expected))
+
+    def test_omitted_budget_uses_documented_default(self):
+        self._assert_cli_budget({}, 1.0)
+
+    def test_null_budget_uses_documented_default(self):
+        self._assert_cli_budget({"max_budget_usd": None}, 1.0)
+
+    def test_explicit_budget_is_preserved(self):
+        self._assert_cli_budget({"max_budget_usd": 0.5}, 0.5)
+
+    def test_zero_is_not_replaced_by_default(self):
+        # Keep explicit values unchanged; CLI validation remains the CLI's responsibility.
+        self._assert_cli_budget({"max_budget_usd": 0.0}, 0.0)
+
+    def test_per_call_budget_overrides_configured_budget(self):
+        self._assert_cli_budget({"max_budget_usd": 0.5}, 0.25, max_budget_usd=0.25)
+
+
 if __name__ == "__main__":
     unittest.main()