add cli argument for model config

Author: m-y-mo

src/seclab_taskflow_agent/cli.py

@@ -107,6 +107,10 @@ def main(
         list[str] | None,
         typer.Argument(help="Remaining prompt text."),
     ] = None,
+    model_config: Annotated[
+        str | None,
+        typer.Option("-m", "--model-config", help="Model configuration module path. Only relevant when running taskflows"),
+    ] = None,
 ) -> None:
     """Run a taskflow or personality-based agent session."""
     # Debug mode from flag or env var
@@ -156,7 +160,7 @@ def main(
         asyncio.run(
             run_main(
                 available_tools, personality, effective_taskflow,
-                cli_globals, user_prompt, resume_session_id=resume,
+                cli_globals, user_prompt, cli_model_config = model_config, resume_session_id=resume,
             ),
             debug=debug,
         )

src/seclab_taskflow_agent/runner.py

@@ -451,6 +451,7 @@ async def run_main(
     taskflow_path: str | None,
     cli_globals: dict[str, str],
     prompt: str | None,
+    cli_model_config: str | None,
     resume_session_id: str | None = None,
 ) -> None:
     """Main entry point for taskflow/personality execution.
@@ -461,6 +462,7 @@ async def run_main(
         taskflow_path: Taskflow module path, or None.
         cli_globals: Global variables from CLI.
         prompt: User prompt text.
+        cli_model_config: Model configuration module path, or None.
         resume_session_id: Session ID to resume from a checkpoint.
     """
     from .session import TaskflowSession
@@ -511,6 +513,8 @@ async def on_handoff_hook(context: RunContextWrapper[TContext], agent: Agent[TCo
 
         # Resolve model config
         model_config_ref = taskflow_doc.model_config_ref
+        if cli_model_config:
+            model_config_ref = cli_model_config
         model_keys: list[str] = []
         model_dict: dict[str, str] = {}
         models_params: dict[str, dict[str, Any]] = {}