feat: add bot option flag

Author: mtth

src/git_draft/__main__.py

@@ -110,6 +110,13 @@ def callback(
         dest="bot",
         help="AI bot name",
     )
+    parser.add_option(
+        "-o",
+        "--bot-option",
+        action="append",
+        dest="bot_options",
+        help="AI bot options",
+    )
     parser.add_option(
         "-e",
         "--edit",
@@ -193,7 +200,7 @@ async def run() -> None:  # noqa: PLR0912 PLR0915
                 bot_config = bot_configs[0]
             elif config.bots:
                 bot_config = config.bots[0]
-            bot = load_bot(bot_config)
+            bot = load_bot(bot_config, overrides=opts.bot_options)
 
             prompt: str | TemplatedPrompt
             if args:

src/git_draft/bots/__init__.py

@@ -1,10 +1,11 @@
 """Bot interfaces and built-in implementations"""
 
+from collections.abc import Sequence
 import importlib
 import os
 import sys
 
-from ..common import BotConfig, reindent
+from ..common import BotConfig, JSONObject, UnreachableError, reindent
 from .common import ActionSummary, Bot, Goal, UserFeedback, Worktree
 
 
@@ -17,10 +18,13 @@
 ]
 
 
-def load_bot(config: BotConfig | None) -> Bot:
+def load_bot(config: BotConfig | None, *, overrides: Sequence[str]=()) -> Bot:
     """Load and return a Bot instance using the provided configuration"""
+    options = {**config.options} if config and config.options else {}
+    options.update(_parse_overrides(overrides))
+
     if not config:
-        return _default_bot()
+        return _default_bot(options)
 
     if config.pythonpath and config.pythonpath not in sys.path:
         sys.path.insert(0, config.pythonpath)
@@ -35,11 +39,23 @@ def load_bot(config: BotConfig | None) -> Bot:
     if not factory:
         raise NotImplementedError(f"Unknown bot factory: {config.factory}")
 
-    kwargs = config.kwargs or {}
-    return factory(**kwargs)
+    return factory(**options)
+
+
+def _parse_overrides(overrides: str) -> JSONObject:
+    options = {}
+    for override in overrides:
+        match override.split("=", 1):
+            case [switch]:
+                options[switch] = True
+            case [flag, value]:
+                options[flag] = value
+            case _:
+                raise UnreachableError()
+    return options
 
 
-def _default_bot() -> Bot:
+def _default_bot(options: JSONObject) -> Bot:
     if not os.environ.get("OPENAI_API_KEY"):
         raise RuntimeError(
             reindent(
@@ -52,7 +68,7 @@ def _default_bot() -> Bot:
         )
 
     try:
-        from .openai_api import new_threads_bot
+        from .openai_api import new_completions_bot
 
     except ImportError:
         raise RuntimeError(
@@ -65,4 +81,4 @@ def _default_bot() -> Bot:
             )
         )
     else:
-        return new_threads_bot()
+        return new_completions_bot(**options)

src/git_draft/common.py

@@ -68,7 +68,7 @@ class BotConfig:
 
     factory: str
     name: str | None = None
-    kwargs: JSONObject | None = None
+    options: JSONObject | None = None
     pythonpath: str | None = None
 
 

tests/git_draft/common_test.py

@@ -30,7 +30,7 @@ def test_load_ok(self) -> None:
             [[bots]]
             name = "bar"
             factory = "bar"
-            kwargs = {one=1}
+            options = {one=1}
         """
         path = sut.Config.folder_path()
         path.mkdir(parents=True, exist_ok=True)
@@ -42,7 +42,7 @@ def test_load_ok(self) -> None:
             log_level=logging.DEBUG,
             bots=[
                 sut.BotConfig(factory="foo:load", pythonpath="./abc"),
-                sut.BotConfig(factory="bar", name="bar", kwargs={"one": 1}),
+                sut.BotConfig(factory="bar", name="bar", options={"one": 1}),
             ],
         )