new: generator can now be specified as an optional field in the yml

Author: evilsocket

examples/changelog/agent.yml

@@ -1,6 +1,6 @@
 agent: >
   You are an helpful assistant that generates changelogs for projects.
-  
+
   ## Guidance
 
   - Do not use the get_new_commits tool since you already have the list of commits.
@@ -44,4 +44,4 @@ tools:
     complete_task: true
     # will just print to stdout
     tool: "echo {{ markdown }}"
-    print: true
+    print: true

nerve/cli/run.py

@@ -39,7 +39,7 @@ def run(
     ] = pathlib.Path("."),
     generator: t.Annotated[
         str,
-        typer.Option("--generator", "-g", help="Generator to use"),
+        typer.Option("--generator", "-g", help="If the agent generator field is not set, use this generator."),
     ] = DEFAULT_GENERATOR,
     conversation_strategy: t.Annotated[
         str,

nerve/models.py

@@ -64,8 +64,8 @@ class Configuration(BaseModel):
     # legacy field used to detect if the user is loading a legacy file
     system_prompt: str | None = Field(default=None, exclude=True)
 
-    # TODO: add optional generator and use -g as override (or if generator not set in yaml)
-
+    # optional generator
+    generator: str | None = None
     # used for versioning the agents
     version: str = "1.0.0"
     # the system prompt, the agent identity

nerve/runtime/agent.py

@@ -44,7 +44,7 @@ def create(
         Create an agent from a generator and configuration.
 
         Args:
-            generator: The generator string to use.
+            generator: The generator string to use if not set in the configuration.
             configuration: The configuration to use.
             start_state: Initial variables for the agent.
             window_strategy: How to handle conversation history.
@@ -67,10 +67,12 @@ def create(
                 )
             )
 
+        generator_to_use = configuration.generator or generator
+
         runtime = Runtime.build(
             working_dir=working_dir,
             name=name,
-            generator=generator,
+            generator=generator_to_use,
             using=configuration.using,
             jail=configuration.jail,
             tools=configuration.tools,
@@ -79,7 +81,7 @@ def create(
         return cls(
             runtime=runtime,
             configuration=configuration,
-            generation_engine=LiteLLMEngine(generator, window_strategy, runtime.tools),
+            generation_engine=LiteLLMEngine(generator_to_use, window_strategy, runtime.tools),
             conv_window_strategy=window_strategy,
         )