code-puppy 0.0.188__tar.gz → 0.0.189__tar.gz

{code_puppy-0.0.188 → code_puppy-0.0.189}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.188
+Version: 0.0.189
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy

{code_puppy-0.0.188 → code_puppy-0.0.189}/code_puppy/main.py

@@ -61,6 +61,12 @@ async def main():
         type=str,
         help="Execute a single prompt and exit (no interactive mode)",
     )
+    parser.add_argument(
+        "--agent",
+        "-a",
+        type=str,
+        help="Specify which agent to use (e.g., --agent code-puppy)",
+    )
     parser.add_argument(
         "command", nargs="*", help="Run a single command (deprecated, use -p instead)"
     )
@@ -161,6 +167,27 @@ async def main():
         return
 
     ensure_config_exists()
+
+    # Handle agent selection from command line
+    if args.agent:
+        from code_puppy.agents.agent_manager import set_current_agent, get_available_agents
+
+        agent_name = args.agent.lower()
+        try:
+            # First check if the agent exists by getting available agents
+            available_agents = get_available_agents()
+            if agent_name not in available_agents:
+                emit_system_message(f"[bold red]Error:[/bold red] Agent '{agent_name}' not found")
+                emit_system_message(f"Available agents: {', '.join(available_agents.keys())}")
+                sys.exit(1)
+
+            # Agent exists, set it
+            set_current_agent(agent_name)
+            emit_system_message(f"🤖 Using agent: {agent_name}")
+        except Exception as e:
+            emit_system_message(f"[bold red]Error setting agent:[/bold red] {str(e)}")
+            sys.exit(1)
+            
     current_version = __version__
 
     no_version_update = os.getenv("NO_VERSION_UPDATE", "").lower() in (

{code_puppy-0.0.188 → code_puppy-0.0.189}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.188"
+version = "0.0.189"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.11"