diff --git a/src/draft/steps/implement_spec/_live_status.py b/src/draft/steps/implement_spec/_live_status.py index 4e48b5d..d0db1ca 100644 --- a/src/draft/steps/implement_spec/_live_status.py +++ b/src/draft/steps/implement_spec/_live_status.py @@ -3,7 +3,7 @@ from pathlib import Path INTERVAL = 20 -MODEL = "claude-3-5-haiku-latest" +MODEL = "haiku" TAIL_BYTES = 4096 MIN_BYTES = 200 MAX_CHARS = 30 diff --git a/src/draft/steps/open_pr/__init__.py b/src/draft/steps/open_pr/__init__.py index 369d37d..0893ccf 100644 --- a/src/draft/steps/open_pr/__init__.py +++ b/src/draft/steps/open_pr/__init__.py @@ -126,7 +126,7 @@ def run(self, ctx, engine, lifecycle, step_metrics): "--permission-mode", "acceptEdits", "--model", - "claude-3-5-haiku-latest", + "haiku", ], timeout=cfg["timeout"], ) diff --git a/tests/draft/test_live_status.py b/tests/draft/test_live_status.py index d124ab3..8ce615a 100644 --- a/tests/draft/test_live_status.py +++ b/tests/draft/test_live_status.py @@ -139,7 +139,7 @@ def test_tick_file_grows_calls_run_llm(tmp_path): s._tick() call_kwargs = engine.run_llm.call_args[1] assert call_kwargs["log_path"] is None - assert call_kwargs["extra_args"] == ["--model", "claude-3-5-haiku-latest"] + assert call_kwargs["extra_args"] == ["--model", "haiku"] assert call_kwargs["allowed_tools"] == [] assert call_kwargs["timeout"] == 10 assert "x" * 10 in call_kwargs["prompt"]