hanzo 0.3.6__py3-none-any.whl → 0.3.7__py3-none-any.whl

hanzo/__init__.py

@@ -3,4 +3,4 @@
 __version__ = "0.3.2"
 __all__ = ["main", "cli"]
 
-from .cli import main, cli
+from .cli import cli, main

hanzo/__main__.py

@@ -3,4 +3,4 @@
 from .cli import main
 
 if __name__ == "__main__":
-    main()
+    main()

hanzo/cli.py

@@ -1,15 +1,26 @@
 """Main CLI entry point for Hanzo."""
 
-import asyncio
 import sys
+import asyncio
 from typing import Optional
 
 import click
 from rich.console import Console
 
-from .commands import agent, auth, chat, cluster, config, mcp, miner, network, repl, tools
-from .interactive.repl import HanzoREPL
+from .commands import (
+    mcp,
+    auth,
+    chat,
+    repl,
+    agent,
+    miner,
+    tools,
+    config,
+    cluster,
+    network,
+)
 from .utils.output import console
+from .interactive.repl import HanzoREPL
 
 # Version
 __version__ = "0.2.10"
@@ -23,7 +34,7 @@ __version__ = "0.2.10"
 @click.pass_context
 def cli(ctx, verbose: bool, json: bool, config: Optional[str]):
     """Hanzo AI - Unified CLI for local, private, and free AI.
-    
+
     Run without arguments to enter interactive mode.
     """
     # Ensure context object exists
@@ -32,14 +43,15 @@ def cli(ctx, verbose: bool, json: bool, config: Optional[str]):
     ctx.obj["json"] = json
     ctx.obj["config"] = config
     ctx.obj["console"] = console
-    
+
     # If no subcommand, enter interactive mode or start compute node
     if ctx.invoked_subcommand is None:
         # Check if we should start as a compute node
         import os
+
         if os.environ.get("HANZO_COMPUTE_NODE") == "1":
             # Start as a compute node
-            from .commands import network
+
             asyncio.run(start_compute_node(ctx))
         else:
             # Enter interactive REPL mode
@@ -90,43 +102,60 @@ def serve(ctx, name: str, port: int):
 
 @cli.command()
 @click.option("--name", "-n", help="Node name (auto-generated if not provided)")
-@click.option("--port", "-p", default=52415, help="Node port (default: 52415 for hanzo/net)")
-@click.option("--network", default="local", help="Network to join (mainnet/testnet/local)")
-@click.option("--models", "-m", multiple=True, help="Models to serve (e.g., llama-3.2-3b)")
+@click.option(
+    "--port", "-p", default=52415, help="Node port (default: 52415 for hanzo/net)"
+)
+@click.option(
+    "--network", default="local", help="Network to join (mainnet/testnet/local)"
+)
+@click.option(
+    "--models", "-m", multiple=True, help="Models to serve (e.g., llama-3.2-3b)"
+)
 @click.option("--max-jobs", type=int, default=10, help="Max concurrent jobs")
 @click.pass_context
 def net(ctx, name: str, port: int, network: str, models: tuple, max_jobs: int):
     """Start the Hanzo Network distributed AI compute node."""
-    start_compute_node(ctx, name, port, network, models, max_jobs)
+    asyncio.run(start_compute_node(ctx, name, port, network, models, max_jobs))
 
 
 @cli.command()
 @click.option("--name", "-n", help="Node name (auto-generated if not provided)")
-@click.option("--port", "-p", default=52415, help="Node port (default: 52415 for hanzo/net)")
-@click.option("--network", default="local", help="Network to join (mainnet/testnet/local)")
-@click.option("--models", "-m", multiple=True, help="Models to serve (e.g., llama-3.2-3b)")
+@click.option(
+    "--port", "-p", default=52415, help="Node port (default: 52415 for hanzo/net)"
+)
+@click.option(
+    "--network", default="local", help="Network to join (mainnet/testnet/local)"
+)
+@click.option(
+    "--models", "-m", multiple=True, help="Models to serve (e.g., llama-3.2-3b)"
+)
 @click.option("--max-jobs", type=int, default=10, help="Max concurrent jobs")
 @click.pass_context
 def node(ctx, name: str, port: int, network: str, models: tuple, max_jobs: int):
     """Alias for 'hanzo net' - Start as a compute node for the Hanzo network."""
-    start_compute_node(ctx, name, port, network, models, max_jobs)
+    asyncio.run(start_compute_node(ctx, name, port, network, models, max_jobs))
 
 
-def start_compute_node(ctx, name: str = None, port: int = 52415, 
-                            network: str = "mainnet", models: tuple = None, 
-                            max_jobs: int = 10):
+async def start_compute_node(
+    ctx,
+    name: str = None,
+    port: int = 52415,
+    network: str = "mainnet",
+    models: tuple = None,
+    max_jobs: int = 10,
+):
     """Start this instance as a compute node using hanzo/net."""
-    from .utils.net_check import check_net_installation, get_missing_dependencies
-    
+    from .utils.net_check import check_net_installation
+
     console = ctx.obj.get("console", Console())
-    
+
     console.print("[bold cyan]Starting Hanzo Net Compute Node[/bold cyan]")
     console.print(f"Network: {network}")
     console.print(f"Port: {port}")
-    
+
     # Check hanzo/net availability
     is_available, net_path, python_exe = check_net_installation()
-    
+
     if not is_available:
         console.print("[red]Error:[/red] hanzo-net is not installed")
         console.print("\nTo install hanzo-net from PyPI:")
@@ -135,23 +164,23 @@ def start_compute_node(ctx, name: str = None, port: int = 52415,
         console.print("  git clone https://github.com/hanzoai/net.git ~/work/hanzo/net")
         console.print("  cd ~/work/hanzo/net && pip install -e .")
         return
-    
+
     try:
-        import subprocess
-        import sys
         import os
-        
+        import sys
+        import subprocess
+
         # Use the checked net_path and python_exe
         if not net_path:
             # net is installed as a package
             console.print("[green]✓[/green] Using installed hanzo/net")
-            
+
             # Set up sys.argv for net's argparse
             original_argv = sys.argv.copy()
             try:
                 # Build argv for net
                 sys.argv = ["hanzo-net"]  # Program name
-                
+
                 # Add options
                 if port != 52415:
                     sys.argv.extend(["--chatgpt-api-port", str(port)])
@@ -159,22 +188,24 @@ def start_compute_node(ctx, name: str = None, port: int = 52415,
                     sys.argv.extend(["--node-id", name])
                 if network != "local":
                     sys.argv.extend(["--discovery-module", network])
-                if models and len(models) > 0:
+                if models:
                     sys.argv.extend(["--default-model", models[0]])
-                
+
                 # Import and run net
                 from net.main import run as net_run
-                
+
                 console.print(f"\n[green]✓[/green] Node initialized")
                 console.print(f"  Port: {port}")
-                console.print(f"  Models: {', '.join(models) if models else 'auto-detect'}")
+                console.print(
+                    f"  Models: {', '.join(models) if models else 'auto-detect'}"
+                )
                 console.print("\n[bold green]Hanzo Net is running![/bold green]")
                 console.print("WebUI: http://localhost:52415")
                 console.print("API: http://localhost:52415/v1/chat/completions")
                 console.print("\nPress Ctrl+C to stop\n")
-                
+
                 # Run net
-                net_run()
+                await net_run()
             finally:
                 sys.argv = original_argv
         else:
@@ -184,28 +215,32 @@ def start_compute_node(ctx, name: str = None, port: int = 52415,
                 console.print(f"[green]✓[/green] Using hanzo/net venv")
             else:
                 console.print("[yellow]⚠[/yellow] Using system Python")
-            
+
             # Change to net directory and run
             original_cwd = os.getcwd()
             try:
                 os.chdir(net_path)
-                
+
                 # Set up environment
                 env = os.environ.copy()
                 if models:
                     env["NET_MODELS"] = ",".join(models)
                 if name:
                     env["NET_NODE_NAME"] = name
-                env["PYTHONPATH"] = os.path.join(net_path, "src") + ":" + env.get("PYTHONPATH", "")
-                
+                env["PYTHONPATH"] = (
+                    os.path.join(net_path, "src") + ":" + env.get("PYTHONPATH", "")
+                )
+
                 console.print(f"\n[green]✓[/green] Starting net node")
                 console.print(f"  Port: {port}")
-                console.print(f"  Models: {', '.join(models) if models else 'auto-detect'}")
+                console.print(
+                    f"  Models: {', '.join(models) if models else 'auto-detect'}"
+                )
                 console.print("\n[bold green]Hanzo Net is running![/bold green]")
                 console.print("WebUI: http://localhost:52415")
                 console.print("API: http://localhost:52415/v1/chat/completions")
                 console.print("\nPress Ctrl+C to stop\n")
-                
+
                 # Build command line args
                 cmd_args = [python_exe, "-m", "net.main"]
                 if port != 52415:
@@ -214,22 +249,20 @@ def start_compute_node(ctx, name: str = None, port: int = 52415,
                     cmd_args.extend(["--node-id", name])
                 if network != "local":
                     cmd_args.extend(["--discovery-module", network])
-                if models and len(models) > 0:
+                if models:
                     cmd_args.extend(["--default-model", models[0]])
-                
+
                 # Run net command with detected python
-                process = subprocess.run(
-                    cmd_args,
-                    env=env,
-                    check=False
-                )
-                
+                process = subprocess.run(cmd_args, env=env, check=False)
+
                 if process.returncode != 0 and process.returncode != -2:  # -2 is Ctrl+C
-                    console.print(f"[red]Net exited with code {process.returncode}[/red]")
-                    
+                    console.print(
+                        f"[red]Net exited with code {process.returncode}[/red]"
+                    )
+
             finally:
                 os.chdir(original_cwd)
-        
+
     except KeyboardInterrupt:
         console.print("\n[yellow]Shutting down node...[/yellow]")
         console.print("[green]✓[/green] Node stopped")
@@ -242,6 +275,7 @@ def start_compute_node(ctx, name: str = None, port: int = 52415,
 def dashboard(ctx):
     """Open interactive dashboard."""
     from .interactive.dashboard import run_dashboard
+
     run_dashboard()
 
 
@@ -258,4 +292,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

hanzo/commands/__init__.py

@@ -1,3 +1,14 @@
 """Command modules for Hanzo CLI."""
 
-__all__ = ["agent", "auth", "chat", "cluster", "config", "mcp", "miner", "network", "repl", "tools"]
+__all__ = [
+    "agent",
+    "auth",
+    "chat",
+    "cluster",
+    "config",
+    "mcp",
+    "miner",
+    "network",
+    "repl",
+    "tools",
+]

hanzo/commands/agent.py

@@ -1,7 +1,7 @@
 """Agent management commands."""
 
 import asyncio
-from typing import List, Optional
+from typing import Optional
 
 import click
 from rich.table import Table
@@ -30,16 +30,12 @@ async def create(ctx, name: str, model: str, description: Optional[str], local:
         console.print("[red]Error:[/red] hanzo-agents not installed")
         console.print("Install with: pip install hanzo[agents]")
         return
-    
+
     base_url = "http://localhost:8000" if local else None
-    
+
     with console.status(f"Creating agent '{name}'..."):
-        agent = create_agent(
-            name=name,
-            model=model,
-            base_url=base_url
-        )
-    
+        agent = create_agent(name=name, model=model, base_url=base_url)
+
     console.print(f"[green]✓[/green] Created agent: {name}")
     console.print(f"  Model: {model}")
     console.print(f"  Mode: {'local' if local else 'cloud'}")
@@ -55,17 +51,17 @@ def list(ctx):
     table.add_column("Model", style="green")
     table.add_column("Status", style="yellow")
     table.add_column("Description")
-    
+
     # Mock data for now
     agents = [
         ("helper", "llama-3.2-3b", "active", "General purpose assistant"),
         ("coder", "codellama-7b", "idle", "Code generation specialist"),
         ("researcher", "llama-3.2-3b", "idle", "Research and analysis"),
     ]
-    
+
     for name, model, status, desc in agents:
         table.add_row(name, model, status, desc)
-    
+
     console.print(table)
 
 
@@ -84,19 +80,20 @@ async def run(ctx, agents: tuple, task: str, parallel: bool, timeout: Optional[i
         console.print("[red]Error:[/red] hanzo-agents not installed")
         console.print("Install with: pip install hanzo[agents]")
         return
-    
+
     agent_list = list(agents)
-    
+
     with console.status(f"Running task with {len(agent_list)} agents..."):
         # Create network with agents
         network = create_network(agents=agent_list)
-        
+
         # Run task
-        result = await asyncio.wait_for(
-            network.run(task),
-            timeout=timeout
-        ) if timeout else await network.run(task)
-    
+        result = (
+            await asyncio.wait_for(network.run(task), timeout=timeout)
+            if timeout
+            else await network.run(task)
+        )
+
     console.print("[green]Task completed![/green]")
     console.print(result)
 
@@ -109,4 +106,4 @@ def delete(ctx, agent: str):
     if click.confirm(f"Delete agent '{agent}'?"):
         console.print(f"[yellow]Deleted agent: {agent}[/yellow]")
     else:
-        console.print("Cancelled")
+        console.print("Cancelled")