duet-cli 0.2.3__tar.gz → 0.2.5__tar.gz

{duet_cli-0.2.3/duet_cli.egg-info → duet_cli-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: duet-cli
-Version: 0.2.3
+Version: 0.2.5
 Summary: Two CLI agents in conversation. One Python file. Stdlib only.
 Author: Volkan Altan
 License-Expression: MIT

{duet_cli-0.2.3 → duet_cli-0.2.5}/duet.py

@@ -3371,6 +3371,10 @@ def _build_arg_parser() -> argparse.ArgumentParser:
                     help="partner agent spec, e.g. codex:coder, claude:reviewer, gemini:coder, copilot:coder (default codex:coder)")
     ap.add_argument("--lead", default="claude:planner",
                     help="lead agent spec, e.g. claude:planner, gemini:reviewer, copilot:planner (default; ignored if --resume-claude given)")
+    ap.add_argument("--lead-model", metavar="MODEL", default=None,
+                    help="model name to pass to the lead agent's backend via --model")
+    ap.add_argument("--partner-model", metavar="MODEL", default=None,
+                    help="model name to pass to the partner agent's backend via --model")
     ap.add_argument("--cwd", default=".", help="working dir for both agents")
     ap.add_argument("--turns", type=int, default=DEFAULT_TURNS, help=f"max turns (default {DEFAULT_TURNS})")
     ap.add_argument("--sentinel", default=DEFAULT_SENTINEL,
@@ -3542,9 +3546,18 @@ def _build_cfg_from_cli(args: argparse.Namespace, ap: argparse.ArgumentParser,
         parser=ap,
         stdin_cache=stdin_cache,
     )
+    agents = [
+        dataclasses.replace(
+            parse_partner(args.lead, default_role="planner"),
+            model=args.lead_model or None,
+        ),
+        dataclasses.replace(
+            parse_partner(args.partner, default_role="coder"),
+            model=args.partner_model or None,
+        ),
+    ]
     agents = apply_resume_overrides(
-        [parse_partner(args.lead, default_role="planner"),
-         parse_partner(args.partner, default_role="coder")],
+        agents,
         resume_claude=args.resume_claude,
         resume_codex=args.resume_codex,
         rename_slots=True,

{duet_cli-0.2.3 → duet_cli-0.2.5/duet_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: duet-cli
-Version: 0.2.3
+Version: 0.2.5
 Summary: Two CLI agents in conversation. One Python file. Stdlib only.
 Author: Volkan Altan
 License-Expression: MIT

{duet_cli-0.2.3 → duet_cli-0.2.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "duet-cli"
-version = "0.2.3"
+version = "0.2.5"
 description = "Two CLI agents in conversation. One Python file. Stdlib only."
 readme = "README.md"
 requires-python = ">=3.9"

{duet_cli-0.2.3 → duet_cli-0.2.5}/tests/test_duet.py

@@ -825,6 +825,45 @@ class TestParsePartner(unittest.TestCase):
         self.assertEqual(agent.name, "copilot-coder")
 
 
+# ---------- _build_cfg_from_cli ----------
+
+
+class TestBuildCfgFromCli(unittest.TestCase):
+    def _parse(self, *argv: str):
+        parser = duet._build_arg_parser()
+        return parser, parser.parse_args(list(argv))
+
+    def test_lead_and_partner_models_attach_to_cli_agents(self) -> None:
+        parser, args = self._parse(
+            "--task", "x",
+            "--lead", "claude:reviewer",
+            "--partner", "codex:coder",
+            "--lead-model", "claude-opus-4-5",
+            "--partner-model", "gpt-5",
+        )
+
+        cfg = duet._build_cfg_from_cli(args, parser, {})
+
+        self.assertEqual(cfg.agents[0].backend, "claude")
+        self.assertEqual(cfg.agents[0].role, "reviewer")
+        self.assertEqual(cfg.agents[0].model, "claude-opus-4-5")
+        self.assertEqual(cfg.agents[1].backend, "codex")
+        self.assertEqual(cfg.agents[1].role, "coder")
+        self.assertEqual(cfg.agents[1].model, "gpt-5")
+
+    def test_empty_cli_model_flags_normalize_to_none(self) -> None:
+        parser, args = self._parse(
+            "--task", "x",
+            "--lead-model", "",
+            "--partner-model", "",
+        )
+
+        cfg = duet._build_cfg_from_cli(args, parser, {})
+
+        self.assertIsNone(cfg.agents[0].model)
+        self.assertIsNone(cfg.agents[1].model)
+
+
 # ---------- apply_resume_overrides ----------
 
 
@@ -878,6 +917,32 @@ class TestApplyResumeOverrides(unittest.TestCase):
         self.assertEqual(agents[1].role, "coder")
         self.assertEqual(agents[1].session_id, "codex-sid")
 
+    def test_resume_codex_move_preserves_model(self) -> None:
+        agents = duet.apply_resume_overrides(
+            [
+                duet.Agent(
+                    name="codex-planner",
+                    backend="codex",
+                    role="planner",
+                    model="gpt-5",
+                ),
+                duet.Agent(
+                    name="claude-coder",
+                    backend="claude",
+                    role="coder",
+                    model="claude-sonnet-4-6",
+                ),
+            ],
+            resume_codex="codex-sid",
+            rename_slots=True,
+        )
+
+        self.assertEqual(agents[0].backend, "claude")
+        self.assertEqual(agents[0].model, "claude-sonnet-4-6")
+        self.assertEqual(agents[1].backend, "codex")
+        self.assertEqual(agents[1].session_id, "codex-sid")
+        self.assertEqual(agents[1].model, "gpt-5")
+
     def test_resume_claude_partner_moves_to_lead(self) -> None:
         agents = duet.apply_resume_overrides(
             [