hanzo 0.3.9__py3-none-any.whl → 0.3.11__py3-none-any.whl

hanzo/cli.py

@@ -2,6 +2,9 @@
 
 import sys
 import asyncio
+import signal
+import os
+import subprocess
 from typing import Optional
 
 import click
@@ -115,7 +118,11 @@ def serve(ctx, name: str, port: int):
 @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."""
-    asyncio.run(start_compute_node(ctx, name, port, network, models, max_jobs))
+    try:
+        asyncio.run(start_compute_node(ctx, name, port, network, models, max_jobs))
+    except KeyboardInterrupt:
+        # Already handled in start_compute_node
+        pass
 
 
 @cli.command()
@@ -133,7 +140,11 @@ def net(ctx, name: str, port: int, network: str, models: tuple, max_jobs: int):
 @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."""
-    asyncio.run(start_compute_node(ctx, name, port, network, models, max_jobs))
+    try:
+        asyncio.run(start_compute_node(ctx, name, port, network, models, max_jobs))
+    except KeyboardInterrupt:
+        # Already handled in start_compute_node
+        pass
 
 
 async def start_compute_node(
@@ -204,8 +215,40 @@ async def start_compute_node(
                 console.print("API: http://localhost:52415/v1/chat/completions")
                 console.print("\nPress Ctrl+C to stop\n")
 
-                # Run net
-                await net_run()
+                # Set up signal handlers for async version
+                stop_event = asyncio.Event()
+                
+                def async_signal_handler(signum, frame):
+                    console.print("\n[yellow]Stopping hanzo net...[/yellow]")
+                    stop_event.set()
+                
+                signal.signal(signal.SIGINT, async_signal_handler)
+                signal.signal(signal.SIGTERM, async_signal_handler)
+                
+                # Run net with proper signal handling
+                try:
+                    net_task = asyncio.create_task(net_run())
+                    stop_task = asyncio.create_task(stop_event.wait())
+                    
+                    # Wait for either net to complete or stop signal
+                    done, pending = await asyncio.wait(
+                        [net_task, stop_task],
+                        return_when=asyncio.FIRST_COMPLETED
+                    )
+                    
+                    # Cancel pending tasks
+                    for task in pending:
+                        task.cancel()
+                        try:
+                            await task
+                        except asyncio.CancelledError:
+                            pass
+                    
+                    # Check if we stopped due to signal
+                    if stop_task in done:
+                        console.print("[green]✓[/green] Node stopped gracefully")
+                except asyncio.CancelledError:
+                    console.print("[yellow]Cancelled[/yellow]")
             finally:
                 sys.argv = original_argv
         else:
@@ -252,12 +295,46 @@ async def start_compute_node(
                 if models:
                     cmd_args.extend(["--default-model", models[0]])
 
-                # Run net command with detected python
-                process = subprocess.run(cmd_args, env=env, check=False)
-
-                if process.returncode != 0 and process.returncode != -2:  # -2 is Ctrl+C
+                # Run net command with detected python in a more signal-friendly way
+                # Create new process group for better signal handling
+                process = subprocess.Popen(
+                    cmd_args, 
+                    env=env,
+                    preexec_fn=os.setsid if hasattr(os, 'setsid') else None
+                )
+                
+                # Set up signal handlers to forward to subprocess group
+                def signal_handler(signum, frame):
+                    if process.poll() is None:  # Process is still running
+                        console.print("\n[yellow]Stopping hanzo net...[/yellow]")
+                        try:
+                            # Send signal to entire process group
+                            if hasattr(os, 'killpg'):
+                                os.killpg(os.getpgid(process.pid), signal.SIGTERM)
+                            else:
+                                process.terminate()
+                            process.wait(timeout=5)  # Wait up to 5 seconds
+                        except subprocess.TimeoutExpired:
+                            console.print("[yellow]Force stopping...[/yellow]")
+                            if hasattr(os, 'killpg'):
+                                os.killpg(os.getpgid(process.pid), signal.SIGKILL)
+                            else:
+                                process.kill()
+                            process.wait()
+                        except ProcessLookupError:
+                            pass  # Process already terminated
+                    raise KeyboardInterrupt
+                
+                # Register signal handlers
+                signal.signal(signal.SIGINT, signal_handler)
+                signal.signal(signal.SIGTERM, signal_handler)
+                
+                # Wait for process to complete
+                returncode = process.wait()
+                
+                if returncode != 0 and returncode != -2:  # -2 is Ctrl+C
                     console.print(
-                        f"[red]Net exited with code {process.returncode}[/red]"
+                        f"[red]Net exited with code {returncode}[/red]"
                     )
 
             finally:

hanzo/commands/auth.py

@@ -144,7 +144,7 @@ async def status(ctx):
                             console.print(
                                 f"  Quota: {usage.get('tokens', 0)} / {quota}"
                             )
-                except:
+                except Exception:
                     console.print("[yellow]![/yellow] Credentials may be expired")
                     console.print("Run 'hanzo auth login' to refresh")
 

hanzo/commands/mcp.py

@@ -144,7 +144,7 @@ async def run(ctx, tool: str, arg: tuple, json_args: str):
             # Try to parse value as JSON first
             try:
                 args[key] = json.loads(value)
-            except:
+            except Exception:
                 args[key] = value
 
     # Create server and run tool
@@ -189,7 +189,7 @@ async def run(ctx, tool: str, arg: tuple, json_args: str):
             # Try to parse as JSON for pretty printing
             data = json.loads(result)
             console.print_json(data=data)
-        except:
+        except Exception:
             # Display as text
             console.print(result)
     else:
@@ -207,7 +207,7 @@ async def run(ctx, tool: str, arg: tuple, json_args: str):
 def install(ctx, path: str):
     """Install MCP server in Claude Desktop."""
     try:
-        from hanzoai.mcp import create_server
+        import hanzoai.mcp
     except ImportError:
         console.print("[red]Error:[/red] hanzo-mcp not installed")
         console.print("Install with: pip install hanzo[mcp]")

hanzo/interactive/repl.py

@@ -167,7 +167,7 @@ hanzo> mcp run read_file --arg path=README.md
             async with httpx.AsyncClient() as client:
                 response = await client.get("http://localhost:8000/health", timeout=1.0)
                 return "running" if response.status_code == 200 else "not responding"
-        except:
+        except Exception:
             return "not running"
 
     async def count_agents(self) -> int:

{hanzo-0.3.9.dist-info → hanzo-0.3.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hanzo
-Version: 0.3.9
+Version: 0.3.11
 Summary: Hanzo AI - Complete AI Infrastructure Platform with CLI, Router, MCP, and Agent Runtime
 Project-URL: Homepage, https://hanzo.ai
 Project-URL: Repository, https://github.com/hanzoai/python-sdk
@@ -14,16 +14,13 @@ Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.8
+Requires-Python: >=3.10
 Requires-Dist: click>=8.1.0
-Requires-Dist: hanzo-net>=0.1.12
 Requires-Dist: httpx>=0.23.0
 Requires-Dist: prompt-toolkit>=3.0.0
 Requires-Dist: pydantic>=2.0.0

{hanzo-0.3.9.dist-info → hanzo-0.3.11.dist-info}/RECORD

@@ -1,28 +1,28 @@
 hanzo/__init__.py,sha256=f6N_RcJZ0F9ADrROlvPi1OrgwjF8cWQm34cml8hb1zk,169
 hanzo/__main__.py,sha256=F3Vz0Ty3bdAj_8oxyETMIqxlmNRnJOAFB1XPxbyfouI,105
-hanzo/cli.py,sha256=Vgo1VGFOa1nurZJIgGvgLxw-jqNKZJVp2nRG-CzbnhI,10340
+hanzo/cli.py,sha256=9dNIUrGkLn1qQYbC3a_2Xun8bdvlIo2yZ-dWQACREMA,13846
 hanzo/mcp_server.py,sha256=XVygFNn-9CVdu8c95sP7fQjIRtA8K7nsGpgQNe44BRg,460
 hanzo/repl.py,sha256=sW1quuqGkJ_AqgjN2vLNdtWgKDlXIkXiO9Bo1QQI0G4,1089
 hanzo/commands/__init__.py,sha256=7rh94TPNhdq4gJBJS0Ayf0fGNChQYCQCJcJPmYYehiQ,182
 hanzo/commands/agent.py,sha256=DXCfuxHfmC90IoIOL6BJyp7h2yNUo-VIxrfl4OMh8CU,3480
-hanzo/commands/auth.py,sha256=gJ3V-E2AGAkKx3sIXKWLZkjzXC4P7yKI_vXt8zcbPLw,11177
+hanzo/commands/auth.py,sha256=JrM-EV4XDHzNDJeGJMjAr69T0Rxez53HEzlNo0jQ8nE,11187
 hanzo/commands/chat.py,sha256=GqBXHpaen3ySEt4q59mmAU4A8PxXZd8ozDw0HptNHYU,6725
 hanzo/commands/cluster.py,sha256=7AVjD1nej1WzBoIz_qcNxRMKHRb05_8qt3z3xRREZ-s,15486
 hanzo/commands/config.py,sha256=xAzM6n9GhdVIqtn7JrHfLRzj1sshmxCujo7iet2hHqE,7490
-hanzo/commands/mcp.py,sha256=FTF3PupRV2WfM65pgoaHhtJjNtshW2Wp3jzED6jyZf0,7417
+hanzo/commands/mcp.py,sha256=u1uEKDY6gUIa7VymEnRzy0ZphdIKYoNwPSeffZaiKnk,7418
 hanzo/commands/miner.py,sha256=_mZT9nQcT2QSSxI0rDDKuSBVdsg_uE_N_j3PXOHoj-Q,11677
 hanzo/commands/network.py,sha256=wJDxGIxJqc6FzQhbAn0Mw-WGCPUeCOsxmdU6GCmOhgM,11408
 hanzo/commands/repl.py,sha256=FwbBmjNDBZCUQlwKbMZl33r3f899rzI2a2-ZRoBe6kE,5977
 hanzo/commands/tools.py,sha256=fG27wRweVmaFJowBpmwp5PgkRUtIF8bIlu_hGWr69Ss,10393
 hanzo/interactive/__init__.py,sha256=ENHkGOqu-JYI05lqoOKDczJGl96oq6nM476EPhflAbI,74
 hanzo/interactive/dashboard.py,sha256=XB5H_PMlReriCip-wW9iuUiJQOAtSATFG8EyhhFhItU,3842
-hanzo/interactive/repl.py,sha256=T_YWXuF7qaEo9JmvdQdt1aMdnKJgE45LED-IpQX40EI,5570
+hanzo/interactive/repl.py,sha256=w69ZSiaJ7g2nYKE4SA0IDsQg-V6eYy3z0-mqw0VCYFU,5580
 hanzo/router/__init__.py,sha256=_cRG9nHC_wwq17iVYZSUNBYiJDdByfLDVEuIQn5-ePM,978
 hanzo/utils/__init__.py,sha256=5RRwKI852vp8smr4xCRgeKfn7dLEnHbdXGfVYTZ5jDQ,69
 hanzo/utils/config.py,sha256=FD_LoBpcoF5dgJ7WL4o6LDp2pdOy8kS-dJ6iRO2GcGM,4728
 hanzo/utils/net_check.py,sha256=YFbJ65SzfDYHkHLZe3n51VhId1VI3zhyx8p6BM-l6jE,3017
 hanzo/utils/output.py,sha256=W0j3psF07vJiX4s02gbN4zYWfbKNsb8TSIoagBSf5vA,2704
-hanzo-0.3.9.dist-info/METADATA,sha256=9GbEl63cI17Q5mkGYQtBzvuXW4uVNAEKmErW4eT2x8I,4348
-hanzo-0.3.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hanzo-0.3.9.dist-info/entry_points.txt,sha256=pQLPMdqOXU_2BfTcMDhkqTCDNk_H6ApvYuSaWcuQOOw,171
-hanzo-0.3.9.dist-info/RECORD,,
+hanzo-0.3.11.dist-info/METADATA,sha256=mcKJuuFGd5nF8_PkPD8M1XzCtpkIQ8zxsoI4B6FTwcc,4217
+hanzo-0.3.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hanzo-0.3.11.dist-info/entry_points.txt,sha256=pQLPMdqOXU_2BfTcMDhkqTCDNk_H6ApvYuSaWcuQOOw,171
+hanzo-0.3.11.dist-info/RECORD,,