new: default task override with new --task argument

Author: evilsocket

docs/index.md

@@ -66,6 +66,12 @@ This will download, extract and install the agent to the folder `$HOME/.nerve/ag
 nerve run changelog
 ```
 
+You can override the default task of any agent:
+
+```bash
+nerve run changelog --task 'use a single sentence for the changelog'
+```
+
 You can uninstall agents with:
 
 ```bash

nerve/cli/run.py

@@ -37,6 +37,10 @@ def run(
         pathlib.Path,
         typer.Argument(help="Agent or workflow to execute"),
     ] = pathlib.Path("."),
+    task: t.Annotated[
+        str | None,
+        typer.Option("--task", "-t", help="Set or override the task for the agent."),
+    ] = None,
     generator: t.Annotated[
         str,
         typer.Option("--generator", "-g", help="If the agent generator field is not set, use this generator."),
@@ -70,7 +74,7 @@ def run(
     ] = DEFAULT_MAX_COST,
     timeout: t.Annotated[
         int | None,
-        typer.Option("--timeout", "-t", help="Timeout in seconds"),
+        typer.Option("--timeout", help="Timeout in seconds"),
     ] = DEFAULT_TIMEOUT,
     log_path: t.Annotated[
         pathlib.Path | None,
@@ -90,6 +94,7 @@ def run(
             generator,
             # convert the conversation strategy string to a valid enum
             conversation.strategy_from_string(conversation_strategy),
+            task,
             ctx.args,
             max_steps,
             max_cost,
@@ -134,6 +139,7 @@ async def _run(
     input_path: pathlib.Path,
     generator: str,
     conv_window_strategy: WindowStrategy,
+    task_override: str | None,
     start_state_args: list[str],
     max_steps: int = 100,
     max_cost: float = 10.0,
@@ -180,4 +186,4 @@ async def _run(
         logger.error(f"path '{input_path}' is not a valid workflow or agent configuration")
         raise typer.Abort()
 
-    await flow.run()
+    await flow.run(task_override)

nerve/runtime/flow.py

@@ -96,13 +96,17 @@ async def from_path(
             timeout=timeout,
         )
 
-    async def _setup_if_needed(self) -> None:
+    async def _setup_if_needed(self, task_override: str | None = None) -> None:
         if self.started_at is None:
             self.started_at = time.time()
 
         if self.curr_actor is None:
             self.curr_actor = self.actors[self.curr_actor_idx]
 
+            if task_override:
+                logger.info(f"🎯 setting task: {task_override}")
+                self.curr_actor.configuration.task = task_override
+
             state.set_tools({tool.__name__: tool for tool in self.curr_actor.runtime.tools})
             state.set_defaults(self.curr_actor.configuration.defaults)
             state.on_task_started(self.curr_actor)
@@ -150,7 +154,7 @@ def done(self) -> bool:
 
         return False
 
-    async def run(self) -> None:
+    async def run(self, task_override: str | None = None) -> None:
         state.on_event(
             "flow_started",
             {
@@ -160,7 +164,7 @@ async def run(self) -> None:
         )
 
         while not self.done():
-            await self._setup_if_needed()
+            await self._setup_if_needed(task_override)
             if self.curr_actor:
                 await self.shell.interact_if_needed(self.curr_actor)
             await self.step()